From 4a72039af56dc0a631fc38a9ce45c338bd45788d Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 5 Jul 2023 15:47:18 -0400 Subject: [PATCH] clean up --- swarms/swarms.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index 6ab87064..a92cc905 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -98,7 +98,7 @@ worker_agent.chain.verbose = True class WorkerNode: def __init__(self, llm, tools, vectorstore): self.llm = llm - self.tools = tools + self.tools = tools # Add this line here self.vectorstore = vectorstore @@ -107,7 +107,7 @@ class WorkerNode: self.agent = AutoGPT.from_llm_and_tools( ai_name=ai_name, ai_role=ai_role, - tools=tools, + tools=self.tools, # Change this line to use self.tools llm=self.llm, memory=self.vectorstore.as_retriever(search_kwargs=search_kwargs), human_in_the_loop=human_in_the_loop, @@ -122,7 +122,6 @@ class WorkerNode: self.agent.run([f"{tree_of_thoughts_prompt} {prompt}"]) - #inti worker node with llm worker_node = WorkerNode(llm=llm, tools=tools, vectorstore=vectorstore)