phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix use guardrail for pre call hook

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-23 09:34:08 -07:00
parent 6e3f27cf69
commit a8e192a868
4 changed files with 30 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/custom_guardrail.py
View file

@ -18,16 +18,16 @@ class CustomGuardrail(CustomLogger):
super().__init__(**kwargs)
def should_run_guardrail(self, data, event_type: GuardrailEventHooks) -> bool:
metadata = data.get("metadata") or {}
requested_guardrails = metadata.get("guardrails") or []
verbose_logger.debug(
"inside should_run_guardrail for guardrail=%s event_type= %s guardrail_supported_event_hooks= %s",
"inside should_run_guardrail for guardrail=%s event_type= %s guardrail_supported_event_hooks= %s requested_guardrails= %s",
self.guardrail_name,
event_type,
self.event_hook,
requested_guardrails,
)
metadata = data.get("metadata") or {}
requested_guardrails = metadata.get("guardrails") or []
if self.guardrail_name not in requested_guardrails:
return False