Update agent.py

agent.py

@@ -359,15 +359,20 @@ class MagAgent:
             "final_answer": "Default answer format: {answer}"
         }
 
-        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}"
+#        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}"
             }
 
+        # Load prompt templates
+        with open("prompts.yaml", 'r') as stream:
+            self.prompt_templates = yaml.safe_load(stream)
+
+        
         self.agent = CodeAgent(
             model=self.model,
             tools=self.tools,