Skip to content

Commit dd8e593

Browse files
committed
Fix HITL Confidence alerts for Pattern1
1 parent 7fcb78a commit dd8e593

File tree

1 file changed

+11
-4
lines changed
  • patterns/pattern-1/src/processresults_function

1 file changed

+11
-4
lines changed

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

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -877,8 +877,11 @@ def process_segments(
877877
for page_num in page_indices
878878
for kv in pagespecific_details['key_value_details'].get(str(page_num), [])
879879
) or float(bp_confidence) < confidence_threshold
880-
logger.info(f"HITL Status Low confidence {low_confidence}")
880+
else:
881+
low_confidence = None
881882

883+
logger.info(f"low_confidence: {low_confidence}")
884+
882885
item.update({
883886
"page_array": page_indices,
884887
"hitl_triggered": low_confidence,
@@ -925,6 +928,10 @@ def process_segments(
925928
"hitl_corrected_result": custom_decimal_output
926929
})
927930
else:
931+
if enable_hitl == 'true':
932+
std_hitl = 'true'
933+
else:
934+
std_hitl = None
928935
# Process standard output if no custom output match
929936
std_bucket, std_key = parse_s3_path(segment['standard_output_path'])
930937
std_output = download_decimal(std_bucket, std_key)
@@ -934,7 +941,7 @@ def process_segments(
934941
page_array = list(range(start_page, end_page + 1))
935942
item.update({
936943
"page_array": page_array,
937-
"hitl_triggered": enable_hitl,
944+
"hitl_triggered": std_hitl,
938945
"extraction_bp_name": "None",
939946
"extracted_result": std_output
940947
})
@@ -944,12 +951,12 @@ def process_segments(
944951
record_number=record_number,
945952
bp_match=segment.get('custom_output_status'),
946953
extraction_bp_name="None",
947-
hitl_triggered=enable_hitl,
954+
hitl_triggered=std_hitl,
948955
page_array=page_array,
949956
review_portal_url=SAGEMAKER_A2I_REVIEW_PORTAL_URL
950957
)
951958

952-
hitl_triggered = enable_hitl
959+
hitl_triggered = std_hitl
953960
if enable_hitl == 'true':
954961
for page_number in range(start_page, end_page + 1):
955962
ImageUri = f"s3://{output_bucket}/{object_key}/pages/{page_number}/image.jpg"

0 commit comments

Comments
 (0)