Quickstart

import { randomUUID } from "node:crypto";

const API = process.env.RETRIEVER_API_URL;
const headers = {
  Authorization: `Bearer ${process.env.RETRIEVER_API_KEY}`,
  "Content-Type": "application/json",
};

async function call(path, init = {}) {
  const res = await fetch(`${API}${path}`, {
    ...init,
    headers: { ...headers, ...init.headers },
    signal: AbortSignal.timeout(30_000),
  });
  const body = await res.json();
  if (!res.ok) throw new Error(`${res.status} ${body.error?.code}: ${body.error?.message}`);
  return body;
}

// Any failure stops the script with the command that resumes it.
function stop(error, resume) {
  console.error(`Stopped: ${error.message}`);
  console.error(`Resume with: ${resume}`);
  process.exit(1);
}

// RUN_ID=... waits for an existing run. IDEMPOTENCY_KEY=... retries a creation
// whose answer was lost: the same key never creates a second run.
let runId = process.env.RUN_ID;
if (!runId) {
  const idempotencyKey = process.env.IDEMPOTENCY_KEY || randomUUID();
  console.log(`IDEMPOTENCY_KEY=${idempotencyKey}`);
  try {
    const created = await call("/v1/runs", {
      method: "POST",
      headers: { "Idempotency-Key": idempotencyKey },
      body: JSON.stringify({
        prompt: "Find 5 B2B SaaS startups based in Lyon, with their website.",
        output: { type: "tables", tables: [{ key: "startups" }] },
      }),
    });
    runId = created.id;
  } catch (error) {
    stop(error, `IDEMPOTENCY_KEY=${idempotencyKey} node quickstart.mjs`);
  }
  console.log(`RUN_ID=${runId}`);
}

// Only polls of a started run count toward the 10 minutes: a queued run
// starts on its own when a slot of your organization frees.
let run;
for (let polls = 0; ; ) {
  try {
    run = await call(`/v1/runs/${runId}`);
  } catch (error) {
    stop(error, `RUN_ID=${runId} node quickstart.mjs`);
  }
  console.log("status", run.status, run.progress?.phase ?? "", run.queue?.reason ?? "");
  if (!["queued", "running", "recovering"].includes(run.status)) break;
  if (run.status !== "queued" && ++polls > 40) {
    stop(new Error("still running after 10 min of execution; it continues on the server"), `RUN_ID=${runId} node quickstart.mjs`);
  }
  await new Promise((r) => setTimeout(r, 15_000));
}
if (run.status !== "completed") throw new Error(run.error?.message ?? run.status);

const page = await call(`/v1/runs/${run.id}/output/tables/startups/rows?limit=10`);
console.table(page.items.map((row) => row.values));

import os, sys, time, uuid, requests

API = os.environ["RETRIEVER_API_URL"]
HEADERS = {"Authorization": f"Bearer {os.environ['RETRIEVER_API_KEY']}"}

def call(method, path, headers=None, **kw):
    res = requests.request(method, f"{API}{path}", headers={**HEADERS, **(headers or {})}, timeout=30, **kw)
    body = res.json()
    if not res.ok:
        err = body.get("error") or {}
        raise RuntimeError(f"{res.status_code} {err.get('code')}: {err.get('message')}")
    return body

def stop(error, resume):
    """Any failure stops the script with the command that resumes it."""
    sys.exit(f"Stopped: {error}\nResume with: {resume}")

# RUN_ID=... waits for an existing run. IDEMPOTENCY_KEY=... retries a creation
# whose answer was lost: the same key never creates a second run.
run_id = os.environ.get("RUN_ID")
if not run_id:
    idempotency_key = os.environ.get("IDEMPOTENCY_KEY") or str(uuid.uuid4())
    print(f"IDEMPOTENCY_KEY={idempotency_key}", flush=True)
    try:
        run_id = call("POST", "/v1/runs", headers={"Idempotency-Key": idempotency_key}, json={
            "prompt": "Find 5 B2B SaaS startups based in Lyon, with their website.",
            "output": {"type": "tables", "tables": [{"key": "startups"}]},
        })["id"]
    except Exception as error:
        stop(error, f"IDEMPOTENCY_KEY={idempotency_key} python quickstart.py")
    print(f"RUN_ID={run_id}", flush=True)

# Only polls of a started run count toward the 10 minutes: a queued run
# starts on its own when a slot of your organization frees.
polls = 0
while True:
    try:
        run = call("GET", f"/v1/runs/{run_id}")
    except Exception as error:
        stop(error, f"RUN_ID={run_id} python quickstart.py")
    print("status", run["status"], (run.get("progress") or {}).get("phase"), (run.get("queue") or {}).get("reason"))
    if run["status"] not in ("queued", "running", "recovering"):
        break
    if run["status"] != "queued":
        polls += 1
        if polls > 40:
            stop("still running after 10 min of execution; it continues on the server", f"RUN_ID={run_id} python quickstart.py")
    time.sleep(15)

if run["status"] != "completed":
    raise RuntimeError((run.get("error") or {}).get("message", run["status"]))

page = call("GET", f"/v1/runs/{run_id}/output/tables/startups/rows", params={"limit": 10})
for row in page["items"]:
    print(row["values"])