From da04ff040b06a545cd909b18e47f42262511578d Mon Sep 17 00:00:00 2001 From: John Wang Date: Thu, 29 Jun 2023 13:09:22 +0800 Subject: [PATCH] fix: remove document from dataset error when vector index npe (#489) --- api/tasks/remove_document_from_index_task.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/tasks/remove_document_from_index_task.py b/api/tasks/remove_document_from_index_task.py index 81e8367b53..1635d38bad 100644 --- 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()