add new GuardrailItem type

This commit is contained in:
Ishaan Jaff 2024-07-03 14:03:34 -07:00
parent 50443d3d48
commit a2b6baab16

View file

@ -0,0 +1,22 @@
from typing import Dict, List, Optional, TypedDict, Union
from pydantic import BaseModel, RootModel
"""
Pydantic object defining how to set guardrails on litellm proxy
litellm_settings:
guardrails:
- prompt_injection:
callbacks: [lakera_prompt_injection, prompt_injection_api_2]
default_on: true
- detect_secrets:
callbacks: [hide_secrets]
default_on: true
"""
class GuardrailItem(BaseModel):
callbacks: List[str]
default_on: bool
guardrail_name: str