diff --git a/tests/integration/tool_runtime/test_mcp.py b/tests/integration/tool_runtime/test_mcp.py index b3d742a93..831186b15 100644 --- a/tests/integration/tool_runtime/test_mcp.py +++ b/tests/integration/tool_runtime/test_mcp.py @@ -31,7 +31,10 @@ def test_mcp_invocation(llama_stack_client, text_model_id, mcp_server): uri = mcp_server["server_url"] # registering should not raise an error anymore even if you don't specify the auth token - llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id) + try: + llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id) + except Exception: + pass llama_stack_client.toolgroups.register( toolgroup_id=test_toolgroup_id,