two version of R2R are hereHEADmaster

1 files changed, 23 insertions, 0 deletions

diff --git a/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/guardrail_registry.py b/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/guardrail_registry.py
new file mode 100644
index 00000000..6b43fd47
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/guardrail_registry.py
@@ -0,0 +1,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,
+}