Veo 视频生成 API 文档¶

import requests
import time
import json
import sys

API_BASE = "https://new.12ai.org"
API_KEY = "YOUR_API_KEY"

headers = {
    "Authorization": f"Bearer {API_KEY}"
}

# 测试图片
IMG1 = "https://example.com/image1.jpg"
IMG2 = "https://example.com/image2.jpg"
IMG3 = "https://example.com/image3.jpg"

# 测试场景定义
TEST_CASES = {
    "1": {
        "name": "文生视频",
        "data": {
            "model": "veo_3_1-fast",
            "prompt": "一只可爱的橘猫在花园里追逐蝴蝶，慢动作镜头，电影级画质",
            "size": "1920x1080"
        },
        "images": []
    },
    "2": {
        "name": "图生视频 · 首尾帧（URL）",
        "data": {
            "model": "veo_3_1-fast-fl",
            "prompt": "平滑过渡，自然变化",
            "size": "1280x720"
        },
        "images": [IMG1, IMG2]
    },
    "3": {
        "name": "图生视频 · 首帧（仅一张）",
        "data": {
            "model": "veo_3_1-fast-fl",
            "prompt": "让画面动起来，镜头缓慢推进",
            "size": "1280x720"
        },
        "images": [IMG1]
    },
    "4": {
        "name": "图生视频 · 参考图（最多 3 张）",
        "data": {
            "model": "veo_3_1-fast",
            "prompt": "根据参考图生成视频",
            "size": "1280x720"
        },
        "images": [IMG1, IMG2, IMG3]
    },
    "5": {
        "name": "竖屏文生视频（9:16）",
        "data": {
            "model": "veo_3_1-fast",
            "prompt": "一个人在城市街道上行走",
            "size": "1080x1920"
        },
        "images": []
    }
}

# 选择测试场景
print("=" * 50)
print("Veo API 功能测试")
print("=" * 50)
for key, case in TEST_CASES.items():
    print(f"  {key}. {case['name']}")
print("=" * 50)

choice = sys.argv[1] if len(sys.argv) > 1 else input("请选择测试场景 (1-5): ").strip()
if choice not in TEST_CASES:
    print(f"无效选择: {choice}")
    sys.exit(1)

selected = TEST_CASES[choice]
print(f"\n>>> 运行测试: {selected['name']}")

# 构建 multipart/form-data 请求
data = selected["data"]
files_list = []

# 图片作为 input_reference[] 字段（URL 以文本形式传递）
for img_url in selected["images"]:
    files_list.append(("input_reference[]", (None, img_url)))

print(f"请求参数: {json.dumps(data, ensure_ascii=False, indent=2)}")
if selected["images"]:
    print(f"图片数量: {len(selected['images'])}")

# 提交任务
response = requests.post(
    f"{API_BASE}/v1/videos",
    headers=headers,
    data=data,
    files=files_list if files_list else None
)
result = response.json()
print(f"\nAPI 返回: {json.dumps(result, ensure_ascii=False, indent=2)}")

# 检查错误
if "error" in result:
    print(f"请求失败: {result['error']}")
    sys.exit(1)

task_id = result.get("id")
if not task_id:
    print(f"未找到 task_id，完整响应: {result}")
    sys.exit(1)

print(f"任务已提交，task_id: {task_id}")

# 轮询查询状态
while True:
    response = requests.get(f"{API_BASE}/v1/videos/{task_id}", headers=headers)
    status = response.json()

    current_status = status.get("status", "")
    progress = status.get("progress", "N/A")
    print(f"状态: {current_status}, 进度: {progress}")

    if current_status == "completed":
        print("视频生成成功！")
        break
    elif current_status == "failed":
        print(f"视频生成失败: {status}")
        break

    time.sleep(5)