Update agent.py

agent.py

@@ -351,12 +351,24 @@ class MagAgent:
             SpeechToTextTool(),
         ]
 
-        # Initialize with default prompts
+        # Initialize with default prompts as nested dictionaries
         self.prompt_templates = {
-            "system_prompt": "You are Magus...",
-            "managed_agent": "Decomposing problem...",
-            "planning": "Step-by-Step Plan...",
-            "final_answer": "Final Verified Answer..."
+            "system_prompt": {
+                "template": "You are Magus...",
+                "variables": []
+            },
+            "managed_agent": {
+                "template": "Decomposing problem...",
+                "variables": ["question_analysis"]
+            },
+            "planning": {
+                "template": "Step-by-Step Plan...",
+                "variables": ["step1", "step2", "step3"]
+            },
+            "final_answer": {
+                "template": "Final Verified Answer...",
+                "variables": ["sources", "answer"]
+            }
         }
 
         try: