Linting and default parameter fix on MockDatabase

Author: Rafael Odon

libs/langchain-mongodb/langchain_mongodb/agent_toolkit/database.py

@@ -57,7 +57,9 @@ def __init__(
             )
         self._include_colls = set(include_collections or [])
         self._ignore_colls = set(ignore_collections or [])
-        self._all_colls = set(self._db.list_collection_names(authorizedCollections=True))
+        self._all_colls = set(
+            self._db.list_collection_names(authorizedCollections=True)
+        )
 
         self._sample_docs_in_coll_info = sample_docs_in_collection_info
         self._indexes_in_coll_info = indexes_in_collection_info

libs/langchain-mongodb/tests/utils.py

@@ -238,7 +238,7 @@ class MockDatabase:
     def __init__(self, client=None):
         self.client = client or MockClient()
 
-    def list_collection_names(self, authorizedCollections:bool) -> list[str]:
+    def list_collection_names(self, authorizedCollections: bool = True) -> list[str]:
         return ["test"]
 
     def __getitem__(self, key: str) -> Any: