mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-21 00:21:49 +00:00
test(mcp): discover concrete tools through a catalog key
This commit is contained in:
parent
15b45839e1
commit
640b0e5fa9
1 changed files with 2 additions and 1 deletions
|
|
@ -179,7 +179,8 @@ def test_request_selected_mcp_guardrail_blocks_direct_and_virtual_calls(gateway:
|
|||
key_selected = scenario.key(object_permission=permission, guardrails=[guardrail])
|
||||
team = scenario.team(guardrails=[guardrail])
|
||||
team_selected = scenario.key(team_id=team, object_permission=permission)
|
||||
names = tool_names(candidate, key, identity)
|
||||
catalog_key = scenario.key(object_permission={"mcp_servers": [identity]})
|
||||
names = tool_names(candidate, catalog_key, identity)
|
||||
assert set(names) == {"add", "multiply", "fail"}
|
||||
for virtual in (False, True):
|
||||
for caller, selected, tool, expected in (
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue