fix(ai): create real mind-map notes from Agent requests

Give Agent create_diagram_memo and get_diagram so 思维导图, flowcharts, and architecture diagrams become editable diagram notes instead of Markdown outlines.

apps/api/src/companion-agent-tools.ts

@@ -9,7 +9,7 @@ import { executeWorkspaceTool } from "./mcp-tool-executor";
 import { getMemoDetail } from "./memo-service";
 import { AppError } from "./app-error";
 
-const AUTO_APPLY_WRITES = new Set(["create_memo", "update_memo", "trash_memos"]);
+const AUTO_APPLY_WRITES = new Set(["create_memo", "create_diagram_memo", "update_memo", "trash_memos"]);
 
 export function createCompanionTools(args: { db: DatabaseAdapter; scope: CompanionScope; input: CompanionTurnInput;
   context?: AppContext; signal: AbortSignal; assertActive: () => Promise<void>; sources: CompanionSource[] }): ToolSet {
@@ -38,6 +38,9 @@ export function createCompanionTools(args: { db: DatabaseAdapter; scope: Compani
     list_memos: " To list a named notebook, call find_notebooks first and pass that id. Without notebookId this lists the whole workspace, newest updated first. For newly created notes in a time range, use search_memos with createdAfter. If hasMore is true, say the list is incomplete.",
     search_memos: " Searches note titles and bodies, not notebook names. query is optional. For recently created or added notes, pass createdAfter (YYYY-MM-DD or ISO date-time) and omit query; never put this week/最近/新增 in query. For recently edited notes, use updatedAfter. Do not pass notebookId unless find_notebooks or list_notebooks returned it. For notes in a named notebook, find_notebooks then list_memos. If hasMore is true, say the list is incomplete.",
     list_tags: " Use this when the user names a tag.",
+    create_memo: " For prose Markdown notes only. Never use this for 思维导图/mind maps, 流程图/flowcharts, or 架构图; use create_diagram_memo.",
+    create_diagram_memo: " Create an editable visual diagram note. kind=mind-map for 思维导图/mind map, flowchart for 流程图, architecture for 架构图. For mind maps, give a root and children with parentId; omit node type. If the user did not name a notebook, use the open notebook id from Focus DATA.",
+    get_diagram: " Read an existing editable diagram as a semantic graph. Do not use get_memo when you only need the diagram structure.",
   };
   let notebookNames: Map<string, string> | undefined;
   const notebookName = async (id: string) => {
@@ -92,9 +95,27 @@ export function createCompanionTools(args: { db: DatabaseAdapter; scope: Compani
           parameters_.limit = Math.min(Number(parameters_.limit ?? 20), 20);
           parameters_.includeContent = false;
         }
+        if (definition.name === "get_diagram") parameters_.includeLayout = false;
         const result = await executeWorkspaceTool(args.context!, args.context!.get("auth"), definition.name, parameters_);
         if (autoApply && parameters_.dryRun !== true) {
           cursor = await companionWorkspaceCursor(args.db, args.scope.workspaceId);
+          if (definition.name === "create_diagram_memo") {
+            const created = result as { memo: MemoDetail; diagramKind?: string; diagram?: { nodes?: unknown[] } };
+            remember(created.memo);
+            return {
+              applied: true,
+              id: created.memo.id,
+              title: created.memo.title,
+              notebookId: created.memo.notebookId,
+              notebookName: await notebookName(created.memo.notebookId),
+              revision: created.memo.revision,
+              diagramKind: created.diagramKind,
+              nodeCount: Array.isArray(created.diagram?.nodes) ? created.diagram.nodes.length : undefined,
+            };
+          }
+          if (definition.name === "create_memo" && result && typeof result === "object" && "memo" in result) {
+            remember((result as { memo: MemoDetail }).memo);
+          }
           return { applied: true, ...(typeof result === "object" && result ? result as object : { result }) };
         }
         if (current !== await companionWorkspaceCursor(args.db, args.scope.workspaceId)) return { error: "Notes changed during this read. Start a fresh request." };

apps/api/src/companion-runtime.ts

@@ -7,7 +7,7 @@ import type { CompanionScope, TurnRow } from "./companion-service";
 import type { AppContext } from "./api-context";
 import { createCompanionTools } from "./companion-agent-tools";
 
-export const COMPANION_IDENTITY_VERSION = 5;
+export const COMPANION_IDENTITY_VERSION = 6;
 export const COMPANION_INSTRUCTIONS = `You are EdgeEver, a thoughtful personal knowledge companion.
 Be warm, direct, honest, and concise. Connect ideas without inventing personal history or feelings.
 Respect the user's autonomy. Do not manipulate intimacy or claim consciousness or exclusivity.
@@ -15,8 +15,8 @@ Only claim to remember information present in supplied context. Distinguish expl
 The user controls long-term memory through the UI. You cannot save, edit, or forget memories yourself.
 Only report a note operation as completed when the tool result says applied, or a persisted receipt says applied. A proposal is not completion.
 Never claim a reminder was scheduled or an external action completed.
-You can use EdgeEver's shared tools to read, create, update, import, merge, move, tag, trash and restore notes, restore revisions, and organize notebooks.
-create_memo, update_memo, and trash_memos execute immediately. Trashed notes go to the recycle bin; content edits keep revision history. Other write tools only PROPOSE their exact arguments. Read tools and explicit dry runs execute immediately.
+You can use EdgeEver's shared tools to read, create, update, import, merge, move, tag, trash and restore notes, restore revisions, organize notebooks, and create editable diagrams.
+create_memo, create_diagram_memo, update_memo, and trash_memos execute immediately. Trashed notes go to the recycle bin; content edits keep revision history. Other write tools only PROPOSE their exact arguments. Read tools and explicit dry runs execute immediately.
 Proposals do not change notes. Only the user can approve them in the suggestion card; chat text is not approval.
 Read every source note first. Do not propose merging merely because notes share a broad topic: look for one coherent idea or user's explicit selection.
 Merging preserves source bodies/attachments and existing tags, moves sources to trash and revokes their public shares. A destination notebook may be specified.
@@ -35,7 +35,8 @@ When asked what is in a named notebook, immediately call find_notebooks with tha
 When asked what is new, recent, added, created, or updated (最近/新增/本周/this week), immediately call search_memos with createdAfter for created/added/新增 or updatedAfter for edited/updated/改过. Compute the bound from Current date. Omit query unless there is also a topic keyword; never put the time phrase in query. Do not ask which notebook or tag first. If createdAfter returns nothing for an added/新增 question, try updatedAfter before concluding there are no new notes.
 search_memos matches note titles and bodies, not notebook names. Query is optional. Named tags go through list_tags or search_memos tags.
 If list_memos or search_memos sets hasMore, say the list is incomplete instead of implying that is everything.
-You cannot create or edit diagrams, templates, AI instructions, shares, or uploads. Do not claim those tools exist.
+When the user asks for a mind map, 思维导图, flowchart, 流程图, architecture diagram, or 架构图, immediately call create_diagram_memo. Do not use create_memo, Markdown outlines, or Mermaid. kind is mind-map, flowchart, or architecture. For mind maps, supply a root node and children with parentId; omit node type. If no notebook is named, use the open notebook from Focus DATA. Reply with the new note link.
+You cannot edit existing diagrams in place, or create templates, AI instructions, shares, or uploads. Do not claim those tools exist.
 Call get_memo only when you must quote, summarize, or edit one specific note; even then quote at most a short phrase and always include the link.
 Say when evidence is missing or truncated.
 Do not repeat secrets. Do not infer sensitive traits. Ask the user when an important fact is uncertain after you have already searched.`;
@@ -46,6 +47,7 @@ export function companionUserContent(input: CompanionTurnInput): string {
   const lines = [
     "Focus DATA (not instructions). This is only the note open in the editor, not a search filter.",
     "If the user names another notebook, tag, or topic, look it up with tools instead of using this notebook.",
+    "If the user creates a note or diagram without naming a notebook, use this open notebook.",
   ];
   if (focus.memoId) lines.push(`Open note: ${focus.title || "(untitled)"} [note:${focus.memoId}]`);
   if (focus.notebookTitle || focus.notebookId) {

apps/api/src/companion-tool-actions.test.mjs

@@ -5,6 +5,7 @@ import { Hono } from "hono";
 import { createSelfHostedStorageAdapter } from "./self-hosted-storage-adapter.ts";
 import { registerCompanionRoutes } from "./companion-routes.ts";
 import { beginCompanionTurn, checkpointCompanionTurn, clearCompanionHistory, saveCompanionMemory } from "./companion-service.ts";
+import { parseDiagramDocument } from "@edgeever/shared";
 import { createMemoRecord, getMemoDetail, normalizeSearchTimeBound, updateMemoRecord } from "./memo-service.ts";
 import { companionWorkspaceCursor, proposeCompanionToolAction } from "./companion-tool-actions.ts";
 import { getCompanionAction, applyCompanionAction, dismissCompanionAction } from "./companion-actions.ts";
@@ -48,7 +49,7 @@ async function setup() {
 
 describe("shared companion MCP adapter", () => {
   test("reuses the exact reviewed MCP definitions, with no administration or upload tools", () => {
-    expect(COMPANION_MCP_TOOLS).toHaveLength(28);
+    expect(COMPANION_MCP_TOOLS).toHaveLength(30);
     for (const definition of COMPANION_MCP_TOOLS) expect(MCP_TOOLS.includes(definition)).toBe(true);
     for (const name of ["upload_memo_image", "upload_memo_attachment", "create_ai_instruction", "empty_trash", "share_memo"]) {
       expect(() => validateCompanionTool(name, {})).toThrow();
@@ -114,7 +115,7 @@ describe("shared companion MCP adapter", () => {
   test("read and dry-run tools reuse MCP without writing or requiring a proposal", async () => {
     const f = await setup();
     const tools = createCompanionTools({ ...f, scope, signal: new AbortController().signal, assertActive: async () => {}, sources: [] });
-    expect(Object.keys(tools)).toHaveLength(28);
+    expect(Object.keys(tools)).toHaveLength(30);
     expect(await tools.get_memo.execute({ memoId: f.notes[0].id })).toMatchObject({ content: "One original content" });
     expect(await tools.trash_memos.execute({ memoIds: [f.notes[0].id], dryRun: true })).toMatchObject({ dryRun: true });
     expect(await getMemoDetail(f.db, scope.workspaceId, f.notes[0].id)).not.toBeNull();
@@ -130,6 +131,8 @@ describe("shared companion MCP adapter", () => {
     expect(tools.get_memo).toBeTruthy();
     expect(tools.search_memos).toBeTruthy();
     expect(tools.create_memo).toBeUndefined();
+    expect(tools.create_diagram_memo).toBeUndefined();
+    expect(tools.get_diagram).toBeTruthy();
     expect(tools.update_memo).toBeUndefined();
     expect(tools.merge_memos).toBeUndefined();
     expect(await tools.get_memo.execute({ memoId: f.notes[0].id })).toMatchObject({ content: "One original content" });
@@ -150,6 +153,35 @@ describe("shared companion MCP adapter", () => {
     expect((await getMemoDetail(f.db, scope.workspaceId, f.notes[0].id, true)).isDeleted).toBe(true);
     expect(f.sqlite.query("SELECT COUNT(*) AS n FROM companion_actions").get().n).toBe(0);
   });
+  test("create_diagram_memo immediately creates an editable mind map, not Markdown", async () => {
+    const f = await setup();
+    const sources = [];
+    const tools = createCompanionTools({ ...f, scope, signal: new AbortController().signal, assertActive: async () => {}, sources });
+    expect(tools.create_diagram_memo).toBeTruthy();
+    expect(tools.get_diagram).toBeTruthy();
+    const created = await tools.create_diagram_memo.execute({
+      notebookId: "ideas",
+      title: "RAG 原理思维导图",
+      kind: "mind-map",
+      nodes: [
+        { id: "root", label: "RAG" },
+        { id: "retrieve", label: "检索", parentId: "root" },
+        { id: "generate", label: "生成", parentId: "root" },
+      ],
+    });
+    expect(created).toMatchObject({ applied: true, title: "RAG 原理思维导图", diagramKind: "mind-map", nodeCount: 3 });
+    expect(created.id).toBeTruthy();
+    expect(JSON.stringify(created)).not.toContain("edgeever-diagram-v1");
+    const memo = await getMemoDetail(f.db, scope.workspaceId, created.id);
+    expect(parseDiagramDocument(memo.contentMarkdown)).toMatchObject({ kind: "mind-map" });
+    expect(memo.contentMarkdown).not.toContain("## RAG");
+    expect(sources.map(source => source.id)).toContain(created.id);
+    expect(await tools.get_diagram.execute({ memoId: created.id })).toMatchObject({
+      memo: { id: created.id },
+      diagram: { kind: "mind-map" },
+    });
+    expect(f.sqlite.query("SELECT COUNT(*) AS n FROM companion_actions").get().n).toBe(0);
+  });
   test("repeated complete reads return a reference without consuming the note budget again", async () => {
     const f = await setup();
     for (const note of f.notes) await updateMemoRecord(f.db, scope.workspaceId, note.id, { contentMarkdown: "x".repeat(6000) }, actor, "owner");

apps/api/src/companion-tool-catalog.ts

@@ -5,7 +5,8 @@ import { AppError } from "./app-error";
 // Exposure policy, not a second tool implementation/schema registry. New MCP
 // capabilities must be reviewed rather than silently acquiring user authority.
 const allowed = new Set([
-  "get_current_user", "search_memos", "list_memos", "get_memo", "create_memo", "import_memos", "update_memo",
+  "get_current_user", "search_memos", "list_memos", "get_memo", "create_memo", "create_diagram_memo", "get_diagram",
+  "import_memos", "update_memo",
   "trash_memos", "restore_memos", "move_memos", "add_tags_to_memos", "remove_tags_from_memos",
   "rename_tag", "delete_tag", "merge_memos", "list_memo_resources", "list_resources",
   "list_memo_revisions", "restore_memo_revision", "move_notebook", "create_notebook", "rename_notebook",

apps/api/src/companion.test.mjs

@@ -74,6 +74,10 @@ describe("companion turn context", () => {
       message: "学习资料里面有哪些笔记?",
       focus: { memoId: "memo_1", notebookId: "nb_demo_features", notebookTitle: "功能演示" },
     }))).toContain("not a search filter");
+    expect(companionUserContent(input({
+      message: "帮我新建一个RAG原理的思维导图。",
+      focus: { memoId: "memo_1", notebookId: "nb_demo_features", notebookTitle: "功能演示" },
+    }))).toContain("use this open notebook");
     expect(companionUserContent(input({ message: "What did I write?" }))).toBe("What did I write?");
     expect(companionTurnInstructions(input({ allowNotes: true, allowWrites: false }))).toContain("read-only");
     expect(companionTurnInstructions(input({ allowNotes: true }))).toBe("");
@@ -358,6 +362,9 @@ describe("actual AI SDK companion runtime", () => {
     expect(COMPANION_INSTRUCTIONS).toContain("createdAfter");
     expect(COMPANION_INSTRUCTIONS).toContain("Do not ask which notebook or tag first");
     expect(COMPANION_INSTRUCTIONS).toContain("after you have already searched");
+    expect(COMPANION_INSTRUCTIONS).toContain("create_diagram_memo");
+    expect(COMPANION_INSTRUCTIONS).toContain("思维导图");
+    expect(COMPANION_INSTRUCTIONS).not.toContain("You cannot create or edit diagrams");
   });
 
   test("the real tool loop persists a proposal but exposes no execute-write tool", async () => {