From 107dde8b55fbbed771e22ee3d37f0cc11ac8f19f Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 6 Jul 2023 14:24:56 -0400 Subject: [PATCH] clean up in init boss mode --- swarms/swarms.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index 4a59aa5d..2c9f962a 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -60,7 +60,9 @@ class Swarms: llm_chain = LLMChain(llm=llm, prompt=prompt) agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=[tool.name for tool in tools]) agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True) - return BossNode(self.openai_api_key, llm, vectorstore, agent_executor, max_iterations=5) + # return BossNode(return BossNode(llm, vectorstore, agent_executor, max_iterations=5) + return BossNode(llm, vectorstore, agent_executor, max_iterations=5) + def run_swarms(self, objective): try: