Quellcode durchsuchen

fix delete dataset when dataset has no document (#2789)

Co-authored-by: jyong <jyong@dify.ai>
Jyong vor 1 Jahr
Ursprung
Commit
796c5626a7
1 geänderte Dateien mit 4 neuen und 0 gelöschten Zeilen
  1. 4 0
      api/tasks/clean_dataset_task.py

+ 4 - 0
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)