diff --git a/tests/guardrails_tests/test_bedrock_guardrails.py b/tests/guardrails_tests/test_bedrock_guardrails.py index 2997e32f09..c4d1655594 100644 --- a/tests/guardrails_tests/test_bedrock_guardrails.py +++ b/tests/guardrails_tests/test_bedrock_guardrails.py @@ -1384,7 +1384,7 @@ async def test_bedrock_guardrail_post_call_success_hook_no_output_text(): guardrailVersion="DRAFT" ) - # Mock Bedrock API response with PII masking + # Mock Bedrock API with no output text mock_bedrock_response = MagicMock() mock_bedrock_response.status_code = 200 mock_bedrock_response.json.return_value = { @@ -1419,4 +1419,6 @@ async def test_bedrock_guardrail_post_call_success_hook_no_output_text(): data=data, response=mock_bedrock_response, user_api_key_dict=mock_user_api_key_dict, - ) \ No newline at end of file + ) + # If no error is raised, then the test passes + print("✅ No output text in response test passed") \ No newline at end of file