mongodb[patch]: include LLM caches in toplevel library import (#18601)

4 months ago · 8bc347c5fc
parent 080904689c
commit 8bc347c5fc
2 changed files with 9 additions and 1 deletions
--- 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",
 ]
--- 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: