From 807314661dbd8933d18a0289df05bd381ffbbc4f Mon Sep 17 00:00:00 2001
From: Piyush Jain <piyushjain@duck.com>
Date: Fri, 4 Oct 2024 16:45:59 -0700
Subject: [PATCH] Added mapping to fix CI for #langchain-aws:227. (#27114)

Fixes unit tests for `ChatBedrockConverse` in
https://github.com/langchain-ai/langchain-aws/pull/227
---
 libs/community/tests/unit_tests/load/test_serializable.py | 5 +++++
 libs/core/langchain_core/load/mapping.py                  | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/libs/community/tests/unit_tests/load/test_serializable.py b/libs/community/tests/unit_tests/load/test_serializable.py
index 9b6dcc536d4..ed8245cf961 100644
--- a/libs/community/tests/unit_tests/load/test_serializable.py
+++ b/libs/community/tests/unit_tests/load/test_serializable.py
@@ -112,6 +112,11 @@ def test_serializable_mapping() -> None:
             "chat_models",
             "ChatGroq",
         ),
+        ("langchain_aws", "chat_models", "ChatBedrockConverse"): (
+            "langchain_aws",
+            "chat_models",
+            "ChatBedrockConverse",
+        ),
     }
     serializable_modules = import_all_modules("langchain")
 
diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py
index 5493882aa99..f40944352aa 100644
--- a/libs/core/langchain_core/load/mapping.py
+++ b/libs/core/langchain_core/load/mapping.py
@@ -520,6 +520,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = {
         "structured",
         "StructuredPrompt",
     ),
+    ("langchain_aws", "chat_models", "ChatBedrockConverse"): (
+        "langchain_aws",
+        "chat_models",
+        "ChatBedrockConverse",
+    ),
 }
 
 # Needed for backwards compatibility for old versions of LangChain where things