From d1ddbb8976d8fce039ea2bf1f4b5a1285f58dacd Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Sun, 5 Apr 2026 16:32:24 +0800
Subject: [PATCH] bugfix

---
 llmage/accounting.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/llmage/accounting.py b/llmage/accounting.py
index 89fcc2a..8bb73f7 100644
--- a/llmage/accounting.py
+++ b/llmage/accounting.py
@@ -172,19 +172,22 @@ where a.llmid = b.id
 				if len(io['output']) == 0:
 					llmusage.accounting_status = 'failed'
 					await sor.U('llmusage', {'id': llmusage.id, 'accounting_status': 'failed'})
+					debug(f'{len(io['output'])} is 0')
 					continue
 				r.usages = io['output'][-1]['usage']
 			if r.usages is None:
 				llmusage.accounting_status = 'failed'
 				await sor.U('llmusage', {'id': llmusage.id, 'accounting_status': 'failed'})
+				debug(f'{r.usages=} is None')
 				continue
 			try:
-				r = await llm_charging(sor, llmusage.ppid, llmusage)
+				d = await llm_charging(sor, r.ppid, r)
 			except Exception as e:
+				debug(f'{r.usages=} is None')
 				continue
-			llmusage.amount = r.amount
-			llmusage.cost = r.cost
-			await sor.U('llmusage', llmusage.copy())
+			r.amount = d.amount
+			r.cost = d.cost
+			await sor.U('llmusage', r.copy())
 			lus.append(r)
 	return lus