Corrected agent block

- Added smolagent WebSearchTool
- updated requirements.txt with smolagents[toolkit]
- updated GAIA benchmark specific promt

app.py

@@ -3,8 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from huggingface_hub import InferenceClient
-from smolagents import CodeAgent,DuckDuckGoSearchTool
+from smolagents import CodeAgent, WebSearchTool, InferenceClientModel
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -12,43 +12,47 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
-
-def web_search(self, query: str) -> str:
-    """A tool that does web search 
-    query: The first argument -> What to search 
-        
-    """
-    search_tool = DuckDuckGoSearchTool()
-    st_response = search_tool(query)
-    return st_response
-
 class BasicAgent:
     def __init__(self):
         print("Srj's Agent initialized.")
-        self.client = InferenceClient(model= "moonshotai/Kimi-K2.5", token = os.environ.get("HF_TOKEN"))
+        self.model = InferenceClientModel(model= "moonshotai/Kimi-K2.5", token = os.environ.get("HF_TOKEN"))
+        self.agent = CodeAgent(tools=[WebSearchTool()], model=self.model, stream_outputs=False, max_tokens=20)
                
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        tools= [web_search]
-        prompt = f"""Question: {question}
-
-Solve step-by-step internally but respond with ONLY the final answer.
-NO "Final Answer:", NO explanation, NO reasoning. You can also use the
-Just the exact answer string (number, city name, or short list). You can use the tools like {tools} to find info online.
-Answer Examples:
+        prompt = f"""You are an expert GAIA benchmark agent. Your job is to solve Level 1 GAIA questions with PERFECT formatting.
+
+CRITICAL RULES (Follow EXACTLY):
+1. Answer with ONLY the final answer - NO "Final Answer:", NO reasoning, NO explanation
+2. Use exact format: number, city name, short phrase, or comma-separated list
+3. NO units unless asked ($, %, etc.)
+4. NO articles ("the", "a", "an")
+5. Spell out numbers in words when asked for text
+6. For cities: full name, no abbreviations (Los Angeles, not LA)
+
+GAIA LEVEL 1 KNOWLEDGE (Your memory):
+- 2024 Summer Olympics: Paris, France (July 26-Aug 11)
+- France capital: Paris (pop ~2 million city proper)
+- USA capital: Washington, D.C.
+- UK capital: London
+- Germany capital: Berlin
+- Japan capital: Tokyo
+- Basic math: 15×3=45, 100/4=25, etc.
+
+FORMAT EXAMPLES (Copy these exactly):
 Capital of France? → Paris
-2024 Olympics city? → Paris
-15 × 3? → 45"""
+2024 Olympics host city? → Paris
+Capital of Japan? → Tokyo
+15 × 3? → 45
+Paris population (millions)? → 2 million
+USA capital? → Washington, D.C.
+
+Question: {question}
+
+Respond with ONLY the final answer."""
         
-        output = self.client.chat.completions.create(
-            messages= [
-            {"role":"User","content":prompt}
-            ],
-            stream= False,
-            max_tokens= 20,
-            extra_body={"thinking":{"type":'disabled'}}
-        )
-        return output.choices[0].message.content.strip()
+        output = self.agent.run(prompt)
+        return output.strip()
     
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):