Просмотр исходного кода

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

tags/0.3.7
John Wang 2 лет назад
Родитель
Сommit
da04ff040b
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 2 добавлений и 1 удалений
  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()

Загрузка…
Отмена
Сохранить