diff --git a/tests/unit/llms/bedrock/chat/test_bedrock_converse_handler.py b/tests/unit/llms/bedrock/chat/test_bedrock_converse_handler.py index 5a75b8d384b..538ac5b55d0 100644 --- a/tests/unit/llms/bedrock/chat/test_bedrock_converse_handler.py +++ b/tests/unit/llms/bedrock/chat/test_bedrock_converse_handler.py @@ -1,5 +1,3 @@ -import litellm - """Tests for `BedrockConverseLLM.completion`. AWS credential resolution is stubbed so nothing reaches STS. @@ -18,6 +16,7 @@ import pytest from botocore.credentials import Credentials from botocore.exceptions import ClientError +import litellm from litellm.llms.bedrock.chat.converse_handler import BedrockConverseLLM from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler from litellm.rust_bridge import configuration