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

Properly handle unexpected decision values

Browse source
This commit is contained in:
clyang 2026-04-05 21:39:13 +08:00
parent 34ee1be71d
commit 4150be56dc
No known key found for this signature in database
GPG key ID: D52FE178A5104AD1
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/proxy/guardrails/guardrail_hooks/xecguard/xecguard.py
View file

@ -358,9 +358,13 @@ class XecGuardGuardrail(CustomGuardrail):
@staticmethod
def _raise_if_unsafe_scan(result: dict) -> None:
"""Raise HTTPException when XecGuard /scan returns UNSAFE."""
decision = result.get("decision", "SAFE")
if decision != "UNSAFE":
decision = result.get("decision", "")
if decision == "SAFE":
return
if decision != "UNSAFE":
verbose_proxy_logger.warning(
"XecGuard returned unexpected decision %r – treating as UNSAFE", decision
)
trace_id = result.get("trace_id", "n/a")
xecguard_result = result.get("xecguard_result", [])
@ -388,9 +392,13 @@ class XecGuardGuardrail(CustomGuardrail):
@staticmethod
def _raise_if_unsafe_grounding(result: dict) -> None:
"""Raise HTTPException when XecGuard /grounding returns UNSAFE."""
decision = result.get("decision", "SAFE")
if decision != "UNSAFE":
decision = result.get("decision", "")
if decision == "SAFE":
return
if decision != "UNSAFE":
verbose_proxy_logger.warning(
"XecGuard returned unexpected decision %r – treating as UNSAFE", decision
)
trace_id = result.get("trace_id", "n/a")
xr = result.get("xecguard_result")