| 1 | import type { ThinkingStage } from '@shared/thinking' |
| 2 | import { COLLECT_STAGE_PROMPT } from './collect' |
| 3 | import { OUTLINE_STAGE_PROMPT } from './outline' |
| 4 | import { DRAFT_STAGE_PROMPT } from './draft' |
| 5 | import { REFINE_STAGE_PROMPT } from './refine' |
| 6 | import { READY_STAGE_PROMPT } from './ready' |
| 7 | |
| 8 | const STAGE_PROMPTS: Record<ThinkingStage, string> = { |
| 9 | collect: COLLECT_STAGE_PROMPT, |
| 10 | outline: OUTLINE_STAGE_PROMPT, |
| 11 | draft: DRAFT_STAGE_PROMPT, |
| 12 | refine: REFINE_STAGE_PROMPT, |
| 13 | ready: READY_STAGE_PROMPT |
| 14 | } |
| 15 | |
| 16 | export function getStagePrompt(stage: ThinkingStage): string { |
| 17 | return STAGE_PROMPTS[stage] || COLLECT_STAGE_PROMPT |
| 18 | } |
| 19 |