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.
51 lines
1.0 KiB
51 lines
1.0 KiB
from swarms.swarms import DialogueSimulator
|
|
from swarms.workers.worker import Worker
|
|
from swarms.models import OpenAIChat
|
|
import os
|
|
|
|
api_key = os.environ.get("OPENAI_API_KEY")
|
|
org_id = os.environ.get("OPENAI_ORG_ID")
|
|
|
|
llm = OpenAIChat(
|
|
model_name="gpt-4",
|
|
openai_api_key=api_key,
|
|
openai_org_id=org_id,
|
|
temperature=0.5
|
|
)
|
|
|
|
worker1 = Worker(
|
|
llm=llm,
|
|
ai_name="Bumble Bee",
|
|
ai_role="Worker in a swarm",
|
|
external_tools=None,
|
|
human_in_the_loop=False,
|
|
temperature=0.5,
|
|
)
|
|
worker2 = Worker(
|
|
llm=llm,
|
|
ai_name="Optimus Prime",
|
|
ai_role="Worker in a swarm",
|
|
external_tools=None,
|
|
human_in_the_loop=False,
|
|
temperature=0.5,
|
|
)
|
|
worker3 = Worker(
|
|
llm=llm,
|
|
ai_name="Megatron",
|
|
ai_role="Worker in a swarm",
|
|
external_tools=None,
|
|
human_in_the_loop=False,
|
|
temperature=0.5,
|
|
)
|
|
|
|
collab = DialogueSimulator(
|
|
[worker1, worker2],
|
|
# DialogueSimulator.select_next_speaker
|
|
)
|
|
|
|
collab.run(
|
|
max_iters=4,
|
|
name="plinus",
|
|
message="how can we enable multi agent collaboration",
|
|
)
|