Update agent.py

agent.py

@@ -351,26 +351,35 @@ class MagAgent:
             SpeechToTextTool(),
         ]
 
-        # Load prompts with required templates
-        default_prompts = {
-            "system_prompt": "Default system instructions...",
-            "managed_agent": "Default subtask template...",
-            "planning": "Default planning template...",
-            "final_answer": "Default answer format: {answer}"
+        # Initialize with default prompts
+        self.prompt_templates = {
+            "system_prompt": "You are Magus...",
+            "managed_agent": "Decomposing problem...",
+            "planning": "Step-by-Step Plan...",
+            "final_answer": "Final Verified 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}"
-#            }
-
-        # Load prompt templates
-        with open("prompts.yaml", 'r') as stream:
-            self.prompt_templates = yaml.safe_load(stream)
+        try:
+            with open("prompts.yaml") as f:
+                user_prompts = yaml.safe_load(f)
+                
+                # 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
+                    })
+                else:
+                    print(f"Invalid prompts.yaml structure. Using defaults. Got type: {type(user_prompts)}")
+                    
+        except Exception as e:
+            print(f"Error loading prompts.yaml: {str(e)}. Using default templates")
+
+        # Final validation
+        required_keys = {'system_prompt', 'managed_agent', 'planning', 'final_answer'}
+        if missing := required_keys - self.prompt_templates.keys():
+            raise ValueError(f"Missing required prompt templates: {missing}")
 
         
         self.agent = CodeAgent(