This commit is contained in:
yumoqing 2026-03-30 13:48:01 +08:00
parent 7a892199f2
commit a315858555

View File

@ -180,6 +180,7 @@ async def query_task_status(request, upappid, apiname, luid, userid, taskid):
'output': rzt 'output': rzt
}) })
if rzt.status == 'SUCCEEDED': if rzt.status == 'SUCCEEDED':
llmusage.usage = rzt['usage']
llms = await sor.R('llm', {'id': llmusage.llmid}) llms = await sor.R('llm', {'id': llmusage.llmid})
if len(llms) == 0: if len(llms) == 0:
e = Exception(f'{llmusage.llmid=} not found in llm') e = Exception(f'{llmusage.llmid=} not found in llm')