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

chore: apply black formatting

Browse source
This commit is contained in:
Panguard AI 2026-05-18 17:36:00 +08:00 • committed by Adam Lin
parent 0dc4417dbf
commit f58694e217
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/guardrails/guardrail_hooks/atr/atr.py
View file

@ -304,7 +304,9 @@ class ATRGuardrail(CustomGuardrail):
# Treat None or unrecognised severity conservatively (rank 0 = critical)
raw_severity = getattr(m, "severity", None)
severity_str = (raw_severity or "").lower() if raw_severity is not None else ""
severity_str = (
(raw_severity or "").lower() if raw_severity is not None else ""
)
rank = _SEVERITY_RANK.get(severity_str, 0)
if rank <= threshold_rank:
result.append(m)

8
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_atr.py
View file

@ -428,7 +428,9 @@ def test_scan_none_severity_treated_conservatively(fake_pyatr, tmp_path):
rules_dir = tmp_path / "rules"
rules_dir.mkdir()
none_severity_match = MagicMock(rule_id="ATR-600", title="Unknown sev", severity=None)
none_severity_match = MagicMock(
rule_id="ATR-600", title="Unknown sev", severity=None
)
engine.evaluate.return_value = [none_severity_match]
guard = ATRGuardrail(
@ -450,7 +452,9 @@ def test_scan_unknown_severity_treated_conservatively(fake_pyatr, tmp_path):
rules_dir = tmp_path / "rules"
rules_dir.mkdir()
unknown_match = MagicMock(rule_id="ATR-601", title="Future sev", severity="informational")
unknown_match = MagicMock(
rule_id="ATR-601", title="Future sev", severity="informational"
)
engine.evaluate.return_value = [unknown_match]
guard = ATRGuardrail(