From d5ed5b7987882c23afce02e00588f39fe098286f Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 25 Sep 2023 13:35:31 -0400 Subject: [PATCH] dialogue simulator Former-commit-id: 8d96cb10aadb5d80ccbe76b233e39c978f15b659 --- swarms/swarms/dialogue_simulator.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/swarms/swarms/dialogue_simulator.py b/swarms/swarms/dialogue_simulator.py index 4873829c..9b7ea1f4 100644 --- a/swarms/swarms/dialogue_simulator.py +++ b/swarms/swarms/dialogue_simulator.py @@ -26,6 +26,12 @@ class DialogueSimulator: #increment time self._step += 1 + def inject(self, name: str, message: str): + for agent in self.agents: + agent.receieve(name, message) + + self._step += 1 + def step(self) -> tuple[str, str]: #choose next speaker speaker_idx = self.select_next_speaker(