Update gaia_agent.py

gaia_agent.py

@@ -31,7 +31,6 @@ import chess
 import chess.engine
 # import tempfile
 # from PIL import Image
-from langchain.chat_models.openai import OpenAIChat
 
 
 
@@ -54,19 +53,15 @@ api_key = os.getenv("HF_TOKEN")
 # search_tool = DuckDuckGoSearchRun()
 
 # Generate the chat interface, including the tools
-# llm = HuggingFaceEndpoint(
-#     #repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-#     #repo_id="deepseek-ai/DeepSeek-R1-0528", 
-#     repo_id="meta-llama/Llama-3.2-1B-Instruct",
-#     huggingfacehub_api_token=api_key,
-#     timeout=300,
-# )
-llm = OpenAIChat(
-    model_name="gpt-4o-mini",
-    temperature=0.2,
-    openai_api_key=OpenAI_key
+llm = HuggingFaceEndpoint(
+    #repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    #repo_id="deepseek-ai/DeepSeek-R1-0528", 
+    repo_id="meta-llama/Llama-3.1-8B-Instruct",
+    huggingfacehub_api_token=api_key,
+    timeout=300,
 )
 
+
 # # Initialize local Ollama model
 # llm  Ollama(model="qwen2.5-coder", base_url="http://127.0.0.1:11434")