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

working lakera ai during call hook

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-20 14:39:04 -07:00
parent 1a142053e5
commit 1fdebfb0b7
5 changed files with 28 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/integrations/custom_guardrail.py
View file

@ -1,4 +1,4 @@
from typing import Literal
from typing import Literal, Optional
from litellm._logging import verbose_logger
from litellm.integrations.custom_logger import CustomLogger
@ -7,9 +7,14 @@ from litellm.types.guardrails import GuardrailEventHooks
class CustomGuardrail(CustomLogger):
def __init__(self, guardrail_name: str, event_hook: GuardrailEventHooks, **kwargs):
def __init__(
self,
guardrail_name: Optional[str] = None,
event_hook: Optional[GuardrailEventHooks] = None,
**kwargs
):
self.guardrail_name = guardrail_name
self.event_hook: GuardrailEventHooks = event_hook
self.event_hook: Optional[GuardrailEventHooks] = event_hook
super().__init__(**kwargs)
def should_run_guardrail(self, data, event_type: GuardrailEventHooks) -> bool: