ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(presidio): remove redundant isinstance check for ModelResponseStream

Browse Source 
ModelResponseStream and ModelResponse are sibling classes (both inherit
from ModelResponseBase), so the guard was always True. Simplify to just
isinstance(response, ModelResponse).
This commit is contained in:
Chesars 2026-02-20 17:59:19 -03:00
parent a2cae0070e
commit b370fcd8de
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/proxy/guardrails/guardrail_hooks/presidio.py
View File

@ -806,9 +806,7 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail):
if self.output_parse_pii is False and litellm.output_parse_pii is False:
return response
if isinstance(response, ModelResponse) and not isinstance(
response, ModelResponseStream
): # /chat/completions requests
if isinstance(response, ModelResponse): # /chat/completions requests
if isinstance(response.choices[0].message.content, str):
verbose_proxy_logger.debug(
f"self.pii_tokens: {self.pii_tokens}; initial response: {response.choices[0].message.content}"