Update gaia_agent.py

gaia_agent.py

@@ -53,13 +53,14 @@ 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 = 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 = OpenAI(model="gpt-4o-mini", temperature=0.2, api_key=OpenAI_key)
 
 # # Initialize local Ollama model
 # llm  Ollama(model="qwen2.5-coder", base_url="http://127.0.0.1:11434")