Update gaia_agent.py

gaia_agent.py

@@ -53,13 +53,22 @@ api_key = os.getenv("HF_TOKEN")
 # search_tool = DuckDuckGoSearchRun()
 
 # Generate the chat interface, including the tools
+# llm = HuggingFaceEndpoint(
+#     repo_id="deepseek-ai/DeepSeek-R1-0528",
+#     huggingfacehub_api_token=api_key,
+#     timeout=300,
+# )
+
 llm = HuggingFaceEndpoint(
-    repo_id="deepseek-ai/DeepSeek-R1-0528",
+    repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
     huggingfacehub_api_token=api_key,
     timeout=300,
+    max_new_tokens=512,
+    temperature=0.2,
 )
 
 
+
 # # Initialize local Ollama model
 # llm  Ollama(model="qwen2.5-coder", base_url="http://127.0.0.1:11434")