From c6481a2f6a04e6feac4c460da2bd6900444a206a 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: 67daba80c6e8d53c83ff712cadda286db7ad3f5d --- 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(