scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(bedrock): update normalize_checks test to keep empty known checks

Browse source
This commit is contained in:
Mateo Wang 2026-06-24 08:34:53 -07:00 • committed by Sameer Kankute
parent 1a889ffa38
commit 1278012a91
No known key found for this signature in database
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_bedrock_invoke_guardrail_checks.py
View file

@ -63,11 +63,14 @@ def test_init_normalizes_checks_and_drops_unknown_keys():
g = BedrockGuardrail(
checks={
"contentFilter": {"categories": [{"category": "VIOLENCE"}]},
"unknownCheck": {"foo": "bar"},
"promptAttack": {}, # empty -> dropped
"unknownCheck": {"foo": "bar"}, # unknown key -> dropped
"promptAttack": {}, # empty known check -> kept (enable with defaults)
}
)
assert g.checks == {"contentFilter": {"categories": [{"category": "VIOLENCE"}]}}
assert g.checks == {
"contentFilter": {"categories": [{"category": "VIOLENCE"}]},
"promptAttack": {},
}
def test_init_empty_checks_falls_back_to_apply_mode():