Update agent.py

agent.py

@@ -388,9 +388,32 @@ class MagAgent:
         )
 
     def _validate_and_format(self, result: str, context: dict) -> str:
-        validation = ValidationPipeline().validate
-
-        def _handle_error(self, error: Exception, context: dict) -> str:
+        """Final validation and formatting pipeline"""
+        try:
+            # Basic result validation
+            if not result:
+                raise ValueError("Empty agent response")
+                
+            # Type checking
+            if not isinstance(result, str):
+                raise TypeError(f"Expected string response, got {type(result)}")
+                
+            # Length validation
+            if len(result) > 4096:
+                result = result[:4090] + "..."
+                
+            # Record successful validation
+            context["validation_checks"].append({
+                "type": "success",
+                "timestamp": datetime.now().isoformat()
+            })
+            
+            return result
+            
+        except Exception as e:
+            return self._handle_error(e, context)
+            
+    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)