From ef3dc143ec773e21f5ef16869b87a81714b1df07 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Sat, 12 Apr 2025 12:04:01 -0700
Subject: [PATCH] fix: test_registration was borked somehow

---
 tests/integration/tool_runtime/test_registration.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/integration/tool_runtime/test_registration.py b/tests/integration/tool_runtime/test_registration.py
index e04b56652..e4241d813 100644
--- a/tests/integration/tool_runtime/test_registration.py
+++ b/tests/integration/tool_runtime/test_registration.py
@@ -12,7 +12,6 @@ import httpx
 import mcp.types as types
 import pytest
 import uvicorn
-from llama_stack_client.types.shared_params.url import URL
 from mcp.server.fastmcp import Context, FastMCP
 from mcp.server.sse import SseServerTransport
 from starlette.applications import Starlette
@@ -97,7 +96,7 @@ def test_register_and_unregister_toolgroup(llama_stack_client, mcp_server):
     llama_stack_client.toolgroups.register(
         toolgroup_id=test_toolgroup_id,
         provider_id=provider_id,
-        mcp_endpoint=URL(uri=f"http://localhost:{port}/sse"),
+        mcp_endpoint=dict(uri=f"http://localhost:{port}/sse"),
     )
 
     # Verify registration