parent
241a1a8125
commit
ceeca90404
@ -0,0 +1,44 @@
|
|||||||
|
from swarms.swarms.multi_agent_debate import MultiAgentDebate, select_speaker
|
||||||
|
from swarms.workers.worker import Worker
|
||||||
|
from langchain.llms import OpenAIChat
|
||||||
|
|
||||||
|
llm = OpenAIChat(model_name="gpt-4", openai_api_key="api-key", 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,
|
||||||
|
)
|
||||||
|
|
||||||
|
#init agents
|
||||||
|
agents = [worker1, worker2, worker3]
|
||||||
|
|
||||||
|
# Initialize multi-agent debate with the selection function
|
||||||
|
debate = MultiAgentDebate(agents, select_speaker)
|
||||||
|
|
||||||
|
# Run task
|
||||||
|
task = "What were the winning boston marathon times for the past 5 years (ending in 2022)? Generate a table of the year, name, country of origin, and times."
|
||||||
|
results = debate.run(task, max_iters=4)
|
||||||
|
|
||||||
|
# Print results
|
||||||
|
for result in results:
|
||||||
|
print(f"Agent {result['agent']} responded: {result['response']}")
|
@ -1,20 +0,0 @@
|
|||||||
from swarms.swarms.multi_agent_debate import MultiAgentDebate, select_speaker
|
|
||||||
from swarms.workers.worker import Worker
|
|
||||||
|
|
||||||
# Initialize agents
|
|
||||||
worker1 = Worker(openai_api_key="", ai_name="Optimus Prime")
|
|
||||||
worker2 = Worker(openai_api_key="", ai_name="Bumblebee")
|
|
||||||
worker3 = Worker(openai_api_key="", ai_name="Megatron")
|
|
||||||
|
|
||||||
agents = [worker1, worker2, worker3]
|
|
||||||
|
|
||||||
# Initialize multi-agent debate with the selection function
|
|
||||||
debate = MultiAgentDebate(agents, select_speaker)
|
|
||||||
|
|
||||||
# Run task
|
|
||||||
task = "What were the winning boston marathon times for the past 5 years (ending in 2022)? Generate a table of the year, name, country of origin, and times."
|
|
||||||
results = debate.run(task, max_iters=4)
|
|
||||||
|
|
||||||
# Print results
|
|
||||||
for result in results:
|
|
||||||
print(f"Agent {result['agent']} responded: {result['response']}")
|
|
Loading…
Reference in new issue