fix retrievers tests

Author: blink1073

libs/langchain-mongodb/tests/integration_tests/test_retrievers_standard.py

@@ -31,10 +31,12 @@
 SEARCH_INDEX_NAME = "text_index"
 
 
-def setup_test() -> tuple[Collection, MongoDBAtlasVectorSearch]:
+def get_collection() -> Collection:
     client = MongoClient(CONNECTION_STRING)
-    coll = client[DB_NAME][COLLECTION_NAME]
+    return client[DB_NAME][COLLECTION_NAME]
 
+
+def setup_test(coll: Collection) -> MongoDBAtlasVectorSearch:
     # Set up the vector search index and add the documents if needed.
     vs = PatchedMongoDBAtlasVectorSearch(
         coll,
@@ -72,7 +74,7 @@ class TestMongoDBAtlasFullTextSearchRetriever(RetrieversIntegrationTests):
 
     @classmethod
     def setup_class(cls):
-        cls._coll, _ = setup_test()
+        cls._coll = get_collection()
 
     @classmethod
     def teardown_class(cls):
@@ -85,6 +87,7 @@ def retriever_constructor(self) -> Type[MongoDBAtlasFullTextSearchRetriever]:
 
     @property
     def retriever_constructor_params(self) -> dict:
+        setup_test(self._coll)
         return {
             "collection": self._coll,
             "search_index_name": SEARCH_INDEX_NAME,
@@ -101,11 +104,10 @@ def retriever_query_example(self) -> str:
 
 class TestMongoDBAtlasHybridSearchRetriever(RetrieversIntegrationTests):
     _coll: Collection
-    _vs: MongoDBAtlasVectorSearch
 
     @classmethod
     def setup_class(cls):
-        cls._coll, cls._vs = setup_test()
+        cls._coll = get_collection()
 
     @classmethod
     def teardown_class(cls):
@@ -118,8 +120,9 @@ def retriever_constructor(self) -> Type[MongoDBAtlasHybridSearchRetriever]:
 
     @property
     def retriever_constructor_params(self) -> dict:
+        vs = setup_test(self._coll)
         return {
-            "vectorstore": self._vs,
+            "vectorstore": vs,
             "collection": self._coll,
             "search_index_name": SEARCH_INDEX_NAME,
             "search_field": PAGE_CONTENT_FIELD,