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

Former-commit-id: 38e01453e18d97654fe3aa19aadd82663cd0455a
---
 dialogue_simulator.py               | 2 +-
 swarms/swarms/dialogue_simulator.py | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/dialogue_simulator.py b/dialogue_simulator.py
index f5a5401f..f02bdc82 100644
--- a/dialogue_simulator.py
+++ b/dialogue_simulator.py
@@ -5,7 +5,7 @@ worker2 = Worker(ai_name="Optimus Prime", openai_api_key="")
 
 collab = DialogueSimulator(
     [worker1, worker2], 
-    DialogueSimulator.select_next_speaker
+    # DialogueSimulator.select_next_speaker
 )
 
 collab.run(
diff --git a/swarms/swarms/dialogue_simulator.py b/swarms/swarms/dialogue_simulator.py
index b216bb96..bc90711a 100644
--- a/swarms/swarms/dialogue_simulator.py
+++ b/swarms/swarms/dialogue_simulator.py
@@ -5,7 +5,12 @@ class DialogueSimulator:
     def __init__(self, agents: List[Worker]):
         self.agents = agents
 
-    def run(self, max_iters: int, name: str = None, message: str = None):
+    def run(
+        self, 
+        max_iters: int, 
+        name: str = None, 
+        message: str = None
+    ):
         step = 0
         if name and message:
             prompt = f"Name {name} and message: {message}"