forked from phoenix/litellm-mirror
fix(presidio_pii_masking.py): support logging_only pii masking
This commit is contained in:
parent
9deb9b4e3f
commit
9d918d2ac7
5 changed files with 145 additions and 8 deletions
|
@ -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", {})
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue