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-01 18:58:55 +08:00
parent 19b05d5d29
commit 5fa4fcf5ec
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
llmage/asyncinference.py
View File

@ -17,14 +17,14 @@ from .accounting import llm_accounting, llm_charging
from .utils import * from .utils import *
async def grab_task_status(request, taskid): async def grab_task_status(request, taskid):
env = ServerEnv() env = ServerEnv()
llm = llmusage = None llm = llmusage = None
async with get_sor_context(env, 'llmage') as sor: async with get_sor_context(env, 'llmage') as sor:
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 None
r = recs[0] r = recs[0]
if r.status == 'SUCCEEDED': if r.status == 'SUCCEEDED':
io = json.loads(r.ioinfo) io = json.loads(r.ioinfo)
return io['output'][-1] return io['output'][-1]
@ -57,8 +57,8 @@ async def grab_task_status(request, taskid):
if changed.output.video: if changed.output.video:
break; break;
x = await uapi.call(llm.upappid, apiname[1], userid, params=d) x = await uapi.call(llm.upappid, apiname[1], userid, params=d)
g = json.load(x) g = json.load(x)
if g.get('video'): if g.get('video'):
changed.output.video = g.get('video') changed.output.video = g.get('video')
break break
await asyncio.sleep(0.2) await asyncio.sleep(0.2)