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

updated the comments

Browse source
This commit is contained in:
Omar Abdelwahab 2025-11-14 16:56:04 -08:00
parent daa5a79b24
commit 0d4fa16ab9
4 changed files with 10 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/inference/test_tools_with_schemas.py
View file

@ -191,7 +191,7 @@ class TestMCPToolsInChatCompletion:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# Use the dedicated authorization parameter
# Get the tools from MCP
tools_response = llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,

5
tests/integration/tool_runtime/test_mcp.py
View file

@ -35,11 +35,8 @@ def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server):
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter (no more provider_data headers)
# Use the dedicated authorization parameter (no more provider_data headers)
# This tests direct tool_runtime.invoke_tool API calls
# 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

16
tests/integration/tool_runtime/test_mcp_json_schema.py
View file

@ -120,7 +120,7 @@ class TestMCPSchemaPreservation:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# Use the dedicated authorization parameter
# List runtime tools
response = llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
@ -160,7 +160,7 @@ class TestMCPSchemaPreservation:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# Use the dedicated authorization parameter
# List tools
response = llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
@ -206,7 +206,7 @@ class TestMCPSchemaPreservation:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# Use the dedicated authorization parameter
response = llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
authorization=AUTH_TOKEN,
@ -249,8 +249,7 @@ class TestMCPToolInvocation:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# List tools to populate the tool index
# Use the dedicated authorization parameter
llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
authorization=AUTH_TOKEN,
@ -291,8 +290,7 @@ class TestMCPToolInvocation:
mcp_endpoint=dict(uri=uri),
)
# Phase 2: Use the dedicated authorization parameter
# List tools to populate the tool index
# Use the dedicated authorization parameter
llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
authorization=AUTH_TOKEN,
@ -343,9 +341,7 @@ class TestAgentWithMCPTools:
mcp_endpoint=dict(uri=uri),
)
# 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
# Use the dedicated authorization parameter
tools_list = llama_stack_client.tool_runtime.list_tools(
tool_group_id=test_toolgroup_id,
authorization=AUTH_TOKEN,