# a_run_swarm ## ``autogen.a_run_swarm`async`[ ``` a_run_swarm(initial_agent, messages, agents, user_agent=None, swarm_manager_args=None, max_rounds=20, context_variables=None, after_work=TERMINATE, exclude_transit_message=True) ``` Source code in `autogen/agentchat/contrib/swarm_agent.py` | | | | --- | --- | | ```
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
``` | ```
@export_module("autogen")
async def a_run_swarm(
initial_agent: ConversableAgent,
messages: list[dict[str, Any]] | str,
agents: list[ConversableAgent],
user_agent: UserProxyAgent | None = None,
swarm_manager_args: dict[str, Any] | None = None,
max_rounds: int = 20,
context_variables: ContextVariables | None = None,
after_work: AfterWorkOption
| Callable[[ConversableAgent, list[dict[str, Any]], GroupChat], AfterWorkOption | ConversableAgent | str]
| None = AfterWorkOption.TERMINATE,
exclude_transit_message: bool = True,
) -> AsyncRunResponseProtocol:
iostream = AsyncThreadIOStream()
response = AsyncRunResponse(iostream, agents) # type: ignore[arg-type]
async def stream_run(
iostream: AsyncThreadIOStream = iostream,
response: AsyncRunResponse = response,
) -> None:
with IOStream.set_default(iostream):
try:
chat_result, returned_context_variables, last_speaker = await a_initiate_swarm_chat(
initial_agent=initial_agent,
messages=messages,
agents=agents,
user_agent=user_agent,
swarm_manager_args=swarm_manager_args,
max_rounds=max_rounds,
context_variables=context_variables,
after_work=after_work,
exclude_transit_message=exclude_transit_message,
)
IOStream.get_default().send(
RunCompletionEvent( # type: ignore[call-arg]
history=chat_result.chat_history,
summary=chat_result.summary,
cost=chat_result.cost,
last_speaker=last_speaker.name,
context_variables=returned_context_variables,
)
)
except Exception as e:
response.iostream.send(ErrorEvent(error=e)) # type: ignore[call-arg]
task = asyncio.create_task(stream_run())
# prevent the task from being garbage collected
response._task_ref = task # type: ignore[attr-defined]
return response
``` |