This commit is contained in:
yumoqing 2026-01-06 17:40:07 +08:00
parent 190fb2e463
commit 95eba74491
2 changed files with 6 additions and 5 deletions

View File

@ -15,15 +15,16 @@ async def checkCustomerBalance(llmid, userorgid):
e = Exception(f'llm({llmid}) not exists') e = Exception(f'llm({llmid}) not exists')
exception(f'{e}') exception(f'{e}')
raise e raise e
llm = llms[0] llm = llms[0].copy()
if llm.ownerid == userorgid: if llm.ownerid == userorgid:
return True return True
balance = await getCustomerBalance(sor, userorgid) balance = await getCustomerBalance(sor, userorgid)
if balance is None: bal = 0 if balance is None else balance
balance = 0.00
if llm.min_balance is None: if llm.min_balance is None:
llm.min_balance = 0.00 llm.min_balance = 0.00
return llm.ppid and llm.min_balance < balance ret = llm.ppid and llm.min_balance < bal
debug(f'{llms=},{userorgid=},{balance=},{ret=}')
return ret
return False return False
async def llm_accounting(request, llmid, async def llm_accounting(request, llmid,

View File

@ -1,5 +1,5 @@
{% if get_user() %} {% if get_user() %}
{% set userorgid = get_userorgid %} {% set userorgid = get_userorgid() %}
{% if checkCustomerBalance(params_kw.id, userorgid) %} {% if checkCustomerBalance(params_kw.id, userorgid) %}
{% set llm = get_llm(params_kw.id) %} {% set llm = get_llm(params_kw.id) %}
{% set kdbs = get_user_kdbs(request) %} {% set kdbs = get_user_kdbs(request) %}