🔧 fixfixfix

shroominic · Jun 15, 2024 · 249a83d · 249a83d
1 parent 34c1317
commit 249a83d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/funcchain/model/abilities.py b/src/funcchain/model/abilities.py
@@ -78,10 +78,13 @@ def gather_llm_type(llm: BaseChatModel, func_check: bool = True) -> str:
             if llm.model in model:
                 return "vision_model"
 
-    from langchain_groq import ChatGroq
+    try:
+        from langchain_groq import ChatGroq
 
-    if isinstance(llm, ChatGroq):
-        return "json_model"
+        if isinstance(llm, ChatGroq):
+            return "json_model"
+    except ImportError:
+        pass
 
     return "chat_model"