update import

ag2ai · Nov 20, 2024 · 2df1b5b · 2df1b5b
1 parent b2615b3
commit 2df1b5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/website/blog/2024-11-17-Swarm/index.mdx b/website/blog/2024-11-17-Swarm/index.mdx
@@ -70,7 +70,7 @@ history, context, last_agent = initiate_swarm_chat(
 This example of managing refunds demonstrates the context handling, swarm and agent-level conditional and after work hand offs, and the human-in-the-loop feature.
 
 ```python
-from autogen.agentchat.contrib.swarm_agent import initiate_swarm_chat, SwarmAgent, SwarmResult, ON_CONDITION, AFTER_WORK, AfterWorkOption
+from autogen import initiate_swarm_chat, SwarmAgent, SwarmResult, ON_CONDITION, AFTER_WORK, AfterWorkOption
 from autogen import UserProxyAgent
 import os
 

diff --git a/website/docs/topics/swarm.ipynb b/website/docs/topics/swarm.ipynb
@@ -147,7 +147,7 @@
    "source": [
     "import random\n",
     "\n",
-    "from autogen.agentchat.contrib.swarm_agent import (\n",
+    "from autogen import (\n",
     "    AFTER_WORK,\n",
     "    ON_CONDITION,\n",
     "    AfterWorkOption,\n",