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: