From 65d5d0d1bf2c30a7f78352a4f023d731646b54f7 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Tue, 25 Mar 2025 16:06:03 -0700
Subject: [PATCH] fix: fix imports for mcp registration in notebook (#1787)

# What does this PR do?
- as title

[//]: # (If resolving an issue, uncomment and update the line below)
[//]: # (Closes #[issue-number])

## Test Plan
notebook

[//]: # (## Documentation)
---
 docs/getting_started.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/getting_started.ipynb b/docs/getting_started.ipynb
index 05b0d1357..cdf899411 100644
--- a/docs/getting_started.ipynb
+++ b/docs/getting_started.ipynb
@@ -2519,11 +2519,11 @@
       "outputs": [],
       "source": [
         "# NBVAL_SKIP\n",
-        "from llama_stack_client.types.shared_params.url import URL\n",
+        "from llama_stack_client.types.tool_group import McpEndpoint\n",
         "client.toolgroups.register(\n",
         "    toolgroup_id=\"mcp::filesystem\",\n",
         "    provider_id=\"model-context-protocol\",\n",
-        "    mcp_endpoint=URL(uri=\"http://localhost:8000/sse\"),\n",
+        "    mcp_endpoint=McpEndpoint(uri=\"http://localhost:8000/sse\"),\n",
         ")"
       ]
     },