From e18929f9277b92e0540a1b36976e068b3426363b Mon Sep 17 00:00:00 2001
From: Piyush Jain <piyushjain@duck.com>
Date: Thu, 19 Sep 2024 12:16:28 -0700
Subject: [PATCH] Fix load of bedrock chat models.

---
 libs/core/langchain_core/load/load.py    | 1 +
 libs/core/langchain_core/load/mapping.py | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/libs/core/langchain_core/load/load.py b/libs/core/langchain_core/load/load.py
index ba150a5a0095b..b941553ea284a 100644
--- a/libs/core/langchain_core/load/load.py
+++ b/libs/core/langchain_core/load/load.py
@@ -19,6 +19,7 @@
     "langchain_anthropic",
     "langchain_groq",
     "langchain_google_genai",
+    "langchain_aws",
 ]
 
 ALL_SERIALIZABLE_MAPPINGS = {
diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py
index 17e50947f3ca5..0c67ffd2a4469 100644
--- a/libs/core/langchain_core/load/mapping.py
+++ b/libs/core/langchain_core/load/mapping.py
@@ -1026,4 +1026,9 @@
         "image",
         "ImagePromptTemplate",
     ),
+    ("langchain", "chat_models", "bedrock", "ChatBedrock"): (
+        "langchain_aws",
+        "chat_models",
+        "ChatBedrock",
+    ),
 }