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

---
 llmage/asyncinference.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py
index 0ca810f..93d612a 100644
--- a/llmage/asyncinference.py
+++ b/llmage/asyncinference.py
@@ -28,6 +28,7 @@ async def grab_task_status(request, taskid):
 		if r.status in [ 'SUCCEEDED', 'FAILED' ]:
 			io = json.loads(r.ioinfo)
 			return io['output'][-1]
+
 		llmusage = r
 		recs = await sor.R('llm', {'id': r.llmid})
 		if len(recs) == 0:
@@ -35,6 +36,11 @@ async def grab_task_status(request, taskid):
 			return {"status": "FAILED", "error": f"{taskid} {r.llmid=} not exist in llm"}
 		llm = recs[0]
 
+	if llm is None:
+		e = Exception(f'id={r.llmid} Error happend!')
+		exception(f'{e}')
+		return {"status": "FAILED", "error": f"{taskid} {r.llmid=} {e}")
+
 	async with get_sor_context(env, 'llmage') as sor:
 		uapi = UAPI(request, sor)
 		apinames = llm.apiname.split(',')