From 20d948893620572edad92296775c8d17c3591e86 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 5 Jul 2023 16:46:30 -0400
Subject: [PATCH] clean up

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

diff --git a/swarms/swarms.py b/swarms/swarms.py
index 37caff7f..478cbcd5 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("search", return_direct=True)
+@tool
 class WorkerNode:
     def __init__(self, llm, tools, vectorstore):
         self.llm = llm