diff --git a/skillagent/skillengine.py b/skillagent/skillengine.py index 6971af8..04686ed 100644 --- a/skillagent/skillengine.py +++ b/skillagent/skillengine.py @@ -1,5 +1,6 @@ import os import re +import json import asyncio import yaml from functools import partial @@ -180,9 +181,9 @@ class IndustrialSkillEngine: return base_content - async def reference(self, user_prompt:str, context: str=None, is_retry: bool=False): - f = partial(self.run, user_prompt, context=context, is_retry=is_retry) - asyncio.create_task(f()) + async def reference(self, params_kw): + f = partial(self.run, params_kw) + asyncio.create_task(self.run(params_kw)) while True: data = await self.task_queue.get() if not data: