Fix error with changed LiteLLM exception type, closes #620

neuml · Dec 22, 2023 · 2eca3e8 · 2eca3e8
1 parent f9229bc
commit 2eca3e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/python/txtai/pipeline/llm/litellm.py b/src/python/txtai/pipeline/llm/litellm.py
@@ -33,11 +33,12 @@ def ismodel(path):
             True if this is a LiteLLM model, False otherwise
         """
 
+        # pylint: disable=W0702
         if isinstance(path, str) and LITELLM:
             with open(os.devnull, "w", encoding="utf-8") as f, contextlib.redirect_stdout(f):
                 try:
                     return api.get_llm_provider(path)
-                except ValueError:
+                except:
                     return False
 
         return False