Update app.py

app.py

@@ -70,51 +70,55 @@ SYSTEM_PROMPT = (
 # =========================================================
 class BasicAgent:
     """
-    Agente smolagents com DuckDuckGoSearchTool,
-    otimizado para GAIA EXACT MATCH.
+    GAIA Agent powered by Mixtral + DuckDuckGo Search
     """
 
     def __init__(self):
-        print("Initializing GAIA agent with web search...")
+        print("Initializing improved GAIA agent...")
 
-        # ---- Token HF vindo do secret HF_TOKEN ----
-        hf_token = os.getenv("HF_TOKEN")
-        if not hf_token:
-            print("⚠️  HF_TOKEN not found in environment! InferenceClient may fail.")
+        HF_TOKEN = os.getenv("HF_TOKEN")
+        if not HF_TOKEN:
+            print("❌ ERROR: HF_TOKEN secret not found!")
+            raise ValueError("HF_TOKEN missing")
 
-        # Cliente de inferência com token
-        client = InferenceClient(token=hf_token) if hf_token else InferenceClient()
+        # search tool
+        self.search_tool = DuckDuckGoSearchTool()
 
-        # Modelo remoto via Inference API
-        self.model = InferenceClientModel(client=client)
+        # LLM client (Mixtral)
+        client = InferenceClient(
+            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+            token=HF_TOKEN,
+        )
 
-        # Ferramenta de busca
-        self.search_tool = DuckDuckGoSearchTool()
+        # Model wrapper
+        self.model = InferenceClientModel(
+            client=client,
+            system_prompt="""
+You are a GAIA evaluation agent.
+
+Rules:
+- Use web search when needed.
+- Output ONLY the final answer.
+- No explanations, no reasoning, no URLs, no extra words.
+- EXACT MATCH evaluation.
+""",
+        )
 
-        # CodeAgent com ferramenta de busca
+        # The real agent
         self.agent = CodeAgent(
             model=self.model,
             tools=[self.search_tool],
-            max_steps=8,  # permite alguns passos (buscar + refinar)
+            max_steps=5,
         )
 
     def __call__(self, question: str) -> str:
-        print(f"\nProcessing question: {question[:80]}...")
         try:
-            # Prompt completo passado para o agente
-            full_prompt = (
-                f"{SYSTEM_PROMPT}\n\n"
-                f"Question: {question}\n\n"
-                "Follow the rules above and return ONLY the exact final answer."
+            result = self.agent.run(
+                f"Answer this GAIA question. Return ONLY the final answer:\n{question}"
             )
-
-            raw = self.agent.run(full_prompt)
-            final = clean_answer(raw)
-            print(f"Raw answer: {raw}")
-            print(f"Final cleaned answer: {final}")
-            return final
+            return clean_answer(result)
         except Exception as e:
-            print(f"Error inside BasicAgent.__call__: {e}")
+            print(f"Agent error: {e}")
             return ""