main
Kye 2 years ago
parent 13185b19d2
commit 4a72039af5

@ -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)

Loading…
Cancel
Save