From dc72b85cc058cc89940835162a850b3e39eada09 Mon Sep 17 00:00:00 2001 From: Amna Mubashar Date: Tue, 13 Aug 2024 17:54:17 +0200 Subject: [PATCH] Update test_chat_generator.py --- integrations/amazon_bedrock/tests/test_chat_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/integrations/amazon_bedrock/tests/test_chat_generator.py b/integrations/amazon_bedrock/tests/test_chat_generator.py index bcb221413..64e9ce2ef 100644 --- a/integrations/amazon_bedrock/tests/test_chat_generator.py +++ b/integrations/amazon_bedrock/tests/test_chat_generator.py @@ -194,7 +194,7 @@ def test_long_prompt_is_truncated(mock_boto3_session): assert prompt_after_resize == truncated_prompt_text -def test_long_prompt_is_not_truncated_when_truncate_false(): +def test_long_prompt_is_not_truncated_when_truncate_false(mock_boto3_session): """ Test that a long prompt is not truncated and _ensure_token_limit is not called when truncate is set to False """