diff --git a/libs/standard-tests/langchain_tests/unit_tests/chat_models.py b/libs/standard-tests/langchain_tests/unit_tests/chat_models.py index 7a93f3989c2..a470a9b59d5 100644 --- a/libs/standard-tests/langchain_tests/unit_tests/chat_models.py +++ b/libs/standard-tests/langchain_tests/unit_tests/chat_models.py @@ -717,4 +717,9 @@ class ChatModelUnitTests(ChatModelTests): with mock.patch.dict(os.environ, env_params): ser = dumpd(model) assert ser == snapshot(name="serialized") - assert model.dict() == load(dumpd(model)).dict() + assert ( + model.dict() + == load( + dumpd(model), valid_namespaces=model.get_lc_namespace()[:1] + ).dict() + )