From 26e62c9bd8585f85bcf0afa9422f05bdbec3587f Mon Sep 17 00:00:00 2001 From: Jugal Bhatt Date: Wed, 13 Aug 2025 11:53:11 -0700 Subject: [PATCH] Update Test for Anthropic Messages Configuration - Renamed `AmazonAnthropicClaude3MessagesConfig` to `AmazonAnthropicClaudeMessagesConfig` in the test file to align with recent refactoring. - Adjusted the instantiation of the configuration class in the test to reflect the new naming convention. --- .../test_anthropic_claude3_transformation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py b/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py index 661962bc52..0d21c16376 100644 --- a/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py +++ b/tests/test_litellm/llms/bedrock/messages/invoke_transformations/test_anthropic_claude3_transformation.py @@ -12,7 +12,7 @@ sys.path.insert(0, os.path.abspath("../../../../../..")) from litellm.litellm_core_utils.litellm_logging import Logging as LiteLLMLoggingObj from litellm.llms.bedrock.messages.invoke_transformations.anthropic_claude3_transformation import ( - AmazonAnthropicClaude3MessagesConfig, + AmazonAnthropicClaudeMessagesConfig, AmazonAnthropicClaudeMessagesStreamDecoder, ) @@ -21,7 +21,7 @@ from litellm.llms.bedrock.messages.invoke_transformations.anthropic_claude3_tran async def test_bedrock_sse_wrapper_encodes_dict_chunks(): """Verify that `bedrock_sse_wrapper` converts dictionary chunks to properly formatted Server-Sent Events and forwards non-dict chunks unchanged.""" - cfg = AmazonAnthropicClaude3MessagesConfig() + cfg = AmazonAnthropicClaudeMessagesConfig() async def _dummy_stream(): # type: ignore[return-type] yield {"type": "message_delta", "text": "hello"}