Merge pull request #30 from ag2ai/swarmag2

Swarm V2
ag2ai · Nov 21, 2024 · c1ebda0 · c1ebda0
2 parents 6abff18 + 8e6710b
commit c1ebda0
Show file tree

Hide file tree

Showing 2 changed files with 407 additions and 2 deletions.
diff --git a/autogen/agentchat/__init__.py b/autogen/agentchat/__init__.py
@@ -7,12 +7,22 @@
 from .agent import Agent
 from .assistant_agent import AssistantAgent
 from .chat import ChatResult, initiate_chats
+
+# Imported last to avoid circular imports
+from .contrib.swarm_agent import (
+    AFTER_WORK,
+    ON_CONDITION,
+    AfterWorkOption,
+    SwarmAgent,
+    SwarmResult,
+    initiate_swarm_chat,
+)
 from .conversable_agent import ConversableAgent, register_function
 from .groupchat import GroupChat, GroupChatManager
 from .user_proxy_agent import UserProxyAgent
 from .utils import gather_usage_summary
 
-__all__ = (
+__all__ = [
     "Agent",
     "ConversableAgent",
     "AssistantAgent",
@@ -23,4 +33,10 @@
     "initiate_chats",
     "gather_usage_summary",
     "ChatResult",
-)
+    "initiate_swarm_chat",
+    "SwarmAgent",
+    "SwarmResult",
+    "ON_CONDITION",
+    "AFTER_WORK",
+    "AfterWorkOption",
+]