bugfix

2026-04-02 15:34:32 +08:00 · 2026-04-02 15:34:32 +08:00 · dfd16547f7
commit dfd16547f7
parent 3f3ddc69b2
1 changed files with 1 additions and 1 deletions
--- a/llmage/asyncinference.py
+++ b/llmage/asyncinference.py
@ -25,7 +25,7 @@ async def grab_task_status(request, taskid):
 			exception(f'{taskid=} not found in llmusage')
 			return {"status": "FAILED", "error": f"{taskid} not exist"}
 		r = recs[0]
-		if r.status == 'SUCCEEDED':
+		if r.status in [ 'SUCCEEDED', 'FAILED' ]:
 			io = json.loads(r.ioinfo)
 			return io['output'][-1]
 		llmusage = r