Fix

Upsonic · May 16, 2024 · d5dde5f · d5dde5f
1 parent 181731b
commit d5dde5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/upsonic/remote/on_prem.py b/upsonic/remote/on_prem.py
@@ -1522,11 +1522,11 @@ def type_for_prompt(self):
 
     def return_openai_llm(self, model=None):
 
-        from langchain_openai import OpenAI
+        from langchain_openai import ChatOpenAI
         import httpx
         httpx_client = httpx.Client(verify=False)
 
-        llm = OpenAI(openai_api_key=self.password, http_client=httpx_client, openai_api_base=self.api_url+'/openai/', model_name=model)        
+        llm = ChatOpenAI(openai_api_key=self.password, http_client=httpx_client, api_base=self.api_url+'/openai/', model_name=model)        
         return llm