PII Detection

from khaos.pii import PIIDetector

detector = PIIDetector()

# Scan a string for PII
result = detector.scan("Contact me at john@example.com or 555-123-4567")
print(f"PII found: {result.has_pii}")
print(f"Matches: {len(result.matches)}")
for match in result.matches:
    print(f"  {match.pattern_name} ({match.category}): {match.matched_text}")

# Quick boolean check
has_pii = detector.quick_check("No sensitive data here")
print(has_pii)  # False

# Mask PII in output
masked = detector.mask_text("SSN: 123-45-6789")
print(masked)  # "SSN: ***-**-****"

from khaos.pii import PIIPattern, PIICategory, PIIDetector
from khaos.pii.patterns import RiskLevel

# Define a custom pattern
employee_id_pattern = PIIPattern(
    name="employee_id",
    category=PIICategory.PERSONAL_ID,
    risk_level=RiskLevel.HIGH,
    pattern=r"EMP-\d{6}",
    mask_char="X",
    description="Internal employee ID format",
)

# Use it in a detector
detector = PIIDetector(custom_patterns=[employee_id_pattern])
result = detector.scan("Employee EMP-123456 reported the issue")
print(result.has_pii)  # True

from khaos.pii.detector import (
    DEFAULT_DETECTOR,
    AUTH_DETECTOR,
    FINANCIAL_DETECTOR,
    CRITICAL_DETECTOR,
)

# Use a pre-configured detector
result = AUTH_DETECTOR.scan(agent_response)
if result.has_pii:
    print(f"Credentials detected: {result.matches}")

# Financial-only scanning
result = FINANCIAL_DETECTOR.scan(agent_response)
if result.critical_count > 0:
    print("CRITICAL: Financial data exposed")

from khaos.pii import PIIDetector, mask_pii

detector = PIIDetector()

# Redact PII from a single response string
clean_response = detector.mask_text(agent_response)

# Redact PII across transcript messages
clean_transcript = [
    {**msg, "content": mask_pii(msg["content"])}
    for msg in transcript
]