Train an Adapter

IMPORT_ID="${IMPORT_ID:?set IMPORT_ID to an activated import}"

curl -fsS -X POST "https://api.zerodrift.ai/api/policies/import/$IMPORT_ID/train" \
  -H "x-api-key: $ZERODRIFT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"cost_cap_usd": 25, "examples_per_rule": 8}'

# Poll until training_status or status is succeeded or failed
while true; do
  RESP=$(curl -fsS "https://api.zerodrift.ai/api/policies/import/$IMPORT_ID/train" \
    -H "x-api-key: $ZERODRIFT_API_KEY")
  STATUS=$(printf '%s' "$RESP" | jq -r '.training_status // .status')
  echo "$STATUS"
  case "$STATUS" in
    succeeded|failed) break ;;
    queued) ;;
    *) printf '%s\n' "$RESP" >&2; exit 1 ;;
  esac
  sleep 10
done

printf '%s\n' "$RESP"
test "$STATUS" = "succeeded"

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\",
    \"document_category\": \"scenario_retail_investor_letter\",
    \"validation_scope\": {\"imports\": [\"$IMPORT_ID\"]}
  }"

import os
import time

import requests

API = "https://api.zerodrift.ai"
HEADERS = {
    "x-api-key": os.environ["ZERODRIFT_API_KEY"],
    "Content-Type": "application/json",
}
IMPORT_ID = os.environ["IMPORT_ID"]
URL = f"{API}/api/policies/import/{IMPORT_ID}/train"

started = requests.post(
    URL,
    headers=HEADERS,
    json={"cost_cap_usd": 25, "examples_per_rule": 8},
    timeout=30,
)
started.raise_for_status()
print(started.json())

while True:
    training = requests.get(URL, headers=HEADERS, timeout=30)
    training.raise_for_status()
    body = training.json()
    status = body.get("training_status") or body.get("status")
    print(status, body.get("stage"))
    if status in ("succeeded", "failed"):
        break
    time.sleep(10)

if (body.get("training_status") or body.get("status")) == "failed":
    raise SystemExit(body.get("error", "Training failed"))

enforced = requests.post(
    f"{API}/api/v3/content/validate",
    headers=HEADERS,
    json={
        "content": "Our fund guarantees 20% returns with zero risk!",
        "model_engine": "anchor_3_0",
        "mode": "sync",
        "document_category": "scenario_retail_investor_letter",
        "validation_scope": {"imports": [IMPORT_ID]},
    },
    timeout=30,
)
enforced.raise_for_status()
job = enforced.json()
if job.get("mode") == "async":
    poll = job["poll"]["url"]
    while True:
        job = requests.get(f"{API}{poll}", headers=HEADERS, timeout=30).json()
        if job.get("status") in ("completed", "done", "failed"):
            break
        time.sleep(2)
print(job.get("status"), (job.get("result") or job).get("overall_status"), job.get("served_by_adapter"))

const API = "https://api.zerodrift.ai";
const HEADERS = {
  "x-api-key": process.env.ZERODRIFT_API_KEY!,
  "Content-Type": "application/json",
};
const IMPORT_ID = process.env.IMPORT_ID!;
const URL = `${API}/api/policies/import/${IMPORT_ID}/train`;

const started = await fetch(URL, {
  method: "POST",
  headers: HEADERS,
  body: JSON.stringify({ cost_cap_usd: 25, examples_per_rule: 8 }),
});
if (!started.ok) throw new Error(`train ${started.status}`);
console.log(await started.json());

let body: {
  status?: string;
  training_status?: string;
  stage?: string;
  error?: string | null;
};
for (;;) {
  const r = await fetch(URL, { headers: HEADERS });
  if (!r.ok) throw new Error(`status ${r.status}`);
  body = await r.json();
  const status = body.training_status ?? body.status;
  console.log(status, body.stage);
  if (status === "succeeded" || status === "failed") break;
  await new Promise((res) => setTimeout(res, 10_000));
}

if ((body.training_status ?? body.status) === "failed") {
  throw new Error(body.error ?? "Training failed");
}

const enforced = await fetch(`${API}/api/v3/content/validate`, {
  method: "POST",
  headers: HEADERS,
  body: JSON.stringify({
    content: "Our fund guarantees 20% returns with zero risk!",
    model_engine: "anchor_3_0",
    mode: "sync",
    document_category: "scenario_retail_investor_letter",
    validation_scope: { imports: [IMPORT_ID] },
  }),
});
if (!enforced.ok) throw new Error(`enforce ${enforced.status}`);
let job = await enforced.json();
if (job.mode === "async") {
  const pollUrl = job.poll.url;
  for (;;) {
    const polled = await fetch(`${API}${pollUrl}`, { headers: HEADERS });
    job = await polled.json();
    if (["completed", "done", "failed"].includes(job.status)) break;
    await new Promise((res) => setTimeout(res, 2000));
  }
}
const verdict = job.result ?? job;
console.log(job.status, verdict.overall_status, job.served_by_adapter);