diff --git a/api/db/services/llm_service.py b/api/db/services/llm_service.py index 7f66ca6a..52585183 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 12707768..36b4700f 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