From 8c595bebc1297e6a93ac28a1609f5bcba0ac8936 Mon Sep 17 00:00:00 2001 From: Omar Abdelwahab Date: Fri, 14 Nov 2025 14:24:17 -0800 Subject: [PATCH] using llama stack client's list_tools --- tests/integration/tool_runtime/test_mcp.py | 11 ++++------- .../integration/tool_runtime/test_mcp_json_schema.py | 8 +++++--- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/tests/integration/tool_runtime/test_mcp.py b/tests/integration/tool_runtime/test_mcp.py index 8d48a1da9..e22e96579 100644 --- a/tests/integration/tool_runtime/test_mcp.py +++ b/tests/integration/tool_runtime/test_mcp.py @@ -38,13 +38,10 @@ def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server): # Phase 2: Use the dedicated authorization parameter (no more provider_data headers) # This tests direct tool_runtime.invoke_tool API calls - # Without authorization, should get Unauthorized error - with pytest.raises(Exception, match="Unauthorized"): - llama_stack_client.tools.list(toolgroup_id=test_toolgroup_id) - - # With authorization parameter, should succeed - tools_list = llama_stack_client.tools.list( - toolgroup_id=test_toolgroup_id, + # Note: tools.list() is the ToolGroups API and doesn't have authorization parameter + # Use tool_runtime.list_tools() for authorization support + tools_list = llama_stack_client.tool_runtime.list_tools( + tool_group_id=test_toolgroup_id, authorization=AUTH_TOKEN, # Use dedicated authorization parameter ) assert len(tools_list) == 2 diff --git a/tests/integration/tool_runtime/test_mcp_json_schema.py b/tests/integration/tool_runtime/test_mcp_json_schema.py index 4fbc72fb6..9c0bbcf0b 100644 --- a/tests/integration/tool_runtime/test_mcp_json_schema.py +++ b/tests/integration/tool_runtime/test_mcp_json_schema.py @@ -343,9 +343,11 @@ class TestAgentWithMCPTools: mcp_endpoint=dict(uri=uri), ) - # Use the dedicated authorization parameter - tools_list = llama_stack_client.tools.list( - toolgroup_id=test_toolgroup_id, + # Phase 2: Use the dedicated authorization parameter + # Note: tools.list() is the ToolGroups API and doesn't have authorization parameter + # Use tool_runtime.list_tools() instead + tools_list = llama_stack_client.tool_runtime.list_tools( + tool_group_id=test_toolgroup_id, authorization=AUTH_TOKEN, ) tool_defs = [