From 0820ff22702f5b129c658c4dbf929da5a77d53e3 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 11 Aug 2023 22:59:35 -0400
Subject: [PATCH] clean up tools
Former-commit-id: cbcf2cc56acfabaf66310c6089e11e96d92d77d4
---
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