diff --git a/litellm/utils.py b/litellm/utils.py index d5d4e61fff..4f990bb5f4 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -290,7 +290,7 @@ if TYPE_CHECKING: from litellm.llms.base_llm.ocr.transformation import BaseOCRConfig from litellm.llms.base_llm.search.transformation import BaseSearchConfig from litellm.llms.base_llm.text_to_speech.transformation import BaseTextToSpeechConfig - from litellm.llms.bedrock.bedrock_model_info import BedrockModelInfo + from litellm.llms.bedrock.common_utils import BedrockModelInfo from litellm.llms.cohere.common_utils import CohereModelInfo from litellm.llms.mistral.ocr.transformation import MistralOCRConfig # Type stubs for lazy-loaded functions and classes @@ -4953,7 +4953,7 @@ def _get_base_bedrock_model(model_name) -> str: Handle model names like - "us.meta.llama3-2-11b-instruct-v1:0" -> "meta.llama3-2-11b-instruct-v1" AND "meta.llama3-2-11b-instruct-v1:0" -> "meta.llama3-2-11b-instruct-v1" """ - from litellm.llms.bedrock.bedrock_model_info import BedrockModelInfo + from litellm.llms.bedrock.common_utils import BedrockModelInfo return BedrockModelInfo.get_base_model(model_name) @@ -7778,7 +7778,7 @@ class ProviderConfigManager: # The 'BEDROCK' provider corresponds to Amazon's implementation of Anthropic Claude v3. # This mapping ensures that the correct configuration is returned for BEDROCK. elif litellm.LlmProviders.BEDROCK == provider: - from litellm.llms.bedrock.bedrock_model_info import BedrockModelInfo + from litellm.llms.bedrock.common_utils import BedrockModelInfo return BedrockModelInfo.get_bedrock_provider_config_for_messages_api(model) elif litellm.LlmProviders.VERTEX_AI == provider: