Jony
ba920f2b66
Replace the prior release tree with the current plugin, frontend, tests, and documentation. Document retired node IDs and the public Gitea update source.
217 lines
8.5 KiB
Python
217 lines
8.5 KiB
Python
"""
|
||
Seedance 视频生成客户端
|
||
使用 new-api 原生格式：POST /v1/video/generations → GET /v1/video/generations/{task_id}
|
||
"""
|
||
|
||
import json
|
||
from typing import Any, Callable, Dict, Optional
|
||
|
||
import aiohttp
|
||
|
||
from ..utils.config import get_api_key_or_raise, get_base_url_by_route
|
||
from ..utils.http_error import async_request_with_retry
|
||
from ..utils.video_task import (
|
||
PollDeadline,
|
||
check_interrupt,
|
||
download_video_to_file,
|
||
extract_error_message,
|
||
extract_progress,
|
||
extract_status,
|
||
interruptible_sleep,
|
||
is_failure_status,
|
||
is_success_status,
|
||
run_with_interrupt,
|
||
)
|
||
|
||
|
||
class SeedanceClient:
|
||
"""Seedance 视频生成客户端（new-api 原生三段式）
|
||
|
||
注意：新旧格式模型（seedance-2-0-260128-d 等）共用同一套端点，
|
||
区别仅在于请求体结构（顶层 content vs metadata.content），
|
||
由调用方（节点层）通过 use_new_format 控制请求体拼装方式。
|
||
"""
|
||
|
||
# 提交任务（新旧格式模型共用）
|
||
CREATE_ENDPOINT = "/v1/video/generations"
|
||
# 查询任务状态：{task_id} 占位（新旧格式模型共用）
|
||
STATUS_ENDPOINT = "/v1/video/generations/{task_id}"
|
||
|
||
POLL_INITIAL_INTERVAL = 4 # 首次轮询等待秒数
|
||
POLL_MAX_INTERVAL = 15 # 最大轮询间隔秒数
|
||
|
||
# new-api 返回的成功状态值
|
||
SUCCESS_STATUSES = {"succeeded", "success", "completed", "done", "finished"}
|
||
FAILURE_STATUSES = {"failed", "fail", "error", "expired"}
|
||
|
||
def __init__(self):
|
||
self.api_key = get_api_key_or_raise()
|
||
self.base_url = get_base_url_by_route()
|
||
|
||
def _headers(self) -> Dict[str, str]:
|
||
return {
|
||
"Authorization": f"Bearer {self.api_key}",
|
||
"Content-Type": "application/json",
|
||
}
|
||
|
||
# ── 1. 提交任务 ────────────────────────────────────────────────────
|
||
|
||
async def submit_async(
|
||
self,
|
||
body: Dict[str, Any],
|
||
session: aiohttp.ClientSession,
|
||
use_new_format: bool = False,
|
||
) -> str:
|
||
"""提交视频生成任务，返回 task_id
|
||
|
||
use_new_format 仅用于调试日志标注请求体格式，不影响端点选择
|
||
（新旧格式模型统一走 CREATE_ENDPOINT）。
|
||
"""
|
||
url = f"{self.base_url}{self.CREATE_ENDPOINT}"
|
||
|
||
print(f"[Seedance] 提交 → {url} (body格式: {'新' if use_new_format else '旧'})")
|
||
|
||
check_interrupt()
|
||
resp = await run_with_interrupt(async_request_with_retry(
|
||
session, "POST", url, json=body, headers=self._headers(), prefix="Seedance 提交: "
|
||
))
|
||
check_interrupt()
|
||
text = await resp.text()
|
||
data = json.loads(text)
|
||
|
||
# new-api 返回字段：id / task_id
|
||
task_id = data.get("id") or data.get("task_id")
|
||
if not task_id:
|
||
raise RuntimeError("API 未返回任务 ID")
|
||
return task_id
|
||
|
||
# ── 2. 轮询状态 ────────────────────────────────────────────────────
|
||
|
||
async def poll_async(
|
||
self,
|
||
task_id: str,
|
||
session: aiohttp.ClientSession,
|
||
on_progress: Optional[Callable[[int], None]] = None,
|
||
use_new_format: bool = False,
|
||
) -> str:
|
||
"""轮询任务状态，成功后返回视频 URL（新旧格式模型统一走 STATUS_ENDPOINT）"""
|
||
url = f"{self.base_url}{self.STATUS_ENDPOINT.format(task_id=task_id)}"
|
||
interval = self.POLL_INITIAL_INTERVAL
|
||
deadline = PollDeadline(label="Seedance")
|
||
|
||
while True:
|
||
deadline.check()
|
||
check_interrupt()
|
||
async with session.get(url, headers=self._headers()) as resp:
|
||
text = await resp.text()
|
||
if resp.status != 200:
|
||
try:
|
||
err = json.loads(text)
|
||
msg = (err.get("error", {}).get("message")
|
||
or err.get("message")
|
||
or text)
|
||
except Exception:
|
||
msg = text
|
||
raise RuntimeError(f"状态查询失败 ({resp.status}): {msg}")
|
||
result = json.loads(text)
|
||
|
||
# new-api 包装格式：真实数据在 result["data"] 里
|
||
inner = result.get("data") or result
|
||
|
||
status = extract_status(result)
|
||
|
||
# 解析进度
|
||
progress_pct = extract_progress(result)
|
||
|
||
print(f"[Seedance] 生成中 {progress_pct}%")
|
||
if on_progress:
|
||
on_progress(progress_pct)
|
||
|
||
if is_success_status(status):
|
||
# 响应结构：result["data"] = inner，inner["data"] = platform_data
|
||
# 视频 URL 在 inner["result_url"] 或 inner["data"]["content"]["video_url"]
|
||
platform_data = inner.get("data") or {}
|
||
content = platform_data.get("content") or {}
|
||
video_url = (
|
||
inner.get("result_url")
|
||
or content.get("video_url")
|
||
or platform_data.get("video_url")
|
||
or inner.get("url")
|
||
)
|
||
if not video_url:
|
||
raise RuntimeError("任务成功但未找到视频 URL")
|
||
# 末帧图片 URL 在 inner["data"]["content"]["last_frame_url"]
|
||
last_frame_url = (
|
||
content.get("last_frame_url")
|
||
or platform_data.get("last_frame_url")
|
||
or inner.get("last_frame_url")
|
||
)
|
||
return video_url, last_frame_url
|
||
|
||
if is_failure_status(status, result):
|
||
reason = extract_error_message(result)
|
||
raise RuntimeError(f"视频生成失败：{reason}")
|
||
|
||
await interruptible_sleep(interval)
|
||
interval = min(interval * 1.5, self.POLL_MAX_INTERVAL)
|
||
|
||
# ── 3. 下载视频 ────────────────────────────────────────────────────
|
||
|
||
async def download_async(
|
||
self,
|
||
video_url: str,
|
||
save_path: str,
|
||
session: aiohttp.ClientSession,
|
||
) -> str:
|
||
"""下载视频到本地，返回本地路径"""
|
||
print(f"[Seedance] 下载视频...")
|
||
return await download_video_to_file(
|
||
session, video_url, save_path, label="Seedance",
|
||
)
|
||
|
||
# ── 全流程入口（供节点调用）────────────────────────────────────────
|
||
|
||
async def generate_async(
|
||
self,
|
||
body: Dict[str, Any],
|
||
save_path: str,
|
||
on_stage: Optional[Callable[[str], None]] = None,
|
||
on_progress: Optional[Callable[[int], None]] = None,
|
||
use_new_format: bool = False,
|
||
) -> tuple:
|
||
"""提交 → 轮询 → 下载，返回 (本地视频路径, 末帧图片URL或None)"""
|
||
connector = aiohttp.TCPConnector(ssl=False, force_close=True)
|
||
async with aiohttp.ClientSession(connector=connector) as session:
|
||
|
||
# 提交
|
||
check_interrupt()
|
||
if on_stage:
|
||
on_stage("submitting")
|
||
task_id = await self.submit_async(body, session, use_new_format=use_new_format)
|
||
print(f"[Seedance] 任务已提交 → {task_id}")
|
||
if on_stage:
|
||
on_stage(f"submitted:{task_id}")
|
||
|
||
# 轮询
|
||
video_url, last_frame_url = await self.poll_async(task_id, session, on_progress=on_progress, use_new_format=use_new_format)
|
||
|
||
# 下载（带"Video not ready"重试）
|
||
if on_stage:
|
||
on_stage("downloading")
|
||
max_retries = 5
|
||
retry_delay = 3.0
|
||
for attempt in range(max_retries):
|
||
try:
|
||
path = await self.download_async(video_url, save_path, session)
|
||
if on_stage:
|
||
on_stage("done")
|
||
return path, last_frame_url
|
||
except Exception as e:
|
||
error_msg = str(e)
|
||
if "Video not ready" in error_msg and attempt < max_retries - 1:
|
||
print(f"[Seedance] 视频未就绪，{retry_delay}秒后重试 ({attempt + 1}/{max_retries})...")
|
||
await interruptible_sleep(retry_delay)
|
||
check_interrupt()
|
||
continue
|
||
raise
|