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-14 15:19:13 +08:00
parent 034e7a072f
commit 4c827cd5d0
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
llmage/asyncinference.py
View File

@ -38,7 +38,7 @@ async def get_asynctask_status(request, taskid):
output = await get_lastoutput(r.ioinfo) output = await get_lastoutput(r.ioinfo)
t = timestampAdd(r.use_time, 600) t = timestampAdd(r.use_time, 600)
now = time.time() now = time.time()
if r.status not in ['FAILED', 'SUCCEEDED'] and now > t: if r.status not in ['UNKNOWN', 'FAILED', 'SUCCEEDED'] and now > t:
asyncio.create_task(query_task_status(request, r.id)) asyncio.create_task(query_task_status(request, r.id))
return output return output
return { return {
@ -135,6 +135,8 @@ async def get_llm_llmusage(luid):
exception(f'{e}') exception(f'{e}')
raise e raise e
llmusage = recs[0] llmusage = recs[0]
if llmusage.status == 'UNKNOWN':
return
if llmusage.status == 'SUCCEEDED': if llmusage.status == 'SUCCEEDED':
return return
if llmusage.status == 'FAILED': if llmusage.status == 'FAILED':
@ -159,7 +161,7 @@ async def query_task_status(request, luid, onetime=False):
for apiname in apinames: for apiname in apinames:
while True: while True:
lastoutout = await get_lastoutput(llmusage.ioinfo) lastoutout = await get_lastoutput(llmusage.ioinfo)
if lastoutout['status'] in ['FAILED', 'SUCCEEDED']: if lastoutout['status'] in ['UNKNOWN', 'FAILED', 'SUCCEEDED']:
critical(f"{lastoutout['status']=}") critical(f"{lastoutout['status']=}")
return return
ns = {'taskid': taskid} ns = {'taskid': taskid}
@ -189,7 +191,7 @@ async def query_task_status(request, luid, onetime=False):
ns['usages'] = json.dumps(new_output['usage']) ns['usages'] = json.dumps(new_output['usage'])
await append_new_llmoutput(llmusage.ioinfo, new_output) await append_new_llmoutput(llmusage.ioinfo, new_output)
await modify_llmusage(ns) await modify_llmusage(ns)
if llmusage.status in ['FAILED', 'SUCCEEDED']: if llmusage.status in ['UNKNOWN', 'FAILED', 'SUCCEEDED']:
critical(f'finished .. {llmusage.status=}') critical(f'finished .. {llmusage.status=}')
return return