From e246f91e773479aff2aa67ce986c355411b2c037 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Fri, 24 Jul 2026 14:34:16 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20asyncinference=20KeyError=20status?= =?UTF-8?q?=E2=86=92.get(status)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- llmage/asyncinference.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py index 9e79af1..f494414 100644 --- a/llmage/asyncinference.py +++ b/llmage/asyncinference.py @@ -177,8 +177,8 @@ async def query_task_status(request, luid, onetime=False): for apiname in apinames: while True: lastoutout = await get_lastoutput(llmusage.ioinfo) - if lastoutout['status'] in ['UNKNOWN', 'FAILED', 'SUCCEEDED']: - critical(f"{lastoutout['status']=}") + if lastoutout.get('status', '') in ['UNKNOWN', 'FAILED', 'SUCCEEDED']: + critical(f"{lastoutout.get('status', '')=}") return ns = {'taskid': taskid} 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") critical(f'{e}') raise e - if lastoutout['status'] != new_output.get('status'): + if lastoutout.get('status', '') != new_output.get('status'): llmusage.status = new_output['status'] ns = { 'id': llmusage.id,