Merge branch 'fix/assessment-error-handling-3' into 'develop'

Fix/assessment error handling 3

See merge request genaiic-reusable-assets/engagement-artifacts/genaiic-idp-accelerator!301

Author: rstrahan

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

@@ -106,8 +106,11 @@ def check_explainability_count(self):
     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]):
+        assessment_enabled = self.assessment_config.get('assessment.enabled', False)
+        logger.info(f"Assessment Enabled: {assessment_enabled}")
+        if (assessment_enabled and
+                expected_attributes and
+                (not self.explainability_info or not self.explainability_info[0])):
             return {
                 'is_valid': False,
                 'failed_attributes': list(expected_attributes),

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

@@ -70,24 +70,26 @@ def handler(event, context):
     logger.info(f"Total extraction time: {t1-t0:.2f} seconds")
 
     # Assessment validation
-    logger.info("--- Start: Assessment Validation ---")
-    for section in updated_document.sections:
-        if section.section_id == section_id and section.extraction_result_uri:
-            logger.info(f"Loading assessment results from: {section.extraction_result_uri}")
-            # Load extraction data with assessment results
-            extraction_data = s3.get_json_content(section.extraction_result_uri)
-            validator = AssessmentValidator(extraction_data,
-                                            assessment_config=config.get('assessment', {}),
-                                            enable_missing_check=False,
-                                            enable_count_check=False)
-            validation_results = validator.validate_all()
-            if not validation_results['is_valid']:
-                # Handle validation failure
-                updated_document.status = Status.FAILED
-                validation_errors = validation_results['validation_errors']
-                updated_document.errors.extend(validation_errors)
-                logger.error(f"Validation Error: {validation_errors}")
-    logger.info("---   End: Assessment Validation ---")
+    assessment_enabled = config.get('assessment.enabled', False)
+    if not assessment_enabled:
+        logger.info("Assessment is disabled.")
+    else:
+        for section in updated_document.sections:
+            if section.section_id == section_id and section.extraction_result_uri:
+                logger.info(f"Loading assessment results from: {section.extraction_result_uri}")
+                # Load extraction data with assessment results
+                extraction_data = s3.get_json_content(section.extraction_result_uri)
+                validator = AssessmentValidator(extraction_data,
+                                                assessment_config=config.get('assessment', {}),
+                                                enable_missing_check=False,
+                                                enable_count_check=False)
+                validation_results = validator.validate_all()
+                if not validation_results['is_valid']:
+                    # Handle validation failure
+                    updated_document.status = Status.FAILED
+                    validation_errors = validation_results['validation_errors']
+                    updated_document.errors.extend(validation_errors)
+                    logger.error(f"Validation Error: {validation_errors}")
 
     # Prepare output with automatic compression if needed
     result = {