estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Feat/delete file when clean document (#5882)

Browse Source
This commit is contained in:
Jyong
2024-07-15 19:57:05 +08:00
committed by GitHub
parent b47fa27a35
commit d320d1468d
4 changed files with 52 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/services/dataset_service.py
View File

@@ -524,7 +524,14 @@ class DocumentService:
@staticmethod
def delete_document(document):
# trigger document_was_deleted signal
document_was_deleted.send(document.id, dataset_id=document.dataset_id, doc_form=document.doc_form)
file_id = None
if document.data_source_type == 'upload_file':
if document.data_source_info:
data_source_info = document.data_source_info_dict
if data_source_info and 'upload_file_id' in data_source_info:
file_id = data_source_info['upload_file_id']
document_was_deleted.send(document.id, dataset_id=document.dataset_id,
doc_form=document.doc_form, file_id=file_id)
db.session.delete(document)
db.session.commit()