mirror of
https://github.com/hwchase17/langchain.git
synced 2026-01-05 07:55:18 +00:00
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
This commit is contained in:
@@ -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")
|
||||
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user