github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 17:08:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Fixes unit tests for `ChatBedrockConverse` in
https://github.com/langchain-ai/langchain-aws/pull/227
This commit is contained in:
Piyush Jain 2024-10-04 16:45:59 -07:00 committed by GitHub
parent 6cfd1e846a
commit 807314661d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/tests/unit_tests/load/test_serializable.py
View File

@ -112,6 +112,11 @@ def test_serializable_mapping() -> None:
"chat_models", "chat_models",
"ChatGroq", "ChatGroq",
), ),
("langchain_aws", "chat_models", "ChatBedrockConverse"): (
"langchain_aws",
"chat_models",
"ChatBedrockConverse",
),
} }
serializable_modules = import_all_modules("langchain") serializable_modules = import_all_modules("langchain")

5
libs/core/langchain_core/load/mapping.py
View File

@ -520,6 +520,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = {
"structured", "structured",
"StructuredPrompt", "StructuredPrompt",
), ),
("langchain_aws", "chat_models", "ChatBedrockConverse"): (
"langchain_aws",
"chat_models",
"ChatBedrockConverse",
),
} }
# Needed for backwards compatibility for old versions of LangChain where things # Needed for backwards compatibility for old versions of LangChain where things