PII Detection & Masking

# Enable PII masking in ask
memvid ask memory.mv2 --question "What's John's contact info?" --mask-pii

John can be reached at john.smith@acme.com or by phone at (555) 867-5309.
His SSN for payroll is 123-45-6789.

John can be reached at [EMAIL] or by phone at [PHONE].
His SSN for payroll is [SSN].

# Search results with masking
memvid find memory.mv2 --query "contact information" --mask-pii

from memvid import use

mem = use('basic', 'memory.mv2')

# Ask with PII masking
response = mem.ask(
    "What is the customer's contact information?",
    mask_pii=True
)
print(response.answer)
# "Customer can be reached at [EMAIL] or [PHONE]"

# Check if content contains PII
from memvid import contains_pii, mask_pii

text = "Email me at test@example.com"
if contains_pii(text):
    safe_text = mask_pii(text)
    print(safe_text)  # "Email me at [EMAIL]"

import { use } from '@anthropics/memvid'

const mem = await use('basic', 'memory.mv2')

// Ask with PII masking
const response = await mem.ask(
  "What is the customer's contact information?",
  { maskPii: true }
)
console.log(response.answer)
// "Customer can be reached at [EMAIL] or [PHONE]"

from memvid import contains_pii

# Returns True if any PII detected
contains_pii("Call 555-123-4567")  # True
contains_pii("Hello world")        # False

from memvid import mask_pii

original = """
Contact: john@example.com
Phone: (555) 123-4567
SSN: 123-45-6789
API Key: sk-abc123xyz
"""

masked = mask_pii(original)
print(masked)

Contact: [EMAIL]
Phone: [PHONE]
SSN: [SSN]
API Key: [API_KEY]

from memvid import detect_pii

text = "Email john@test.com or call 555-1234"
pii_items = detect_pii(text)

for item in pii_items:
    print(f"Type: {item.type}, Value: {item.value}, Position: {item.start}-{item.end}")

# Type: email, Value: john@test.com, Position: 6-19
# Type: phone, Value: 555-1234, Position: 28-36

# Support bot with PII protection
response = mem.ask(
    ticket_content,
    mask_pii=True  # Don't expose customer PII
)

# Log safely
logger.info(f"Response: {response.answer}")  # No PII in logs

# Search medical records
results = mem.find("patient symptoms", mask_pii=True)

# Safe to display - no PHI exposed
for result in results:
    print(result.snippet)  # "[EMAIL]", "[PHONE]", "[SSN]" redacted

# Export masked data for dev/test
for frame in mem.timeline():
    masked_content = mask_pii(frame.text)
    dev_mem.put(masked_content)

def search_with_audit(query):
    results = mem.find(query)

    # Log masked version
    audit_log.info(f"Query: {mask_pii(query)}")
    audit_log.info(f"Results: {len(results)}")

    return results

# CLI: use --mask-pii flag
memvid ask memory.mv2 -q "..." --mask-pii

# Python: mask_pii=True parameter
mem.ask("...", mask_pii=True)

user@domain.com
user.name@subdomain.domain.co.uk
user+tag@domain.com

555-123-4567
(555) 123-4567
555.123.4567
+1 555 123 4567
15551234567

123-45-6789
123 45 6789
123456789

4111-1111-1111-1111
4111 1111 1111 1111
4111111111111111

192.168.1.1
10.0.0.1
172.16.0.1

sk-abc123...
api_key_xxx...
Bearer eyJhbGciOiJ...
ghp_xxxxxxxxxxxx

# Might be flagged as phone
text = "Order #555-123-4567"  # Could be order number

# Might be flagged as SSN
text = "Version 123-45-6789"  # Could be version string

# Both masked the same way
"Call me at 555-123-4567"  # Real phone
"The code is 555-123-4567"  # Not a phone, but still masked

# Always mask when displaying to users
response = mem.ask(question, mask_pii=True)
display_to_user(response.answer)

# Raw data for internal processing
results = mem.find(query)  # No masking

# Masked for display
masked_results = [mask_pii(r.text) for r in results]

import logging

def safe_log(message):
    logging.info(mask_pii(message))

# Encrypt at rest + mask on output
memvid lock sensitive.mv2 --out sensitive.mv2e

# When using:
memvid unlock sensitive.mv2e --out temp.mv2
memvid ask temp.mv2 -q "..." --mask-pii
memvid lock temp.mv2 --out sensitive.mv2e
rm temp.mv2

from memvid import detect_pii

# Test with your actual data patterns
test_data = [
    "Contact: support@yourcompany.com",
    "Phone: 1-800-YOUR-NUM",
    "Account: CUST-12345",
]

for text in test_data:
    pii = detect_pii(text)
    print(f"Found: {len(pii)} PII items in: {text}")