From cb14bb341d4caadbe7904b03f1b7e33f0c424554 Mon Sep 17 00:00:00 2001 From: Arshdeep Singh Date: Thu, 25 Sep 2025 13:00:05 -0400 Subject: [PATCH] fix: update PresidioAnalyzeRequest type to support Union entities --- litellm/proxy/guardrails/guardrail_hooks/presidio.py | 2 +- litellm/types/proxy/guardrails/guardrail_hooks/presidio.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/guardrails/guardrail_hooks/presidio.py b/litellm/proxy/guardrails/guardrail_hooks/presidio.py index d2ec0471e7..e3f114c746 100644 --- a/litellm/proxy/guardrails/guardrail_hooks/presidio.py +++ b/litellm/proxy/guardrails/guardrail_hooks/presidio.py @@ -173,7 +173,7 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail): analyze_payload["ad_hoc_recognizers"] = self.ad_hoc_recognizers if self.pii_entities_config: - analyze_payload["entities"] = list(self.pii_entities_config.keys()) # type: ignore + analyze_payload["entities"] = list(self.pii_entities_config.keys()) ################################################################## ######### End of adding config params diff --git a/litellm/types/proxy/guardrails/guardrail_hooks/presidio.py b/litellm/types/proxy/guardrails/guardrail_hooks/presidio.py index 5e480db482..89e0f39c43 100644 --- a/litellm/types/proxy/guardrails/guardrail_hooks/presidio.py +++ b/litellm/types/proxy/guardrails/guardrail_hooks/presidio.py @@ -9,7 +9,7 @@ class PresidioAnalyzeRequest(TypedDict, total=False): text: str language: Optional[str] ad_hoc_recognizers: Optional[List[str]] - entities: Optional[List[PiiEntityType]] + entities: Optional[List[Union[PiiEntityType, str]]] class PresidioAnalyzeResponseItem(TypedDict, total=False):