feat(home): add AI story expand button and modal

This commit is contained in:
saturn
2026-03-24 23:53:26 +08:00
parent 4e469074e0
commit fd8f5f8635
28 changed files with 615 additions and 3 deletions

View File

@@ -0,0 +1,50 @@
import { describe, expect, it, vi } from 'vitest'
import { expandHomeStory } from '@/lib/home/ai-story-expand'
vi.mock('@/lib/task/client', () => ({
resolveTaskResponse: vi.fn(),
}))
import { resolveTaskResponse } from '@/lib/task/client'
function buildJsonResponse(body: unknown, status = 200): Response {
return new Response(JSON.stringify(body), {
status,
headers: { 'Content-Type': 'application/json' },
})
}
describe('expandHomeStory', () => {
it('posts the prompt to the user ai-story-expand route and returns expanded text', async () => {
const apiFetch = vi.fn(async () => buildJsonResponse({ async: true, taskId: 'task-1' }))
vi.mocked(resolveTaskResponse).mockResolvedValue({
expandedText: '扩写后的故事正文',
})
const result = await expandHomeStory({
apiFetch,
prompt: '宫廷复仇女主回京',
})
expect(apiFetch).toHaveBeenCalledWith('/api/user/ai-story-expand', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
prompt: '宫廷复仇女主回京',
}),
})
expect(result).toEqual({
expandedText: '扩写后的故事正文',
})
})
it('fails explicitly when the route does not return expandedText', async () => {
const apiFetch = vi.fn(async () => buildJsonResponse({ async: true, taskId: 'task-1' }))
vi.mocked(resolveTaskResponse).mockResolvedValue({})
await expect(expandHomeStory({
apiFetch,
prompt: '宫廷复仇女主回京',
})).rejects.toThrow('AI story expand response missing expandedText')
})
})

View File

@@ -59,6 +59,10 @@ vi.mock('@/lib/home/create-project-launch', () => ({
createHomeProjectLaunch: vi.fn(),
}))
vi.mock('@/lib/home/ai-story-expand', () => ({
expandHomeStory: vi.fn(),
}))
describe('resolveTextareaTargetHeight', () => {
it('keeps the home quick-start input at least three rows tall', () => {
expect(resolveTextareaTargetHeight({

View File

@@ -7,6 +7,7 @@ describe('resolveTaskIntent', () => {
expect(resolveTaskIntent(TASK_TYPE.IMAGE_CHARACTER)).toBe('generate')
expect(resolveTaskIntent(TASK_TYPE.IMAGE_LOCATION)).toBe('generate')
expect(resolveTaskIntent(TASK_TYPE.VIDEO_PANEL)).toBe('generate')
expect(resolveTaskIntent(TASK_TYPE.AI_STORY_EXPAND)).toBe('generate')
})
it('maps regenerate and modify task types', () => {

View File

@@ -0,0 +1,84 @@
import type { Job } from 'bullmq'
import { beforeEach, describe, expect, it, vi } from 'vitest'
import { TASK_TYPE, type TaskJobData } from '@/lib/task/types'
const aiRuntimeMock = vi.hoisted(() => ({
executeAiTextStep: vi.fn(async () => ({
text: '扩写后的完整故事内容',
reasoning: '',
})),
}))
const workerMock = vi.hoisted(() => ({
reportTaskProgress: vi.fn(async () => undefined),
assertTaskActive: vi.fn(async () => undefined),
}))
vi.mock('@/lib/ai-runtime', () => aiRuntimeMock)
vi.mock('@/lib/llm-observe/internal-stream-context', () => ({
withInternalLLMStreamCallbacks: vi.fn(async (_callbacks: unknown, fn: () => Promise<unknown>) => await fn()),
}))
vi.mock('@/lib/prompt-i18n', () => ({
PROMPT_IDS: { NP_AI_STORY_EXPAND: 'np_ai_story_expand' },
buildPrompt: vi.fn(() => 'story-expand-prompt'),
}))
vi.mock('@/lib/workers/shared', () => ({ reportTaskProgress: workerMock.reportTaskProgress }))
vi.mock('@/lib/workers/utils', () => ({ assertTaskActive: workerMock.assertTaskActive }))
vi.mock('@/lib/workers/handlers/llm-stream', () => ({
createWorkerLLMStreamContext: vi.fn(() => ({ streamRunId: 'run-1', nextSeqByStepLane: {} })),
createWorkerLLMStreamCallbacks: vi.fn(() => ({
onStage: vi.fn(),
onChunk: vi.fn(),
onComplete: vi.fn(),
onError: vi.fn(),
flush: vi.fn(async () => undefined),
})),
}))
import { handleAiStoryExpandTask } from '@/lib/workers/handlers/ai-story-expand'
function buildJob(payload: Record<string, unknown>): Job<TaskJobData> {
return {
data: {
taskId: 'task-ai-story-expand-1',
type: TASK_TYPE.AI_STORY_EXPAND,
locale: 'zh',
projectId: 'home-ai-write',
targetType: 'HomeAiStoryExpand',
targetId: 'user-1',
payload,
userId: 'user-1',
},
} as unknown as Job<TaskJobData>
}
describe('worker ai-story-expand behavior', () => {
beforeEach(() => {
vi.clearAllMocks()
})
it('missing prompt -> explicit error', async () => {
const job = buildJob({ prompt: ' ', analysisModel: 'provider::analysis-model' })
await expect(handleAiStoryExpandTask(job)).rejects.toThrow('prompt is required')
})
it('missing analysis model -> explicit error', async () => {
const job = buildJob({ prompt: '宫廷复仇女主回京' })
await expect(handleAiStoryExpandTask(job)).rejects.toThrow('analysisModel is required')
})
it('success path -> returns expanded text without touching episode persistence', async () => {
const job = buildJob({ prompt: '宫廷复仇女主回京', analysisModel: 'provider::analysis-model' })
const result = await handleAiStoryExpandTask(job)
expect(result).toEqual({
expandedText: '扩写后的完整故事内容',
})
expect(aiRuntimeMock.executeAiTextStep).toHaveBeenCalledWith(expect.objectContaining({
userId: 'user-1',
model: 'provider::analysis-model',
projectId: 'home-ai-write',
action: 'ai_story_expand',
}))
})
})