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