From 84eb6b2454260d722ed361a6c231a92215741954 Mon Sep 17 00:00:00 2001 From: pengx17 Date: Mon, 29 Apr 2024 11:23:02 +0000 Subject: [PATCH] fix(presets): mindmap expand (#6921) 1. mindMapToMarkdown is --- packages/presets/src/ai/entries/edgeless/actions-config.ts | 2 +- packages/presets/src/ai/utils/edgeless.ts | 6 ++---- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/packages/presets/src/ai/entries/edgeless/actions-config.ts b/packages/presets/src/ai/entries/edgeless/actions-config.ts index 83719b95572e..82a37632c147 100644 --- a/packages/presets/src/ai/entries/edgeless/actions-config.ts +++ b/packages/presets/src/ai/entries/edgeless/actions-config.ts @@ -291,7 +291,7 @@ const generateGroup: AIItemGroupConfig = { return Promise.resolve({ input: firstSelected.text?.toString() ?? '', - content: mindMapToMarkdown(mindmap), + mindmap: mindMapToMarkdown(mindmap), }); }), }, diff --git a/packages/presets/src/ai/utils/edgeless.ts b/packages/presets/src/ai/utils/edgeless.ts index 9abc1d632b13..a91bc104f52c 100644 --- a/packages/presets/src/ai/utils/edgeless.ts +++ b/packages/presets/src/ai/utils/edgeless.ts @@ -11,14 +11,12 @@ export function mindMapToMarkdown(mindmap: MindmapElementModel) { node: MindmapElementModel['tree']['children'][number], indent: number = 0 ) => { - markdownStr += `${node.element as ShapeElementModel}\n`; - const text = (node.element as ShapeElementModel).text?.toString() ?? ''; - markdownStr += `${' '.repeat(indent)}${text}\n`; + markdownStr += `${' '.repeat(indent)}- ${text}\n`; if (node.children) { - node.children.forEach(traverse, indent + 1); + node.children.forEach(node => traverse(node, indent + 2)); } };