phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 02:12:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cleanup: remove unusable code

Browse source 
Signed-off-by: Calum Murray <cmurray@redhat.com>
This commit is contained in:
Calum Murray 2025-07-18 14:38:23 -04:00
parent ac0e3e133c
commit 211800146e
No known key found for this signature in database
GPG key ID: B67F01AEB13FE187
1 changed files with 2 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_stack/providers/utils/tools/mcp.py
View file

@ -48,16 +48,9 @@ async def client_wrapper(endpoint: str, headers: dict[str, str]) -> AsyncGenerat
for i, strategy in enumerate(connection_strategies): for i, strategy in enumerate(connection_strategies):
try: try:
if strategy == MCPProtol.STREAMABLE_HTTP: client = streamablehttp_client
client = streamablehttp_client if strategy == MCPProtol.SSE:
elif strategy == MCPProtol.SSE:
client = sse_client client = sse_client
else:
# this should not happen
logger.warning(
"tried to establish MCP connection with UNKNOWN protocol, defaulting to try with STREAMABLE_HTTP"
)
client = streamablehttp_client
async with client(endpoint, headers=headers) as client_streams: async with client(endpoint, headers=headers) as client_streams:
async with ClientSession(read_stream=client_streams[0], write_stream=client_streams[1]) as session: async with ClientSession(read_stream=client_streams[0], write_stream=client_streams[1]) as session:
await session.initialize() await session.initialize()