Update app.py

app.py

@@ -5,7 +5,8 @@ import inspect
 import pandas as pd
 
 # 🔹 NOVO: imports do smolagents
-from smolagents import CodeAgent, InferenceClientModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
+
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -52,9 +53,13 @@ def clean_answer(text: str) -> str:
 # =========================================================
 
 SYSTEM_PROMPT = (
-    "You are an exam-taking assistant.\n"
-    "For each question, reply with ONLY the final answer, with no explanation, "
-    "no reasoning, no extra words, no quotes, and no labels like 'Final answer'.\n"
+    "You are an AI agent solving GAIA-style questions.\n"
+    "You have access to a web search tool (DuckDuckGoSearchTool).\n"
+    "For each question, you MUST search the web when needed to obtain accurate, "
+    "up-to-date factual information before answering.\n"
+    "Use the search tool, read the results, reason, and then produce ONLY the final answer.\n"
+    "Do NOT output explanations, steps, reasoning, citations, links, or any extra words.\n"
+    "Do NOT output labels like 'Final answer', 'Answer:', etc.\n"
     "If the answer is a number, output just the number. "
     "If it is a word or short phrase, output just that.\n"
     "Your output will be compared to the ground truth using EXACT MATCH."
@@ -63,25 +68,29 @@ SYSTEM_PROMPT = (
 
 class BasicAgent:
     """
-    Agente simples baseado em smolagents:
+    Agente simples baseado em smolagents, mas agora com ferramenta de busca:
     - Usa InferenceClientModel (Inference API da Hugging Face)
-    - Não utiliza tools adicionais
+    - Usa DuckDuckGoSearchTool para buscar informações na web
     - Retorna uma string já limpa para EXACT MATCH
     """
 
     def __init__(self):
-        print("Initializing smolagents BasicAgent...")
+        print("Initializing smolagents BasicAgent with web search...")
+
+        # Ferramenta de busca (DuckDuckGo)
+        self.search_tool = DuckDuckGoSearchTool()
 
-        # Passamos o system_prompt para o modelo, não para o CodeAgent
+        # Modelo remoto via Inference API (usa HF_TOKEN configurado no Space)
+        # Passamos o SYSTEM_PROMPT aqui
         self.model = InferenceClientModel(
             system_prompt=SYSTEM_PROMPT
         )
 
-        # CodeAgent sem ferramentas (agente simples)
+        # CodeAgent com ferramenta de busca
         self.agent = CodeAgent(
             model=self.model,
-            tools=[],    # agente simples: sem tools
-            max_steps=1  # sem tools, 1 passo é suficiente
+            tools=[self.search_tool],
+            max_steps=5,   # permite alguns passos (pensar + buscar + responder)
         )
 
     def __call__(self, question: str) -> str: