Tarefas assíncronas

# Etapa 1: enviar a tarefa assíncrona de vídeo
curl -X POST https://aihubmix.com/ai/v1/videos \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "wan2.6-t2v",
    "prompt": "A cat playing jazz on a piano, warm lighting, cinematic shot",
    "seconds": "5",
    "size": "1280x720"
  }'

# Etapa 2: consultar a cada 15 segundos, até a tarefa ser concluída, falhar ou ser cancelada
curl https://aihubmix.com/ai/v1/tasks/{task_id} \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY"

# Etapa 3: baixar um único arquivo gerado
curl https://aihubmix.com/ai/v1/tasks/{task_id}/content \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  --output result.mp4

{
  "id": "task_01K0...",
  "object": "video",
  "model": "wan2.6-t2v",
  "status": "in_progress",
  "output": [],
  "error": null,
  "created_at": 1784707200,
  "completed_at": null,
  "expires_at": null
}

import base64
import os
import time

import requests

BASE_URL = "https://aihubmix.com"
API_KEY = os.environ["AIHUBMIX_API_KEY"]
HEADERS = {
    "Authorization": f"Bearer {API_KEY}",
    "Content-Type": "application/json",
}

# 1. Criar a tarefa
response = requests.post(
    f"{BASE_URL}/ai/v1/videos",
    headers=HEADERS,
    json={
        "model": "wan2.6-t2v",
        "prompt": "A cat playing jazz on a piano",
        "seconds": "5",
        "size": "1280x720",
    },
    timeout=60,
)
response.raise_for_status()
task = response.json()
task_id = task["id"]

# 2. Fazer polling, até a tarefa ser concluída, falhar ou ser cancelada
while task["status"] not in {"completed", "failed", "cancelled"}:
    time.sleep(15)
    response = requests.get(
        f"{BASE_URL}/ai/v1/tasks/{task_id}",
        headers=HEADERS,
        timeout=30,
    )
    response.raise_for_status()
    task = response.json()
    print("status:", task["status"])

# 3. Obter o resultado
if task["output"]:
    for index, item in enumerate(task["output"]):
        if encoded := item.get("b64_json"):
            with open(f"result-{index}.bin", "wb") as file:
                file.write(base64.b64decode(encoded))
            continue
        result = requests.get(
            item["content_url"],
            headers=HEADERS,
            timeout=120,
        )
        result.raise_for_status()
        with open(f"result-{index}.bin", "wb") as file:
            file.write(result.content)
elif task["status"] == "failed":
    raise RuntimeError(task.get("error"))

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

const BASE_URL = "https://aihubmix.com";
const HEADERS = {
  Authorization: `Bearer ${process.env.AIHUBMIX_API_KEY}`,
  "Content-Type": "application/json",
};

// 1. Criar a tarefa
const created = await fetch(`${BASE_URL}/ai/v1/videos`, {
  method: "POST",
  headers: HEADERS,
  body: JSON.stringify({
    model: "wan2.6-t2v",
    prompt: "A cat playing jazz on a piano",
    seconds: "5",
    size: "1280x720",
  }),
});
let task = await created.json();

// 2. Fazer polling, até a tarefa ser concluída, falhar ou ser cancelada
const finished = new Set(["completed", "failed", "cancelled"]);
while (!finished.has(task.status)) {
  await new Promise((resolve) => setTimeout(resolve, 15_000));
  const polled = await fetch(`${BASE_URL}/ai/v1/tasks/${task.id}`, {
    headers: HEADERS,
  });
  task = await polled.json();
  console.log("status:", task.status);
}

// 3. Obter o resultado
if (task.output?.length) {
  for (const [index, item] of task.output.entries()) {
    if (item.b64_json) {
      await writeFile(`result-${index}.bin`, Buffer.from(item.b64_json, "base64"));
      continue;
    }
    const result = await fetch(item.content_url, { headers: HEADERS });
    await writeFile(`result-${index}.bin`, Buffer.from(await result.arrayBuffer()));
  }
} else if (task.status === "failed") {
  throw new Error(JSON.stringify(task.error));
}

# 1. Criar a tarefa e anotar o id retornado
curl -X POST https://aihubmix.com/ai/v1/videos \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "wan2.6-t2v",
    "prompt": "A cat playing jazz on a piano",
    "seconds": "5",
    "size": "1280x720"
  }'

# 2. Consultar o status a cada 15 segundos
curl https://aihubmix.com/ai/v1/tasks/task_01K0ABCDEF \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY"

# 3. Baixar o resultado depois que o status mudar para completed
curl https://aihubmix.com/ai/v1/tasks/task_01K0ABCDEF/content \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  --output result.mp4