Przeglądaj źródła

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

Co-authored-by: jyong <jyong@dify.ai>
tags/0.5.9
Jyong 1 rok temu
rodzic
commit
796c5626a7
No account linked to committer's email address
1 zmienionych plików z 4 dodań i 0 usunięć
  1. 4
    0
      api/tasks/clean_dataset_task.py

+ 4
- 0
api/tasks/clean_dataset_task.py Wyświetl plik

@@ -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)


Ładowanie…
Anuluj
Zapisz