diff --git a/libs/langchain/langchain/chat_models/base.py b/libs/langchain/langchain/chat_models/base.py index b838576e6ca..9a3f8b7151a 100644 --- a/libs/langchain/langchain/chat_models/base.py +++ b/libs/langchain/langchain/chat_models/base.py @@ -366,6 +366,11 @@ def _init_chat_model_helper( # TODO: update to use model= once ChatBedrock supports return ChatBedrock(model_id=model, **kwargs) + elif model_provider == "bedrock_converse": + _check_pkg("langchain_aws") + from langchain_aws import ChatBedrockConverse + + return ChatBedrockConverse(model=model, **kwargs) else: supported = ", ".join(_SUPPORTED_PROVIDERS) raise ValueError( @@ -388,6 +393,7 @@ _SUPPORTED_PROVIDERS = { "huggingface", "groq", "bedrock", + "bedrock_converse", }