Some clean up and back to GAIA extra system prompt as the tested simple "write numbers in letters" works well.

agent.py

@@ -1,7 +1,6 @@
 import os
 from smolagents import (
     CodeAgent,
-    PromptTemplates,
     DuckDuckGoSearchTool,
     VisitWebpageTool, 
     PythonInterpreterTool,
@@ -23,24 +22,19 @@ class QAgent:
         api_key: Optional[str] = None,
         provider: Optional[str] = None,  # for InferenceClientModel
         timeout: Optional[int] = None,   # for InferenceClientModel
-        # system_prompt: str = "",
         temperature: float = 0.2,
         verbose: bool = False # Verbose logging or not
     ):
         """ 
         QAgent description 
         """
-        # Enhance system prompt for GAIA test
-        extra="If the answer is a number, write it out in words."
-        # "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 comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
-        # custom_prompt = f"{extra}\n\n{CODE_SYSTEM_PROMPT}"
+        # Enhance system prompt for GAIA questions
+        extra="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 comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
+        # "If the answer is a number, write it out in words." --> tested, it works!
 
         
         print(f"Begin QAgent init with: ")
         self.verbose = verbose
-        # self.system_prompt = system_prompt or ""
-        # print(f"self.system_prompt = {self.system_prompt},  system_prompt = {system_prompt}")
-
 
 
         # if model_type == "HfApiModel":
@@ -116,17 +110,13 @@ class QAgent:
 
         # Setup imports
         self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
-        # custom_templates = PromptTemplates(
-        #     system_prompt=custom_prompt,
-        #     planning=PlanningPromptTemplate(template="Remember: reflect deeply."),
-        # )
+
         # Create CodeAgent
         print(f"Begin creating CodeAgent")
         self.agent = CodeAgent(
             tools=self.tools,
             model=self.model,
             instructions=extra,
-            # prompt_templates=custom_templates,
             # additional_authorized_imports=self.imports,
             # executor_type=executor_type,
             # executor_kwargs=executor_kwargs,