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

feat: check ppid has today pricing data before allowing inference

Browse Source
This commit is contained in:
Hermes Agent 2026-06-22 17:55:29 +08:00
parent 6a3cace0e8
commit 1357eabce7
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
llmage/accounting.py
View File

@ -57,8 +57,16 @@ async def checkCustomerBalance(llmid, userid, userorgid, catelogid=None):
bal = 0 if balance is None else balance
if llm.min_balance is None:
llm.min_balance = 0.00
ret = llm.ppid and llm.min_balance < bal
debug(f'{llm.ppid=}, {llm.min_balance=}, {bal=}')
if not llm.ppid:
debug(f'{llm.ppid=} is empty, model unavailable')
return False
try:
await env.get_ppid_pricing(llm.ppid)
except Exception as e:
debug(f'{llm.ppid=} has no pricing data for today: {e}')
return False
ret = llm.min_balance < bal
debug(f'{llm.ppid=}, {llm.min_balance=}, {bal=}, pricing OK')
return ret
async def llm_accounting(llmusage):