# 需要安装 curl 和 jq；把占位符替换为平台 API Key。
export TOKENOPS_API_KEY="<API-KEY>"
BASE_URL="https://model-api.skyengine.com.cn"

# 1. 提交一次并保存原始 jobId。
SUBMITTED=$(curl --fail-with-body --max-time 60 -sS \
  "$BASE_URL/v1/tob/diffusion" \
  -H "Authorization: Bearer $TOKENOPS_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"model":"midjourney","text":"A red ceramic mug on a white table --v 7 --ar 1:1 --raw --seed 42"}') || exit 1
JOB_ID=$(printf '%s' "$SUBMITTED" | jq -er '.id | select(type == "string" and length > 0)') || exit 1
echo "jobId=$JOB_ID"

# 2. 最多查询 360 次，间隔 5 秒；超时后继续查询同一个 jobId。
for attempt in $(seq 1 360); do
  RESULT=$(curl --fail-with-body --max-time 60 -sS \
    "$BASE_URL/v1/tob/job/$JOB_ID" \
    -H "Authorization: Bearer $TOKENOPS_API_KEY") || exit 1
  STATUS=$(printf '%s' "$RESULT" | jq -r '.status')
  case "$STATUS" in
    2) printf '%s' "$RESULT" | jq -er '.urls | select(type == "array" and length > 0) | .[]'; exit $? ;;
    3) printf '%s' "$RESULT" | jq -r '.comment'; exit 1 ;;
    0|1) sleep 5 ;;
    *) echo "Unexpected status: $STATUS"; exit 1 ;;
  esac
done
echo "仍未完成，请继续查询 jobId=$JOB_ID；不要因此重新提交生成请求。"

import os
import time
from urllib.parse import quote
import requests

BASE_URL = os.getenv("TOKENOPS_BASE_URL", "https://model-api.skyengine.com.cn").rstrip("/")
session = requests.Session()
session.headers["Authorization"] = f"Bearer {os.environ['TOKENOPS_API_KEY']}"


def submit(operation, payload):
    response = session.post(f"{BASE_URL}/v1/tob/{operation}", json=payload, timeout=60)
    response.raise_for_status()
    result = response.json()
    if not result.get("id"):
        raise RuntimeError("提交响应缺少任务 id")
    return result["id"]


def wait_job(job_id, timeout=1800):
    deadline = time.monotonic() + timeout
    while time.monotonic() < deadline:
        response = session.get(
            f"{BASE_URL}/v1/tob/job/{quote(job_id, safe='')}", timeout=60
        )
        response.raise_for_status()
        result = response.json()
        if result["status"] == 2:
            if not result.get("urls"):
                raise RuntimeError("任务成功但缺少图片地址，请保留任务 id 排查")
            return result
        if result["status"] == 3:
            raise RuntimeError(result.get("comment") or "生成失败")
        if result["status"] not in (0, 1):
            raise RuntimeError(f"未知任务状态：{result['status']}")
        time.sleep(5)
    raise TimeoutError(f"任务仍未完成；保留 {job_id}，稍后继续查询")


job_id = submit("diffusion", {
    "model": "midjourney",
    "text": "A red ceramic mug on a white table --v 7 --ar 1:1 --raw --seed 42",
})
print("jobId:", job_id)  # 在等待之前持久保存，方便超时或断线后恢复查询。
result = wait_job(job_id)
for url in result["urls"]:
    print(url)

// Node.js 18+；先设置 TOKENOPS_API_KEY 环境变量。
const baseUrl = (process.env.TOKENOPS_BASE_URL || "https://model-api.skyengine.com.cn").replace(/\/$/, "");
const apiKey = process.env.TOKENOPS_API_KEY;
if (!apiKey) throw new Error("请设置 TOKENOPS_API_KEY");

async function request(path, body) {
  const response = await fetch(`${baseUrl}${path}`, {
    method: body === undefined ? "GET" : "POST",
    headers: {
      Authorization: `Bearer ${apiKey}`,
      "Content-Type": "application/json",
    },
    body: body === undefined ? undefined : JSON.stringify(body),
    signal: AbortSignal.timeout(60000),
  });
  if (!response.ok) throw new Error(`HTTP ${response.status}: ${await response.text()}`);
  return response.json();
}

async function waitJob(jobId, timeoutMs = 1800000) {
  const deadline = Date.now() + timeoutMs;
  while (Date.now() < deadline) {
    const result = await request(`/v1/tob/job/${encodeURIComponent(jobId)}`);
    if (result.status === 2) {
      if (!Array.isArray(result.urls) || result.urls.length === 0) throw new Error("任务成功但缺少图片地址");
      return result;
    }
    if (result.status === 3) throw new Error(result.comment || "生成失败");
    if (result.status !== 0 && result.status !== 1) throw new Error("未知任务状态");
    await new Promise(resolve => setTimeout(resolve, 5000));
  }
  throw new Error(`任务仍未完成；保留 ${jobId}，稍后继续查询`);
}

const submitted = await request("/v1/tob/diffusion", {
  model: "midjourney",
  text: "A red ceramic mug on a white table --v 7 --ar 1:1 --raw --seed 42",
});
if (!submitted.id) throw new Error("提交响应缺少任务 id");
console.log("jobId:", submitted.id);
const result = await waitJob(submitted.id);
result.urls.forEach(url => console.log(url));

curl --fail-with-body -sS "https://model-api.skyengine.com.cn/v1/tob/variation" \
  -H "Authorization: Bearer <API-KEY>" \
  -H "Content-Type: application/json" \
  -d '{"jobId":"<source_job_id>","imageNo":0,"type":1}'

# 复用快速开始中的 submit、wait_job。
remix_id = submit("remix", {
    "jobId": job_id,
    "imageNo": 0,
    "remixPrompt": "A blue ceramic mug on a white table",
    "mode": 1,
})
print("remix jobId:", remix_id)
remix_result = wait_job(remix_id)
print(remix_result["urls"])

// 复用快速开始中的 request、waitJob。
const pan = await request("/v1/tob/pan", {
  jobId: submitted.id,
  imageNo: 0,
  direction: 1,
  scale: 1.5,
});
console.log("pan jobId:", pan.id);
const panResult = await waitJob(pan.id);
console.log(panResult.urls);