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