Explorar o código

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

John Wang %!s(int64=2) %!d(string=hai) anos
pai
achega
da04ff040b
Modificáronse 1 ficheiros con 2 adicións e 1 borrados
  1. 2 1
      api/tasks/remove_document_from_index_task.py

+ 2 - 1
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()