swarms/dialogue_simulator.py

9 lines
338 B

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?")