Unverified Commit da04ff04 authored by John Wang's avatar John Wang Committed by GitHub

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

parent b9b0866a
...@@ -41,7 +41,8 @@ def remove_document_from_index_task(document_id: str): ...@@ -41,7 +41,8 @@ def remove_document_from_index_task(document_id: str):
kw_index = IndexBuilder.get_index(dataset, 'economy') kw_index = IndexBuilder.get_index(dataset, 'economy')
# delete from vector index # 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 # delete from keyword index
segments = db.session.query(DocumentSegment).filter(DocumentSegment.document_id == document.id).all() segments = db.session.query(DocumentSegment).filter(DocumentSegment.document_id == document.id).all()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment