scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

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