From b851d0a59658fa13b26e031fa0c1dc2c42d1569f Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 25 Sep 2023 13:36:57 -0400 Subject: [PATCH] clean up Former-commit-id: 0fd5d9a634c92524b8464700d8b541d13d3513a3 --- dialogue_simulator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dialogue_simulator.py b/dialogue_simulator.py index cbbb3f5a..ec3ebcd3 100644 --- a/dialogue_simulator.py +++ b/dialogue_simulator.py @@ -5,6 +5,7 @@ worker2 = Worker(ai_name="Optimus Prime", openai_api_key="") collab = DialogueSimulator([worker1, worker2], DialogueSimulator.select_next_speaker) collab.reset() +collab.inject(name=worker1.ai_name, "what is your name") # collab.start("My name is Plinus and I am a worker", "How are you?") @@ -12,7 +13,7 @@ max_iters = 6 n = 0 while n < max_iters: - name, message = collab.step() + name, message = simulator.step() print(f"({name}): {message}") print("\n") n += 1 \ No newline at end of file