Update

HACK-THE-WORLD · Sep 21, 2024 · a6367e2 · a6367e2
1 parent a5a04a3
commit a6367e2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/Gepetto/gepetto/models/local_ollama.py b/Gepetto/gepetto/models/local_ollama.py
@@ -26,7 +26,7 @@ def supported_models():
         if OLLAMA_MODELS is None:
             try:
                 OLLAMA_MODELS = [m["name"] for m in create_client().list()["models"]]
-            except _httpx.ConnectError:
+            except (_httpx.ConnectError, ollama.ResponseError):
                 OLLAMA_MODELS = []
         return OLLAMA_MODELS
 

diff --git a/Gepetto/gepetto/models/openai.py b/Gepetto/gepetto/models/openai.py
@@ -97,9 +97,7 @@ def query_model_async(self, query, cb, additional_model_options=None):
         :param additional_model_options: Additional parameters used when creating the model object. Typically, for
         OpenAI, response_format={"type": "json_object"}.
         """
-        if additional_model_options is None:
-            additional_model_options = {}
-            t = threading.Thread(target=self.query_model, args=[query, cb, additional_model_options])
-            t.start()
+        t = threading.Thread(target=self.query_model, args=[query, cb, additional_model_options])
+        t.start()
 
 gepetto.models.model_manager.register_model(GPT)