Spaces:
Runtime error
Runtime error
File size: 1,259 Bytes
4962437 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
from langchain.llms import OpenAIChat
from swarms.swarms import GroupChat, GroupChatManager
from swarms.workers import Worker
llm = OpenAIChat(
model_name='gpt-4',
openai_api_key="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",
temperature=0.5
)
node = Worker(
llm=llm,
ai_name="Optimus Prime",
ai_role="Worker in a swarm",
external_tools=None,
human_in_the_loop=False,
temperature=0.5,
)
node2 = Worker(
llm=llm,
ai_name="Optimus Prime",
ai_role="Worker in a swarm",
external_tools=None,
human_in_the_loop=False,
temperature=0.5,
)
node3 = Worker(
llm=llm,
ai_name="Optimus Prime",
ai_role="Worker in a swarm",
external_tools=None,
human_in_the_loop=False,
temperature=0.5,
)
nodes = [
node,
node2,
node3
]
messages = [
{
"role": "system",
"context": f"Create an a small feedforward in pytorch",
}
]
group = GroupChat(
workers=nodes,
messages=messages,
max_rounds=3,
)
manager = GroupChatManager(
groupchat=group,
max_consecutive_auto_reply=3,
human_input_mode="NEVER",
system_message="Group chat manager"
)
output = group.run(
messages,
sender=node,
config=group,
)
print(output)
|