This commit is contained in:
yumoqing 2026-02-12 15:28:36 +08:00
parent f1e761adda
commit 4a938728b8

View File

@ -87,6 +87,7 @@ class IndustrialSkillEngine:
self.state = {"current_skill": None, "history": [], "pending_params": []} self.state = {"current_skill": None, "history": [], "pending_params": []}
async def write_output(self, data=None): async def write_output(self, data=None):
debug('-----------write_output() called: {data=}')
await self.task_queue.put(data) await self.task_queue.put(data)
# --- 1. 工业级初始化:依赖检查与索引 --- # --- 1. 工业级初始化:依赖检查与索引 ---
@ -186,12 +187,14 @@ class IndustrialSkillEngine:
return base_content return base_content
async def reference(self, params_kw): async def inference(self, params_kw):
f = partial(self.run, params_kw) f = partial(self.run, params_kw)
asyncio.create_task(self.run(params_kw)) asyncio.create_task(self.run(params_kw))
while True: while True:
data = await self.task_queue.get() data = await self.task_queue.get()
if not data: if not data:
debug('End Data')
break; break;
debug(f'{data=}, {type(data)=}') debug(f'{data=}, {type(data)=}')
yield json.dump(data, ensure_ascii=False) + '\n' yield json.dump(data, ensure_ascii=False) + '\n'