This commit is contained in:
Bagatur
2024-09-04 11:28:04 -07:00
parent fdf6fbde18
commit dba308447d
5 changed files with 26 additions and 3 deletions

View File

@@ -102,6 +102,16 @@ def test_serializable_mapping() -> None:
"modifier",
"RemoveMessage",
),
("langchain", "chat_models", "mistralai", "MistralAI"): (
"langchain_mistralai",
"chat_models",
"ChatMistralAI",
),
("langchain_groq", "chat_models", "ChatGroq"): (
"langchain_groq",
"chat_models",
"ChatGroq",
),
}
serializable_modules = import_all_modules("langchain")