tool

2 years ago · 306b8ff15e
parent 20d9488936
commit 306b8ff15e
1 changed files with 1 additions and 1 deletions
--- 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