From 7ef71d48854d6caaefa87a58c8105cf467151980 Mon Sep 17 00:00:00 2001 From: Patrick Lafleur Date: Wed, 1 Oct 2025 12:08:03 -0400 Subject: [PATCH] Fix text --- tests/guardrails_tests/test_bedrock_guardrails.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/guardrails_tests/test_bedrock_guardrails.py b/tests/guardrails_tests/test_bedrock_guardrails.py index 1b1fea74af..2997e32f09 100644 --- a/tests/guardrails_tests/test_bedrock_guardrails.py +++ b/tests/guardrails_tests/test_bedrock_guardrails.py @@ -1407,11 +1407,16 @@ async def test_bedrock_guardrail_post_call_success_hook_no_output_text(): "stopReason": "tool_use" } - data = {} # request data not used by our condition + data = { + "model": "gpt-4o", + "messages": [ + {"role": "user", "content": "Hello"}, + ], + } mock_user_api_key_dict = UserAPIKeyAuth() return await guardrail.async_post_call_success_hook( data=data, - response=mock_bedrock_response, # dict with "outputs" + response=mock_bedrock_response, user_api_key_dict=mock_user_api_key_dict, ) \ No newline at end of file