github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-23 15:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain[patch]: init_chat_model support ChatBedrockConverse (#24564)

Browse Source
This commit is contained in:
Bagatur 2024-07-23 11:07:38 -07:00 committed by GitHub
parent 9b7db08184
commit 55e66aa40c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/chat_models/base.py
View File

@ -366,6 +366,11 @@ def _init_chat_model_helper(
# TODO: update to use model= once ChatBedrock supports # TODO: update to use model= once ChatBedrock supports
return ChatBedrock(model_id=model, **kwargs) 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: else:
supported = ", ".join(_SUPPORTED_PROVIDERS) supported = ", ".join(_SUPPORTED_PROVIDERS)
raise ValueError( raise ValueError(
@ -388,6 +393,7 @@ _SUPPORTED_PROVIDERS = {
"huggingface", "huggingface",
"groq", "groq",
"bedrock", "bedrock",
"bedrock_converse",
} }