yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-04-24 18:16:11 +08:00
parent f1e5cb88a4
commit 1eea3f69f3
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
llmage/accounting.py
View File

@ -52,7 +52,7 @@ async def checkCustomerBalance(llmid, userorgid):
bal = 0 if balance is None else balance bal = 0 if balance is None else balance
if llm.min_balance is None: if llm.min_balance is None:
llm.min_balance = 0.00 llm.min_balance = 0.00
debug(f'{userorgid=}, {balance=}, {llm.min_balance=}, {llm.ppid=}') # debug(f'{userorgid=}, {balance=}, {llm.min_balance=}, {llm.ppid=}')
ret = llm.ppid and llm.min_balance < bal ret = llm.ppid and llm.min_balance < bal
return ret return ret
debug(f'{userorgid=} checkCustomerBalance() failed') debug(f'{userorgid=} checkCustomerBalance() failed')
@ -149,6 +149,11 @@ async def llm_accounting(llmusage):
} }
await sor.U('llmusage', ns) await sor.U('llmusage', ns)
async def update_llmusage(ns):
env = ServerEnv()
async with get_sor_context(env, 'llmage') as sor:
await sor.U('llmusage', ns)
async def get_accounting_llmusages(luid=None): async def get_accounting_llmusages(luid=None):
env = ServerEnv() env = ServerEnv()
lus = [] lus = []
@ -176,8 +181,7 @@ where a.llmid = b.id
continue continue
r.usages = output.get('usage') r.usages = output.get('usage')
if r.usages is None: if r.usages is None:
llmusage.accounting_status = 'failed' await llm_accoung_failed(r.id)
await sor.U('llmusage', {'id': llmusage.id, 'accounting_status': 'failed'})
debug(f'{r.usages=} is None') debug(f'{r.usages=} is None')
continue continue
d = None d = None
@ -186,7 +190,7 @@ where a.llmid = b.id
r.usages = json.loads(r.usages) r.usages = json.loads(r.usages)
d = await llm_charging(r.ppid, r) d = await llm_charging(r.ppid, r)
except Exception as e: except Exception as e:
await sor.U('llmusage', {'id': r.id, 'accounting_status': 'failed'}) await llm_accoung_failed(r.id)
exception(f'{r.ppid=}, {r.usages=} llm_charging() failed,{e}') exception(f'{r.ppid=}, {r.usages=} llm_charging() failed,{e}')
continue continue
r.amount = d.amount r.amount = d.amount