fix-prompt.js 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  1. const fs = require('fs');
  2. const file = 'src/modules/book-generator/langgraph-generator.ts';
  3. let content = fs.readFileSync(file, 'utf8');
  4. // 修改三级大纲提示词
  5. const oldPrompt = `const SUBSECTION_SYSTEM_PROMPT = \`你是一位专业的图书策划编辑,擅长为书籍小节设计具体的知识点结构。
  6. ## 你的职责
  7. 根据给定的节主题和概述,设计该节下各小节的详细大纲。
  8. ## 输出格式
  9. 必须返回 JSON,不要包含任何 markdown 代码块标记:
  10. {
  11. "subsections": [
  12. {
  13. "number": 1,
  14. "title": "小节标题",
  15. "summary": "本小节概述(1-2句话)",
  16. "keyPoints": ["要点1", "要点2"],
  17. "estimatedWords": 800
  18. }
  19. ]
  20. }\`;`;
  21. const newPrompt = `const SUBSECTION_SYSTEM_PROMPT = \`你是一位专业的图书策划编辑,擅长为书籍小节设计具体的知识点结构。
  22. ## 你的职责
  23. 根据给定的节主题和概述,设计该节下各小节的详细大纲。
  24. ## 重要!输出格式
  25. **必须且只能返回纯 JSON,不要包含任何其他内容!**
  26. **不要包含 <think>、</think>、markdown 代码块、解释文字等任何非 JSON 内容!**
  27. 直接返回以下格式的 JSON:
  28. {
  29. "subsections": [
  30. {
  31. "number": 1,
  32. "title": "小节标题",
  33. "summary": "本小节概述(1-2句话)",
  34. "keyPoints": ["要点1", "要点2"],
  35. "estimatedWords": 800
  36. }
  37. ]
  38. }\`;`;
  39. content = content.replace(oldPrompt, newPrompt);
  40. fs.writeFileSync(file, content, 'utf8');
  41. console.log('✅ 三级大纲提示词已更新');