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

cleanup: fix nits from review

Browse source 
Signed-off-by: Calum Murray <cmurray@redhat.com>
This commit is contained in:
Calum Murray 2025-07-03 11:30:44 -04:00
parent ffbea13e07
commit 033160b277
No known key found for this signature in database
GPG key ID: B67F01AEB13FE187
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -44,8 +44,7 @@ async def client_wrapper(endpoint: str, headers: dict[str, str]) -> AsyncGenerat
if err.response.status_code == 401:
raise AuthenticationRequiredError(exc) from exc
except* McpError:
logger.debug("failed to connect via streamable http, falling back to sse")
# fallback to sse
logger.warning("failed to connect via streamable http, falling back to sse")
try:
async with sse_client(endpoint, headers=headers) as streams:
async with ClientSession(*streams) as session:
@ -65,7 +64,7 @@ async def client_wrapper(endpoint: str, headers: dict[str, str]) -> AsyncGenerat
async def list_mcp_tools(endpoint: str, headers: dict[str, str]) -> ListToolDefsResponse:
tools = []
async with client_wrapper(endpoint, headers) as session:
logger.info("listing mcp tools...")
logger.debug("listing mcp tools")
tools_result = await session.list_tools()
for tool in tools_result.tools:
parameters = []
@ -94,7 +93,7 @@ async def invoke_mcp_tool(
endpoint: str, headers: dict[str, str], tool_name: str, kwargs: dict[str, Any]
) -> ToolInvocationResult:
async with client_wrapper(endpoint, headers) as session:
logger.info("invoking mcp tool")
logger.debug("invoking mcp tool")
result = await session.call_tool(tool_name, kwargs)
content: list[InterleavedContentItem] = []