hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Better verification for instance connection

Browse Source
This commit is contained in:
jeffser 2024-07-23 19:20:27 -06:00
parent 3e7fd1140c
commit a765e8cf2e
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/window.py
View File

@ -782,13 +782,11 @@ Generate a title following these rules:
json.dump({'remote_url': self.remote_url, 'remote_bearer_token': self.remote_bearer_token, 'run_remote': self.run_remote, 'local_port': local_instance.port, 'run_on_background': self.run_on_background, 'model_tweaks': self.model_tweaks, 'ollama_overrides': local_instance.overrides}, f, indent=6) json.dump({'remote_url': self.remote_url, 'remote_bearer_token': self.remote_bearer_token, 'run_remote': self.run_remote, 'local_port': local_instance.port, 'run_on_background': self.run_on_background, 'model_tweaks': self.model_tweaks, 'ollama_overrides': local_instance.overrides}, f, indent=6)
def verify_connection(self): def verify_connection(self):
response = connection_handler.simple_get(connection_handler.url) response = connection_handler.simple_get(f"{connection_handler.url}/api/tags")
if response.status_code == 200: if response.status_code == 200:
if "Ollama is running" in response.text: self.save_server_config()
self.save_server_config() self.update_list_local_models()
self.update_list_local_models() return response.status_code == 200
return True
return False
def add_code_blocks(self): def add_code_blocks(self):
text = self.bot_message.get_text(self.bot_message.get_start_iter(), self.bot_message.get_end_iter(), True) text = self.bot_message.get_text(self.bot_message.get_start_iter(), self.bot_message.get_end_iter(), True)