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/litellm_core_utils/litellm_logging.py
+++ b/litellm/litellm_core_utils/litellm_logging.py
@ -655,6 +655,16 @@ class Logging:
                result=result, litellm_logging_obj=self
            )

+            ## LOGGING HOOK ##
+
+            for callback in callbacks:
+                if isinstance(callback, CustomLogger):
+                    self.model_call_details["input"], result = callback.logging_hook(
+                        kwargs=self.model_call_details,
+                        result=result,
+                        call_type=self.call_type,
+                    )
+
            for callback in callbacks:
                try:
                    litellm_params = self.model_call_details.get("litellm_params", {})