final assignment gaia agent tools submission enabled system prompt litellm login

agents/gaia.py

@@ -1,5 +1,5 @@
 # --- Gaia Agent Definition ---
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool
 from tools.gaiatool import fetch_questions
 from smolagents import LiteLLMModel, OpenAIServerModel, HfApiModel, InferenceClientModel
 import os
@@ -21,7 +21,7 @@ class GaiaAgent:
         #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_base="https://api.openai.com/v1",
             api_key=os.environ["OPENAI_API_KEY"],
             temperature=0.0
         )
@@ -29,25 +29,22 @@ class GaiaAgent:
         #model_id = "meta-llama/Llama-3.3-70B-Instruct"
         #self.model = InferenceClientModel(model_id=model_id)
 
-        # Initialize the DuckDuckGo search tool
-        self.search_tool = DuckDuckGoSearchTool()
+        search= DuckDuckGoSearchTool()
+        web_browser = VisitWebpageTool()
 
         self.agent = CodeAgent(
             model=self.model,
-            tools=[self.search_tool],
+            tools=[search, web_browser],
             add_base_tools=True,
-            additional_authorized_imports=["pandas"]
+            additional_authorized_imports=["pandas", "re"]
         )
 
         system_prompt = """You are a general AI assistant. I will ask you a question. Report your thoughts, and
 finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
 YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated
-list of numbers and/or strings.
-If you are asked for a number, don’t use comma to write your number neither use units such as $ or
-percent sign unless specified otherwise.
-If you are asked for a string, don’t use articles, neither abbreviations (e.g. for cities), and write the
-digits in plain text unless specified otherwise.
-If you are asked for a comma separated list, apply the above rules depending of whether the element
+list of numbers and/or strings. If you are asked for a number, don’t use comma to write your number neither use units such as $ or
+percent sign unless specified otherwise. If you are asked for a string, don’t use articles, neither abbreviations (e.g. for cities), and write the
+digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element
 to be put in the list is a number or a string."""
 
         self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt

tools/gaiatool.py

@@ -1,6 +1,7 @@
 from smolagents import tool
 import requests
 
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -30,4 +31,4 @@ def fetch_questions(questions_url: str) -> list:
         print(f"An unexpected error occurred fetching questions: {e}")
         return f"An unexpected error occurred fetching questions: {e}", None
 
-    return questions_data
+    return questions_data