PII LLM Call

pii-llm.py

"""
Create a key here" https://platform.openai.com/settings/organization/api-keys

Setup:
    pip install openai

Environment variable:
    export OPENAI_API_KEY="your_api_key"
    
Use GPT-5 Mini, which would cost about ~$0.000085 per request 
    
"""

from openai import OpenAI

client = OpenAI()

# -------------------------
# SYSTEM PROMPT
# -------------------------

SYSTEM_PROMPT = """
You remove personally identifiable information (PII) from text.

Replace:
- Names with [PERSON]
- Phone numbers with [PHONE]
- Email addresses with [EMAIL]
- Street addresses with [ADDRESS]

Return only the cleaned text.
"""


# -------------------------
# USER INPUT
# -------------------------

post = """
Had a great time at dinner with Sarah Johnson last night.
Call me at 917-555-1234 or email me at randy@example.com.
"""


# -------------------------
# LLM CALL
# -------------------------

response = client.responses.create(
    model="gpt-5-mini",
    input=[
        {
            "role": "system",
            "content": SYSTEM_PROMPT,
        },
        {
            "role": "user",
            "content": post,
        },
    ],
)


"""
Expected output:

Had a great time at dinner with [PERSON] last night.
Call me at [PHONE] or email me at [EMAIL].
"""