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

fix(ztds): add guardrail_name attribute and standalone fallback handling

Browse source
This commit is contained in:
Ilya Sibiryakov 2026-09-27 09:26:15 +03:00
parent 7dd8632471
commit d8aec05451
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -22,7 +22,8 @@ except ImportError:
# Standalone fallback when running outside full LiteLLM package
class CustomGuardrail:
def __init__(self, **kwargs: Any) -> None:
pass
for k, v in kwargs.items():
setattr(self, k, v)
class ZTDSGuardrail(CustomGuardrail):
@ -52,6 +53,7 @@ class ZTDSGuardrail(CustomGuardrail):
**kwargs: Any,
):
super().__init__(guardrail_name=guardrail_name, **kwargs)
self.guardrail_name = guardrail_name or kwargs.get("guardrail_name", "ztds")
self.enabled_entities = enabled_entities or list(self.PATTERNS.keys())
self.reverse_on_output = reverse_on_output
self.enforce_zero_egress = enforce_zero_egress