mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 10:44:24 +00:00
ed283bc5b4
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 12s
* test_default_on_guardrail * update debug on custom guardrail * refactor guardrails init * guardrail registry * allow switching guardrails default_on * fix circle import issue * fix bedrock applying guardrails where content is a list * fix unused import * docs default on guardrail * docs fix per api key
23 lines
860 B
Python
23 lines
860 B
Python
# 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,
|
|
}
|