This commit is contained in:
Bagatur 2024-09-04 11:44:59 -07:00
parent 4ba14ae9e5
commit 1241a004cb

View File

@ -73,7 +73,6 @@ def test_import_all_modules() -> None:
("langchain", "chat_models", "google_palm", "ChatGooglePalm"),
("langchain", "chat_models", "openai", "ChatOpenAI"),
("langchain", "chat_models", "vertexai", "ChatVertexAI"),
("langchain", "chat_models", "mistralai", "ChatMistralAI"),
]
)
@ -103,7 +102,7 @@ def test_serializable_mapping() -> None:
"modifier",
"RemoveMessage",
),
("langchain", "chat_models", "mistralai", "MistralAI"): (
("langchain", "chat_models", "mistralai", "ChatMistralAI"): (
"langchain_mistralai",
"chat_models",
"ChatMistralAI",