fix(presidio_pii_masking.py): support logging_only pii masking

This commit is contained in:
Krrish Dholakia 2024-07-11 18:04:12 -07:00
parent 9deb9b4e3f
commit 9d918d2ac7
5 changed files with 145 additions and 8 deletions

View file

@ -1310,6 +1310,18 @@ class Logging:
result=result, litellm_logging_obj=self
)
## LOGGING HOOK ##
for callback in callbacks:
if isinstance(callback, CustomLogger):
self.model_call_details["input"], result = (
await callback.async_logging_hook(
kwargs=self.model_call_details,
result=result,
call_type=self.call_type,
)
)
for callback in callbacks:
# check if callback can run for this request
litellm_params = self.model_call_details.get("litellm_params", {})