Enforce a Chatbot

curl -X POST "https://api.zerodrift.ai/api/v3/content/validate" \
  -H "x-api-key: $ZERODRIFT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "content": "Our fund guarantees 20% returns with zero risk!",
    "model_engine": "anchor_3_0",
    "mode": "sync",
    "metadata": {
      "conversation_id": "conv_123",
      "surface": "chat"
    }
  }'

import os

import requests

API = "https://api.zerodrift.ai"
HEADERS = {"x-api-key": os.environ["ZERODRIFT_API_KEY"]}
FALLBACK = (
    "I can't help with that here. I've passed your question to a specialist "
    "who will follow up."
)


def payload(job: dict) -> dict:
    inner = job.get("result")
    return inner if isinstance(inner, dict) else job


def confident(value) -> bool:
    if isinstance(value, str):
        return value == "high"
    try:
        return float(value) >= 0.85
    except (TypeError, ValueError):
        return False


def enforce_reply(reply: str, conversation_id: str) -> str:
    r = requests.post(
        f"{API}/api/v3/content/validate",
        headers=HEADERS,
        json={
            "content": reply,
            "model_engine": "anchor_3_0",
            "mode": "sync",
            "metadata": {
                "conversation_id": conversation_id,
                "surface": "chat",
            },
        },
        timeout=30,
    )
    r.raise_for_status()
    job = r.json()

    # Long replies can fall back to async; poll or fail closed.
    if job.get("mode") == "async":
        return FALLBACK

    body = payload(job)
    status = body.get("overall_status")
    if status == "approved":
        return reply
    if status == "do not send":
        return FALLBACK

    text = reply
    items = body.get("violations_by_line") or body.get("violations") or []
    if not items:
        return FALLBACK
    for v in items:
        fix = v.get("best_fix") or v.get("fix") or {}
        action = fix.get("action") or "replace"
        suggested = fix.get("suggested_text")
        conf = fix.get("confidence", v.get("confidence"))
        if action != "replace" or not suggested or not confident(conf):
            return FALLBACK
        quote = v.get("quote") or v.get("line_text")
        if v.get("line_number"):
            lines = text.split("\n")
            lines[v["line_number"] - 1] = suggested
            text = "\n".join(lines)
        elif quote and quote in text:
            text = text.replace(quote, suggested)
        else:
            return FALLBACK
    return text


print(enforce_reply("Our fund guarantees 20% returns with zero risk!", "conv_123"))

const API = "https://api.zerodrift.ai";
const FALLBACK =
  "I can't help with that here. I've passed your question to a specialist who will follow up.";

function payload(job: any) {
  return job?.result && typeof job.result === "object" ? job.result : job;
}

function confident(value: unknown) {
  if (typeof value === "string") return value === "high";
  const n = Number(value);
  return Number.isFinite(n) && n >= 0.85;
}

export async function enforceReply(
  reply: string,
  conversationId: string,
): Promise<string> {
  const r = await fetch(`${API}/api/v3/content/validate`, {
    method: "POST",
    headers: {
      "x-api-key": process.env.ZERODRIFT_API_KEY!,
      "Content-Type": "application/json",
    },
    body: JSON.stringify({
      content: reply,
      model_engine: "anchor_3_0",
      mode: "sync",
      metadata: { conversation_id: conversationId, surface: "chat" },
    }),
  });
  if (!r.ok) throw new Error(`ZeroDrift ${r.status}`);
  const job = await r.json();

  // Long replies can fall back to async; poll or fail closed.
  if (job.mode === "async") return FALLBACK;
  const body = payload(job);
  if (body.overall_status === "approved") return reply;
  if (body.overall_status === "do not send") return FALLBACK;

  let text = reply;
  const items = body.violations_by_line ?? body.violations ?? [];
  if (!items.length) return FALLBACK;
  for (const v of items) {
    const fix = v.best_fix ?? v.fix ?? {};
    const action = fix.action ?? "replace";
    const suggested = fix.suggested_text;
    const conf = fix.confidence ?? v.confidence;
    if (action !== "replace" || !suggested || !confident(conf)) return FALLBACK;
    if (v.line_number) {
      const lines = text.split("\n");
      lines[v.line_number - 1] = suggested;
      text = lines.join("\n");
    } else {
      const quote = v.quote ?? v.line_text;
      if (quote && text.includes(quote)) {
        text = text.replaceAll(quote, suggested);
      } else {
        return FALLBACK;
      }
    }
  }
  return text;
}

console.log(
  await enforceReply("Our fund guarantees 20% returns with zero risk!", "conv_123"),
);