From 2df1b5b210a20ca8d43a2b86436bceb74ae63b99 Mon Sep 17 00:00:00 2001
From: Yiran Wu <32823396+kevin666aa@users.noreply.github.com>
Date: Wed, 20 Nov 2024 16:15:34 -0500
Subject: [PATCH] update import

---
 website/blog/2024-11-17-Swarm/index.mdx | 2 +-
 website/docs/topics/swarm.ipynb         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/website/blog/2024-11-17-Swarm/index.mdx b/website/blog/2024-11-17-Swarm/index.mdx
index 6cb9b85ded..80e4ff43f0 100644
--- 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
index f960144bf1..0e7c5353c6 100644
--- 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",