Updated assessment error handling

Author: webarch-ai

patterns/pattern-2/src/assessment_function/assessment_validator.py

@@ -103,34 +103,32 @@ def check_explainability_count(self):
             'error_message': None
         }
 
-    def check_top_k_config(self):
-        """Check if top_k configuration is valid"""
-        top_k = self.assessment_config.get("top_k")
-        
-        if top_k is not None:
-            if not isinstance(top_k, (int, float)) or top_k > 100:
-                raise ValueError("top_k must be a number and not greater than 100")
-                # return {
-                #     'is_valid': False,
-                #     'failed_attributes': ['top_k'],
-                #     'error_message': 'top_k must be a number and not greater than 100'
-                # }
-        
+    def check_explainability_exists(self):
+        """Check if explainability_info has at least one element when expected attributes exist"""
+        expected_attributes = set(self.inference_result.keys())
+
+        if expected_attributes and (not self.explainability_info or not self.explainability_info[0]):
+            return {
+                'is_valid': False,
+                'failed_attributes': list(expected_attributes),
+                'error_message': 'No assessment information found for the extracted attributes.'
+            }
+
         return {
             'is_valid': True,
             'failed_attributes': [],
             'error_message': None
         }
-    
+
     def validate_all(self):
         """Run all validations and return comprehensive results"""
         validation_results = {
             'is_valid': True,
             'failed_attributes': [],
             'validation_errors': []
         }
-        
-        # Check missing explainability (only if enabled)
+
+        # Check missing explainability if enabled
         if self.enable_missing_check:
             missing_result = self.check_missing_explainability()
             if not missing_result['is_valid']:

patterns/pattern-2/src/processresults_function/index.py

@@ -512,6 +512,9 @@ def handler(event, context):
 
     logger.info(f"Response: {json.dumps(response, default=str)}")
 
+    if document.errors:
+        validation_errors.extend(document.errors)
+
     # Raise exception if there were validation errors
     if validation_errors:
         document.status = Status.FAILED