Preventing authorized collections listing issues in other parts of the code.

Author: Rafael Odon

libs/langchain-mongodb/langchain_mongodb/cache.py

@@ -52,7 +52,9 @@ def __init__(
         self.__database_name = database_name
         self.__collection_name = collection_name
 
-        if self.__collection_name not in self.database.list_collection_names():
+        if self.__collection_name not in self.database.list_collection_names(
+            authorizedCollections=True
+        ):
             self.database.create_collection(self.__collection_name)
             # Create an index on key and llm_string
             self.collection.create_index([self.PROMPT, self.LLM])

libs/langchain-mongodb/langchain_mongodb/graphrag/graph.py

@@ -142,7 +142,9 @@ def __init__(
                 driver=DRIVER_METADATA,
             )
             db = client[database_name]
-            if collection_name not in db.list_collection_names():
+            if collection_name not in db.list_collection_names(
+                authorizedCollections=True
+            ):
                 validator = {"$jsonSchema": self._schema} if validate else None
                 collection = client[database_name].create_collection(
                     collection_name,

libs/langchain-mongodb/langchain_mongodb/index.py

@@ -60,7 +60,9 @@ def create_vector_search_index(
     """
     logger.info("Creating Search Index %s on %s", index_name, collection.name)
 
-    if collection.name not in collection.database.list_collection_names():
+    if collection.name not in collection.database.list_collection_names(
+        authorizedCollections=True
+    ):
         collection.database.create_collection(collection.name)
 
     result = collection.create_search_index(
@@ -219,7 +221,9 @@ def create_fulltext_search_index(
     """
     logger.info("Creating Search Index %s on %s", index_name, collection.name)
 
-    if collection.name not in collection.database.list_collection_names():
+    if collection.name not in collection.database.list_collection_names(
+        authorizedCollections=True
+    ):
         collection.database.create_collection(collection.name)
 
     if isinstance(field, str):

libs/langgraph-store-mongodb/langgraph/store/mongodb/base.py

@@ -251,7 +251,9 @@ def from_conn_string(
                 driver=DRIVER_METADATA,
             )
             db = client[db_name]
-            if collection_name not in db.list_collection_names():
+            if collection_name not in db.list_collection_names(
+                authorizedCollections=True
+            ):
                 db.create_collection(collection_name)
             collection = client[db_name][collection_name]