export BASE_URL="https://api.apiyi.com"
export API_TOKEN="sk-xxxxxxxx"

# 创建任务
RESP=$(curl -sS -X POST "$BASE_URL/v1/videos" \
  -H "Authorization: Bearer $API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "veo-3.1-fast-generate-preview",
    "prompt": "黄昏海边的灯塔,镜头缓慢推进,海浪轻拍礁石,电影级光影",
    "duration": "4",
    "size": "1280x720",
    "metadata": {"resolution": "720p", "aspectRatio": "16:9"}
  }')
TASK_ID=$(echo "$RESP" | python3 -c 'import sys,json;print(json.load(sys.stdin)["task_id"])')
echo "task_id=$TASK_ID"

# 轮询
while :; do
  S=$(curl -sS -H "Authorization: Bearer $API_TOKEN" "$BASE_URL/v1/videos/$TASK_ID")
  ST=$(echo "$S" | python3 -c 'import sys,json;print(json.load(sys.stdin)["status"])')
  echo "status=$ST"
  [ "$ST" = "completed" ] && break
  [ "$ST" = "failed" ] && { echo "$S"; exit 1; }
  sleep 8
done

# 下载
curl -sSL -H "Authorization: Bearer $API_TOKEN" \
  "$BASE_URL/v1/videos/$TASK_ID/content" -o demo.mp4

import time, json, urllib.request

BASE = "https://api.apiyi.com"
KEY  = "sk-..."

def req(method, path, body=None):
    h = {"Authorization": f"Bearer {KEY}"}
    data = None
    if body is not None:
        h["Content-Type"] = "application/json"
        data = json.dumps(body).encode()
    r = urllib.request.Request(BASE + path, data=data, method=method, headers=h)
    return json.loads(urllib.request.urlopen(r, timeout=120).read())

create = req("POST", "/v1/videos", {
    "model": "veo-3.1-fast-generate-preview",
    "prompt": "傍晚海边的灯塔,镜头缓慢推进,稳定运镜",
    "duration": "4",
    "size": "1280x720",
    "metadata": {"resolution": "720p", "aspectRatio": "16:9"},
})
tid = create["task_id"]

while True:
    s = req("GET", f"/v1/videos/{tid}")
    if s["status"] in {"completed", "succeeded"}:
        break
    if s["status"] == "failed":
        raise RuntimeError(s)
    time.sleep(8)

# 下载(content endpoint 偶尔会在 status=completed 后短暂返回 400,加重试)
import time as _t
for i in range(5):
    try:
        with urllib.request.urlopen(
            urllib.request.Request(
                f"{BASE}/v1/videos/{tid}/content",
                headers={"Authorization": f"Bearer {KEY}"},
            ), timeout=180,
        ) as resp, open("demo.mp4", "wb") as f:
            while chunk := resp.read(1 << 16):
                f.write(chunk)
        break
    except urllib.error.HTTPError as e:
        if i == 4:
            raise
        _t.sleep(4)

curl -X POST "$BASE_URL/v1/videos" \
  -H "Authorization: Bearer $API_TOKEN" \
  -F "model=veo-3.1-fast-generate-preview" \
  -F "prompt=镜头从灯塔基座缓慢上升至塔顶" \
  -F "input_reference=@./lighthouse.png;type=image/png" \
  -F "seconds=8" \
  -F "size=1280x720" \
  -F "resolution=720p" \
  -F "aspectRatio=16:9"

# 创建 4K 任务(注意:本轮实测耗时 5-6 分钟,先把超时调大)
RESP=$(curl -sS -X POST "$BASE_URL/v1/videos" \
  -H "Authorization: Bearer $API_TOKEN" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "veo-3.1-generate-preview",
    "prompt": "黄昏海边的灯塔,镜头缓慢推进,海浪轻拍礁石,远处低沉的风声和海鸟声,电影级光影,稳定运镜",
    "duration": "8",
    "size": "3840x2160",
    "metadata": {
      "resolution": "4k",
      "aspectRatio": "16:9",
      "seed": 20260521,
      "negativePrompt": "blurry, watermark, distorted, low quality"
    }
  }')
TASK_ID=$(echo "$RESP" | python3 -c 'import sys,json;print(json.load(sys.stdin)["task_id"])')
echo "task_id=$TASK_ID"

# 轮询(4K 建议 10 秒间隔 + 10 分钟硬超时)
START=$(date +%s)
while :; do
  S=$(curl -sS -H "Authorization: Bearer $API_TOKEN" "$BASE_URL/v1/videos/$TASK_ID")
  ST=$(echo "$S" | python3 -c 'import sys,json;print(json.load(sys.stdin)["status"])')
  ELAPSED=$(( $(date +%s) - START ))
  echo "[${ELAPSED}s] status=$ST"
  [ "$ST" = "completed" ] && break
  [ "$ST" = "failed" ]    && { echo "$S"; exit 1; }
  [ $ELAPSED -gt 600 ]    && { echo "timeout"; exit 2; }
  sleep 10
done

# 下载(status 刚翻 completed 偶发 400,等几秒再下;--retry 3 兜底)
sleep 4
curl -sSL --retry 3 --retry-delay 4 \
  -H "Authorization: Bearer $API_TOKEN" \
  "$BASE_URL/v1/videos/$TASK_ID/content" \
  -o "veo_4k_${TASK_ID}.mp4"

ls -lh "veo_4k_${TASK_ID}.mp4"   # 实测 25-40MB