From b714ad363fed32caca20ecfd40c3c9a0a7b94960 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fernando=20Su=C3=A1rez=20Jim=C3=A9nez?= <fsuarezj@gmail.com>
Date: Sun, 18 Feb 2024 00:30:04 +0300
Subject: [PATCH] Update custom_agent.ipynb (#10926)

Replacing import source from pydantic to llama_index.core.bridge.pydantic to avoid error explained in issue #10921
---
 docs/examples/agent/custom_agent.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/examples/agent/custom_agent.ipynb b/docs/examples/agent/custom_agent.ipynb
index c55a7c1a4c..7a26db4db1 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",
-- 
GitLab