Idempotency

import os, time, uuid
import requests

API = "https://api.onlyx.ai/v1"
HEADERS = {"Authorization": f"Bearer {os.environ['ONLYX_API_KEY']}"}


def send_once(conversation_id, body, key=None, attempts=5):
    key = key or str(uuid.uuid4())  # store this with your draft before the first attempt
    for attempt in range(attempts):
        try:
            r = requests.post(
                f"{API}/conversations/{conversation_id}/messages",
                headers={**HEADERS, "Idempotency-Key": key},
                json=body,
                timeout=30,
            )
        except requests.RequestException:
            time.sleep(2 ** attempt)
            continue  # same key: safe
        if r.status_code in (429, 409, 500, 502, 503, 504):
            try:
                code = r.json().get("error", {}).get("code")
            except ValueError:
                code = None
            if code == "IDEMPOTENCY_KEY_REUSED":
                raise RuntimeError("This key was used for a different message.")
            if r.status_code == 409 and code != "IDEMPOTENCY_IN_PROGRESS":
                r.raise_for_status()  # e.g. SALES_OPTED_OUT: do not retry
            time.sleep(int(r.headers.get("Retry-After", 2 ** attempt)))
            continue
        r.raise_for_status()
        replayed = r.headers.get("Idempotent-Replayed") == "true"
        return r.json(), replayed
    raise RuntimeError(f"No answer after {attempts} attempts; retry later with key {key}")

const API = "https://api.onlyx.ai/v1";
const sleep = (ms) => new Promise((r) => setTimeout(r, ms));

export async function sendOnce(conversationId, body, key = crypto.randomUUID(), attempts = 5) {
  // Store `key` with your draft before the first attempt.
  for (let attempt = 0; attempt < attempts; attempt++) {
    let res;
    try {
      res = await fetch(`${API}/conversations/${conversationId}/messages`, {
        method: "POST",
        headers: {
          Authorization: `Bearer ${process.env.ONLYX_API_KEY}`,
          "Content-Type": "application/json",
          "Idempotency-Key": key,
        },
        body: JSON.stringify(body),
      });
    } catch {
      await sleep(2 ** attempt * 1000);
      continue; // same key: safe
    }
    if ([429, 409, 500, 502, 503, 504].includes(res.status)) {
      const { error = {} } = await res.json().catch(() => ({}));
      if (error.code === "IDEMPOTENCY_KEY_REUSED") throw new Error("This key was used for a different message.");
      if (res.status === 409 && error.code !== "IDEMPOTENCY_IN_PROGRESS") throw new Error(`${error.code}: ${error.message}`);
      await sleep(Number(res.headers.get("Retry-After") ?? 2 ** attempt) * 1000);
      continue;
    }
    if (!res.ok) throw new Error(`${res.status} ${await res.text()}`);
    return { message: await res.json(), replayed: res.headers.get("Idempotent-Replayed") === "true" };
  }
  throw new Error(`No answer after ${attempts} attempts; retry later with key ${key}`);
}