scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-16 04:29:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 138

style(presidio): apply ruff formatting to cluster resolution

Browse source
This commit is contained in:
musi22 2026-09-21 23:34:09 +05:30
parent 07db16d71f
commit 36e21f22c7
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/proxy/guardrails/guardrail_hooks/presidio.py
View file

@ -702,9 +702,7 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail):
valid_candidates: Final = tuple(
c
for c in analyze_results
if c.get("start") is not None
and c.get("end") is not None
and int(c["start"]) < int(c["end"])
if c.get("start") is not None and c.get("end") is not None and int(c["start"]) < int(c["end"])
)
if not valid_candidates:
return ()
@ -739,12 +737,14 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail):
int(x["end"]) - int(x["start"]),
),
)
resolved: Final = MappingProxyType({
"entity_type": best_candidate.get("entity_type") or "UNKNOWN",
"start": cluster_start,
"end": cluster_end,
"score": max(float(s.get("score") or 0) for s in cluster),
})
resolved: Final = MappingProxyType(
{
"entity_type": best_candidate.get("entity_type") or "UNKNOWN",
"start": cluster_start,
"end": cluster_end,
"score": max(float(s.get("score") or 0) for s in cluster),
}
)
return resolved
return tuple(_resolve_cluster(c) for c in clusters)