fix: remove document from dataset error when vector index npe (#489)

3 years ago · da04ff040b
parent b9b0866a46
commit da04ff040b
1 changed files with 2 additions and 1 deletions
--- a/api/tasks/remove_document_from_index_task.py
+++ b/api/tasks/remove_document_from_index_task.py
@ -41,7 +41,8 @@ def remove_document_from_index_task(document_id: str):
        kw_index = IndexBuilder.get_index(dataset, 'economy')

        # delete from vector index
-        vector_index.delete_by_document_id(document.id)
+        if vector_index:
+            vector_index.delete_by_document_id(document.id)

        # delete from keyword index
        segments = db.session.query(DocumentSegment).filter(DocumentSegment.document_id == document.id).all()