tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-20 02:28:39 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files
litellm/proxy/guardrails/guardrail_hooks/presidio.py
+1 -1
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
litellm/types/proxy/guardrails/guardrail_hooks/presidio.py
+1 -1
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):