Update app.py

app.py

@@ -154,6 +154,26 @@ system_prompt = (
     "```py\nfinal_answer('...')\n```"
 )
 
+prompt_templates = {
+    "system_prompt": system_prompt,
+    "default": "Response: {{question}}",
+    "managed_agent": {
+        "execute": "```py\nfinal_answer('...')\n```",
+        "report": "```py\nfinal_answer('Report not implemented')\n```",
+        "task": "```py\nfinal_answer('Task not implemented')\n```"
+    },
+    "planning": {
+        "initial_plan": "```py\nfinal_answer('Plan not needed')\n```",
+        "update_plan_pre_messages": "```py\nfinal_answer('Update plan')\n```",
+        "update_plan_post_messages": "```py\nfinal_answer('Post update')\n```"
+    },
+    "final_answer": {
+        "pre_messages": "```py\nfinal_answer('Pre-message')\n```",
+        "template": "Final response: {{answer}}",
+        "post_messages": "```py\nfinal_answer('Post-message')\n```"
+    }
+}
+
 agent = CodeAgent(
     model=model,
     tools=[
@@ -163,10 +183,7 @@ agent = CodeAgent(
         get_weather,
         final_answer,
     ],
-    prompt_templates={
-        "system_prompt": system_prompt,
-        "default": "Response: {{question}}"
-    },
+    prompt_templates=prompt_templates,
     max_steps=3,
     verbosity_level=2,
 )
@@ -185,4 +202,4 @@ if __name__ == "__main__":
         outputs="text",
         title="Weather Agent",
         description="Enter a city name to get time, air quality, and weather info."
-    ).launch()
+    ).launch()