From efcb90944f04f9b64b2bd466d61708c3db4bd4c5 Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Mon, 30 Mar 2026 12:23:11 +0800
Subject: [PATCH] bugfix

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

diff --git a/llmage/asyncinference.py b/llmage/asyncinference.py
index 58bc26c..0c23018 100644
--- a/llmage/asyncinference.py
+++ b/llmage/asyncinference.py
@@ -139,6 +139,12 @@ async def add_new_llmusage_output(luid, newd):
 async def query_task_status(request, upappid, apiname, luid, userid, taskid):
 	env = request._run_ns
 	async with get_sor_context(env, 'llmage') as sor:
+		recs = await sor.R('llmusage', {'id': luid})
+		if len(recs) == 0:
+			e = Exception(f'{luid=} is not found in llmusage')
+			exception(f'{e}')
+			raise e
+		llmusage = recs[0]
 		uapi = UAPI(request, sor)
 		apinames = apiname.split(',')
 		for apiname in apinames: