Update app.py

app.py

@@ -52,6 +52,23 @@ with open("prompts.yaml", 'r') as stream:
 
 if "final_answer" not in prompt_templates:
     prompt_templates["final_answer"] = "This is the final answer: {answer}"
+
+required_templates = ["planning", "managed_agent"]
+
+for template in required_templates:
+    if template not in prompt_templates:
+        # Define a default template for the missing one(s)
+        if template == "planning":
+            prompt_templates["planning"] = """
+            You are an expert assistant capable of breaking down complex tasks into smaller steps. 
+            For each task, plan your approach step-by-step. Each step should include a 'Thought:' 
+            explaining the reasoning, 'Code:' to write the Python code, and 'Observation:' to explain the result.
+            """
+        elif template == "managed_agent":
+            prompt_templates["managed_agent"] = """
+            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,