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

fix: asyncinference KeyError status→.get(status)

Browse Source
This commit is contained in:
yumoqing 2026-07-24 14:34:16 +08:00
parent c099af6bc7
commit e246f91e77
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llmage/asyncinference.py
View File

@ -177,8 +177,8 @@ 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 ['UNKNOWN', 'FAILED', 'SUCCEEDED']: if lastoutout.get('status', '') in ['UNKNOWN', 'FAILED', 'SUCCEEDED']:
critical(f"{lastoutout['status']=}") critical(f"{lastoutout.get('status', '')=}")
return return
ns = {'taskid': taskid} ns = {'taskid': taskid}
new_output = b = d = None new_output = b = d = None
@ -197,7 +197,7 @@ async def query_task_status(request, luid, onetime=False):
e = Exception(f"{new_output=} {upappid=}, {apiname=} has not status field") e = Exception(f"{new_output=} {upappid=}, {apiname=} has not status field")
critical(f'{e}') critical(f'{e}')
raise e raise e
if lastoutout['status'] != new_output.get('status'): if lastoutout.get('status', '') != new_output.get('status'):
llmusage.status = new_output['status'] llmusage.status = new_output['status']
ns = { ns = {
'id': llmusage.id, 'id': llmusage.id,