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