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-05-21 13:33:32 +08:00
parent 449e9db1f4
commit 6031efaa67
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
llmage/utils.py
View File

@ -57,21 +57,24 @@ async def tpac_accounting(apikey, userid, llmid, amount, usage):
'amount': amount, 'amount': amount,
'usage': usage 'usage': usage
} }
url = config.thirdparty_accounting_center.get_user_balance_url
hc = StreamHttpClient() hc = StreamHttpClient()
status = 'failed'
try: try:
b = hc.request('POST', url, data=d): b = hc.request('POST', url, data=d):
d = json.loads(b.decode('utf-8')) d = json.loads(b.decode('utf-8'))
if d['status'] == 'ok': if d['status'] == 'ok':
env = ServerEnv() status = 'accounted'
async with get_sor_context(env, 'llmage') as sor:
await sor.U('llmusage', {'id': llmid, 'tpac_accounting_status': 'accounted')
return
exception(f'{apikey=}, {userid=}, {llmid=}, {amount=}, {usage=} tpac accounting error') exception(f'{apikey=}, {userid=}, {llmid=}, {amount=}, {usage=} tpac accounting error')
return
except Exception as e: except Exception as e:
exception(f'{apikey=}, {userid=}, {llmid=}, {amount=}, {usage=} tpac accounting error:{e}') exception(f'{apikey=}, {userid=}, {llmid=}, {amount=}, {usage=} tpac accounting error:{e}')
env = ServerEnv()
async with get_sor_context(env, 'llmage') as sor:
await sor.U('llmusage', {
'id': llmid,
'accounting_status': status
})
return return
exception(f'{apikey=}, {userid=}, {llmid=}, {amount=}, {usage=} tpac_accounting error:update llmusage error')
async def append_new_llmoutput(webpath, output): async def append_new_llmoutput(webpath, output):
fs = FileStorage() fs = FileStorage()