Vercel AI Gateway

npm install ai@^6

import { Firewall, FirewallBlockedException, HookLabel } from "@silmaril-security/sdk"
import { gateway, generateText, wrapLanguageModel } from "ai"

const shadowMode = true

const fw = new Firewall({
  apiKey: process.env.SILMARIL_API_KEY!,
  apiUrl: process.env.SILMARIL_API_URL!,
  shadowMode,
})

const model = wrapLanguageModel({
  model: gateway("anthropic/claude-sonnet-4.6"),
  middleware: fw.asMiddleware({ scanOutput: false }),
})

const response = await generateText({
  model,
  prompt: "Summarize the benefits of request validation.",
})

const result = await fw.classify(response.text, { hook: HookLabel.LLM_OUTPUT })
const blocked =
  result.prediction === "MALICIOUS" || result.governance?.action === "block"

if (blocked && !shadowMode) {
  throw new FirewallBlockedException({
    score: result.score,
    threshold: result.threshold,
    promptText: response.text,
    hook: HookLabel.LLM_OUTPUT,
    result,
  })
}

console.log(response.text)