commit error

main
Kye 2 years ago
parent 1774fe2592
commit 2af7fab21c

@ -18,7 +18,7 @@ class WorkerNode(BaseTool):
vectorstore: VectorStore = Field(description="Vector store")
def __init__(self, llm, tools, vectorstore):
super().__init__()
super().__init__(llm=llm, tools=tools, vectorstore=vectorstore)
self.llm = llm
self.tools = tools
self.vectorstore = vectorstore

@ -33,7 +33,7 @@ class Swarms:
def initialize_worker_node(self, llm, tools, vectorstore):
worker_node = WorkerNode(llm=llm, tools=tools, vectorstore=vectorstore)
worker_node.create_agent(ai_name="AI Assistant", ai_role="Assistant", human_in_the_loop=True, search_kwargs={})
worker_node.create_agent(ai_name="AI Assistant", ai_role="Assistant", human_in_the_loop=False, search_kwargs={})
return worker_node
def initialize_boss_node(self, llm, vectorstore):

Loading…
Cancel
Save