diff --git a/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py b/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py index f7dc376f8..2fda60b0b 100644 --- a/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py +++ b/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py @@ -36,7 +36,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime): raise ValueError("mcp_endpoint is required") tools = [] - async with sse_client(mcp_endpoint.uri) as streams: + async with sse_client(mcp_endpoint["uri"]) as streams: async with ClientSession(*streams) as session: await session.initialize() tools_result = await session.list_tools() @@ -56,7 +56,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime): description=tool.description, parameters=parameters, metadata={ - "endpoint": mcp_endpoint.uri, + "endpoint": mcp_endpoint["uri"], }, ) )