fix image files not deleted on indexing_estimate #9541 (#10798)

Co-authored-by: root <root@localhost.localdomain>
2024-11-21 13:03:16 +08:00
parent 2ae6460f46
commit 94c9cadbd8
3 changed files with 16 additions and 0 deletions
--- a/api/tasks/clean_dataset_task.py
+++ b/api/tasks/clean_dataset_task.py
@@ -78,6 +78,7 @@ def clean_dataset_task(
                            "Delete image_files failed when storage deleted, \
                                          image_upload_file_is: {}".format(upload_file_id)
                        )
+                    db.session.delete(image_file)
                db.session.delete(segment)

        db.session.query(DatasetProcessRule).filter(DatasetProcessRule.dataset_id == dataset_id).delete()