Masking

1
import re
2
from deepeval.tracing import observe, trace_manager
3
4
def masking_function(data):
5
    if isinstance(data, str):
6
        data = re.sub(r'\b(?:\d{4}[- ]?){3}\d{4}\b', '[REDACTED CARD]', data)
7
        return data
8
    return data
9
10
trace_manager.configure(mask=masking_function)
11
12
@observe()
13
def llm_app(query: str):
14
    return "4242-4242-4242-4242"
15
16
llm_app("Test Masking")

1
import re
2
from typing import Any
3
4
def mask_credit_card(data: Any) -> Any:
5
    if isinstance(data, str):
6
        return re.sub(r'\b(?:\d{4}[- ]?){3}\d{4}\b', '[REDACTED CARD]', data)
7
    elif isinstance(data, list):
8
        return [mask_credit_card(item) for item in data]
9
    elif isinstance(data, dict):
10
        return {k: mask_credit_card(v) for k, v in data.items()}
11
    else:
12
        return data
13
14
15
print(mask_credit_card("My card number is 4111 1111 1111 1234."))
16
# Output: My card number is [REDACTED CARD].

1
import re
2
from typing import Any
3
4
def mask_email(data: Any) -> Any:
5
    if isinstance(data, str):
6
        return re.sub(r'\b([\w.%+-])[^\s@]*(@[\w.-]+\.\w+)', r'\1*****\2', data)
7
    elif isinstance(data, list):
8
        return [mask_email(item) for item in data]
9
    elif isinstance(data, dict):
10
        return {k: mask_email(v) for k, v in data.items()}
11
    else:
12
        return data
13
14
print(mask_email("Contact me at johndoe@example.com."))
15
# Output: Contact me at j*****@example.com.

1
import re
2
from typing import Any
3
4
def mask_bank_account(data: Any) -> Any:
5
    if isinstance(data, str):
6
        return re.sub(
7
            r'\b\d{6,}(?!\d)',
8
            lambda m: '*' * (len(m.group()) - 4) + m.group()[-4:],
9
            data
10
        )
11
    elif isinstance(data, list):
12
        return [mask_bank_account(item) for item in data]
13
    elif isinstance(data, dict):
14
        return {k: mask_bank_account(v) for k, v in data.items()}
15
    else:
16
        return data
17
18
19
print(mask_bank_account("My account is 9876543210."))
20
# Output: My account is ******3210.

1
import re
2
from typing import Any
3
4
def mask_passport_number(data: Any) -> Any:
5
    if isinstance(data, str):
6
        return re.sub(r'\b([A-Z])([A-Z0-9]{6,8})([A-Z0-9])\b', r'\1*******\3', data)
7
    elif isinstance(data, list):
8
        return [mask_passport_number(item) for item in data]
9
    elif isinstance(data, dict):
10
        return {k: mask_passport_number(v) for k, v in data.items()}
11
    return data
12
13
print(mask_passport_number("Passport: A12345678."))
14
# Output: Passport: A*******8.