diff --git a/tests/integration/inference/test_tools_with_schemas.py b/tests/integration/inference/test_tools_with_schemas.py index 116e8ff4c..f6f15c0be 100644 --- a/tests/integration/inference/test_tools_with_schemas.py +++ b/tests/integration/inference/test_tools_with_schemas.py @@ -193,11 +193,6 @@ class TestMCPToolsInChatCompletion: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } - # Get the tools from MCP tools_response = llama_stack_client.tool_runtime.list_tools( tool_group_id=test_toolgroup_id, diff --git a/tests/integration/tool_runtime/test_mcp_json_schema.py b/tests/integration/tool_runtime/test_mcp_json_schema.py index 62e9844b4..cb713adec 100644 --- a/tests/integration/tool_runtime/test_mcp_json_schema.py +++ b/tests/integration/tool_runtime/test_mcp_json_schema.py @@ -123,10 +123,7 @@ class TestMCPSchemaPreservation: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + # List runtime tools response = llama_stack_client.tool_runtime.list_tools( @@ -166,10 +163,7 @@ class TestMCPSchemaPreservation: provider_id="model-context-protocol", mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + # List tools response = llama_stack_client.tool_runtime.list_tools( @@ -216,10 +210,7 @@ class TestMCPSchemaPreservation: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + response = llama_stack_client.tool_runtime.list_tools( tool_group_id=test_toolgroup_id, @@ -263,10 +254,7 @@ class TestMCPToolInvocation: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + # List tools to populate the tool index llama_stack_client.tool_runtime.list_tools( @@ -309,10 +297,7 @@ class TestMCPToolInvocation: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + # List tools to populate the tool index llama_stack_client.tool_runtime.list_tools( @@ -365,10 +350,7 @@ class TestAgentWithMCPTools: mcp_endpoint=dict(uri=uri), ) - # Authorization now passed as request body parameter - # Removed auth_headers - using authorization parameter instead - # (no longer needed) - } + tools_list = llama_stack_client.tools.list( toolgroup_id=test_toolgroup_id,