diff --git a/docs/examples/agent/custom_agent.ipynb b/docs/examples/agent/custom_agent.ipynb
index c55a7c1a4cfc7f6a700124d8b997a2254786b9dd..7a26db4db1cfea9e46b9c6ee57cf1d293e24a58a 100644
--- a/docs/examples/agent/custom_agent.ipynb
+++ b/docs/examples/agent/custom_agent.ipynb
@@ -86,7 +86,7 @@
     "from llama_index.core.query_engine import RouterQueryEngine\n",
     "from llama_index.core import ChatPromptTemplate, PromptTemplate\n",
     "from llama_index.core.selectors import PydanticSingleSelector\n",
-    "from pydantic import Field, BaseModel"
+    "from llama_index.core.bridge.pydantic import Field, BaseModel"
    ]
   },
   {
@@ -162,7 +162,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from pydantic import PrivateAttr\n",
+    "from llama_index.core.bridge.pydantic import PrivateAttr\n",
     "\n",
     "\n",
     "class RetryAgentWorker(CustomSimpleAgentWorker):\n",