Skip to content

Commit dd58d4f

Browse files
committed
pangea-sdk: make entities optional in AI Guard
1 parent 74a86ab commit dd58d4f

File tree

2 files changed

+25
-3
lines changed

2 files changed

+25
-3
lines changed

packages/pangea-sdk/CHANGELOG.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
77

88
## Unreleased
99

10+
### Changed
11+
12+
- AI Guard: `entities` from Redact is now an optional field.
13+
1014
## 6.10.0 - 2025-10-29
1115

1216
### Changed

packages/pangea-sdk/pangea/services/ai_guard.py

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,24 @@ class SecretsEntityResult(APIResponseModel):
283283
"""Detected redaction rules."""
284284

285285

286+
class RedactEntityResultItem(APIResponseModel):
287+
action: str
288+
"""The action taken on this Entity"""
289+
290+
redacted: bool
291+
292+
type: str
293+
294+
value: str
295+
296+
start_pos: Optional[int] = None
297+
298+
299+
class RedactEntityResult(APIResponseModel):
300+
entities: Optional[list[RedactEntityResultItem]] = None
301+
"""Detected redaction rules."""
302+
303+
286304
class LanguageDetectionResult(APIResponseModel):
287305
action: Optional[str] = None
288306
"""The action taken by this Detector"""
@@ -438,13 +456,13 @@ class GuardDetectors(APIResponseModel):
438456

439457
code: Optional[GuardDetector[CodeDetectionResult]] = None
440458
competitors: Optional[GuardDetector[object]] = None
441-
confidential_and_pii_entity: Optional[GuardDetector[PiiEntityResult]] = None
442-
custom_entity: Optional[GuardDetector[object]] = None
459+
confidential_and_pii_entity: Optional[GuardDetector[RedactEntityResult]] = None
460+
custom_entity: Optional[GuardDetector[RedactEntityResult]] = None
443461
language: Optional[GuardDetector[LanguageDetectionResult]] = None
444462
malicious_entity: Optional[GuardDetector[MaliciousEntityResult]] = None
445463
malicious_prompt: Optional[GuardDetector[PromptInjectionResult]] = None
446464
prompt_hardening: Optional[GuardDetector[object]] = None
447-
secret_and_key_entity: Optional[GuardDetector[SecretsEntityResult]] = None
465+
secret_and_key_entity: Optional[GuardDetector[RedactEntityResult]] = None
448466
topic: Optional[GuardDetector[TopicDetectionResult]] = None
449467

450468

0 commit comments

Comments
 (0)