diff --git a/litellm/types/guardrails.py b/litellm/types/guardrails.py index 7dd06a79b17..a5b5f5562c3 100644 --- a/litellm/types/guardrails.py +++ b/litellm/types/guardrails.py @@ -19,4 +19,19 @@ litellm_settings: class GuardrailItem(BaseModel): callbacks: List[str] default_on: bool + logging_only: Optional[bool] guardrail_name: str + + def __init__( + self, + callbacks: List[str], + default_on: bool, + guardrail_name: str, + logging_only: Optional[bool] = None, + ): + super().__init__( + callbacks=callbacks, + default_on=default_on, + logging_only=logging_only, + guardrail_name=guardrail_name, + )