i/next-ai-draw-io
1
0
Fork 0
mirror of https://github.com/DayuanJiang/next-ai-draw-io.git synced 2026-01-02 22:32:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: use cached chartXML for edit_diagram to avoid Vercel timeout

Browse Source 
DrawIO iframe export was unreliable on Vercel due to network latency,
causing edit_diagram tool to hang. Now uses chartXML from context directly,
falling back to export only when no cached XML exists.
This commit is contained in:
dayuan.jiang
2025-12-05 00:43:21 +09:00
parent 46cbc3354c
commit 7b8bd8c621
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
components/chat-panel.tsx
View File

@@ -99,8 +99,18 @@ export default function ChatPanel({
let currentXml = "";
try {
// Fetch without saving to history - edit_diagram shouldn't create history entry
console.log("[edit_diagram] Starting...");
// Use chartXML from context directly - more reliable than export
// especially on Vercel where DrawIO iframe may have latency issues
if (chartXML) {
currentXml = chartXML;
console.log("[edit_diagram] Using cached chartXML, length:", currentXml.length);
} else {
// Fallback to export only if no cached XML
console.log("[edit_diagram] No cached XML, fetching from DrawIO...");
currentXml = await onFetchChart(false);
console.log("[edit_diagram] Got XML from export, length:", currentXml.length);
}
const { replaceXMLParts } = await import("@/lib/utils");
const editedXml = replaceXMLParts(currentXml, edits);
@@ -112,8 +122,9 @@ export default function ChatPanel({
toolCallId: toolCall.toolCallId,
output: `Successfully applied ${edits.length} edit(s) to the diagram.`,
});
console.log("[edit_diagram] Success");
} catch (error) {
console.error("Edit diagram failed:", error);
console.error("[edit_diagram] Failed:", error);
const errorMessage =
error instanceof Error
@@ -127,7 +138,7 @@ export default function ChatPanel({
Current diagram XML:
\`\`\`xml
${currentXml}
${currentXml || "No XML available"}
\`\`\`
Please retry with an adjusted search pattern or use display_diagram if retries are exhausted.`,