Update agent.py

agent.py

@@ -461,14 +461,34 @@ class MagAgent:
 
     def _build_task_prompt(self, question: str, task_id: str) -> str:
         """Constructs task-specific prompts using templates"""
-        system_template = self.prompt_templates["system_prompt"]["template"]
-        managed_agent_template = self.prompt_templates["managed_agent"]["template"].format(
-            task=f"Task ID: {task_id}\nQuestion: {question}\nDate: {datetime.now().strftime('%Y-%m-%d')}",
-            question_analysis=question,
+        managed_agent = self.prompt_templates["managed_agent"]
+        
+        task_context = managed_agent["task_template"].format(
+            task_id=task_id,
+            question=question,
+            current_date=datetime.now().strftime("%Y-%m-%d"),
+            answer_format="concise, accurate, properly punctuated",
+            validation_level="strict"
+        )
+        
+        analysis = managed_agent["analysis_template"].format(
+            question_analysis=self._analyze_question(question),
+            entities=self._extract_entities(question),
+            temporal_constraints=self._find_temporal_limits(question)
+        )
+        
+        subtasks = managed_agent["subtask_template"].format(
             subtasks=self._generate_subtasks(question),
-            validation_rules=self._get_validation_rules()
+            tools=", ".join([tool.name for tool in self.tools])
+        )
+        
+        validation = managed_agent["validation_template"].format(
+            validation_rules=self._get_validation_rules(question),
+            max_retries=3,
+            fallback_protocol="Multi-source verification"
         )
-        return f"{system_template}\n\n{managed_agent_template}"
+        
+        return "\n\n".join([task_context, analysis, subtasks, validation])
     
     async def _execute_agent(self, question: str, task_id: str) -> str:
         return await asyncio.to_thread(
@@ -501,7 +521,16 @@ class MagAgent:
             
         except Exception as e:
             return self._handle_error(e, context)
-            
+
+    def _extract_entities(self, question: str) -> str:
+        """Extract key entities from question"""
+        return ", ".join(re.findall(r'\b[A-Z][a-z]+\b', question))
+    
+    def _find_temporal_limits(self, question: str) -> str:
+        """Find date ranges in question"""
+        dates = re.findall(r'\b\d{4}\b', question)
+        return f"{min(dates)}-{max(dates)}" if dates else "No date constraints"
+        
     def _handle_error(self, error: Exception, context: dict) -> str:
         """Central error handling with context-aware formatting"""
         error_type = error.__class__.__name__