update

app.py

@@ -3,23 +3,100 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, GradioUI, InferenceClientModel
+from smolagents import (
+    CodeAgent,
+    ToolCallingAgent,
+    DuckDuckGoSearchTool,
+    VisitWebpageTool,
+    WikipediaSearchTool,
+    GradioUI,
+    InferenceClientModel,
+    LiteLLMModel,
+)
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# GAIA benchmark prompt — enforces exact-match answer formatting
+GAIA_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 on whether the element is a number or a string."
+)
+
+
 class BasicAgent:
     def __init__(self):
+        print("Initializing BasicAgent...")
+        model = self._get_model()
+
+        web_agent = ToolCallingAgent(
+            tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
+            model=model,
+            name="web_search_agent",
+            description=(
+                "A web search agent that can search the internet and visit web pages. "
+                "Give it a clear, specific research question and it will return the relevant information found."
+            ),
+            max_steps=10,
+        )
+
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
-            model=InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+            tools=[
+                DuckDuckGoSearchTool(),
+                WikipediaSearchTool(),
+                VisitWebpageTool(),
+            ],
+            model=model,
+            managed_agents=[web_agent],
+            additional_authorized_imports=[
+                "json", "re", "math", "statistics", "datetime",
+                "csv", "collections", "itertools", "functools",
+                "unicodedata", "string", "fractions", "decimal",
+            ],
+            planning_interval=2,
+            max_steps=20,
         )
+        print("BasicAgent initialized successfully.")
+
+    @staticmethod
+    def _get_model():
+
+        if os.getenv("OPENAI_API_KEY"):
+            print("Using OpenAI GPT-4o")
+            return LiteLLMModel(model_id="gpt-4o", temperature=0.1)
+        if os.getenv("ANTHROPIC_API_KEY"):
+            print("Using Anthropic Claude Sonnet")
+            return LiteLLMModel(
+                model_id="anthropic/claude-sonnet-4-20250514",
+                temperature=0.1,
+            )
+        print("Using Qwen/Qwen2.5-72B-Instruct via HF Inference")
+        return InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        response = self.agent(question)
-        print(f"Agent returning response: {response}")
-        return response
+        prompt = (
+            f"{GAIA_SYSTEM_PROMPT}\n\n"
+            f"Now here is the question you need to answer.\n\n"
+            f"Question: {question}"
+        )
+        response = self.agent.run(prompt)
+        answer = self._extract_final_answer(str(response))
+        print(f"Agent returning answer: {answer}")
+        return answer
+
+    @staticmethod
+    def _extract_final_answer(response: str) -> str:
+        """Extract the FINAL ANSWER from the agent response for exact-match scoring."""
+        if "FINAL ANSWER:" in response:
+            return response.split("FINAL ANSWER:")[-1].strip()
+        return response.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """

requirements.txt

@@ -1,4 +1,7 @@
 gradio
 requests
-smolagents
-ddgs
+pandas
+smolagents[toolkit]
+litellm
+wikipedia-api
+markdownify