Update agent.py

agent.py

@@ -1,26 +1,18 @@
 """
-GAIA Level-1 agent powered by smolagents.
-
-* Planner/esecutore: CodeAgent (smolagents)
-* LLM backend      : GPT-4.1 via OpenAI
-* Tools            : DuckDuckGo (builtin), WikipediaTool, ArxivTool
-* Output           : UNA sola riga (exact-match)
+GAIA Level-1 agent powered by smolagents ≥ 0.6
 """
 from __future__ import annotations
 import os, textwrap
-from smolagents import CodeAgent, DuckDuckGoSearchTool, OpenAIModel
+from smolagents import CodeAgent, DuckDuckGoSearchTool
 from tools import WikipediaTool, ArxivTool
 import openai
 
-# ─── API key check ──────────────────────────────────────────────
+# ─── API key ───────────────────────────────────────
 openai.api_key = os.getenv("OPENAI_API_KEY") or ""
 if not openai.api_key:
-    raise EnvironmentError(
-        "OPENAI_API_KEY non impostata: aggiungila nei Secrets dello Space "
-        "o in un file .env locale."
-    )
+    raise EnvironmentError("OPENAI_API_KEY non trovata – aggiungila nei Secrets dello Space.")
 
-# ─── Prompt di sistema rigido (exact-match) ─────────────────────
+# ─── System prompt (exact-match) ───────────────────
 SYSTEM_PROMPT = textwrap.dedent("""
 You are a helpful assistant tasked with answering questions using a set of tools.
 Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
@@ -29,38 +21,27 @@ If you are asked for a string, don't use articles, neither abbreviations, and wr
 Return ONLY the final answer line.
 """).strip()
 
-# ─── Costruzione del “core” CodeAgent ───────────────────────────
-model = OpenAIModel(
-    model_id="gpt-4.1",
-    temperature=0,
-    system_prompt=SYSTEM_PROMPT
-)
-
-tools = [
-    DuckDuckGoSearchTool(),  # incorporato in smolagents
-    WikipediaTool(),
-    ArxivTool(),
-]
+# ─── Build CodeAgent ───────────────────────────────
+MODEL_ID = os.getenv("GAIA_MODEL_ID", "gpt-4o")   # override se hai “gpt-4.1”
 
 core_agent = CodeAgent(
-    model=model,
-    tools=tools,
-    max_steps=6,          # previene loop infiniti
-    scratchpad="minimal"  # log conciso
-    #stream_outputs=False
+    model=MODEL_ID,
+    tools=[
+        DuckDuckGoSearchTool(),
+        WikipediaTool(),
+        ArxivTool(),
+    ],
+    max_steps=6,
+    temperature=0,
+    system_prompt=SYSTEM_PROMPT,
+    stream_outputs=False,
 )
 
-# ─── Thin wrapper usato da app.py ───────────────────────────────
-class BasicAgent:  # (mantiene lo stesso nome già importato in app.py)
+# ─── Thin wrapper per app.py ───────────────────────
+class BasicAgent:
     def __init__(self):
-        print("✅ smolagents BasicAgent inizializzato")
+        print("✅ smolagents BasicAgent avviato")
 
     def __call__(self, question: str) -> str:
-        """
-        Esegue CodeAgent e restituisce SOLO la prima riga,
-        così il grader riceve una stringa exact-match.
-        """
-        raw_answer: str = core_agent.run(question)
-        answer = raw_answer.strip().split("\n", 1)[0]
-        print(f"[ANSWER] {answer}")
-        return answer
+        raw = core_agent.run(question)
+        return raw.strip().split("\n", 1)[0]