Reverse Unscrubbing: Restoring Deterministic PII Tokens in RAM

import OpenAI from 'openai';
import { sanitize, restore } from '@privacyscrubber/sdk';

const openai = new OpenAI();

// 1. Raw prompt containing sensitive data
const rawPrompt = 'Draft a formal confirmation for client Eleanor Vance (Email: evance@heritage.org, Phone: 212-555-0198).';

// 2. Pre-Prompt Scrubbing in RAM (<0.8ms)
const { scrubbedText, tokenMap } = sanitize(rawPrompt, {
  profile: 'General'
});

console.log('Sanitized Prompt Dispatched to OpenAI:');
console.log(scrubbedText);
// "Draft a formal confirmation for client [NAME_1] (Email: [EMAIL_1], Phone: [PHONE_1])."

// 3. Call OpenAI model with sanitized tokens
const completion = await openai.chat.completions.create({
  model: 'gpt-4o',
  messages: [{ role: 'user', content: scrubbedText }]
});

const aiResponse = completion.choices[0].message.content;
console.log('\nRaw Tokenized Model Response:');
console.log(aiResponse);
// "Dear [NAME_1], we have confirmed your contact details [EMAIL_1]..."

// 4. Reverse Unscrub in local RAM (<0.3ms)
const restoredResponse = restore(aiResponse, tokenMap).restoredText;

console.log('\nFinal Restored Cleartext Response:');
console.log(restoredResponse);
// "Dear Eleanor Vance, we have confirmed your contact details evance@heritage.org..."