bugfix

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