From b370fcd8de0993084c219ad6dd0146a80e6095cf Mon Sep 17 00:00:00 2001 From: Chesars Date: Fri, 20 Feb 2026 17:59:19 -0300 Subject: [PATCH] fix(presidio): remove redundant isinstance check for ModelResponseStream ModelResponseStream and ModelResponse are sibling classes (both inherit from ModelResponseBase), so the guard was always True. Simplify to just isinstance(response, ModelResponse). --- litellm/proxy/guardrails/guardrail_hooks/presidio.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/presidio.py b/litellm/proxy/guardrails/guardrail_hooks/presidio.py index d3f1fd17816..b4b25e2d909 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/presidio.py +++ b/litellm/proxy/guardrails/guardrail_hooks/presidio.py @@ -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}"