diff --git a/api/services/vector_service.py b/api/services/vector_service.py index 49c3b858311..5b5088ec5a1 100644 --- a/api/services/vector_service.py +++ b/api/services/vector_service.py @@ -333,7 +333,7 @@ class VectorService: # Add documents to vector store if any if documents and dataset.is_multimodal: - vector.add_texts(documents, duplicate_check=True) + vector.create_multimodal(documents) # Single commit for all operations db.session.commit() diff --git a/api/tests/unit_tests/services/test_vector_service.py b/api/tests/unit_tests/services/test_vector_service.py index a5873870103..e7ebada6bea 100644 --- a/api/tests/unit_tests/services/test_vector_service.py +++ b/api/tests/unit_tests/services/test_vector_service.py @@ -639,8 +639,8 @@ def test_update_multimodel_vector_adds_bindings_and_vectors_and_skips_missing_up assert len(bindings) == 1 assert bindings[0]["attachment_id"] == "file-1" - vector_instance.add_texts.assert_called_once() - documents = vector_instance.add_texts.call_args.args[0] + vector_instance.create_multimodal.assert_called_once() + documents = vector_instance.create_multimodal.call_args.args[0] assert len(documents) == 1 assert documents[0].page_content == "img.png" assert documents[0].metadata["doc_id"] == "file-1"