1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
# litellm/proxy/guardrails/guardrail_registry.py
from litellm.types.guardrails import SupportedGuardrailIntegrations
from .guardrail_initializers import (
initialize_aim,
initialize_aporia,
initialize_bedrock,
initialize_guardrails_ai,
initialize_hide_secrets,
initialize_lakera,
initialize_presidio,
)
guardrail_registry = {
SupportedGuardrailIntegrations.APORIA.value: initialize_aporia,
SupportedGuardrailIntegrations.BEDROCK.value: initialize_bedrock,
SupportedGuardrailIntegrations.LAKERA.value: initialize_lakera,
SupportedGuardrailIntegrations.AIM.value: initialize_aim,
SupportedGuardrailIntegrations.PRESIDIO.value: initialize_presidio,
SupportedGuardrailIntegrations.HIDE_SECRETS.value: initialize_hide_secrets,
SupportedGuardrailIntegrations.GURDRAILS_AI.value: initialize_guardrails_ai,
}
|
