Skip to content

Commit 26b949f

Browse files
committed
fixed the test and lambda function copy
1 parent a5b8c4a commit 26b949f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

lib/idp_common_pkg/tests/unit/summarization/test_summarization_service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -450,8 +450,8 @@ def test_process_document(
450450
# Verify executor was used to process sections in parallel
451451
assert mock_executor_instance.submit.call_count == 2
452452

453-
# Verify write_content was called for combined results
454-
assert mock_write_content.call_count == 2
453+
# Verify write_content was called for combined results (JSON, fulltext, and markdown)
454+
assert mock_write_content.call_count == 3
455455

456456
# Verify document has summarization_result
457457
assert result.summarization_result is not None

src/lambda/chat_with_document_resolver/index.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def handler(event, context):
143143
elif error_code == 'NoSuchBucket':
144144
raise Exception(f"Bucket not found: {output_bucket}")
145145
else:
146-
raise Exception(f"Error accessing S3: {error_message}")
146+
raise Exception(error_message)
147147

148148
except Exception as e:
149149
logger.error(f"Unexpected error: {str(e)}")

0 commit comments

Comments
 (0)