From 8bc347c5fcccb3f22c39aefde50714bb14ed172a Mon Sep 17 00:00:00 2001
From: Jib <jib.adegunloye@mongodb.com>
Date: Tue, 5 Mar 2024 19:35:13 -0500
Subject: [PATCH] mongodb[patch]: include LLM caches in toplevel library import
 (#18601)

---
 libs/partners/mongodb/langchain_mongodb/__init__.py    | 3 +++
 libs/partners/mongodb/tests/unit_tests/test_imports.py | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

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: