core[patch]: Fixed bedrock chat model load. (#26643)

Fixes `load` for `ChatBedrock`, in-turn fixes unit test errors for `standard-tests` in `langchain-aws`, when working with v0.2 branch. ``` FAILED tests/unit_tests/test_standard.py::TestBedrockStandard::test_serdes - ValueError: Trying to deserialize something that cannot be deserialized in current version of langchain-core: ('langchain', 'cha... FAILED tests/unit_tests/test_standard.py::TestBedrockAsConverseStandard::test_serdes - ValueError: Trying to deserialize something that cannot be deserialized in current version of langchain-core: ('langchain', 'cha... ```
langchain-ai · Sep 19, 2024 · c5eca37 · c5eca37
1 parent 70c992b
commit c5eca37
Show file tree

Hide file tree

Showing 4 changed files with 333 additions and 374 deletions.
diff --git a/libs/core/langchain_core/load/load.py b/libs/core/langchain_core/load/load.py
@@ -18,6 +18,7 @@
     "langchain_community",
     "langchain_anthropic",
     "langchain_groq",
+    "langchain_aws",
 ]
 
 ALL_SERIALIZABLE_MAPPINGS = {

diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py
@@ -1017,4 +1017,9 @@
         "image",
         "ImagePromptTemplate",
     ),
+    ("langchain", "chat_models", "bedrock", "ChatBedrock"): (
+        "langchain_aws",
+        "chat_models",
+        "ChatBedrock",
+    ),
 }