cURL / HTTP

# pip install requests
import os, time, requests

BASE = "https://test-demo.pinerouter.ai"
H = {"Authorization": f"Bearer {os.environ['API_KEY']}"}

def run_job(model, input, every=5, max_wait=1800):
    r = requests.post(f"{BASE}/v1/jobs", headers=H, json={"model": model, "input": input}, timeout=60)
    body = r.json()
    if r.status_code != 200 or body.get("code") != 200:
        raise RuntimeError(f"start failed ({r.status_code}): {body}")
    task = body["data"]["id"]
    waited = 0
    while waited < max_wait:
        time.sleep(every)
        waited += every
        r = requests.get(f"{BASE}/v1/jobs/{task}", headers=H, timeout=60)
        if r.status_code == 429 or r.status_code >= 500:
            continue  # poll again; never start the job a second time
        d = r.json()["data"]
        if d["status"] == "succeeded":
            return task, d["outputs"]
        if d["status"] == "failed":
            raise RuntimeError(f"job {task} failed: {d.get('error')}")
    raise TimeoutError(f"job {task} still running; poll it later")

def download(urls, folder="."):
    paths = []
    for i, url in enumerate(urls):
        name = os.path.join(folder, f"{i}_{url.split('/')[-1].split('?')[0]}")
        with requests.get(url, stream=True, timeout=300) as r, open(name, "wb") as f:
            r.raise_for_status()
            for chunk in r.iter_content(1 << 20):
                f.write(chunk)
        paths.append(name)
    return paths

task, urls = run_job("z-image", {"prompt": "A red paper boat on a calm lake", "aspect_ratio": "1:1"})
print(download(urls))

// Node.js 18 or newer (built-in fetch)
import { writeFile } from "node:fs/promises";

const BASE = "https://test-demo.pinerouter.ai";
const H = { Authorization: `Bearer ${process.env.API_KEY}`, "Content-Type": "application/json" };
const sleep = (ms: number) => new Promise((r) => setTimeout(r, ms));

export async function runJob(model: string, input: Record<string, unknown>, { every = 5000, maxWait = 1_800_000 } = {}) {
  const r = await fetch(`${BASE}/v1/jobs`, { method: "POST", headers: H, body: JSON.stringify({ model, input }) });
  const body = await r.json();
  if (r.status !== 200 || body.code !== 200) throw new Error(`start failed (${r.status}): ${JSON.stringify(body)}`);
  const task: string = body.data.id;
  for (let waited = 0; waited < maxWait; waited += every) {
    await sleep(every);
    const p = await fetch(`${BASE}/v1/jobs/${task}`, { headers: H });
    if (p.status === 429 || p.status >= 500) continue; // poll again; never start the job a second time
    const d = (await p.json()).data;
    if (d.status === "succeeded") return { task, urls: d.outputs as string[] };
    if (d.status === "failed") throw new Error(`job ${task} failed: ${d.error}`);
  }
  throw new Error(`job ${task} still running; poll it later`);
}

export async function download(urls: string[]) {
  const paths: string[] = [];
  for (const [i, url] of urls.entries()) {
    const name = `${i}_${new URL(url).pathname.split("/").pop()}`;
    await writeFile(name, Buffer.from(await (await fetch(url)).arrayBuffer()));
    paths.push(name);
  }
  return paths;
}

const { urls } = await runJob("z-image", { prompt: "A red paper boat on a calm lake", aspect_ratio: "1:1" });
console.log(await download(urls));