phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix to allow mcp_endpoint from yaml config

Browse source
This commit is contained in:
Philip 2025-03-12 14:37:19 -04:00
parent 00da911167
commit b3bd1b6d6d
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py
View file

@ -36,7 +36,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
raise ValueError("mcp_endpoint is required") raise ValueError("mcp_endpoint is required")
tools = [] 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: async with ClientSession(*streams) as session:
await session.initialize() await session.initialize()
tools_result = await session.list_tools() tools_result = await session.list_tools()
@ -56,7 +56,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
description=tool.description, description=tool.description,
parameters=parameters, parameters=parameters,
metadata={ metadata={
"endpoint": mcp_endpoint.uri, "endpoint": mcp_endpoint["uri"],
}, },
) )
) )