bugfix
This commit is contained in:
parent
86fb13e168
commit
cd332481b3
@ -85,12 +85,21 @@ class IndustrialSkillEngine:
|
||||
config = getConfig()
|
||||
skills_dir = config.skills_dir
|
||||
self.root = Path(skills_dir).resolve()
|
||||
self.llm = LLMHandler(request, llmid, apikey=apikey)
|
||||
self.llmid = llmid
|
||||
# self.llm = LLMHandler(request, llmid, apikey=apikey)
|
||||
self.llm = None
|
||||
self.registry = {}
|
||||
self.task_queue = asyncio.Queue(maxsize=20)
|
||||
# 状态机:记录当前任务执行到的步骤
|
||||
self.state = {"current_skill": None, "history": [], "pending_params": []}
|
||||
|
||||
async def init_llm(self):
|
||||
if self.llm:
|
||||
return
|
||||
if self.llmid is None:
|
||||
pass
|
||||
self.llm = LLMHandler(request, llmid, apikey=apikey)
|
||||
|
||||
async def write_output(self, data):
|
||||
await self.task_queue.put(data)
|
||||
|
||||
@ -205,6 +214,7 @@ class IndustrialSkillEngine:
|
||||
return base_content
|
||||
|
||||
async def inference(self, params_kw):
|
||||
await self.init_llm()
|
||||
f = partial(self.run, params_kw)
|
||||
asyncio.create_task(self.run(params_kw))
|
||||
while True:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user