Update gaia_agent.py

gaia_agent.py

@@ -31,7 +31,7 @@ import chess
 import chess.engine
 # import tempfile
 # from PIL import Image
-from langchain.chat_models import ChatOpenAI
+from langchain.chat_models.openai import OpenAIChat
 
 
 
@@ -61,11 +61,11 @@ api_key = os.getenv("HF_TOKEN")
 #     huggingfacehub_api_token=api_key,
 #     timeout=300,
 # )
-llm = ChatOpenAI(
-        model_name="gpt-4o-mini",  # use model_name instead of model
-        temperature=0.2,
-        openai_api_key=OpenAI_key
-    )
+llm = OpenAIChat(
+    model_name="gpt-4o-mini",
+    temperature=0.2,
+    openai_api_key=OpenAI_key
+)
 
 # # Initialize local Ollama model
 # llm  Ollama(model="qwen2.5-coder", base_url="http://127.0.0.1:11434")