From 9dd73339ceb6883c3940ba1e360b70609f88ccca Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Mon, 30 Mar 2026 15:32:59 +0800
Subject: [PATCH] bugfix

---
 llmage/asyncinference.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py
index b620b40..1c9d0d3 100644
--- a/llmage/asyncinference.py
+++ b/llmage/asyncinference.py
@@ -205,8 +205,12 @@ async def query_task_status(request, upappid, apiname, luid, userid, taskid):
 						changed.cost = 0
 				await add_new_llmusage_output(luid, changed)
 				if llmusage.accounting_status != 'accounted' and changed.amount > 0.00001:
-					await llm_accounting(request, llmusage)
-					debug(f'{changed=} accounted ')
+					llmusage.amount = changed.amount
+					llmusage.cost   = changed.cost
+					try:
+						await llm_accounting(request, llmusage)
+					except Exception as e:
+						debug(f'{changed=} accounting failed,{e=} ')
 				if changed.status in ['FAILED', 'SUCCEEDED']:
 					return
 				await asyncio.sleep(llm.query_period or 30)