pipeline-app/docs/model-selection-integration.md

模型选择功能集成指南

功能概述

在分镜生成（storyboard_generating）后插入模型选择步骤，允许客户根据费用预估选择视频生成模型。

已完成的代码修改

1. 新增模块

• app/cost_estimator.py - 多模型费用预估（Wan 2.2, Wan 2.7, Vidu 2.0）
• app/model_selector.py - 模型选择交互处理器

2. 修改模块

• app/ktv_adapter.py - 添加 model_selecting handler，修改 scene_video_generating 支持多模型

产线定义更新步骤

步骤 1: 插入 model_selecting 步骤

在数据库的 pipeline_steps 表中，在 storyboard_generating (step_order=13) 之后插入新步骤：

-- 查找当前 storyboard_generating 的 step_order
SELECT step_order FROM pipeline_steps 
WHERE pipeline_id = 'ktv_pipeline' AND step_name = 'storyboard_generating';

-- 假设结果是 13，需要将后续步骤的 step_order +1
UPDATE pipeline_steps 
SET step_order = step_order + 1 
WHERE pipeline_id = 'ktv_pipeline' AND step_order > 13;

-- 插入 model_selecting 步骤
INSERT INTO pipeline_steps (
    pipeline_id, step_name, step_type, step_order, 
    description, handler_function, step_config
) VALUES (
    'ktv_pipeline',
    'model_selecting',
    'interactive',
    14,
    '客户根据费用预估选择视频生成模型',
    'handle_model_selecting',
    '{"deps": ["storyboard_generating"], "timeout_hours": 24}'
);

-- 更新 scene_video_generating 的依赖
UPDATE pipeline_steps 
SET step_config = JSON_SET(
    step_config, 
    '$.deps', 
    JSON_ARRAY('model_selecting', 'character_image_generating')
)
WHERE pipeline_id = 'ktv_pipeline' AND step_name = 'scene_video_generating';

步骤 2: 验证步骤顺序

SELECT step_name, step_order, step_type 
FROM pipeline_steps 
WHERE pipeline_id = 'ktv_pipeline' 
ORDER BY step_order;

预期结果：

1  - audio_preparing
2  - demucs_separating
3  - lyric_calibrating
...
13 - storyboard_generating
14 - model_selecting          <-- 新增
15 - scene_video_generating   <-- 更新依赖
16 - scene_video_evaluating
...

费用预估模型配置

当前支持的模型（在 cost_estimator.py 中定义）：

模型	质量评分	生成速度	价格/帧	特点
Wan 2.2	7.5/10	fast	$0.002	本地GPU，性价比高
Wan 2.7	8.5/10	medium	$0.003	本地GPU，质量更好
Vidu 2.0	9.0/10	slow	$0.008	云端API，最高质量

工作流程

1. 分镜生成完成 → 计算总帧数和时长
2. 生成费用预估 → 为每个模型计算成本
3. 创建 human_task → 显示选项给客户
4. 客户选择模型 → 通过前端界面选择
5. 任务继续执行 → scene_video_generating 使用选中的模型

注意事项

1. Vidu 2.0 暂未实现 - 当前选择 Vidu 2.0 会降级到 Wan 2.2，需要后续集成 Vidu API
2. 超时设置 - model_selecting 步骤默认 24 小时超时，可在 step_config 中调整
3. 默认选择 - 如果客户未选择，默认使用 Wan 2.2（最便宜选项）

验证清单

• 数据库步骤已更新
• 代码已提交 (commit: model-selection)
• 测试产线执行到 storyboard_generating
• 验证 human_task 创建成功
• 验证客户选择后任务继续执行
• 验证 scene_video_generating 使用了正确的模型