From d8aec05451d78eeea61d811c0576e12b86de2437 Mon Sep 17 00:00:00 2001 From: Ilya Sibiryakov Date: Sun, 27 Sep 2026 09:26:15 +0300 Subject: [PATCH] fix(ztds): add guardrail_name attribute and standalone fallback handling --- litellm/proxy/guardrails/guardrail_hooks/ztds.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/ztds.py b/litellm/proxy/guardrails/guardrail_hooks/ztds.py index 85efbca26ad..52e5aec0575 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/ztds.py +++ b/litellm/proxy/guardrails/guardrail_hooks/ztds.py @@ -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