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)
This commit is contained in:
Xi Yan 2025-03-25 16:06:03 -07:00 committed by GitHub
parent c8f740353b
commit 65d5d0d1bf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -2519,11 +2519,11 @@
"outputs": [], "outputs": [],
"source": [ "source": [
"# NBVAL_SKIP\n", "# 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", "client.toolgroups.register(\n",
" toolgroup_id=\"mcp::filesystem\",\n", " toolgroup_id=\"mcp::filesystem\",\n",
" provider_id=\"model-context-protocol\",\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",
")" ")"
] ]
}, },