dify

dify/api/events/event_handlers/clean_when_dataset_deleted.py

@@ -0,0 +1,18 @@
+from events.dataset_event import dataset_was_deleted
+from models import Dataset
+from tasks.clean_dataset_task import clean_dataset_task
+
+
+@dataset_was_deleted.connect
+def handle(sender: Dataset, **kwargs):
+    dataset = sender
+    if not dataset.doc_form or not dataset.indexing_technique:
+        return
+    clean_dataset_task.delay(
+        dataset.id,
+        dataset.tenant_id,
+        dataset.indexing_technique,
+        dataset.index_struct,
+        dataset.collection_binding_id,
+        dataset.doc_form,
+    )