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

fix(presidio_pii_masking.py): fix conditional check

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-13 22:11:03 -08:00
parent f68b656040
commit 6b91f48c64
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/hooks/presidio_pii_masking.py
View file

@ -154,8 +154,8 @@ class _OPTIONAL_PresidioPIIMasking(CustomLogger):
if litellm.output_parse_pii == False:
return response
if isinstance(response, ModelResponse) and isinstance(
response.choices, StreamingChoices
if isinstance(response, ModelResponse) and not isinstance(
response.choices[0], StreamingChoices
): # /chat/completions requests
if isinstance(response.choices[0].message.content, str):
verbose_proxy_logger.debug(