Feat : better interpreter loading of code block

This commit is contained in:
martin legrand
2025-02-23 12:40:30 +01:00
parent 298c26a63f
commit 43ba140a21
6 changed files with 38 additions and 14 deletions
+2 -3
View File
@@ -15,8 +15,7 @@ class Provider:
self.available_providers = {
"ollama": self.ollama_fn,
"server": self.server_fn,
"test": self.test_fn
"test": self.test_fn,
}
if self.server != "":
print("Provider initialized at ", self.server)
@@ -99,7 +98,7 @@ class Provider:
thought += chunk['message']['content']
except ollama.ResponseError as e:
if e.status_code == 404:
ollama.pull(self._model)
ollama.pull(self.model)
if "refused" in str(e):
raise Exception("Ollama connection failed. is the server running ?")
raise e