From 8d6d7f68879fe67f420a5c06dee9e5574b2d920a Mon Sep 17 00:00:00 2001 From: KevinHuSh Date: Tue, 7 May 2024 20:46:45 +0800 Subject: [PATCH] fix task losting isssue (#665) ### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- api/db/services/llm_service.py | 8 +++++--- api/db/services/task_service.py | 6 +++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/api/db/services/llm_service.py b/api/db/services/llm_service.py index 7f66ca6aa..525851835 100644 --- a/api/db/services/llm_service.py +++ b/api/db/services/llm_service.py @@ -128,9 +128,11 @@ class TenantLLMService(CommonService): else: assert False, "LLM type error" - num = cls.model.update(used_tokens=cls.model.used_tokens + used_tokens)\ - .where(cls.model.tenant_id == tenant_id, cls.model.llm_name == mdlnm)\ - .execute() + num = 0 + for u in cls.query(tenant_id = tenant_id, llm_name=mdlnm): + num += cls.model.update(used_tokens = u.used_tokens + used_tokens)\ + .where(cls.model.tenant_id == tenant_id, cls.model.llm_name == mdlnm)\ + .execute() return num diff --git a/api/db/services/task_service.py b/api/db/services/task_service.py index 12707768b..36b4700f0 100644 --- a/api/db/services/task_service.py +++ b/api/db/services/task_service.py @@ -155,8 +155,8 @@ def queue_tasks(doc, bucket, name): else: tsks.append(new_task()) - for t in tsks: - REDIS_CONN.queue_product(SVR_QUEUE_NAME, message=t) - bulk_insert_into_db(Task, tsks, True) DocumentService.begin2parse(doc["id"]) + + for t in tsks: + REDIS_CONN.queue_product(SVR_QUEUE_NAME, message=t) \ No newline at end of file