feat(guardrails): Flag for PII Masking on Logging

Fixes https://github.com/BerriAI/litellm/issues/4580
2024-07-11 16:09:34 -07:00 · 2024-07-11 16:09:34 -07:00 · 9deb9b4e3f
commit 9deb9b4e3f
parent ca76d2fd72
7 changed files with 107 additions and 6 deletions
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@ -1467,12 +1467,14 @@ class ProxyConfig:
                            + CommonProxyErrors.not_premium_user.value
                        )

-                    initialize_guardrails(
+                    guardrail_name_config_map = initialize_guardrails(
                        guardrails_config=value,
                        premium_user=premium_user,
                        config_file_path=config_file_path,
                        litellm_settings=litellm_settings,
                    )
+
+                    litellm.guardrail_name_config_map = guardrail_name_config_map
                elif key == "callbacks":

                    initialize_callbacks_on_proxy(