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

refactor: introduce common 'ResourceNotFoundError' exception

Browse source 
Signed-off-by: Nathan Weinberg <nweinber@redhat.com>
This commit is contained in:
Nathan Weinberg 2025-08-03 17:54:44 -04:00
parent dfce05d0c5
commit 0eb55866f6
2 changed files with 33 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/integration/tool_runtime/test_registration.py
View file

@ -4,9 +4,12 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import re
import pytest
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.apis.common.errors import ToolGroupNotFoundError
from tests.common.mcp import MCP_TOOLGROUP_ID, make_mcp_server
@ -48,8 +51,18 @@ def test_register_and_unregister_toolgroup(llama_stack_client):
llama_stack_client.toolgroups.unregister(toolgroup_id=test_toolgroup_id)
# Verify it is unregistered
with pytest.raises(Exception, match=f"Tool group '{test_toolgroup_id}' not found"):
with pytest.raises(
ToolGroupNotFoundError,
match=re.escape(
f"Tool Group '{test_toolgroup_id}' not found. Use 'client.toolgroups.list()' to list available Tool Groups."
),
):
llama_stack_client.toolgroups.get(toolgroup_id=test_toolgroup_id)
with pytest.raises(Exception, match=f"Tool group '{test_toolgroup_id}' not found"):
with pytest.raises(
ToolGroupNotFoundError,
match=re.escape(
f"Tool Group '{test_toolgroup_id}' not found. Use 'client.toolgroups.list()' to list available Tool Groups."
),
):
llama_stack_client.tools.list(toolgroup_id=test_toolgroup_id)