yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-04-02 15:20:51 +08:00
parent d90c4ab198
commit 3f3ddc69b2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llmage/asyncinference.py
View File

@ -23,7 +23,7 @@ async def grab_task_status(request, taskid):
recs = await sor.R('llmusage', {'taskid': taskid})
if len(recs) == 0:
exception(f'{taskid=} not found in llmusage')
return None
return {"status": "FAILED", "error": f"{taskid} not exist"}
r = recs[0]
if r.status == 'SUCCEEDED':
io = json.loads(r.ioinfo)
@ -32,7 +32,7 @@ async def grab_task_status(request, taskid):
recs = await sor.R('llm', {'id': r.llmid})
if len(recs) == 0:
exception(f'id={r.llmid} not found in llm')
return None
return {"status": "FAILED", "error": f"{taskid} {r.llmid=} not exist in llm"}
llm = recs[0]
async with get_sor_context(env, 'llmage') as sor:
@ -71,7 +71,7 @@ async def grab_task_status(request, taskid):
'output': {'status': 'FAILED', 'error': str(e)}
}
await add_new_llmusage_output(llmusage.id, changed)
return
return {'status': 'FAILED', 'error': str(e)}
if changed.status == 'SUCCEEDED':
llmusage.usage = changed.output.usage
if llm.ppid: