feat: initial release v0.3.0
This commit is contained in:
saturn
@@ -0,0 +1,277 @@
|
||||
import { describe, expect, it, vi } from 'vitest'
|
||||
import { runScriptToStoryboardOrchestrator } from '@/lib/novel-promotion/script-to-storyboard/orchestrator'
|
||||
|
||||
describe('script-to-storyboard orchestrator retry', () => {
|
||||
it('retries retryable step failures up to 3 attempts', async () => {
|
||||
const attemptsByAction = new Map<string, number>()
|
||||
const phase1Metas: Array<{ stepId: string; stepAttempt?: number }> = []
|
||||
const runStep = vi.fn(async (meta, _prompt, action: string) => {
|
||||
attemptsByAction.set(action, (attemptsByAction.get(action) || 0) + 1)
|
||||
|
||||
if (action === 'storyboard_phase1_plan') {
|
||||
phase1Metas.push({ stepId: meta.stepId, stepAttempt: meta.stepAttempt })
|
||||
const attempt = attemptsByAction.get(action) || 0
|
||||
if (attempt < 3) {
|
||||
throw new TypeError('terminated')
|
||||
}
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
|
||||
if (action === 'storyboard_phase2_cinematography') {
|
||||
return { text: JSON.stringify([{ panel_number: 1, composition: '居中' }]), reasoning: '' }
|
||||
}
|
||||
if (action === 'storyboard_phase2_acting') {
|
||||
return { text: JSON.stringify([{ panel_number: 1, characters: [] }]), reasoning: '' }
|
||||
}
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
})
|
||||
|
||||
const result = await runScriptToStoryboardOrchestrator({
|
||||
clips: [
|
||||
{
|
||||
id: 'clip-1',
|
||||
content: '文本',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
],
|
||||
novelPromotionData: {
|
||||
characters: [{ name: '角色A', appearances: [] }],
|
||||
locations: [{ name: '场景A', images: [] }],
|
||||
},
|
||||
promptTemplates: {
|
||||
phase1PlanTemplate: '{clip_content} {clip_json} {characters_lib_name} {locations_lib_name} {characters_introduction} {characters_appearance_list} {characters_full_description}',
|
||||
phase2CinematographyTemplate: '{panels_json} {panel_count} {locations_description} {characters_info}',
|
||||
phase2ActingTemplate: '{panels_json} {panel_count} {characters_info}',
|
||||
phase3DetailTemplate: '{panels_json} {characters_age_gender} {locations_description}',
|
||||
},
|
||||
runStep,
|
||||
})
|
||||
|
||||
expect(result.summary.clipCount).toBe(1)
|
||||
expect(runStep).toHaveBeenCalled()
|
||||
expect(attemptsByAction.get('storyboard_phase1_plan')).toBe(3)
|
||||
expect(phase1Metas).toEqual([
|
||||
{ stepId: 'clip_clip-1_phase1', stepAttempt: undefined },
|
||||
{ stepId: 'clip_clip-1_phase1', stepAttempt: 2 },
|
||||
{ stepId: 'clip_clip-1_phase1', stepAttempt: 3 },
|
||||
])
|
||||
})
|
||||
|
||||
it('does not retry non-retryable step failure', async () => {
|
||||
let callCount = 0
|
||||
const runStep = vi.fn(async () => {
|
||||
callCount += 1
|
||||
throw new Error('SENSITIVE_CONTENT: blocked')
|
||||
})
|
||||
|
||||
await expect(
|
||||
runScriptToStoryboardOrchestrator({
|
||||
clips: [
|
||||
{
|
||||
id: 'clip-1',
|
||||
content: '文本',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
],
|
||||
novelPromotionData: {
|
||||
characters: [{ name: '角色A', appearances: [] }],
|
||||
locations: [{ name: '场景A', images: [] }],
|
||||
},
|
||||
promptTemplates: {
|
||||
phase1PlanTemplate: '{clip_content} {clip_json} {characters_lib_name} {locations_lib_name} {characters_introduction} {characters_appearance_list} {characters_full_description}',
|
||||
phase2CinematographyTemplate: '{panels_json} {panel_count} {locations_description} {characters_info}',
|
||||
phase2ActingTemplate: '{panels_json} {panel_count} {characters_info}',
|
||||
phase3DetailTemplate: '{panels_json} {characters_age_gender} {locations_description}',
|
||||
},
|
||||
runStep,
|
||||
}),
|
||||
).rejects.toThrow('SENSITIVE_CONTENT')
|
||||
|
||||
expect(callCount).toBe(1)
|
||||
})
|
||||
|
||||
it('does not retry Ark invalid parameter error even when message contains json', async () => {
|
||||
let callCount = 0
|
||||
const runStep = vi.fn(async () => {
|
||||
callCount += 1
|
||||
throw new Error(
|
||||
'Ark Responses 调用失败: 400 - {"error":{"code":"InvalidParameter","message":"json: unknown field \\"reasoning_effort\\""}}',
|
||||
)
|
||||
})
|
||||
|
||||
await expect(
|
||||
runScriptToStoryboardOrchestrator({
|
||||
clips: [
|
||||
{
|
||||
id: 'clip-1',
|
||||
content: '文本',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
],
|
||||
novelPromotionData: {
|
||||
characters: [{ name: '角色A', appearances: [] }],
|
||||
locations: [{ name: '场景A', images: [] }],
|
||||
},
|
||||
promptTemplates: {
|
||||
phase1PlanTemplate: '{clip_content} {clip_json} {characters_lib_name} {locations_lib_name} {characters_introduction} {characters_appearance_list} {characters_full_description}',
|
||||
phase2CinematographyTemplate: '{panels_json} {panel_count} {locations_description} {characters_info}',
|
||||
phase2ActingTemplate: '{panels_json} {panel_count} {characters_info}',
|
||||
phase3DetailTemplate: '{panels_json} {characters_age_gender} {locations_description}',
|
||||
},
|
||||
runStep,
|
||||
}),
|
||||
).rejects.toThrow('unknown field')
|
||||
|
||||
expect(callCount).toBe(1)
|
||||
})
|
||||
|
||||
it('enforces topology: phase3 runs after both phase2 steps complete', async () => {
|
||||
const actionOrder: string[] = []
|
||||
const runStep = vi.fn(async (_meta, _prompt, action: string) => {
|
||||
actionOrder.push(action)
|
||||
if (action === 'storyboard_phase1_plan') {
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
if (action === 'storyboard_phase2_cinematography') {
|
||||
return { text: JSON.stringify([{ panel_number: 1, composition: '居中' }]), reasoning: '' }
|
||||
}
|
||||
if (action === 'storyboard_phase2_acting') {
|
||||
return { text: JSON.stringify([{ panel_number: 1, characters: [] }]), reasoning: '' }
|
||||
}
|
||||
if (action === 'storyboard_phase3_detail') {
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
throw new Error(`unexpected action: ${action}`)
|
||||
})
|
||||
|
||||
const result = await runScriptToStoryboardOrchestrator({
|
||||
clips: [
|
||||
{
|
||||
id: 'clip-1',
|
||||
content: '文本',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
],
|
||||
novelPromotionData: {
|
||||
characters: [{ name: '角色A', appearances: [] }],
|
||||
locations: [{ name: '场景A', images: [] }],
|
||||
},
|
||||
promptTemplates: {
|
||||
phase1PlanTemplate: '{clip_content} {clip_json} {characters_lib_name} {locations_lib_name} {characters_introduction} {characters_appearance_list} {characters_full_description}',
|
||||
phase2CinematographyTemplate: '{panels_json} {panel_count} {locations_description} {characters_info}',
|
||||
phase2ActingTemplate: '{panels_json} {panel_count} {characters_info}',
|
||||
phase3DetailTemplate: '{panels_json} {characters_age_gender} {locations_description}',
|
||||
},
|
||||
runStep,
|
||||
})
|
||||
|
||||
expect(result.summary.clipCount).toBe(1)
|
||||
const phase3Index = actionOrder.indexOf('storyboard_phase3_detail')
|
||||
const phase2CineIndex = actionOrder.indexOf('storyboard_phase2_cinematography')
|
||||
const phase2ActingIndex = actionOrder.indexOf('storyboard_phase2_acting')
|
||||
expect(phase3Index).toBeGreaterThan(phase2CineIndex)
|
||||
expect(phase3Index).toBeGreaterThan(phase2ActingIndex)
|
||||
})
|
||||
|
||||
it('limits clip fan-out by configured concurrency', async () => {
|
||||
let activePhase1 = 0
|
||||
let maxActivePhase1 = 0
|
||||
|
||||
const runStep = vi.fn(async (_meta, _prompt, action: string) => {
|
||||
if (action === 'storyboard_phase1_plan') {
|
||||
activePhase1 += 1
|
||||
maxActivePhase1 = Math.max(maxActivePhase1, activePhase1)
|
||||
await new Promise((resolve) => setTimeout(resolve, 5))
|
||||
activePhase1 -= 1
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
if (action === 'storyboard_phase2_cinematography') {
|
||||
return {
|
||||
text: JSON.stringify([{
|
||||
panel_number: 1,
|
||||
composition: '居中',
|
||||
lighting: '顶光',
|
||||
color_palette: '冷色',
|
||||
atmosphere: '紧张',
|
||||
technical_notes: 'note',
|
||||
}]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
if (action === 'storyboard_phase2_acting') {
|
||||
return { text: JSON.stringify([{ panel_number: 1, characters: [] }]), reasoning: '' }
|
||||
}
|
||||
if (action === 'storyboard_phase3_detail') {
|
||||
return {
|
||||
text: JSON.stringify([{ panel_number: 1, description: '镜头', location: '场景A', source_text: '原文', characters: [] }]),
|
||||
reasoning: '',
|
||||
}
|
||||
}
|
||||
throw new Error(`unexpected action: ${action}`)
|
||||
})
|
||||
|
||||
const result = await runScriptToStoryboardOrchestrator({
|
||||
concurrency: 1,
|
||||
clips: [
|
||||
{
|
||||
id: 'clip-1',
|
||||
content: '文本1',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
{
|
||||
id: 'clip-2',
|
||||
content: '文本2',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
{
|
||||
id: 'clip-3',
|
||||
content: '文本3',
|
||||
characters: JSON.stringify([{ name: '角色A' }]),
|
||||
location: '场景A',
|
||||
screenplay: null,
|
||||
},
|
||||
],
|
||||
novelPromotionData: {
|
||||
characters: [{ name: '角色A', appearances: [] }],
|
||||
locations: [{ name: '场景A', images: [] }],
|
||||
},
|
||||
promptTemplates: {
|
||||
phase1PlanTemplate: '{clip_content} {clip_json} {characters_lib_name} {locations_lib_name} {characters_introduction} {characters_appearance_list} {characters_full_description}',
|
||||
phase2CinematographyTemplate: '{panels_json} {panel_count} {locations_description} {characters_info}',
|
||||
phase2ActingTemplate: '{panels_json} {panel_count} {characters_info}',
|
||||
phase3DetailTemplate: '{panels_json} {characters_age_gender} {locations_description}',
|
||||
},
|
||||
runStep,
|
||||
})
|
||||
|
||||
expect(result.summary.clipCount).toBe(3)
|
||||
expect(maxActivePhase1).toBe(1)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user