.

src/paper/paper-gen/chapter.ts

@@ -69,7 +69,7 @@ function formatChapters(chapters: any[]) {
 }
 export async function genChapters(tools: LLMTools, major, title, desc) {
     const pRetry = (await eval("import('p-retry')")).default
-    const chapters = await pRetry(() => _genChapters(tools, major, title, desc), {
+    let chapters = await pRetry(() => _genChapters(tools, major, title, desc), {
         retries: 5,
         onFailedAttempt: (e) => console.log(e.stack)
     })
@@ -79,5 +79,22 @@ export async function genChapters(tools: LLMTools, major, title, desc) {
     //         return pRetry(() => genSections(major, tools, title, desc, chapters, index), { retries: 5 })
     //     })
     // )
-    return formatChapters(chapters)
+
+    chapters = formatChapters(chapters)
+
+    if (!chapters.find((i) => i.title.includes('参考文献'))) {
+        chapters.push({
+            title: '参考文献',
+            children: []
+        })
+    }
+
+    if (!chapters.find((i) => i.title.includes('附录'))) {
+        chapters.push({
+            title: '附录',
+            children: []
+        })
+    }
+
+    return chapters
 }

src/paper/paper-gen/general.ts

@@ -72,6 +72,7 @@ ${chapters2md(chapters)}
 你要严格按照markdown格式返回内容。
 不要擅作主张输出你对该内容的总结或者介绍等其他文字，不需要出现介绍下一章或其他承上启下的文字。
 你要严格遵守我的指令，按照我的指定的章节和小节完成论文的写作。
+你的写作应该有逻辑性，不要让人觉得你是一个机器人。
 请直接开始写作，不要对内容作任何解释。`
         paper.write(
             `\n\n# 第${NumberToChinese.convertNumber(i + 1, { removeLeadingOne: i + 1 >= 10 })}章 ${chapter.title}`