vector_search_index

deepset-ai · Feb 15, 2024 · 2d3a6f1 · 2d3a6f1
1 parent 26f34b5
commit 2d3a6f1
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/integrations/mongodb_atlas/pyproject.toml b/integrations/mongodb_atlas/pyproject.toml
@@ -25,11 +25,7 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
 dependencies = [
-<<<<<<< HEAD
-  "haystack-ai>=2.0.0.b6",
-=======
   "haystack-ai>=2.0.0b6",
->>>>>>> main
   "pymongo[srv]",
 ]
 

diff --git a/integrations/mongodb_atlas/tests/test_document_store.py b/integrations/mongodb_atlas/tests/test_document_store.py
@@ -20,9 +20,10 @@ def document_store(request):
     store = MongoDBAtlasDocumentStore(
         database_name="haystack_integration_test",
         collection_name=request.node.name + str(uuid4()),
+        vector_search_index="vector_search_index",
+        recreate_collection=True,
     )
-    yield store
-    store.collection.drop()
+    return store
 
 
 @pytest.mark.skipif(
@@ -56,6 +57,7 @@ def test_to_dict(self, _):
         document_store = MongoDBAtlasDocumentStore(
             database_name="database_name",
             collection_name="collection_name",
+            vector_search_index="vector_search_index",
         )
         assert document_store.to_dict() == {
             "type": "haystack_integrations.document_stores.mongodb_atlas.document_store.MongoDBAtlasDocumentStore",

diff --git a/integrations/mongodb_atlas/tests/test_embedding_retrieval.py b/integrations/mongodb_atlas/tests/test_embedding_retrieval.py
@@ -1,7 +1,6 @@
 # SPDX-FileCopyrightText: 2023-present deepset GmbH <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
-
 from typing import List
 from uuid import uuid4
 import os