WaveSpeedAI 替代方案

import { writeFile } from "node:fs/promises";

const baseURL = "https://api.api-stock.com/api/v1";
const apiKey = process.env.API_KEY;
if (!apiKey) throw new Error("Set API_KEY on the server");

async function request(path, options = {}) {
  const response = await fetch(baseURL + path, {
    ...options,
    headers: {
      Authorization: `Bearer ${apiKey}`,
      "Content-Type": "application/json",
    },
    signal: AbortSignal.timeout(30_000),
  });
  if (!response.ok) throw new Error(await response.text());
  return response.json();
}

const { taskId } = await request("/generation/create", {
  method: "POST",
  body: JSON.stringify({
  "model": "seedream-4-5",
  "prompt": "A ceramic mug on a wooden table, soft morning light",
  "resolution": "2K",
  "size": "1:1"
}),
});
if (!taskId) throw new Error("Missing taskId");

const deadline = Date.now() + 10 * 60_000;
let fileUrl;
while (Date.now() < deadline) {
  const { data } = await request(`/task/status/${encodeURIComponent(taskId)}`);
  if (data.status === "finished") {
    fileUrl = data.files?.[0]?.fileUrl;
    if (!fileUrl) throw new Error("Task finished without a file");
    break;
  }
  if (data.status === "failed" || data.status === "expired") {
    throw new Error(data.errorMessage || data.status);
  }
  await new Promise((resolve) => setTimeout(resolve, 2000));
}
if (!fileUrl) throw new Error(`Still processing: save taskId ${taskId} and resume polling`);
const file = await fetch(fileUrl, { signal: AbortSignal.timeout(60_000) });
if (!file.ok) throw new Error(`Download failed: ${file.status}`);
const mime = file.headers.get("content-type")?.split(";")[0];
const extension = { "image/png": "png", "image/jpeg": "jpg", "image/webp": "webp" }[mime] || "bin";
await writeFile(`generated-image.${extension}`, new Uint8Array(await file.arrayBuffer()));