From 50063f66981a34deb177fdb733084c57fcd6ba5f Mon Sep 17 00:00:00 2001 From: Ilya Sibiryakov Date: Sun, 27 Sep 2026 09:19:43 +0300 Subject: [PATCH] feat(guardrails): add initialize_ztds callback initializer --- litellm/proxy/guardrails/guardrail_initializers.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/litellm/proxy/guardrails/guardrail_initializers.py b/litellm/proxy/guardrails/guardrail_initializers.py index c422902d30d..5c04e4cf4ca 100644 --- a/litellm/proxy/guardrails/guardrail_initializers.py +++ b/litellm/proxy/guardrails/guardrail_initializers.py @@ -276,3 +276,14 @@ def initialize_panw_prisma_airs(litellm_params, guardrail): litellm.logging_callback_manager.add_litellm_callback(_panw_callback) return _panw_callback + + +def initialize_ztds(litellm_params: LitellmParams, guardrail: Guardrail): + from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail + + _ztds_object = ZTDSGuardrail( + reverse_on_output=getattr(litellm_params, "reverse_on_output", True), + ) + litellm.logging_callback_manager.add_litellm_callback(_ztds_object) + return _ztds_object +