浏览代码

update empty document caused delete exist collection (#6392)

tags/0.6.15
Jyong 1年前
父节点
当前提交
443e96777b
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 7 次插入6 次删除
  1. 7
    6
      api/tasks/document_indexing_update_task.py

+ 7
- 6
api/tasks/document_indexing_update_task.py 查看文件

@@ -46,14 +46,15 @@ def document_indexing_update_task(dataset_id: str, document_id: str):
index_processor = IndexProcessorFactory(index_type).init_index_processor()

segments = db.session.query(DocumentSegment).filter(DocumentSegment.document_id == document_id).all()
index_node_ids = [segment.index_node_id for segment in segments]
if segments:
index_node_ids = [segment.index_node_id for segment in segments]

# delete from vector index
index_processor.clean(dataset, index_node_ids)
# delete from vector index
index_processor.clean(dataset, index_node_ids)

for segment in segments:
db.session.delete(segment)
db.session.commit()
for segment in segments:
db.session.delete(segment)
db.session.commit()
end_at = time.perf_counter()
logging.info(
click.style('Cleaned document when document update data source or process rule: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))

正在加载...
取消
保存