cleanup

Former-commit-id: 82f2d0fee7

dialogue_simulator.py

@@ -0,0 +1,9 @@
+from swarms import DialogueSimulator, Worker
+
+worker1 = Worker(ai_name="Optimus Prime", openai_api_key="")
+worker2 = Worker(ai_name="Optimus Prime", openai_api_key="")
+
+collab = DialogueSimulator([worker1, worker2], DialogueSimulator.select_next_speaker)
+collab.reset()
+
+collab.start("My name is Plinus and I am a worker", "How are you?")

mmi.py

@@ -1,7 +0,0 @@
-from swarms import DialogueSimulator, Worker
-
-worker = Worker(ai_name="Optimus Prime", openai_api_key="")
-
-collab = DialogueSimulator(worker, DialogueSimulator.select_next_speaker)
-
-collab.start("My name is Plinus and I am a worker", "How are you?")

swarms/swarms/dialogue_simulator.py

@@ -1,7 +1,9 @@
+from swarms.swarms.dialogue_simulator import Worker
+
 class DialogueSimulator:
     def __init__(
         self,
-        agents,
+        agents[Worker],
         selection_func
     ):
         self.agents = agents