From 306b8ff15e272f731f27747c12b7c7dbfac13380 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 5 Jul 2023 16:48:41 -0400
Subject: [PATCH] tool

---
 swarms/swarms.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swarms/swarms.py b/swarms/swarms.py
index 478cbcd5..41c5decc 100644
--- a/swarms/swarms.py
+++ b/swarms/swarms.py
@@ -58,7 +58,7 @@ index = faiss.IndexFlatL2(embedding_size)
 vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
 
 # ---------- Worker Node ----------
-@tool
+@tool("workerAgent", return_direct=True)
 class WorkerNode:
     def __init__(self, llm, tools, vectorstore):
         self.llm = llm