fix: ChatOpenAI setting

nobu007 · May 18, 2024 · 66ef291 · 66ef291
1 parent a7630e7
commit 66ef291
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/codeinterpreterapi/llm/llm.py b/src/codeinterpreterapi/llm/llm.py
@@ -25,11 +25,11 @@ def get_llm(cls) -> BaseChatModel:
                 timeout=settings.REQUEST_TIMEOUT,
             )  # type: ignore
         if settings.OPENAI_API_KEY:
-            from langchain_openai import ChatGoogleGenerativeAI
+            from langchain_openai import ChatOpenAI
 
-            return ChatGoogleGenerativeAI(
+            return ChatOpenAI(
                 model=model,
-                api_key=settings.GEMINI_API_KEY,
+                api_key=settings.OPENAI_API_KEY,
                 timeout=settings.REQUEST_TIMEOUT,
                 temperature=settings.TEMPERATURE,
                 max_retries=settings.MAX_RETRY,