🔧 fix fix fix

shroominic · Mar 10, 2024 · 8f2ecbc · 8f2ecbc
1 parent f2fa148
commit 8f2ecbc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
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 .patches import ChatOpenAI
+    from langchain_openai import ChatOpenAI
 
     if not isinstance(llm, BaseChatModel):
         return "base_model"

diff --git a/src/funcchain/model/patches/__init__.py b/src/funcchain/model/patches/__init__.py
@@ -1,11 +1,7 @@
-from .azure import AzureChatOpenAI
 from .llamacpp import ChatLlamaCpp
 from .ollama import ChatOllama
-from .openai import ChatOpenAI
 
 __all__ = [
-    "AzureChatOpenAI",
-    "ChatOpenAI",
     "ChatLlamaCpp",
     "ChatOllama",
 ]