fix test

Author: Gaudy Blanco

tests/integration/vector_stores/test_factory.py

@@ -131,7 +131,10 @@ def __init__(self, **kwargs):
         def connect(self, **kwargs):
             pass
 
-        def load_documents(self, documents, overwrite=True):
+        def create_index(self, **kwargs):
+            pass
+
+        def load_documents(self, documents):
             pass
 
         def similarity_search_by_vector(self, query_embedding, k=10, **kwargs):

tests/integration/vector_stores/test_lancedb.py

@@ -64,6 +64,7 @@ def test_vector_store_operations(self, sample_documents):
                 )
             )
             vector_store.connect(db_uri=temp_dir)
+            vector_store.create_index()
             vector_store.load_documents(sample_documents[:2])
 
             if vector_store.index_name:
@@ -83,6 +84,7 @@ def test_vector_store_operations(self, sample_documents):
             assert isinstance(results[0].score, float)
 
             # Test append mode
+            vector_store.create_index()
             vector_store.load_documents([sample_documents[2]])
             result = vector_store.search_by_id("3")
             assert result.id == "3"
@@ -121,6 +123,7 @@ def test_empty_collection(self):
                 id="tmp",
                 vector=[0.1, 0.2, 0.3, 0.4, 0.5],
             )
+            vector_store.create_index()
             vector_store.load_documents([sample_doc])
             vector_store.db_connection.open_table(
                 vector_store.index_name if vector_store.index_name else ""
@@ -137,6 +140,7 @@ def test_empty_collection(self):
                 id="1",
                 vector=[0.1, 0.2, 0.3, 0.4, 0.5],
             )
+            vector_store.create_index()
             vector_store.load_documents([doc])
 
             result = vector_store.search_by_id("1")
@@ -157,7 +161,7 @@ def test_filter_search(self, sample_documents_categories):
             )
 
             vector_store.connect(db_uri=temp_dir)
-
+            vector_store.create_index()
             vector_store.load_documents(sample_documents_categories)
 
             # Filter to include only documents about animals
@@ -186,6 +190,7 @@ def test_vector_store_customization(self, sample_documents):
                 ),
             )
             vector_store.connect(db_uri=temp_dir)
+            vector_store.create_index()
             vector_store.load_documents(sample_documents[:2])
 
             if vector_store.index_name:
@@ -205,6 +210,7 @@ def test_vector_store_customization(self, sample_documents):
             assert isinstance(results[0].score, float)
 
             # Test append mode
+            vector_store.create_index()
             vector_store.load_documents([sample_documents[2]])
             result = vector_store.search_by_id("3")
             assert result.id == "3"