Merge pull request ag2ai#356 from ag2ai/swarmmanagerautomissing

Swarm: Restore lost SWARM_MANAGER return
harishmohanraj · Jan 6, 2025 · a1c21dd · a1c21dd
2 parents 1623ecb + c98d19e
commit a1c21dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/autogen/agentchat/contrib/swarm_agent.py b/autogen/agentchat/contrib/swarm_agent.py
@@ -352,6 +352,8 @@ def _determine_next_agent(
             return None if user_agent is None else user_agent
         elif after_work_condition == AfterWorkOption.STAY:
             return last_speaker
+        elif after_work_condition == AfterWorkOption.SWARM_MANAGER:
+            return "auto"
     else:
         raise ValueError("Invalid After Work condition or return value from callable")