Update tools.py

tools.py

@@ -1,16 +1,56 @@
-
-from typing import Any
+# tools.py
+import os
+import openai
 from smolagents.tools import Tool
 
-class FinalAnswerTool(Tool):
-    name = "final_answer"
-    description = "Returns the final answer generated by the LLM."
-    inputs = {"answer": {"type": "any", "description": "The final answer to output"}}
-    output_type = "any"
+# Carichiamo la tua chiave
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+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 commas, symbols or units (e.g. %, $, km) unless explicitly asked.
+- If you are asked for a string, don't use articles ("a", "the"), abbreviations (e.g. "NYC"), or extra words; write digits in plain text unless specified otherwise.
+- If you are asked for a comma separated list, apply the above rules to each element.
+
+Examples:
+Q: How many legs does a spider have?
+Thought: It's basic biology.
+FINAL ANSWER: 8
+
+Q: What are the primary colors in light?
+Thought: Physics concept.
+FINAL ANSWER: red, green, blue
+
+Q: Who won the Best Picture Oscar in 1994?
+Thought: Film history.
+FINAL ANSWER: Forrest Gump
+
+Now it's your turn.
+"""
+
+class AnswerTool(Tool):
+    name = "answer_tool"
+    description = "Answer GAIA Level 1 questions in exact-match format."
+    inputs = {"question": {"type": "string", "description": "The GAIA question text."}}
+    output_type = "string"
 
-    def forward(self, answer: Any) -> Any:
-        # Just pass through whatever the LLM returns.
-        return answer
+    def forward(self, question: str) -> str:
+        # Build chat messages
+        messages = [
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user",   "content": question},
+        ]
+        resp = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",  # o "gpt-4" se hai ancora quota
+            messages=messages,
+            temperature=0.0,
+            max_tokens=64,
+        )
+        text = resp.choices[0].message.content.strip()
+        if "FINAL ANSWER:" in text:
+            return text.split("FINAL ANSWER:")[-1].strip()
+        return text
 
-    def __init__(self, *args, **kwargs):
-        self.is_initialized = False