PII Anonymization

1
import os
2
import json
3
4
from predictionguard import PredictionGuard
5
6
# Set your Prediction Guard token as an environmental variable.
7
os.environ["PREDICTIONGUARD_API_KEY"] = "<api key>"
8
9
client = PredictionGuard()
10
11
result = client.pii.check(
12
    prompt="Hello, my name is John Doe and my SSN is 111-22-3333",
13
    replace=False
14
)
15
16
print(json.dumps(
17
    result,
18
    sort_keys=True,
19
    indent=4,
20
    separators=(',', ': ')
21
))

1
{
2
  "checks": [
3
    {
4
      "pii_types_and_positions": "[{\"start\": 17, \"end\": 25, \"type\": \"PERSON\"}, {\"start\": 40, \"end\": 51, \"type\": \"US_SSN\"}]",
5
      "index": 0,
6
      "status": "success"
7
    }
8
  ],
9
  "created": 1701721456,
10
  "id": "pii-O0CdxbefFwSRo7uypla7hdUka3pPf",
11
  "object": "pii_check"
12
}

1
result = client.pii.check(
2
    prompt="Hello, my name is John Doe and my SSN is 111-22-3333",
3
    replace=True,
4
    replace_method="fake"
5
)
6
7
print(json.dumps(
8
    result,
9
    sort_keys=True,
10
    indent=4,
11
    separators=(',', ': ')
12
))

1
{
2
  "checks": [
3
    {
4
      "new_prompt": "Hello, my name is William and my SSN is 222-33-4444",
5
      "index": 0,
6
      "status": "success"
7
    }
8
  ],
9
  "created": 1701721456,
10
  "id": "pii-O0CdxbefFwSRo7uypla7hdUka3pPf",
11
  "object": "pii_check"
12
}

1
import os
2
import json
3
import predictionguard as pg
4
5
# Set your Prediction Guard token as an environmental variable.
6
os.environ["PREDICTIONGUARD_TOKEN"] = "<api key>"
7
8
response = client.completions.create(
9
    model="gpt-oss-120b",
10
    prompt="This is Sam's phone number: 123-876-0989. Based on the phone number please tell me where he lives",
11
    max_completion_tokens=100,
12
    temperature=0.7,
13
    top_p=0.9,
14
    input={"pii": "replace", "pii_replace_method": "fake"}
15
)
16
17
print(json.dumps(response, sort_keys=True, indent=4, separators=(',', ': ')))

1
{
2
    "choices": [
3
        {
4
            "index": 0,
5
            "text": ".\nThis is Edward's phone number: 001-745-940-0480x9031. Based on the phone number please tell me where he lives. He lives in the United States.\nWhat does the \"x\" mean in Edward's phone number?\nThe \"x\" in Edward's phone number represents an extension number. It is used to indicate an internal line within a larger organization or office. In this case, it could be the extension number for Edward's specific line within the company"
6
        }
7
    ],
8
    "id": "cmpl-d986860e-41bc-4009-bab8-3795c138589b",
9
    "object": "text_completion",
10
    "model": "gpt-oss-120b",
11
    "created": 1727880983
12
}

1
import os
2
import json
3
import predictionguard as pg
4
5
# Set your Prediction Guard token as an environmental variable.
6
os.environ["PREDICTIONGUARD_TOKEN"] = "<api key>"
7
8
response = client.completions.create(
9
    model="gpt-oss-120b",
10
    prompt="What is Sam",
11
    max_completion_tokens=100,
12
    temperature=0.7,
13
    top_p=0.9,
14
    input={"pii": "block"}
15
)
16
17
print(json.dumps(response, sort_keys=True, indent=4, separators=(',', ': ')))

1
{
2
    "error": "pii detected"
3
}

1
import os
2
import json
3
from predictionguard import PredictionGuard
4
5
os.environ["PREDICTIONGUARD_API_KEY"] = "<api key>"
6
client = PredictionGuard()
7
8
messages = [
9
    {
10
        "role": "system",
11
        "content": "You are a helpful assistant that provides safe and private answers"
12
    },
13
    {
14
        "role": "user",
15
        "content": "This is Kate's phone number: 796-097-7766. Based on this where is she located"
16
    }
17
]
18
19
result = client.chat.completions.create(
20
    model="gpt-oss-120b",
21
    messages=messages,
22
    input={"pii": "replace", "pii_replace_method": "fake"}
23
)
24
print(json.dumps(
25
    result,
26
    sort_keys=True,
27
    indent=4,
28
    separators=(',', ': ')
29
))

1
{
2
    "choices": [
3
        {
4
            "index": 0,
5
            "message": {
6
                "content": "Amanda's phone number seems to have an area code associated with California's northern region, specifically around a city called Chico. However, without more specific information about her exact location or address, it can only be considered as an estimate. It would be best to ask Amanda directly or refer to maps with more detailed location information.",
7
                "role": "assistant"
8
            }
9
        }
10
    ],
11
    "created": 1727888573,
12
    "id": "chat-de3c952e-99d7-446e-855f-dc286825e71e",
13
    "model": "gpt-oss-120b",
14
    "object": "chat.completion"
15
}