Vercel AI SDK Stream Sanitizer: ZTDS™ In-Memory Masking

import { streamText } from 'ai';
import { openai } from '@ai-sdk/openai';
import { withPrivacyScrubber } from '@privacyscrubber/sdk/vercel-ai';

// 1. Wrap the provider model (1-line drop-in, zero cloud egress):
const safeModel = withPrivacyScrubber(openai('gpt-4o'), {
  profile: 'General', // or 'Dev', 'Financial', 'Medical'
  detectSecrets: true
});

export async function POST(req: Request) {
  const { messages } = await req.json();

  // 2. Stream completions: prompts are sanitized and responses de-tokenized in-flight
  const result = streamText({
    model: safeModel,
    messages,
    temperature: 0.2
  });

  return result.toDataStreamResponse();
}

import { streamText, wrapLanguageModel } from 'ai';
import { openai } from '@ai-sdk/openai';
import { createPrivacyMiddleware } from '@privacyscrubber/sdk/vercel-ai';

const safeModel = wrapLanguageModel({
  model: openai('gpt-4o'),
  middleware: createPrivacyMiddleware({
    profile: 'General',
    detectSecrets: true,
    onSanitized: (meta) => {
      console.log(`[ZTDS] Redacted ${meta.tokenCount} PII entities before model dispatch`);
    }
  })
});

export async function POST(req: Request) {
  const { messages } = await req.json();
  const result = streamText({ model: safeModel, messages });
  return result.toDataStreamResponse();
}