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

fixes and enable tool_runtime tests

Browse source
This commit is contained in:
Ashwin Bharambe 2025-05-24 07:27:53 -07:00
parent 9f7ed4be43
commit bc7901e3bd
4 changed files with 38 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -7,7 +7,14 @@
from contextlib import asynccontextmanager
from typing import Any
import exceptiongroup
try:
# for python < 3.11
import exceptiongroup
BaseExceptionGroup = exceptiongroup.BaseExceptionGroup
except ImportError:
pass
import httpx
from mcp import ClientSession
from mcp import types as mcp_types
@ -34,7 +41,7 @@ async def sse_client_wrapper(endpoint: str, headers: dict[str, str]):
await session.initialize()
yield session
except BaseException as e:
if isinstance(e, exceptiongroup.BaseExceptionGroup):
if isinstance(e, BaseExceptionGroup):
for exc in e.exceptions:
if isinstance(exc, httpx.HTTPStatusError) and exc.response.status_code == 401:
raise AuthenticationRequiredError(exc) from exc