AWS Comprehend PII Alternative: In-Memory Latency Reduction (<1ms)

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

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });

// Replace heavy @aws-sdk/client-comprehend with 1 in-memory call
const rawPrompt = 'Customer Rebecca Vance (Phone: 415-555-0144, SSN: 901-22-3344) requested balance confirmation.';

// 1. Sanitize in local RAM in <0.82ms (Zero IAM roles, zero AWS network calls)
const { scrubbedText, tokenMap } = sanitize(rawPrompt, {
  profile: 'General',
  detectSecrets: true
});

console.log('Sanitized output:', scrubbedText);
// "Customer [NAME_1] (Phone: [PHONE_1], SSN: [SSN_1]) requested balance confirmation."

// 2. Invoke AI model with sanitized prompt (0 PII egress):
const completion = await openai.chat.completions.create({
  model: 'gpt-4o',
  messages: [{ role: 'user', content: scrubbedText }]
});
const llmResponse = completion.choices[0].message.content || '';

// 3. Restore authentic values in local RAM:
const cleartextResponse = restore(llmResponse, tokenMap).restoredText;
console.log(cleartextResponse);