From 557d952eef690c093155dd60d0a1ce439d340991 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Mon, 25 Sep 2023 13:32:27 -0400
Subject: [PATCH] clean up

Former-commit-id: 4c521437a0b346f19985a39f41c9fda0259f8e7c
---
 dialogue_simulator.py               | 4 ++--
 swarms/swarms/dialogue_simulator.py | 6 +++---
 swarms/workers/worker.py            | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/dialogue_simulator.py b/dialogue_simulator.py
index 72c4e45c..214458a3 100644
--- a/dialogue_simulator.py
+++ b/dialogue_simulator.py
@@ -4,7 +4,7 @@ worker1 = Worker(ai_name="Plinus", openai_api_key="")
 worker2 = Worker(ai_name="Optimus Prime", openai_api_key="")
 
 collab = DialogueSimulator([worker1, worker2], DialogueSimulator.select_next_speaker)
-collab.step()
+collab.reset()
 
 # collab.start("My name is Plinus and I am a worker", "How are you?")
 
@@ -12,7 +12,7 @@ max_iters = 6
 n = 0
 
 while n < max_iters:
-    collab.start("My name is Plinus and I am a worker", "How are you?")
+    name, message = simulator.step()
     print(f"({name}): {message}")
     print("\n")
     n += 1
\ No newline at end of file
diff --git a/swarms/swarms/dialogue_simulator.py b/swarms/swarms/dialogue_simulator.py
index 8f0cbf17..4873829c 100644
--- a/swarms/swarms/dialogue_simulator.py
+++ b/swarms/swarms/dialogue_simulator.py
@@ -12,9 +12,9 @@ class DialogueSimulator:
         self._step = 0
         self.select_next_speaker = selection_func
     
-    # def reset(self):
-    #     for agent in self.agents:
-    #         agent.reset()
+    def reset(self):
+        for agent in self.agents:
+            agent.reset()
         
     def start(self, name: str, message: str):
         #init conv with a message from name
diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py
index acbcf18f..dea29cd8 100644
--- a/swarms/workers/worker.py
+++ b/swarms/workers/worker.py
@@ -115,7 +115,7 @@ class Worker:
         self.message_history.append(f"{name}: {message}")
 
     def send(self) -> str:
-        self.agent.run(self.message_history)
+        self.agent.run(task=self.message_history)
 
     def add(self, task, priority=0):
         self.task_queue.append((priority, task))