3D 生成接口

curl -X POST https://aihubmix.com/v1/3d/generations \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "hy-3d-3.1",
    "prompt": "一辆复古蒸汽朋克风格的摩托车，黄铜质感，细节丰富",
    "enable_pbr": true
  }'

curl -X POST https://aihubmix.com/v1/3d/generations \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "hy-3d-3.1",
    "input_references": [
      { "image_url": "https://example.com/cat.png" }
    ]
  }'

curl -X POST https://aihubmix.com/v1/3d/generations \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "hy-3d-3.1",
    "input_references": [
      { "image_base64": "<BASE64_ENCODED_IMAGE>" }
    ]
  }'

curl -X POST https://aihubmix.com/v1/3d/generations \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "hy-3d-3.1",
    "generate_type": "Sketch",
    "prompt": "科幻风格的飞行器，金属外壳",
    "input_references": [
      { "image_url": "https://example.com/sketch.png" }
    ]
  }'

curl -X POST https://aihubmix.com/v1/3d/generations \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "hy-3d-3.1",
    "prompt": "一个中世纪骑士头盔",
    "generate_type": "Normal",
    "enable_pbr": true,
    "face_count": 40000,
    "format": "fbx"
  }'

import requests
import time

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

# 第一步：创建 3D 生成任务
response = requests.post(
    f"{BASE_URL}/v1/3d/generations",
    headers=HEADERS,
    json={
        "model": "hy-3d-3.1",
        "prompt": "一只戴帽子的柴犬",
        "enable_pbr": True
    }
)
result = response.json()
generation_id = result["id"]
print(f"任务已创建，generation_id: {generation_id}")

# 第二步：轮询查询状态
while True:
    status_response = requests.get(
        f"{BASE_URL}/v1/3d/generations/{generation_id}",
        headers=HEADERS
    )
    status_data = status_response.json()
    current_status = status_data["status"]
    print(f"当前状态: {current_status}")

    if current_status == "completed":
        print("3D 生成完成！")
        break
    elif current_status == "failed":
        error = status_data.get("error") or {}
        print(f"生成失败: {error.get('message', '未知错误')}")
        exit(1)

    time.sleep(15)  # 每 15 秒查询一次

# 第三步：下载产物（content_url 为临时地址，直接下载）
for item in status_data["output"]:
    file_url = item["content_url"]
    file_name = f"model.{item['type']}"
    file_response = requests.get(file_url)
    with open(file_name, "wb") as f:
        f.write(file_response.content)
    print(f"已保存 {file_name}（{len(file_response.content) / 1024 / 1024:.1f} MB）")

const API_KEY = "your_aihubmix_api_key";
const BASE_URL = "https://aihubmix.com";

async function generate3D() {
  // 第一步：创建任务
  const createResponse = await fetch(`${BASE_URL}/v1/3d/generations`, {
    method: "POST",
    headers: {
      "Authorization": `Bearer ${API_KEY}`,
      "Content-Type": "application/json"
    },
    body: JSON.stringify({
      model: "hy-3d-3.1",
      prompt: "一只戴帽子的柴犬",
      enable_pbr: true
    })
  });
  const { id: generationId } = await createResponse.json();
  console.log(`任务已创建: ${generationId}`);

  // 第二步：轮询状态
  let result;
  while (true) {
    await new Promise(resolve => setTimeout(resolve, 15000));
    const statusResponse = await fetch(
      `${BASE_URL}/v1/3d/generations/${generationId}`,
      { headers: { "Authorization": `Bearer ${API_KEY}` } }
    );
    result = await statusResponse.json();
    console.log(`当前状态: ${result.status}`);
    if (result.status === "completed" || result.status === "failed") break;
  }

  if (result.status === "failed") {
    console.error(`生成失败: ${result.error?.message}`);
    return;
  }

  // 第三步：下载产物
  const fs = require("fs");
  for (const item of result.output) {
    const fileResponse = await fetch(item.content_url);
    const buffer = Buffer.from(await fileResponse.arrayBuffer());
    fs.writeFileSync(`model.${item.type}`, buffer);
    console.log(`已保存 model.${item.type}`);
  }
}

generate3D();