final assignment gaia agent tools submission enabled system prompt litellm

agents/gaia.py

@@ -1,7 +1,7 @@
 # --- Gaia Agent Definition ---
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from tools.gaiatool import fetch_questions
-from smolagents import LiteLLMModel
+from smolagents import LiteLLMModel, OpenAIServerModel, HfApiModel
 import os
 
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -18,7 +18,13 @@ class GaiaAgent:
         self.questions_url = f"{api_url}/questions"
 
         # Initialize the model
-        self.model = LiteLLMModel(model_id="gemini/gemini-1.5-flash-latest", api_key=os.getenv("GEMINI_API_KEY"))
+        #self.model = LiteLLMModel(model_id="gemini/gemini-1.5-flash-latest", api_key=os.getenv("GEMINI_API_KEY"))
+        self.model = OpenAIServerModel(
+            model_id="gpt-4o-mini",
+            api_base="https://api.openai.com/v1",
+            api_key=os.environ["OPENAI_API_KEY"],
+            temperature=0.0
+        )
 
         # Initialize the DuckDuckGo search tool
         self.search_tool = DuckDuckGoSearchTool()