From 796c5626a7437ad8d77b1654dcaf401dd3105855 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Mon, 11 Mar 2024 23:57:38 +0800 Subject: [PATCH] fix delete dataset when dataset has no document (#2789) Co-authored-by: jyong --- api/tasks/clean_dataset_task.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/api/tasks/clean_dataset_task.py b/api/tasks/clean_dataset_task.py index 37e109c847..b9737d7ddd 100644 --- a/api/tasks/clean_dataset_task.py +++ b/api/tasks/clean_dataset_task.py @@ -44,6 +44,10 @@ def clean_dataset_task(dataset_id: str, tenant_id: str, indexing_technique: str, documents = db.session.query(Document).filter(Document.dataset_id == dataset_id).all() segments = db.session.query(DocumentSegment).filter(DocumentSegment.dataset_id == dataset_id).all() + if documents is None or len(documents) == 0: + logging.info(click.style('No documents found for dataset: {}'.format(dataset_id), fg='green')) + return + index_processor = IndexProcessorFactory(doc_form).init_index_processor() index_processor.clean(dataset, None)