diff --git a/skillagent/agent.py b/skillagent/agent.py index 08bf058..d489754 100644 --- a/skillagent/agent.py +++ b/skillagent/agent.py @@ -86,10 +86,11 @@ class DummyLLM(LLM): # --------------------------- # Agent 实现 # --------------------------- -class Agent: - def __init__(self, llm: LLM, skillkit): - self.llm = llm - self.skillkit = skillkit +class SkillAgent: + def __init__(self, llmid:str, apikey:str, rootdirs=[]): + self.llm = DummyLLM(llmid, apikey) + # self.skillkit = skillkit + self.skillkit = SkillkitWrapper(rootdirs) self.skills = None self.loaded = False @@ -255,7 +256,7 @@ class Agent: async def skillagent(llm, apikey, user_skillroot, sys_skillroot=None): llm = DummyLLM('8L4hFJ4QpSMyu1UP03Juo', 'eYgNuD6sVQgbj-khOOUNU') skillkit = SkillkitWrapper(user_skillroot) - agent = Agent(llm, skillkit) + agent = SkillAgent(llmid, apikey) while True: print('What you want to do?')