From 5058acb814830547c5f7f67033b3285a7e2f17fc Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 5 Jul 2023 18:46:14 -0400 Subject: [PATCH] clean up init --- swarms/swarms.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/swarms/swarms.py b/swarms/swarms.py index 21d50b6a..e8f830a1 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -32,7 +32,13 @@ class Swarms: return FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) def initialize_worker_node(self, llm, tools, vectorstore): - worker_node = WorkerNode(llm=llm, tools=tools, vectorstore=vectorstore) + worker_node = WorkerNode( + llm=llm, + tools=tools, + vectorstore=vectorstore, + name="WorkerNode", + description="A worker node that can perform complex tasks" + ) worker_node.create_agent(ai_name="AI Assistant", ai_role="Assistant", human_in_the_loop=False, search_kwargs={}) return worker_node