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

pull/5066/head^2
Jib 6 months ago committed by GitHub
parent 080904689c
commit 8bc347c5fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -1,3 +1,4 @@
from langchain_mongodb.cache import MongoDBAtlasSemanticCache, MongoDBCache
from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory
from langchain_mongodb.vectorstores import ( from langchain_mongodb.vectorstores import (
MongoDBAtlasVectorSearch, MongoDBAtlasVectorSearch,
@ -6,4 +7,6 @@ from langchain_mongodb.vectorstores import (
__all__ = [ __all__ = [
"MongoDBAtlasVectorSearch", "MongoDBAtlasVectorSearch",
"MongoDBChatMessageHistory", "MongoDBChatMessageHistory",
"MongoDBCache",
"MongoDBAtlasSemanticCache",
] ]

@ -1,6 +1,11 @@
from langchain_mongodb import __all__ from langchain_mongodb import __all__
EXPECTED_ALL = ["MongoDBAtlasVectorSearch", "MongoDBChatMessageHistory"] EXPECTED_ALL = [
"MongoDBAtlasVectorSearch",
"MongoDBChatMessageHistory",
"MongoDBCache",
"MongoDBAtlasSemanticCache",
]
def test_all_imports() -> None: def test_all_imports() -> None:

Loading…
Cancel
Save