From d913756844b5616330def450717729a8ed4a0375 Mon Sep 17 00:00:00 2001 From: Omar Abdelwahab Date: Thu, 13 Nov 2025 11:54:09 -0800 Subject: [PATCH] updated test_tools_with_schemas --- tests/integration/inference/test_tools_with_schemas.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/inference/test_tools_with_schemas.py b/tests/integration/inference/test_tools_with_schemas.py index 5b6e69ae3..8ceea3930 100644 --- a/tests/integration/inference/test_tools_with_schemas.py +++ b/tests/integration/inference/test_tools_with_schemas.py @@ -183,11 +183,11 @@ class TestMCPToolsInChatCompletion: uri = mcp_with_schemas["server_url"] try: - llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id) + llama_stack_client.toolgroups.unregister_toolgroup(toolgroup_id=test_toolgroup_id) except Exception: pass - llama_stack_client.toolgroups.register( + llama_stack_client.toolgroups.register_tool_group( toolgroup_id=test_toolgroup_id, provider_id="model-context-protocol", mcp_endpoint=dict(uri=uri),