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

Fix to issue 1586

Browse source
This commit is contained in:
Philip 2025-03-13 16:15:11 -04:00
parent a1bb7c8d82
commit fbe3c9c588
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -36,7 +36,12 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
raise ValueError("mcp_endpoint is required")
tools = []
async with sse_client(mcp_endpoint.uri) as streams:
if hasattr(mcp_endpoint, "__getitem__"):
if mcp_endpoint["uri"] is not None:
uri = mcp_endpoint["uri"]
else:
uri = mcp_endpoint.uri
async with sse_client(uri) as streams:
async with ClientSession(*streams) as session:
await session.initialize()
tools_result = await session.list_tools()
@ -56,7 +61,7 @@ class ModelContextProtocolToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime):
description=tool.description,
parameters=parameters,
metadata={
"endpoint": mcp_endpoint.uri,
"endpoint": uri,
},
)
)