Initial commit

ag2ai · Nov 21, 2024 · 40a8cef · 40a8cef
1 parent 996f923
commit 40a8cef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),