Update agent.py

agent.py

@@ -351,8 +351,14 @@ class MagAgent:
             SpeechToTextTool(),
         ]
 
-        with open("prompts.yaml") as f:
-            self.prompt_templates = yaml.safe_load(f)
+        try:
+            with open("prompts.yaml") as f:
+                self.prompt_templates = yaml.safe_load(f)
+        except Exception as e:
+            self.prompt_templates = {
+                "base_prompt": "Default base prompt...",
+                "task_prompt": "Default task template: {question}"
+            }
 
         self.agent = CodeAgent(
             model=self.model,
@@ -381,6 +387,16 @@ class MagAgent:
             "validation_checks": []
         }
 
+    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}"
+    
     async def _execute_agent(self, question: str, task_id: str) -> str:
         return await asyncio.to_thread(
             self.agent.run,