From 8f2ecbcd6d7d43d74c98dc4074e05cc81778158d Mon Sep 17 00:00:00 2001
From: Shroominic <db2002dominic@gmail.com>
Date: Sun, 10 Mar 2024 13:21:44 +0700
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix=20fix=20fix?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/funcchain/model/abilities.py        | 2 +-
 src/funcchain/model/patches/__init__.py | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/src/funcchain/model/abilities.py b/src/funcchain/model/abilities.py
index 82c3b56..f13026a 100644
--- 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
index 3c8b5fe..a9d3461 100644
--- 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",
 ]