使用 Python 接入图生视频 AI API

import os
import time
import requests

base_url = "https://sora2-api.com"
headers = {
    "Authorization": f"Bearer {os.environ['TIKDEK_API_KEY']}",
    "Content-Type": "application/json",
}

created = requests.post(
    f"{base_url}/api/v1/video/sora-video",
    headers=headers,
    json={
        "model": "sora-2-pro",
        "prompt": "Slow camera push-in, rain moving across the frame",
        "url": "https://example.com/reference-frame.jpg",
        "aspectRatio": "16:9",
        "webHook": "-1",
        "isPublic": False,
    },
    timeout=30,
)
created.raise_for_status()
task_id = created.json()["data"]["id"]

for _ in range(90):
    response = requests.post(
        f"{base_url}/api/video-generations/check-result",
        headers=headers,
        json={"taskId": task_id},
        timeout=30,
    )
    response.raise_for_status()
    task = response.json()["data"]

    if task["status"] == "succeeded":
        video = requests.get(task["result_url"], timeout=120)
        video.raise_for_status()
        open("result.mp4", "wb").write(video.content)
        break

    if task["status"] == "failed":
        raise RuntimeError(task.get("error_message") or "Generation failed")

    time.sleep(4)
else:
    raise TimeoutError(f"Task {task_id} did not finish in time")