scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

fix(guardrails): address TrendAI lint budget

Browse source
This commit is contained in:
Oliver Fei 2026-09-24 16:01:49 -04:00
parent b6e4e45a39
commit 721d1efe03
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/guardrails/guardrail_hooks/trendai/__init__.py
View file

@ -16,7 +16,7 @@ def _normalize_event_hook(
if isinstance(mode, str):
return GuardrailEventHooks(mode)
if isinstance(mode, list):
return [GuardrailEventHooks(item) for item in mode]
return [GuardrailEventHooks(item) for item in mode] # mutable-ok: guardrail event hooks require a list
return mode
@ -41,7 +41,11 @@ def initialize_guardrail(litellm_params: "LitellmParams", guardrail: "Guardrail"
return callback
guardrail_initializer_registry: Final = {GUARDRAIL_NAME: initialize_guardrail}
guardrail_class_registry: Final = {GUARDRAIL_NAME: TrendAIGuardrail}
guardrail_initializer_registry: Final = { # mutable-ok: registry discovery requires a dict
GUARDRAIL_NAME: initialize_guardrail
}
guardrail_class_registry: Final = { # mutable-ok: registry discovery requires a dict
GUARDRAIL_NAME: TrendAIGuardrail
}
__all__ = ("TrendAIGuardrail",)