dify/api/events/event_handlers/clean_when_document_deleted.py

12 lines
390 B
Python

from events.document_event import document_was_deleted
from tasks.clean_document_task import clean_document_task
@document_was_deleted.connect
def handle(sender, **kwargs):
document_id = sender
dataset_id = kwargs.get('dataset_id')
doc_form = kwargs.get('doc_form')
file_id = kwargs.get('file_id')
clean_document_task.delay(document_id, dataset_id, doc_form, file_id)