🔄 Update import path

shroominic · Feb 17, 2024 · e581086 · e581086
1 parent 1dbdc17
commit e581086
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/funcchain/model/abilities.py b/src/funcchain/model/abilities.py
@@ -29,7 +29,7 @@
 
 
 def gather_llm_type(llm: BaseChatModel, func_check: bool = True) -> str:
-    from langchain_openai.chat_models import ChatOpenAI
+    from .patches import ChatOpenAI
 
     if not isinstance(llm, BaseChatModel):
         return "base_model"