Update app.py

app.py

@@ -51,7 +51,14 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 if "final_answer" not in prompt_templates:
-    prompt_templates["final_answer"] = "This is the final answer: {answer}"
+    prompt_templates["final_answer"] = {
+        "template": "This is the final answer: {answer}",
+        "pre_messages": "You are about to receive the final answer."
+    }
+else:
+    # Ensure 'pre_messages' exists within 'final_answer' template
+    if "pre_messages" not in prompt_templates["final_answer"]:
+        prompt_templates["final_answer"]["pre_messages"] = "You are about to receive the final answer."
 
 required_templates = ["planning", "managed_agent"]
 
@@ -69,6 +76,8 @@ for template in required_templates:
             You are a managed agent who will execute a task with steps defined for you. After each step,
             you should provide 'Thought:', 'Code:', and 'Observation:' sequences to guide the process.
             """
+
+            
     
 agent = CodeAgent(
     model=model,