jinhai-2012 commited on
Commit
35dcaff
·
1 Parent(s): 3808212

Update progress time info (#4193)

Browse files

### What problem does this PR solve?

Ignore the millisecond and microsecond value.

### Type of change

- [x] Refactoring

Signed-off-by: jinhai <[email protected]>

api/db/services/task_service.py CHANGED
@@ -85,7 +85,7 @@ class TaskService(CommonService):
85
  if not docs:
86
  return None
87
 
88
- msg = f"\n{datetime.now().strftime('%H:%M:%S.%f')} Task has been received."
89
  prog = random.random() / 10.0
90
  if docs[0]["retry_count"] >= 3:
91
  msg = "\nERROR: Task is abandoned after 3 times attempts."
 
85
  if not docs:
86
  return None
87
 
88
+ msg = f"\n{datetime.now().strftime('%H:%M:%S')} Task has been received."
89
  prog = random.random() / 10.0
90
  if docs[0]["retry_count"] >= 3:
91
  msg = "\nERROR: Task is abandoned after 3 times attempts."
rag/svr/task_executor.py CHANGED
@@ -117,7 +117,7 @@ def set_progress(task_id, from_page=0, to_page=-1, prog=None, msg="Processing...
117
  if msg:
118
  msg = f"Page({from_page + 1}~{to_page + 1}): " + msg
119
  if msg:
120
- msg = datetime.now().strftime("%H:%M:%S.%f") + " " + msg
121
  d = {"progress_msg": msg}
122
  if prog is not None:
123
  d["progress"] = prog
 
117
  if msg:
118
  msg = f"Page({from_page + 1}~{to_page + 1}): " + msg
119
  if msg:
120
+ msg = datetime.now().strftime("%H:%M:%S") + " " + msg
121
  d = {"progress_msg": msg}
122
  if prog is not None:
123
  d["progress"] = prog