Update agent.py

agent.py

@@ -377,11 +377,12 @@ class MagAgent:
                 
                 # Validate loaded prompts structure
                 if isinstance(user_prompts, dict):
-                    # Merge with defaults, preserving missing keys
-                    self.prompt_templates.update({
-                        k: v for k, v in user_prompts.items()
-                        if k in self.prompt_templates
-                    })
+                    for key in self.prompt_templates.keys():
+                        if key in user_prompts:
+                            # Merge nested template dictionaries
+                            self.prompt_templates[key].update(
+                                user_prompts[key]
+                            )
                 else:
                     print(f"Invalid prompts.yaml structure. Using defaults. Got type: {type(user_prompts)}")
                     
@@ -423,13 +424,14 @@ class MagAgent:
 
     def _build_task_prompt(self, question: str, task_id: str) -> str:
         """Constructs task-specific prompts using templates"""
-        base_template = self.prompt_templates.get("base_prompt", "")
-        task_template = self.prompt_templates.get("task_prompt", "").format(
-            question=question,
-            task_id=task_id,
-            current_date=datetime.now().strftime("%Y-%m-%d")
-        )  
-        return f"{base_template}\n\n{task_template}"
+        system_template = self.prompt_templates["system_prompt"]["template"]
+        managed_agent_template = self.prompt_templates["managed_agent"]["template"].format(
+            question_analysis=question,
+            subtasks=self._generate_subtasks(question),
+            validation_rules=self._get_validation_rules()
+    )
+    
+    return f"{system_template}\n\n{managed_agent_template}"
     
     async def _execute_agent(self, question: str, task_id: str) -> str:
         return await asyncio.to_thread(