examples fix

pull/64/head
Kye 1 year ago
parent 241a1a8125
commit ceeca90404

@ -106,7 +106,7 @@ for result in results:
```python
from langchain.llms import ChatOpenAI
from swarms import Worker
from swarms.workers import Worker
llm = ChatOpenAI(
model_name='gpt-4',

@ -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…
Cancel
Save