scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(health): key guardrails on test_model turns are an enterprise path, run that unit test as a premium user

Browse source
This commit is contained in:
Devin AI 2026-11-05 00:37:02 +00:00
parent 198a5c925b
commit f703af6621
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/unit/proxy/health_endpoints/test_health_endpoints.py
View file

@ -6355,7 +6355,10 @@ async def test_test_model_turns_pick_up_the_keys_guardrails_and_the_pinned_team_
hooks=hooks,
)
with pytest.raises(HTTPException, match="stopped before the provider"):
with (
patch("litellm.proxy.proxy_server.premium_user", True),
pytest.raises(HTTPException, match="stopped before the provider"),
):
await call(CHAT_COMPLETIONS_SCRIPT, CHAT_COMPLETIONS.turn_1, "turn-a")
assert hooks.guardrails == (["key-prompt-guard", "pii-mask"],)