phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(presidio_pii_masking.py): support logging_only pii masking

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-11 18:04:12 -07:00
parent abd682323c
commit 1a57e49e46
5 changed files with 145 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/integrations/custom_logger.py
View file

@ -90,9 +90,11 @@ class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callbac
):
pass
async def async_logging_hook(self):
"""For masking logged request/response"""
pass
async def async_logging_hook(
self, kwargs: dict, result: Any, call_type: str
) -> Tuple[dict, Any]:
"""For masking logged request/response. Return a modified version of the request/result."""
return kwargs, result
def logging_hook(
self, kwargs: dict, result: Any, call_type: str