From 920697c616051553e1e9ffd3e90fbef9e628d96c Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 5 Jul 2023 19:21:39 -0400
Subject: [PATCH] workernode cleanup

---
 swarms/agents/workers/worker_agent.py | 55 +++++++++++++++++++++++++--
 1 file changed, 51 insertions(+), 4 deletions(-)

diff --git a/swarms/agents/workers/worker_agent.py b/swarms/agents/workers/worker_agent.py
index 01f524e4..1432fd18 100644
--- a/swarms/agents/workers/worker_agent.py
+++ b/swarms/agents/workers/worker_agent.py
@@ -6,6 +6,7 @@ from langchain.callbacks.manager import (
     CallbackManagerForToolRun,
 )
 
+
 class WorkerNode(BaseTool):
     """Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on """
     name = "WorkerNode"
@@ -28,9 +29,7 @@ class WorkerNode(BaseTool):
         )
         self.agent.chain.verbose = True
 
-    # @tool
-        # name="Worker AutoBot Agent",
-        # description="Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on",
+    @tool("perform_task")
     def _run(
         self, prompt: str, run_manager: Optional[CallbackManagerForToolRun] = None
     ) -> str:
@@ -45,4 +44,52 @@ class WorkerNode(BaseTool):
         self, prompt: str, run_manager: Optional[AsyncCallbackManagerForToolRun] = None
     ) -> str:
         """Use the tool asynchronously."""
-        raise NotImplementedError("WorkerNode does not support async")
\ No newline at end of file
+        raise NotImplementedError("WorkerNode does not support async")
+
+
+
+
+
+
+
+
+# class WorkerNode(BaseTool):
+#     """Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on """
+#     name = "WorkerNode"
+#     description = "A worker node that can perform complex tasks"
+    
+#     def __init__(self, llm, tools, vectorstore):
+#         super().__init__()
+#         self.llm = llm
+#         self.tools = tools
+#         self.vectorstore = vectorstore
+
+#     def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs):
+#         self.agent = AutoGPT.from_llm_and_tools(
+#             ai_name=ai_name,
+#             ai_role=ai_role,
+#             tools=self.tools,
+#             llm=self.llm,
+#             memory=self.vectorstore.as_retriever(search_kwargs=search_kwargs),
+#             human_in_the_loop=human_in_the_loop,
+#         )
+#         self.agent.chain.verbose = True
+
+#     # @tool
+#         # name="Worker AutoBot Agent",
+#         # description="Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on",
+#     def _run(
+#         self, prompt: str, run_manager: Optional[CallbackManagerForToolRun] = None
+#     ) -> str:
+#         """Use the tool."""
+#         tree_of_thoughts_prompt = """
+#         Imagine three different experts are answering this question. All experts will write down each chain of thought of each step of their thinking, then share it with the group. Then all experts will go on to the next step, etc. If any expert realises they're wrong at any point then they leave. The question is...
+#         """
+#         self.agent.run([f"{tree_of_thoughts_prompt}{prompt}"])
+#         return "Task completed by WorkerNode"
+
+#     async def _arun(
+#         self, prompt: str, run_manager: Optional[AsyncCallbackManagerForToolRun] = None
+#     ) -> str:
+#         """Use the tool asynchronously."""
+#         raise NotImplementedError("WorkerNode does not support async")
\ No newline at end of file