Mise en place de Prompt_templates suivant instructions par Perplexity

agent.py

@@ -10,6 +10,7 @@ from smolagents import (
     # HfApiModel, # import bug from smolagents after adding duckduckgo-search in requirements
     tool
 )
+from smolagents.prompts import CODE_SYSTEM_PROMPT
 from typing import List, Dict, Any, Optional
 
 
@@ -22,17 +23,22 @@ class QAgent:
         api_key: Optional[str] = None,
         provider: Optional[str] = None,  # for InferenceClientModel
         timeout: Optional[int] = None,   # for InferenceClientModel
-        system_prompt: str = "",
+        # system_prompt: str = "",
         temperature: float = 0.2,
         verbose: bool = False # Verbose logging or not
     ):
         """ 
         QAgent description 
         """
+        # Enhance system prompt for GAIA test
+        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."
+        custom_prompt = f"{extra}\n\n{CODE_SYSTEM_PROMPT}"
+
+        
         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}")
+        # self.system_prompt = system_prompt or ""
+        # print(f"self.system_prompt = {self.system_prompt},  system_prompt = {system_prompt}")
 
 
 
@@ -110,7 +116,8 @@ class QAgent:
         # Setup imports
         self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
         custom_templates = PromptTemplates(
-            system_prompt="Votre system prompt personnalisé ici"
+            system_prompt=custom_prompt,
+            planning=PlanningPromptTemplate(template="Remember: reflect deeply."),
         )
         # Create CodeAgent
         print(f"Begin creating CodeAgent")