From b3bd1b6d6de21fa5f0649998685e522ebabfda60 Mon Sep 17 00:00:00 2001 From: Philip Date: Wed, 12 Mar 2025 14:37:19 -0400 Subject: [PATCH] Fix to allow mcp_endpoint from yaml config --- .../model_context_protocol/model_context_protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"], }, ) )