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

feat(guardrails): add initialize_ztds callback initializer

Browse source
This commit is contained in:
Ilya Sibiryakov 2026-09-27 09:19:43 +03:00
parent 8e1c9a68f9
commit 50063f6698
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/proxy/guardrails/guardrail_initializers.py
View file

@ -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