From f087ab43fdecaf12ac9129d4611bf0e63bfa02f9 Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Thu, 19 Sep 2024 14:57:20 -0700 Subject: [PATCH] core[patch]: Fix load of ChatBedrock (#26679) Complementary PR to master for #26643. Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> --- libs/core/langchain_core/load/mapping.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py index 17e50947f3c..0c67ffd2a44 100644 --- a/libs/core/langchain_core/load/mapping.py +++ b/libs/core/langchain_core/load/mapping.py @@ -1026,4 +1026,9 @@ _JS_SERIALIZABLE_MAPPING: dict[tuple[str, ...], tuple[str, ...]] = { "image", "ImagePromptTemplate", ), + ("langchain", "chat_models", "bedrock", "ChatBedrock"): ( + "langchain_aws", + "chat_models", + "ChatBedrock", + ), }