async mongo document loader

langchain/document_loaders/__init__.py

@@ -49,6 +49,7 @@
 from langchain.document_loaders.markdown import UnstructuredMarkdownLoader
 from langchain.document_loaders.mediawikidump import MWDumpLoader
 from langchain.document_loaders.modern_treasury import ModernTreasuryLoader
+from langchain.document_loaders.mongodb import MongodbLoader
 from langchain.document_loaders.notebook import NotebookLoader
 from langchain.document_loaders.notion import NotionDirectoryLoader
 from langchain.document_loaders.notiondb import NotionDBLoader
@@ -150,6 +151,7 @@
     "MWDumpLoader",
     "MathpixPDFLoader",
     "ModernTreasuryLoader",
+    "MongodbLoader",
     "NotebookLoader",
     "NotionDBLoader",
     "NotionDirectoryLoader",

tests/integration_tests/document_loaders/test_mongodb.py

@@ -1,17 +1,47 @@
+import os
+
+import pytest
+
 from langchain.document_loaders.mongodb import MongodbLoader
 
+if "MONGODB_API_KEY" in os.environ:
+    mongo_api_key_set = True
+    mongo_api_key = os.environ["MONGODB_API_KEY"]
+else:
+    mongo_api_key_set = False
 
-def test_load() -> None:
-    connection_string = "mongodb://localhost:27017/"
-    db_name = "sample_restaurants"
-    collection_name = "restaurants"
+
+@pytest.fixture
+def mongodb_loader():
+    connection_string = "mongodb://localhost:27017"
+    db_name = "test_db"
+    collection_name = "test_collection"
 
     loader = MongodbLoader(connection_string, db_name, collection_name)
-
-    result = loader.load()
+    return loader
+
+
+def test_mongodb_loader_init(mongodb_loader):
+    assert isinstance(mongodb_loader, MongodbLoader)
+
+
+def test_db_and_collection_names(mongodb_loader):
+    assert mongodb_loader.db_name != ""
+    assert mongodb_loader.collection_name != ""
+
+
+@pytest.mark.skipif(not mongo_api_key_set, reason="MONGODB_API_KEY not provided.")
+def test_load() -> None:
+    if "MONGODB_API_KEY" in os.environ:
+        mongo_api_key = os.environ["MONGODB_API_KEY"]
+        db_name = os.environ["MONGODB_DB_NAME"]
+        collection_name = os.environ["MONGODB_COLLECTION_NAME"]
+
+        loader = MongodbLoader(mongo_api_key, db_name, collection_name)
+
+        result = loader.load()
 
-    print(result[0])
-    print(len(result))
-
-    assert len(result) > 0
+        print(result[0])
+        print(len(result))
 
+        assert len(result) > 0