You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
swarms/dialogue_simulator.py

18 lines
508 B

1 year ago
from swarms import DialogueSimulator, Worker
1 year ago
worker1 = Worker(ai_name="Plinus", openai_api_key="")
1 year ago
worker2 = Worker(ai_name="Optimus Prime", openai_api_key="")
collab = DialogueSimulator([worker1, worker2], DialogueSimulator.select_next_speaker)
1 year ago
collab.step()
1 year ago
1 year ago
# collab.start("My name is Plinus and I am a worker", "How are you?")
max_iters = 6
n = 0
while n < max_iters:
collab.start("My name is Plinus and I am a worker", "How are you?")
print(f"({name}): {message}")
print("\n")
n += 1