Update tools.py

tools.py

@@ -1,33 +1,27 @@
-# tools.py
 import os
 import openai
 from smolagents.tools import Tool
 
-# Carichiamo la tua chiave
+# Load your OpenAI API key
 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.
 """
 
@@ -42,7 +36,7 @@ class AnswerTool(Tool):
             {"role": "system", "content": SYSTEM_PROMPT},
             {"role": "user",   "content": question},
         ]
-        # → qui usiamo la nuova API v1
+        # Use the v1 OpenAI API with GPT-4o
         resp = openai.chat.completions.create(
             model="gpt-4o",
             messages=messages,
@@ -52,4 +46,4 @@ class AnswerTool(Tool):
         text = resp.choices[0].message.content.strip()
         if "FINAL ANSWER:" in text:
             return text.split("FINAL ANSWER:")[-1].strip()
-        return text
+        return text