From 0fbd3fe842df9e51bb312a4bf65362bd888a1712 Mon Sep 17 00:00:00 2001
From: "dayuan.jiang" <jdy.toh@gmail.com>
Date: Sun, 14 Dec 2025 12:33:47 +0900
Subject: [PATCH] docs: fix append_diagram instructions for consistency

- Change 'Do NOT include' to 'Do NOT start with' (clearer intent)
- Add <mxCell id="0"> to prohibited start patterns
- Change 'closing tags </root></mxGraphModel>' to just '</root>' (wrapWithMxFile handles the rest)
---
 app/api/chat/route.ts     | 4 ++--
 components/chat-panel.tsx | 4 ++--
 lib/system-prompts.ts     | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/app/api/chat/route.ts b/app/api/chat/route.ts
index 9d06327..19e5560 100644
--- a/app/api/chat/route.ts
+++ b/app/api/chat/route.ts
@@ -444,9 +444,9 @@ IMPORTANT: Keep edits concise:
 WHEN TO USE: Only call this tool after display_diagram was truncated (you'll see an error message about truncation).
 
 CRITICAL INSTRUCTIONS:
-1. Do NOT include <mxGraphModel> or <root> tags - they already exist in the partial
+1. Do NOT start with <mxGraphModel>, <root>, or <mxCell id="0"> - they already exist in the partial
 2. Continue from EXACTLY where your previous output stopped
-3. Generate the remaining XML including closing tags </root></mxGraphModel>
+3. End with the closing </root> tag to complete the diagram
 4. If still truncated, call append_diagram again with the next fragment
 
 Example: If previous output ended with '<mxCell id="x" style="rounded=1', continue with ';" vertex="1">...' and complete the remaining elements.`,
diff --git a/components/chat-panel.tsx b/components/chat-panel.tsx
index f21f939..1fbb601 100644
--- a/components/chat-panel.tsx
+++ b/components/chat-panel.tsx
@@ -244,9 +244,9 @@ ${partialEnding}
 \`\`\`
 
 NEXT STEP: Call append_diagram with the continuation XML.
-- Do NOT include <mxGraphModel> or <root> tags
+- Do NOT start with <mxGraphModel>, <root>, or <mxCell id="0"> (they already exist)
 - Start from EXACTLY where you stopped
-- Continue until complete with </root></mxGraphModel>`,
+- End with the closing </root> tag to complete the diagram`,
                     })
                     return
                 }
diff --git a/lib/system-prompts.ts b/lib/system-prompts.ts
index e99ac96..d878acb 100644
--- a/lib/system-prompts.ts
+++ b/lib/system-prompts.ts
@@ -186,9 +186,9 @@ const EXTENDED_ADDITIONS = `
 **WHEN TO USE:** Only call this tool when display_diagram output was truncated (you'll see an error message about truncation).
 
 **CRITICAL RULES:**
-1. Do NOT include <mxGraphModel>, <root>, or <mxCell id="0"> - they already exist in the partial
+1. Do NOT start with <mxGraphModel>, <root>, or <mxCell id="0"> - they already exist in the partial
 2. Continue from EXACTLY where your previous output stopped
-3. Generate the remaining XML including closing tags </root></mxGraphModel>
+3. End with the closing </root> tag to complete the diagram
 4. If still truncated, call append_diagram again with the next fragment
 
 **Example:** If previous output ended with \`<mxCell id="x" style="rounded=1\`, continue with \`;" vertex="1">...\` and complete the remaining elements.