From 86507af770cae6b63fa0c9eab139461aa697ba5f Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Fri, 13 Dec 2024 17:15:08 +0800 Subject: [PATCH] Set task progress on exception (#4025) ### What problem does this PR solve? Set task progress on exception ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/svr/task_executor.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index da62b7245cb..aa951b74f15 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -502,12 +502,19 @@ def handle_task(): with mt_lock: DONE_TASKS += 1 CURRENT_TASK = None - logging.info(f"handle_task got TaskCanceledException for task {json.dumps(task)}") + try: + set_progress(task["id"], prog=-1, msg="handle_task got TaskCanceledException") + except Exception: + pass logging.debug("handle_task got TaskCanceledException", exc_info=True) except Exception: with mt_lock: FAILED_TASKS += 1 CURRENT_TASK = None + try: + set_progress(task["id"], prog=-1, msg="handle_task got exception, please check log") + except Exception: + pass logging.exception(f"handle_task got exception for task {json.dumps(task)}") if PAYLOAD: PAYLOAD.ack()