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

fix: update PresidioAnalyzeRequest type to support Union entities

Browse Source
This commit is contained in:
Arshdeep Singh 2025-09-25 13:00:05 -04:00
parent 471f011ae5
commit cb14bb341d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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

2
litellm/types/proxy/guardrails/guardrail_hooks/presidio.py
View File

@ -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):