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