CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-07-02 01:55:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: keep PII detection separate from refusal metrics

Browse Source
This commit is contained in:
Edneam
2026-05-14 22:42:28 +05:30
parent d734067ef6
commit be7fb1f370
4 changed files with 69 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/refusal_classifier/test_pii_detector.py
+6
View File
@@ -33,6 +33,12 @@ class TestPIIDetector:
assert detector.detected_types("Contact me at jane@example.com") == []
assert detector.detected_types("card: 4111 1111 1111 1111") == ["credit_card"]
def test_credit_card_detection_can_be_disabled(self):
detector = PIIDetector(patterns=(), detect_credit_cards=False)
assert detector.detected_types("card: 4111 1111 1111 1111") == []
assert not detector.is_leak("card: 4111 1111 1111 1111")
def test_custom_patterns_can_be_used(self):
detector = PIIDetector(
patterns=(PIIPattern("employee_id", re.compile(r"EMP-\d{4}")),)