Update agent.py

agent.py

@@ -388,4 +388,20 @@ class MagAgent:
         )
 
     def _validate_and_format(self, result: str, context: dict) -> str:
-        validation = ValidationPipeline().validate
+        validation = ValidationPipeline().validate
+
+        def _handle_error(self, error: Exception, context: dict) -> str:
+        """Central error handling with context-aware formatting"""
+        error_type = error.__class__.__name__
+        error_msg = str(error)
+        
+        # Store error in validation context
+        context["validation_checks"].append({
+            "type": "error",
+            "error_type": error_type,
+            "message": error_msg,
+            "timestamp": datetime.now().isoformat()
+        })
+        
+        # Format user-facing message
+        return f"AGENT ERROR: {error_type} - {error_msg}"