From 40a8cef1db7618bf857effd250133357fae388fb Mon Sep 17 00:00:00 2001 From: Mark Sze Date: Thu, 21 Nov 2024 16:46:39 +0000 Subject: [PATCH] Initial commit --- autogen/agentchat/contrib/swarm_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autogen/agentchat/contrib/swarm_agent.py b/autogen/agentchat/contrib/swarm_agent.py index 6ced66f267..d790d32560 100644 --- a/autogen/agentchat/contrib/swarm_agent.py +++ b/autogen/agentchat/contrib/swarm_agent.py @@ -49,7 +49,7 @@ def initiate_swarm_chat( init_agent: "SwarmAgent", messages: Union[List[Dict[str, Any]], str], agents: List["SwarmAgent"], - user_agent: Optional[UserProxyAgent] = None, + user_agent: Optional[Union[UserProxyAgent, str]] = None, max_rounds: int = 20, context_variables: Optional[Dict[str, Any]] = None, after_work: Optional[Union[AFTER_WORK, Callable]] = AFTER_WORK(AfterWorkOption.TERMINATE),