From 35a2beb1951c52d7e073a140d99bff21837c8153 Mon Sep 17 00:00:00 2001 From: Jyong <76649700+JohnJyong@users.noreply.github.com> Date: Tue, 6 Jun 2023 23:16:51 +0800 Subject: [PATCH] delete segment not commit (#309) --- api/tasks/clean_document_task.py | 2 +- api/tasks/document_indexing_update_task.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/tasks/clean_document_task.py b/api/tasks/clean_document_task.py index 63d66e4ea6..2d8a4fa75e 100644 --- a/api/tasks/clean_document_task.py +++ b/api/tasks/clean_document_task.py @@ -43,7 +43,7 @@ def clean_document_task(document_id: str, dataset_id: str): for segment in segments: db.session.delete(segment) - + db.session.commit() end_at = time.perf_counter() logging.info( click.style('Cleaned document when document deleted: {} latency: {}'.format(document_id, end_at - start_at), fg='green')) diff --git a/api/tasks/document_indexing_update_task.py b/api/tasks/document_indexing_update_task.py index 493c05505a..2aa98fb7b7 100644 --- a/api/tasks/document_indexing_update_task.py +++ b/api/tasks/document_indexing_update_task.py @@ -59,7 +59,7 @@ def document_indexing_update_task(dataset_id: str, document_id: str): for segment in segments: db.session.delete(segment) - + db.session.commit() end_at = time.perf_counter() logging.info( click.style('Cleaned document when document update data source or process rule: {} latency: {}'.format(document_id, end_at - start_at), fg='green'))