AIがドキュメントのdiffとして回答を出力する

Author: na-trium-144

app/actions/chatActions.ts

@@ -66,6 +66,13 @@ export async function askAI(params: ChatParams): Promise<ChatResult> {
     `質問に答える際には、ユーザーが閲覧しているセクションの内容を特に考慮してください。`
   );
   prompt.push(``);
+  prompt.push(
+    `質問への回答はユーザー向けのメッセージに加えて、ドキュメント自体を改訂するという形でも可能です。`
+  );
+  prompt.push(
+    `質問内容とドキュメントの内容の関連性が深く、比較的長めの解説をしたい場合、またはドキュメントへの補足がしたい場合は、そちらの形式での回答を検討してください。`
+  );
+  prompt.push(``);
   prompt.push(`# ドキュメント`);
   prompt.push(``);
   for (const section of sectionContent) {
@@ -157,6 +164,27 @@ export async function askAI(params: ChatParams): Promise<ChatResult> {
   prompt.push(
     "  - 水平線(---)はシステムが区切りとして認識するので、ユーザーへの回答中に水平線を使用することはできません。"
   );
+  prompt.push(
+    "- ユーザーへのメッセージの最後の行の次には水平線 --- を出力してください。"
+  );
+  prompt.push(
+    "- それ以降の行に、ドキュメントの一部を改訂したい場合はその差分を"
+  );
+  prompt.push("<<<<<<< SEARCH");
+  prompt.push("修正したい元の文章の塊（一字一句違わずに）");
+  prompt.push("=======");
+  prompt.push("修正後の新しい文章の塊");
+  prompt.push(">>>>>>> REPLACE");
+  prompt.push("の形式で出力してください。");
+  prompt.push(
+    "  - 複数箇所改訂したい場合は上の形式の出力を複数回繰り返してください。"
+  );
+  prompt.push(
+    "  - ドキュメントにテキストを追加したい場合は追加したい箇所の前後のテキストを含めて出力してください。"
+  );
+  prompt.push(
+    "  - セクションid、セクション見出し、およびコードブロックの内側を編集することはできません。それ以外の文章のみを編集してください。"
+  );
   console.log(prompt);
 
   try {
@@ -170,6 +198,14 @@ export async function askAI(params: ChatParams): Promise<ChatResult> {
       targetSectionId = introSectionId(path);
     }
     const responseMessage = text.split(/-{3,}/)[1].trim();
+    const diffMatch = text
+      .split(/-{3,}/)[2]
+      .matchAll(
+        /<{3,}\s*SEARCH\n([\s\S]*?)\n={3,}\n([\s\S]*?)\n>{3,}\s*REPLACE/g
+      );
+    const diff: { search: string; replace: string }[] = diffMatch
+      ? Array.from(diffMatch, (m) => ({ search: m[1], replace: m[2] }))
+      : [];
     const newChat = await addChat(path, targetSectionId, [
       { role: "user", content: userQuestion },
       { role: "ai", content: responseMessage },