diff --git a/libs/partners/mongodb/langchain_mongodb/__init__.py b/libs/partners/mongodb/langchain_mongodb/__init__.py index 53678092c4..ec25697851 100644 --- a/libs/partners/mongodb/langchain_mongodb/__init__.py +++ b/libs/partners/mongodb/langchain_mongodb/__init__.py @@ -1,3 +1,4 @@ +from langchain_mongodb.cache import MongoDBAtlasSemanticCache, MongoDBCache from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory from langchain_mongodb.vectorstores import ( MongoDBAtlasVectorSearch, @@ -6,4 +7,6 @@ from langchain_mongodb.vectorstores import ( __all__ = [ "MongoDBAtlasVectorSearch", "MongoDBChatMessageHistory", + "MongoDBCache", + "MongoDBAtlasSemanticCache", ] diff --git a/libs/partners/mongodb/tests/unit_tests/test_imports.py b/libs/partners/mongodb/tests/unit_tests/test_imports.py index 84244d404d..e0bb498a7d 100644 --- a/libs/partners/mongodb/tests/unit_tests/test_imports.py +++ b/libs/partners/mongodb/tests/unit_tests/test_imports.py @@ -1,6 +1,11 @@ from langchain_mongodb import __all__ -EXPECTED_ALL = ["MongoDBAtlasVectorSearch", "MongoDBChatMessageHistory"] +EXPECTED_ALL = [ + "MongoDBAtlasVectorSearch", + "MongoDBChatMessageHistory", + "MongoDBCache", + "MongoDBAtlasSemanticCache", +] def test_all_imports() -> None: