scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 129

test(mcp): stub premium check in key-guardrail regression test

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-08-31 19:49:39 +00:00
parent feaab767d1
commit a0f2536e8b
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
View file

@ -10914,7 +10914,10 @@ class TestKeyAttachedGuardrailsReachTheMCPPreCallHook:
seen.update(data)
return None
with patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook):
with (
patch("litellm.proxy.utils._premium_user_check"),
patch.object(proxy_logging_obj, "pre_call_hook", side_effect=capture_pre_call_hook),
):
await MCPServerManager().pre_call_tool_check(
name="echo",
arguments={"text": "hi"},