Update agent.py

agent.py

@@ -403,7 +403,7 @@ class MagAgent:
         if missing := required_keys - set(self.prompt_templates.keys()):
             raise ValueError(f"Missing required prompt templates: {missing}")
         
-        required_managed_agent_vars = {'task_id', 'task', 'timestamp', 'question_analysis', 'subtasks', 'validation_rules'}
+        required_managed_agent_vars = {'task_id', 'task', 'timestamp', 'question_analysis', 'subtasks', 'validation_rules', 'report'}
         if missing_vars := required_managed_agent_vars - set(self.prompt_templates["managed_agent"]["variables"]):
             raise ValueError(f"Missing required variables in managed_agent template: {missing_vars}")
         
@@ -430,10 +430,11 @@ class MagAgent:
     def _create_context(self, question: str, task_id: str) -> dict:
         """Create context dictionary for the task."""
         return {
-            "question": question,
+            "task": question,
             "task_id": task_id,
             "timestamp": datetime.now().isoformat(),
-            "validation_checks": []
+            "validation_checks": [],
+            "report": None
         }
     
     def _build_task_prompt(self, question: str, task_id: str) -> str:
@@ -445,7 +446,8 @@ class MagAgent:
             "timestamp": datetime.now().isoformat(),
             "question_analysis": self._generate_analysis(question),
             "subtasks": self._generate_subtasks(question),
-            "validation_rules": self._get_validation_rules(question)
+            "validation_rules": self._get_validation_rules(question),
+            "report": None
         }
         return template.format(**variables)