Merge pull request #4669 from BerriAI/litellm_logging_only_masking

Flag for PII masking on Logging only
2024-07-11 22:03:37 -07:00 · 2024-07-11 22:03:37 -07:00 · d72bcdbce3
commit d72bcdbce3
parent f0b8c0e7fb 8d4e7f9967
11 changed files with 379 additions and 81 deletions
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@ -1469,12 +1469,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(