From dfd16547f79e850014e85996c1a2dd53e9f20a4c Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Thu, 2 Apr 2026 15:34:32 +0800
Subject: [PATCH] bugfix

---
 llmage/asyncinference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py
index 03d075e..0ca810f 100644
--- 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