浏览代码

Fix issue : don't delete DatasetProcessRule, DatasetQuery and AppDatasetJoin when delete dataset with no document (#3354)

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

+ 8
- 8
api/tasks/clean_dataset_task.py 查看文件

@@ -46,16 +46,16 @@ def clean_dataset_task(dataset_id: str, tenant_id: str, indexing_technique: str,

if documents is None or len(documents) == 0:
logging.info(click.style('No documents found for dataset: {}'.format(dataset_id), fg='green'))
return
else:
logging.info(click.style('Cleaning documents for dataset: {}'.format(dataset_id), fg='green'))
index_processor = IndexProcessorFactory(doc_form).init_index_processor()
index_processor.clean(dataset, None)

index_processor = IndexProcessorFactory(doc_form).init_index_processor()
index_processor.clean(dataset, None)
for document in documents:
db.session.delete(document)

for document in documents:
db.session.delete(document)

for segment in segments:
db.session.delete(segment)
for segment in segments:
db.session.delete(segment)

db.session.query(DatasetProcessRule).filter(DatasetProcessRule.dataset_id == dataset_id).delete()
db.session.query(DatasetQuery).filter(DatasetQuery.dataset_id == dataset_id).delete()

正在加载...
取消
保存