phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #4669 from BerriAI/litellm_logging_only_masking

Browse source 
Flag for PII masking on Logging only
This commit is contained in:
Krish Dholakia 2024-07-11 22:03:37 -07:00 • committed by GitHub
commit d72bcdbce3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 379 additions and 81 deletions
Download patch file Download diff file Expand all files Collapse all files

22
litellm/litellm_core_utils/litellm_logging.py
View file

@ -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", {})
@ -1302,6 +1312,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", {})