Audit Bot
ea9e536740
方案A: 后端扩展3处API改动，支持前端模板中心化模型。
1. POST /api/v1/generation/tasks — project_id改为可选，新增template_id/asset_ids/title_ids/voice_ids
2. GET /api/v1/generation/tasks + GET /api/v1/tasks — 新增用户级列表（跨project）
3. POST /api/v1/tasks/{task_id}/retry + POST /api/v1/generation/tasks/{task_id}/retry — 简化重试
改动涉及：
- Domain: GenerationTask新增6个字段，放宽create()校验
- Application: CreateGenerationTaskCommand新增字段
- Ports: GenerationTaskRepository新增list_by_user()
- Adapter: SQLAlchemy模型+仓储实现新字段和list_by_user
- Schema: 双模式校验（project模式/模板模式）
- Routes: generation_tasks + task_center双路由注册
- Migration: 015_add_generation_task_extensions
向后兼容：所有旧端点和参数不变。
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from pydantic import BaseModel, Field, model_validator
|
|
|
|
|
|
class CreateGenerationTaskRequest(BaseModel):
|
|
"""创建生成任务请求。
|
|
|
|
支持两种模式（至少提供一种）：
|
|
- 项目模式：project_id + asset_library_id（向后兼容）
|
|
- 模板模式：template_id + asset_ids / title_ids / voice_ids
|
|
"""
|
|
project_id: str = ""
|
|
asset_library_id: str = ""
|
|
strategy_id: str = ""
|
|
voice_library_id: str = ""
|
|
created_by_user_id: str = ""
|
|
# ── 模板模式新增字段 ──
|
|
template_id: str = ""
|
|
asset_ids: list[str] = Field(default_factory=list)
|
|
title_ids: list[str] = Field(default_factory=list)
|
|
voice_ids: list[str] = Field(default_factory=list)
|
|
|
|
@model_validator(mode="after")
|
|
def _check_at_least_one_mode(self) -> "CreateGenerationTaskRequest":
|
|
has_project = bool(self.project_id.strip())
|
|
has_template = bool(self.template_id.strip())
|
|
if not has_project and not has_template:
|
|
raise ValueError("project_id 或 template_id 至少需要提供一个")
|
|
has_library = bool(self.asset_library_id.strip())
|
|
has_assets = bool(self.asset_ids or self.title_ids or self.voice_ids)
|
|
if not has_library and not has_assets:
|
|
raise ValueError("asset_library_id 或 asset_ids/title_ids/voice_ids 至少需要提供一个")
|
|
return self
|
|
|
|
|
|
class GenerationTaskResponse(BaseModel):
|
|
id: str
|
|
project_id: str
|
|
asset_library_id: str
|
|
strategy_id: str
|
|
voice_library_id: str
|
|
template_id: str = ""
|
|
asset_ids: list[str] = Field(default_factory=list)
|
|
title_ids: list[str] = Field(default_factory=list)
|
|
voice_ids: list[str] = Field(default_factory=list)
|
|
status: str
|
|
progress: float
|
|
result_count: int
|
|
error_message: str
|
|
|
|
|
|
class ListGenerationTasksResponse(BaseModel):
|
|
"""用户级生成任务列表响应（跨 project）。"""
|
|
items: list[GenerationTaskResponse]
|