diff --git a/python/semantic_kernel/connectors/ai/bedrock/services/model_provider/bedrock_model_provider.py b/python/semantic_kernel/connectors/ai/bedrock/services/model_provider/bedrock_model_provider.py index c1275718d69a..d1da79a82b61 100644 --- a/python/semantic_kernel/connectors/ai/bedrock/services/model_provider/bedrock_model_provider.py +++ b/python/semantic_kernel/connectors/ai/bedrock/services/model_provider/bedrock_model_provider.py @@ -37,10 +37,12 @@ class BedrockModelProvider(Enum): @classmethod def to_model_provider(cls, model_id: str) -> "BedrockModelProvider": """Convert a model ID to a model provider.""" - if model_id.startswith("us."): - model_id = model_id.removeprefix("us.") - provider = model_id.split(".")[0] - return cls(provider) + try: + return next(provider for provider in cls if provider.value in model_id) + except StopAsyncIteration: + raise ValueError( + f"Model ID {model_id} does not contain a valid model provider name." + ) # region Text Completion