From d030b4a680c85435caf8554a10a171e782c93057 Mon Sep 17 00:00:00 2001 From: Jin Hai Date: Mon, 23 Dec 2024 21:04:44 +0800 Subject: [PATCH] Update progress time info (#4193) ### What problem does this PR solve? Ignore the millisecond and microsecond value. ### Type of change - [x] Refactoring Signed-off-by: jinhai --- api/db/services/task_service.py | 2 +- rag/svr/task_executor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py index 2f2b60a29..3991b2291 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -85,7 +85,7 @@ class TaskService(CommonService): if not docs: return None - msg = f"\n{datetime.now().strftime('%H:%M:%S.%f')} Task has been received." + msg = f"\n{datetime.now().strftime('%H:%M:%S')} Task has been received." prog = random.random() / 10.0 if docs[0]["retry_count"] >= 3: msg = "\nERROR: Task is abandoned after 3 times attempts." diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index 41f8f0a2a..c1f7f8de3 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -117,7 +117,7 @@ def set_progress(task_id, from_page=0, to_page=-1, prog=None, msg="Processing... if msg: msg = f"Page({from_page + 1}~{to_page + 1}): " + msg if msg: - msg = datetime.now().strftime("%H:%M:%S.%f") + " " + msg + msg = datetime.now().strftime("%H:%M:%S") + " " + msg d = {"progress_msg": msg} if prog is not None: d["progress"] = prog