Update app.py

app.py

@@ -83,6 +83,14 @@ with open("prompts.yaml", 'r') as stream:
 for key in list(prompt_templates.keys()):
     if isinstance(prompt_templates[key], str):
         prompt_templates[key] = {"content": prompt_templates[key]}
+
+# Convert final_answer to the required nested structure
+if 'final_answer' in prompt_templates:
+    prompt_templates['final_answer'] = {
+        'pre_messages': '',
+        'content': prompt_templates['final_answer'],
+        'post_messages': ''
+    }
     
 agent = CodeAgent(
     model=model,