Update multi_agent.py
Browse files- multi_agent.py +17 -25
multi_agent.py
CHANGED
@@ -22,7 +22,7 @@ def run_multi_agent(llm, task):
|
|
22 |
"After each step is done by others, check the progress and "
|
23 |
"instruct the remaining steps. If a step fails, try to "
|
24 |
"workaround",
|
25 |
-
description="Given a task, determine what "
|
26 |
"information is needed to complete the task. "
|
27 |
"After each step is done by others, check the progress and "
|
28 |
"instruct the remaining steps",
|
@@ -32,25 +32,13 @@ def run_multi_agent(llm, task):
|
|
32 |
engineer = autogen.AssistantAgent(
|
33 |
name="Engineer",
|
34 |
llm_config=llm_config,
|
35 |
-
description="
|
36 |
"provided by the planner.",
|
37 |
)
|
38 |
|
39 |
-
writer = autogen.ConversableAgent(
|
40 |
-
name="Writer",
|
41 |
-
llm_config=llm_config,
|
42 |
-
system_message="Writer. "
|
43 |
-
"Please write blogs in markdown format (with relevant titles)"
|
44 |
-
" and put the content in pseudo ```md``` code block. "
|
45 |
-
"You take feedback from the admin and refine your blog.",
|
46 |
-
description="After all the info is available, "
|
47 |
-
"write blogs based on the code execution results and take "
|
48 |
-
"feedback from the admin to refine the blog. ",
|
49 |
-
)
|
50 |
-
|
51 |
executor = autogen.ConversableAgent(
|
52 |
name="Executor",
|
53 |
-
|
54 |
"engineer and report the result.",
|
55 |
human_input_mode="NEVER",
|
56 |
code_execution_config={
|
@@ -59,19 +47,23 @@ def run_multi_agent(llm, task):
|
|
59 |
"use_docker": False,
|
60 |
},
|
61 |
)
|
62 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
groupchat = autogen.GroupChat(
|
64 |
agents=[user_proxy, engineer, writer, executor, planner],
|
65 |
messages=[],
|
66 |
-
max_round=
|
67 |
-
allowed_or_disallowed_speaker_transitions={
|
68 |
-
user_proxy: [engineer, writer, executor, planner],
|
69 |
-
engineer: [user_proxy, executor],
|
70 |
-
writer: [user_proxy, planner],
|
71 |
-
executor: [user_proxy, engineer, planner],
|
72 |
-
planner: [user_proxy, engineer, writer],
|
73 |
-
},
|
74 |
-
speaker_transitions_type="allowed",
|
75 |
)
|
76 |
|
77 |
manager = autogen.GroupChatManager(
|
|
|
22 |
"After each step is done by others, check the progress and "
|
23 |
"instruct the remaining steps. If a step fails, try to "
|
24 |
"workaround",
|
25 |
+
description="Planner. Given a task, determine what "
|
26 |
"information is needed to complete the task. "
|
27 |
"After each step is done by others, check the progress and "
|
28 |
"instruct the remaining steps",
|
|
|
32 |
engineer = autogen.AssistantAgent(
|
33 |
name="Engineer",
|
34 |
llm_config=llm_config,
|
35 |
+
description="An engineer that writes code based on the plan "
|
36 |
"provided by the planner.",
|
37 |
)
|
38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
executor = autogen.ConversableAgent(
|
40 |
name="Executor",
|
41 |
+
system_message="Execute the code written by the "
|
42 |
"engineer and report the result.",
|
43 |
human_input_mode="NEVER",
|
44 |
code_execution_config={
|
|
|
47 |
"use_docker": False,
|
48 |
},
|
49 |
)
|
50 |
+
|
51 |
+
writer = autogen.ConversableAgent(
|
52 |
+
name="Writer",
|
53 |
+
llm_config=llm_config,
|
54 |
+
system_message="Writer."
|
55 |
+
"Please write blogs in markdown format (with relevant titles)"
|
56 |
+
" and put the content in pseudo ```md``` code block. "
|
57 |
+
"You take feedback from the admin and refine your blog.",
|
58 |
+
description="Writer."
|
59 |
+
"Write blogs based on the code execution results and take "
|
60 |
+
"feedback from the admin to refine the blog."
|
61 |
+
)
|
62 |
+
|
63 |
groupchat = autogen.GroupChat(
|
64 |
agents=[user_proxy, engineer, writer, executor, planner],
|
65 |
messages=[],
|
66 |
+
max_round=25,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
67 |
)
|
68 |
|
69 |
manager = autogen.GroupChatManager(
|