From 438768567a1d5d95067af1757a93f0734716494b Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 6 Oct 2023 14:37:29 -0400
Subject: [PATCH] clean up

---
 swarms/workers/base.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/swarms/workers/base.py b/swarms/workers/base.py
index 5220713e..1f841579 100644
--- a/swarms/workers/base.py
+++ b/swarms/workers/base.py
@@ -33,7 +33,7 @@ class AbstractWorker:
     def send(
         self, 
         message: Union[Dict, str], 
-        recipient: "Agent", 
+        recipient, #add AbstractWorker
         request_reply: Optional[bool] = None
     ):
         """(Abstract method) Send a message to another worker."""
@@ -41,7 +41,7 @@ class AbstractWorker:
     async def a_send(
         self, 
         message: Union[Dict, str], 
-        recipient: "Agent", 
+        recipient, #add AbstractWorker
         request_reply: Optional[bool] = None
     ):
         """(Aabstract async method) Send a message to another worker."""
@@ -49,7 +49,7 @@ class AbstractWorker:
     def receive(
         self, 
         message: Union[Dict, str], 
-        sender: "Agent", 
+        sender, #add AbstractWorker
         request_reply: Optional[bool] = None
     ):
         """(Abstract method) Receive a message from another worker."""
@@ -57,7 +57,7 @@ class AbstractWorker:
     async def a_receive(
         self, 
         message: Union[Dict, str], 
-        sender: "Agent", 
+        sender, #add AbstractWorker
         request_reply: Optional[bool] = None
     ):
         """(Abstract async method) Receive a message from another worker."""
@@ -68,7 +68,7 @@ class AbstractWorker:
     def generate_reply(
         self,
         messages: Optional[List[Dict]] = None,
-        sender: Optional["Agent"] = None,
+        sender: Optional["AbstractWorker"] = None,
         **kwargs,
     ) -> Union[str, Dict, None]:
         """(Abstract method) Generate a reply based on the received messages.
@@ -83,7 +83,7 @@ class AbstractWorker:
     async def a_generate_reply(
         self,
         messages: Optional[List[Dict]] = None,
-        sender: Optional["Agent"] = None,
+        sender: Optional["AbstractWorker"] = None,
         **kwargs,
     ) -> Union[str, Dict, None]:
         """(Abstract async method) Generate a reply based on the received messages.