Added mapping to fix CI for #langchain-aws:227. (#27114)

Fixes unit tests for `ChatBedrockConverse` in langchain-ai/langchain-aws#227
langchain-ai · Oct 4, 2024 · 8073146 · 8073146
1 parent 6cfd1e8
commit 8073146
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git 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
@@ -520,6 +520,11 @@
         "structured",
         "StructuredPrompt",
     ),
+    ("langchain_aws", "chat_models", "ChatBedrockConverse"): (
+        "langchain_aws",
+        "chat_models",
+        "ChatBedrockConverse",
+    ),
 }
 
 # Needed for backwards compatibility for old versions of LangChain where things