Prompt template .yaml added

app.py

@@ -8,18 +8,22 @@ from smolagents import (CodeAgent, InferenceClientModel,
                 DuckDuckGoSearchTool, WikipediaSearchTool,
                 PythonInterpreterTool, VisitWebpageTool)
 from smolagents.agents import PromptTemplates
+import yaml
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-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 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."""
-
+# --- System Prompt ---
+with open("prompts.yaml", "r", encoding="utf-8") as f:
+    prompts = yaml.safe_load(f)
 
+prompt_templates = PromptTemplates(
+    system=prompts["system"],
+    planning=prompts["planning"],
+    managed_agent=prompts["managed_agent"],
+    final_answer=prompts["final_answer"],
+)
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -32,10 +36,7 @@ class BasicAgent:
         wiki_tool = WikipediaSearchTool()
         python_tool = PythonInterpreterTool()
         visit_tool = VisitWebpageTool()
-
-        # Create prompt templates with your system prompt
-        prompt_templates = PromptTemplates(system=SYSTEM_PROMPT)
-
+        
         self.agent = CodeAgent(
             model=self.model,
             tools=[search_tool, wiki_tool, python_tool, visit_tool],

prompt.yaml

@@ -0,0 +1,16 @@
+system: |
+  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.
+planning: |
+  Think step by step and plan how to answer the question using available tools.
+managed_agent: |
+  If you are managing another agent. Delegate tasks as needed.
+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.