Upgrade to gpt-4o, add VisitWebpageTool, sharper prompt, more steps

app.py

@@ -9,7 +9,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Agent Definition ---
 def build_agent():
     """Build and return the smolagents CodeAgent with Groq backend."""
-    from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool, tool
+    from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, WikipediaSearchTool, VisitWebpageTool, tool
 
     @tool
     def download_task_file(task_id: str) -> str:
@@ -57,7 +57,7 @@ def build_agent():
         raise ValueError("OPENAI_API_KEY environment variable not set. Add it as a Secret in your HF Space settings.")
 
     model = LiteLLMModel(
-        model_id="openai/gpt-4o-mini",
+        model_id="openai/gpt-4o",
         api_key=openai_api_key,
         temperature=0.0,
     )
@@ -66,6 +66,7 @@ def build_agent():
         tools=[
             DuckDuckGoSearchTool(),
             WikipediaSearchTool(),
+            VisitWebpageTool(),
             download_task_file,
         ],
         model=model,
@@ -73,9 +74,9 @@ def build_agent():
             "requests", "json", "re", "math", "datetime",
             "csv", "io", "os", "pathlib",
             "PIL", "PIL.Image",
-            "pandas",
+            "pandas", "openpyxl",
         ],
-        max_steps=10,
+        max_steps=15,
     )
 
     return agent
@@ -90,10 +91,16 @@ class BasicAgent:
     def __call__(self, question: str) -> str:
         print(f"Question: {question[:100]}...")
         system_note = (
-            "You are a precise research assistant. "
-            "Answer the question with ONLY the final answer — no explanation, no 'FINAL ANSWER:' prefix, "
-            "no extra text. If the answer is a number, give only the number. "
-            "If it is a name, give only the name. Be exact."
+            "You are a precise research assistant solving GAIA benchmark questions. "
+            "Your answers are graded by EXACT STRING MATCH, so formatting is critical.\n\n"
+            "Rules:\n"
+            "- Reply with ONLY the answer, nothing else. No explanation, no 'FINAL ANSWER:' prefix.\n"
+            "- Numbers: use digits (e.g. 42, 3.14). No units unless the question asks for them.\n"
+            "- Lists: comma-separated on one line unless the question specifies otherwise.\n"
+            "- Names/strings: exact spelling, match the question's expected format.\n"
+            "- If a file is attached to the question, use the download_task_file tool first.\n"
+            "- Search the web and visit pages to verify facts before answering.\n"
+            "- Think step by step, but output ONLY the final answer."
         )
         full_prompt = f"{system_note}\n\nQuestion: {question}"
         try:

requirements.txt

@@ -5,3 +5,5 @@ smolagents[litellm]
 ddgs
 wikipedia-api
 Pillow
+openpyxl
+markdownify