From 54692ac691f65d9f90ef696cd4bdf144fb57853e Mon Sep 17 00:00:00 2001 From: yumoqing Date: Fri, 6 Mar 2026 11:42:53 +0800 Subject: [PATCH] bugfix --- skillagent/call.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/skillagent/call.py b/skillagent/call.py index 0eb54aa..a651668 100644 --- a/skillagent/call.py +++ b/skillagent/call.py @@ -2,6 +2,7 @@ from appPublic.log import debug, exception from ahserver.serverenv import ServerEnv from sqlor.dbpools import get_sor_context from .skillengine import IndustrialSkillEngine +from llmage.llmclient import inference_generator ## for video-generate skill @@ -29,9 +30,9 @@ async def text2video(request, params_kw=None): if llmid is None: debug(f'---------text2video(): llmid is None----------') return f'image2video():{params_kw}, llmid not found' - engine = IndustrialSkillEngine(request, llmid) + ret = '' - async for d in engine.inference(params_kw): + async for d in inference_generator(request, params_kw=params_kw): ret += d debug(f'---------text2video():{ret=}----------') return ret @@ -43,10 +44,8 @@ async def image2video(request, params_kw=None): llmid = await get_llmid_by_llmcatelog('图生视频', providerid) if llmid is None: return f'image2video():{params_kw}, llmid not found' - - engine = IndustrialSkillEngine(request, llmid) ret = '' - async for d in engine.inference(params_kw): + async for d in inference_generator(request, params_kw=params_kw): ret += d return ret