diff --git a/tests/integration/inference/test_tools_with_schemas.py b/tests/integration/inference/test_tools_with_schemas.py index 8ceea3930..6a9882a15 100644 --- a/tests/integration/inference/test_tools_with_schemas.py +++ b/tests/integration/inference/test_tools_with_schemas.py @@ -179,6 +179,10 @@ class TestMCPToolsInChatCompletion: if not isinstance(llama_stack_client, LlamaStackAsLibraryClient): pytest.skip("Library client required for local MCP server") + # Check if the client has the register_tool_group method (older client versions) + if not hasattr(llama_stack_client.toolgroups, 'register_tool_group'): + pytest.skip("Client SDK doesn't support dynamic toolgroup registration - toolgroups must be configured in run.yaml") + test_toolgroup_id = "mcp::calc" uri = mcp_with_schemas["server_url"]