From 67fd8365be1fac44296750c09cfc69d70a5bce5c Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 11 Aug 2023 22:59:35 -0400 Subject: [PATCH] clean up tools --- swarms/boss/boss_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/boss/boss_node.py b/swarms/boss/boss_node.py index f91bf615..d76bfb16 100644 --- a/swarms/boss/boss_node.py +++ b/swarms/boss/boss_node.py @@ -141,7 +141,7 @@ class BossNode: prefix = """You are a Boss in a swarm who performs one task based on the following objective: {objective}. Take into account these previously completed tasks: {context}.\n """ prompt = ZeroShotAgent.create_prompt(tools, prefix=prefix, suffix=suffix, input_variables=["objective", "task", "context", "agent_scratchpad"],) llm_chain = LLMChain(llm=self.llm, prompt=prompt) - agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=[tool.name for tool in tools]) + agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=[tool for tool in tools]) self.agent_executor = agent_executor if agent_executor else AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=self.verbose) # Setup BabyAGI