mongodb[patch]: include LLM caches in toplevel library import (langch…

…ain-ai#18601)
FalkorDB · Mar 30, 2024 · 643fb19 · 643fb19
1 parent b1d8fef
commit 643fb19
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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 @@
 __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
@@ -1,6 +1,11 @@
 from langchain_mongodb import __all__
 
-EXPECTED_ALL = ["MongoDBAtlasVectorSearch", "MongoDBChatMessageHistory"]
+EXPECTED_ALL = [
+    "MongoDBAtlasVectorSearch",
+    "MongoDBChatMessageHistory",
+    "MongoDBCache",
+    "MongoDBAtlasSemanticCache",
+]
 
 
 def test_all_imports() -> None: