@platejs/ai 51.0.0 → 51.0.1

package/dist/react/index.js

@@ -34,9 +34,9 @@ __export(react_exports, {
   AIPlugin: () => AIPlugin,
   CopilotPlugin: () => CopilotPlugin,
   acceptAIChat: () => acceptAIChat,
+  acceptAISuggestions: () => acceptAISuggestions,
   acceptCopilot: () => acceptCopilot,
   acceptCopilotNextWord: () => acceptCopilotNextWord,
-  acceptSuggestions: () => acceptSuggestions,
   aiCommentToRange: () => aiCommentToRange,
   applyAISuggestions: () => applyAISuggestions,
   callCompletionApi: () => callCompletionApi,
@@ -54,6 +54,7 @@ __export(react_exports, {
   isCompleteMath: () => isCompleteMath,
   isSameNode: () => isSameNode,
   nodesWithProps: () => nodesWithProps,
+  rejectAISuggestions: () => rejectAISuggestions,
   removeAnchorAIChat: () => removeAnchorAIChat,
   renderCopilotBelowNodes: () => renderCopilotBelowNodes,
   replaceSelectionAIChat: () => replaceSelectionAIChat,
@@ -341,15 +342,41 @@ var aiCommentToRange = (editor, aiComment) => {
 };
 
 // src/react/ai-chat/AIChatPlugin.ts
-var import_react9 = require("@platejs/selection/react");
-var import_platejs14 = require("platejs");
-var import_react10 = require("platejs/react");
+var import_react12 = require("@platejs/selection/react");
+var import_platejs15 = require("platejs");
+var import_react13 = require("platejs/react");
 
 // src/react/ai-chat/transforms/acceptAIChat.ts
-var import_suggestion2 = require("@platejs/suggestion");
-var import_react2 = require("@platejs/suggestion/react");
 var import_platejs7 = require("platejs");
 var import_react3 = require("platejs/react");
+
+// src/react/ai-chat/utils/acceptAISuggestions.ts
+var import_suggestion2 = require("@platejs/suggestion");
+var import_react2 = require("@platejs/suggestion/react");
+var acceptAISuggestions = (editor) => {
+  const suggestions = editor.getApi(import_react2.SuggestionPlugin).suggestion.nodes({
+    transient: true
+  });
+  suggestions.forEach(([suggestionNode]) => {
+    const suggestionData = editor.getApi(import_react2.SuggestionPlugin).suggestion.suggestionData(suggestionNode);
+    if (!suggestionData) return;
+    const description = {
+      createdAt: new Date(suggestionData.createdAt),
+      keyId: (0, import_suggestion2.getSuggestionKey)(suggestionData.id),
+      suggestionId: suggestionData.id,
+      type: suggestionData.type,
+      userId: suggestionData.userId
+    };
+    (0, import_suggestion2.acceptSuggestion)(editor, description);
+  });
+  editor.tf.unsetNodes([(0, import_suggestion2.getTransientSuggestionKey)()], {
+    at: [],
+    mode: "all",
+    match: (n) => !!n[(0, import_suggestion2.getTransientSuggestionKey)()]
+  });
+};
+
+// src/react/ai-chat/transforms/acceptAIChat.ts
 var acceptAIChat = (editor) => {
   const mode = editor.getOption(AIChatPlugin, "mode");
   if (mode === "insert") {
@@ -370,43 +397,287 @@ var acceptAIChat = (editor) => {
   }
   if (mode === "chat") {
     withAIBatch(editor, () => {
-      acceptSuggestions(editor);
+      acceptAISuggestions(editor);
     });
     editor.getApi(AIChatPlugin).aiChat.hide();
   }
 };
-var acceptSuggestions = (editor) => {
-  const suggestions = editor.getApi(import_react2.SuggestionPlugin).suggestion.nodes({
+
+// src/react/ai-chat/transforms/insertBelowAIChat.ts
+var import_react11 = require("@platejs/selection/react");
+var import_cloneDeep2 = __toESM(require("lodash/cloneDeep.js"));
+var import_platejs12 = require("platejs");
+
+// src/react/ai-chat/utils/applyAISuggestions.ts
+var import_markdown2 = require("@platejs/markdown");
+var import_react4 = require("@platejs/selection/react");
+var import_suggestion3 = require("@platejs/suggestion");
+var import_platejs8 = require("platejs");
+var applyAISuggestions = (editor, content) => {
+  editor.getApi({ key: import_platejs8.KEYS.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
+  const isBlockSelecting = editor.getOption(
+    import_react4.BlockSelectionPlugin,
+    "isSelectingSome"
+  );
+  const { chatNodes } = editor.getOptions(AIChatPlugin);
+  if (isBlockSelecting || editor.api.blocks({ mode: "highest" }).length > 1) {
+    const setReplaceIds = (ids) => {
+      editor.setOption(AIChatPlugin, "_replaceIds", ids);
+    };
+    if (editor.getOption(AIChatPlugin, "_replaceIds").length === 0) {
+      setReplaceIds(chatNodes.map((node) => node.id));
+    }
+    const diffNodes = getDiffNodes(editor, content);
+    const replaceNodes = Array.from(
+      editor.api.nodes({
+        at: [],
+        match: (n) => import_platejs8.ElementApi.isElement(n) && editor.getOption(AIChatPlugin, "_replaceIds").includes(n.id)
+      })
+    );
+    replaceNodes.forEach(([node, path], index) => {
+      const replaceNode = node;
+      const diffNode = diffNodes[index];
+      const isSameString = (0, import_suggestion3.SkipSuggestionDeletes)(editor, replaceNode) === (0, import_suggestion3.SkipSuggestionDeletes)(editor, diffNode);
+      const isSameSuggestion = replaceNode.suggestion?.type === diffNode.suggestion?.type;
+      if (index === replaceNodes.length - 1 && diffNodes.length > replaceNodes.length) {
+        editor.tf.replaceNodes(diffNodes.slice(index), {
+          at: path
+        });
+      }
+      if (isSameString && isSameSuggestion && node.id == diffNode.id) {
+        return;
+      }
+      editor.tf.replaceNodes(diffNode, {
+        at: path
+      });
+    });
+    editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.set(diffNodes.map((node) => node.id));
+    setReplaceIds(diffNodes.map((node) => node.id));
+  } else {
+    const diffNodes = getDiffNodes(editor, content);
+    editor.tf.insertFragment(diffNodes);
+    const nodes = Array.from(
+      editor.api.nodes({
+        at: [],
+        mode: "lowest",
+        match: (n) => import_platejs8.TextApi.isText(n) && !!n[(0, import_suggestion3.getTransientSuggestionKey)()]
+      })
+    );
+    const range = editor.api.nodesRange(nodes);
+    editor.tf.setSelection(range);
+    return;
+  }
+};
+var withProps = (diffNodes, chatNodes) => {
+  return diffNodes.map((node, index) => {
+    if (!import_platejs8.ElementApi.isElement(node)) return node;
+    const originalNode = chatNodes?.[index];
+    return {
+      ...node,
+      ...originalNode ?? { id: (0, import_platejs8.nanoid)() },
+      children: node.children
+    };
+  });
+};
+var withTransient = (diffNodes) => {
+  return diffNodes.map((node) => {
+    if (import_platejs8.TextApi.isText(node)) {
+      return {
+        ...node,
+        [(0, import_suggestion3.getTransientSuggestionKey)()]: true
+      };
+    } else {
+      return {
+        ...node,
+        children: withTransient(node.children),
+        [(0, import_suggestion3.getTransientSuggestionKey)()]: true
+      };
+    }
+  });
+};
+var withoutSuggestionAndComments = (nodes) => {
+  return nodes.map((node) => {
+    if (import_platejs8.TextApi.isText(node)) {
+      if (node[import_platejs8.KEYS.suggestion] || node[import_platejs8.KEYS.comment]) {
+        return {
+          text: node.text
+        };
+      }
+      return node;
+    }
+    if (import_platejs8.ElementApi.isElement(node)) {
+      if (node[import_platejs8.KEYS.suggestion]) {
+        const nodeWithoutSuggestion = {};
+        Object.keys(node).forEach((key) => {
+          if (key !== import_platejs8.KEYS.suggestion && !key.startsWith(import_platejs8.KEYS.suggestion)) {
+            nodeWithoutSuggestion[key] = node[key];
+          }
+        });
+        return {
+          ...nodeWithoutSuggestion,
+          children: withoutSuggestionAndComments(node.children)
+        };
+      }
+      return {
+        ...node,
+        children: withoutSuggestionAndComments(node.children)
+      };
+    }
+    return node;
+  });
+};
+var getDiffNodes = (editor, aiContent) => {
+  const chatNodes = withoutSuggestionAndComments(
+    editor.getOption(AIChatPlugin, "chatNodes")
+  );
+  const aiNodes = withProps((0, import_markdown2.deserializeMd)(editor, aiContent), chatNodes);
+  const diffNodes = withTransient(
+    (0, import_suggestion3.diffToSuggestions)(editor, chatNodes, aiNodes, {
+      ignoreProps: ["id", "listStart"]
+    })
+  );
+  return diffNodes;
+};
+
+// src/react/ai-chat/utils/getLastAssistantMessage.ts
+var import_react5 = require("platejs/react");
+function getLastAssistantMessage(editor) {
+  const messages = editor.getOptions(AIChatPlugin).chat.messages;
+  return messages?.findLast((message) => message.role === "assistant");
+}
+function useLastAssistantMessage() {
+  const toolName = (0, import_react5.usePluginOption)(AIChatPlugin, "toolName");
+  const chat = (0, import_react5.usePluginOption)(AIChatPlugin, "chat");
+  if (toolName === "comment") return;
+  return chat.messages?.findLast((message) => message.role === "assistant");
+}
+
+// src/react/ai-chat/utils/rejectAISuggestions.ts
+var import_suggestion4 = require("@platejs/suggestion");
+var import_react6 = require("@platejs/suggestion/react");
+var rejectAISuggestions = (editor) => {
+  const suggestions = editor.getApi(import_react6.SuggestionPlugin).suggestion.nodes({
     transient: true
   });
   suggestions.forEach(([suggestionNode]) => {
-    const suggestionData = editor.getApi(import_react2.SuggestionPlugin).suggestion.suggestionData(suggestionNode);
+    const suggestionData = editor.getApi(import_react6.SuggestionPlugin).suggestion.suggestionData(suggestionNode);
     if (!suggestionData) return;
     const description = {
       createdAt: new Date(suggestionData.createdAt),
-      keyId: (0, import_suggestion2.getSuggestionKey)(suggestionData.id),
+      keyId: (0, import_suggestion4.getSuggestionKey)(suggestionData.id),
       suggestionId: suggestionData.id,
       type: suggestionData.type,
       userId: suggestionData.userId
     };
-    (0, import_suggestion2.acceptSuggestion)(editor, description);
+    (0, import_suggestion4.rejectSuggestion)(editor, description);
   });
-  editor.tf.unsetNodes([(0, import_suggestion2.getTransientSuggestionKey)()], {
+  editor.tf.unsetNodes([(0, import_suggestion4.getTransientSuggestionKey)()], {
     at: [],
     mode: "all",
-    match: (n) => !!n[(0, import_suggestion2.getTransientSuggestionKey)()]
+    match: (n) => !!n[(0, import_suggestion4.getTransientSuggestionKey)()]
   });
 };
 
-// src/react/ai-chat/transforms/insertBelowAIChat.ts
-var import_react5 = require("@platejs/selection/react");
-var import_cloneDeep2 = __toESM(require("lodash/cloneDeep.js"));
+// src/react/ai-chat/utils/resetAIChat.ts
 var import_platejs9 = require("platejs");
+var import_react7 = require("platejs/react");
+var resetAIChat = (editor, { undo = true } = {}) => {
+  const { api, getOptions, setOptions } = (0, import_react7.getEditorPlugin)(
+    editor,
+    {
+      key: import_platejs9.KEYS.aiChat
+    }
+  );
+  api.aiChat.stop();
+  const chat = getOptions().chat;
+  if (chat.messages && chat.messages.length > 0) {
+    chat.setMessages?.([]);
+  }
+  setOptions({
+    _replaceIds: [],
+    chatNodes: [],
+    mode: "insert",
+    toolName: null
+  });
+  if (undo) {
+    editor.getTransforms(AIPlugin).ai.undo();
+  }
+};
+
+// src/react/ai-chat/utils/submitAIChat.ts
+var import_selection2 = require("@platejs/selection");
+var import_react8 = require("@platejs/selection/react");
+var import_platejs10 = require("platejs");
+var import_react9 = require("platejs/react");
+var submitAIChat = (editor, input, {
+  mode,
+  options,
+  prompt,
+  toolName: toolNameProps
+} = {}) => {
+  const { getOptions, setOption } = (0, import_react9.getEditorPlugin)(
+    editor,
+    {
+      key: import_platejs10.KEYS.aiChat
+    }
+  );
+  const { chat, toolName: toolNameOption } = getOptions();
+  const toolName = toolNameProps ?? toolNameOption ?? null;
+  if (!prompt && input?.length === 0) {
+    return;
+  }
+  if (!prompt) {
+    prompt = input;
+  }
+  if (!mode) {
+    mode = (0, import_selection2.isSelecting)(editor) ? "chat" : "insert";
+  }
+  if (mode === "insert") {
+    editor.getTransforms(AIPlugin).ai.undo();
+  }
+  setOption("mode", mode);
+  setOption("toolName", toolName);
+  const blocks = editor.getApi(import_react8.BlockSelectionPlugin).blockSelection.getNodes();
+  const blocksRange = editor.api.nodesRange(blocks);
+  const promptText = getEditorPrompt(editor, {
+    prompt
+  });
+  const selection = blocks.length > 0 ? blocksRange : editor.selection;
+  let chatNodes;
+  if (blocks.length > 0) {
+    chatNodes = blocks.map((block) => block[0]);
+  } else {
+    const selectionBlocks = editor.api.blocks({ mode: "highest" });
+    if (selectionBlocks.length > 1) {
+      chatNodes = selectionBlocks.map((block) => block[0]);
+    } else {
+      chatNodes = editor.api.fragment();
+    }
+  }
+  setOption("chatNodes", chatNodes);
+  setOption("chatSelection", blocks.length > 0 ? null : editor.selection);
+  const ctx = {
+    children: editor.children,
+    selection: selection ?? null,
+    toolName
+  };
+  void chat.sendMessage?.(
+    {
+      text: promptText
+    },
+    {
+      body: {
+        ctx
+      },
+      ...options
+    }
+  );
+};
 
 // src/react/ai-chat/transforms/replaceSelectionAIChat.ts
-var import_react4 = require("@platejs/selection/react");
+var import_react10 = require("@platejs/selection/react");
 var import_cloneDeep = __toESM(require("lodash/cloneDeep.js"));
-var import_platejs8 = require("platejs");
+var import_platejs11 = require("platejs");
 var createFormattedBlocks = ({
   blocks,
   format,
@@ -414,12 +685,12 @@ var createFormattedBlocks = ({
 }) => {
   if (format === "none") return (0, import_cloneDeep.default)(blocks);
   const [sourceNode] = sourceBlock;
-  const firstTextEntry = import_platejs8.NodeApi.firstText(sourceNode);
+  const firstTextEntry = import_platejs11.NodeApi.firstText(sourceNode);
   if (!firstTextEntry) return null;
-  const blockProps = import_platejs8.NodeApi.extractProps(sourceNode);
-  const textProps = import_platejs8.NodeApi.extractProps(firstTextEntry[0]);
+  const blockProps = import_platejs11.NodeApi.extractProps(sourceNode);
+  const textProps = import_platejs11.NodeApi.extractProps(firstTextEntry[0]);
   const applyTextFormatting = (node) => {
-    if (import_platejs8.TextApi.isText(node)) {
+    if (import_platejs11.TextApi.isText(node)) {
       return { ...textProps, ...node };
     }
     if (node.children) {
@@ -443,10 +714,10 @@ var createFormattedBlocks = ({
 var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
   if (!sourceEditor || sourceEditor.api.isEmpty()) return;
   const isBlockSelecting = editor.getOption(
-    import_react4.BlockSelectionPlugin,
+    import_react10.BlockSelectionPlugin,
     "isSelectingSome"
   );
-  editor.getApi({ key: import_platejs8.KEYS.ai }).aiChat.hide();
+  editor.getApi({ key: import_platejs11.KEYS.ai }).aiChat.hide();
   if (!isBlockSelecting) {
     const firstBlock = editor.api.node({
       block: true,
@@ -459,7 +730,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
         sourceBlock: firstBlock
       });
       if (!formattedBlocks2) return;
-      if (firstBlock[0].type === import_platejs8.KEYS.codeLine && sourceEditor.children[0].type === import_platejs8.KEYS.codeBlock && sourceEditor.children.length === 1) {
+      if (firstBlock[0].type === import_platejs11.KEYS.codeLine && sourceEditor.children[0].type === import_platejs11.KEYS.codeBlock && sourceEditor.children.length === 1) {
         editor.tf.insertFragment(formattedBlocks2[0].children);
       } else {
         editor.tf.insertFragment(formattedBlocks2);
@@ -471,14 +742,14 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
     editor.tf.focus();
     return;
   }
-  const blockSelectionApi = editor.getApi(import_react4.BlockSelectionPlugin).blockSelection;
+  const blockSelectionApi = editor.getApi(import_react10.BlockSelectionPlugin).blockSelection;
   const selectedBlocks = blockSelectionApi.getNodes();
   if (selectedBlocks.length === 0) return;
   if (format === "none" || format === "single" && selectedBlocks.length > 1) {
     editor.tf.withoutNormalizing(() => {
-      (0, import_react4.removeBlockSelectionNodes)(editor);
+      (0, import_react10.removeBlockSelectionNodes)(editor);
       editor.tf.withNewBatch(() => {
-        editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(
+        editor.getTransforms(import_react10.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(
           (0, import_cloneDeep.default)(sourceEditor.children),
           {
             at: selectedBlocks[0][1]
@@ -486,7 +757,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
         );
       });
     });
-    editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.focus();
+    editor.getApi(import_react10.BlockSelectionPlugin).blockSelection.focus();
     return;
   }
   const [, firstBlockPath] = selectedBlocks[0];
@@ -497,14 +768,14 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
   });
   if (!formattedBlocks) return;
   editor.tf.withoutNormalizing(() => {
-    (0, import_react4.removeBlockSelectionNodes)(editor);
+    (0, import_react10.removeBlockSelectionNodes)(editor);
     editor.tf.withNewBatch(() => {
-      editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(formattedBlocks, {
+      editor.getTransforms(import_react10.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(formattedBlocks, {
         at: firstBlockPath
       });
     });
   });
-  editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.focus();
+  editor.getApi(import_react10.BlockSelectionPlugin).blockSelection.focus();
 };
 
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
@@ -512,192 +783,96 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
   const { toolName } = editor.getOptions(AIChatPlugin);
   if (toolName === "generate")
     return insertBelowGenerate(editor, sourceEditor, { format });
-  const selectedBlocks = editor.getApi(import_react5.BlockSelectionPlugin).blockSelection.getNodes();
-  const selectedIds = editor.getOptions(import_react5.BlockSelectionPlugin).selectedIds;
+  const selectedBlocks = editor.getApi(import_react11.BlockSelectionPlugin).blockSelection.getNodes();
+  const selectedIds = editor.getOptions(import_react11.BlockSelectionPlugin).selectedIds;
   editor.getTransforms(AIPlugin).ai.undo();
-  const insertBlocksAndSelect = editor.getTransforms(import_react5.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
+  const insertBlocksAndSelect = editor.getTransforms(import_react11.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
   if (!selectedIds || selectedIds.size === 0) return;
   const lastBlock = selectedBlocks.at(-1);
   if (!lastBlock) return;
-  const nextPath = import_platejs9.PathApi.next(lastBlock[1]);
+  const nextPath = import_platejs12.PathApi.next(lastBlock[1]);
   const nodes = selectedBlocks.map((block) => block[0]);
   insertBlocksAndSelect(nodes, {
     at: nextPath,
-    insertedCallback: () => {
-      withAIBatch(editor, () => {
-        acceptSuggestions(editor);
-      });
-    }
-  });
-  editor.getApi(AIChatPlugin).aiChat.hide({ focus: false });
-};
-var insertBelowGenerate = (editor, sourceEditor, { format = "single" } = {}) => {
-  if (!sourceEditor || sourceEditor.api.isEmpty()) return;
-  const isBlockSelecting = editor.getOption(
-    import_react5.BlockSelectionPlugin,
-    "isSelectingSome"
-  );
-  editor.getApi({ key: import_platejs9.KEYS.ai }).aiChat.hide();
-  const insertBlocksAndSelect = editor.getTransforms(import_react5.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
-  if (isBlockSelecting) {
-    const selectedBlocks = editor.getApi(import_react5.BlockSelectionPlugin).blockSelection.getNodes();
-    const selectedIds = editor.getOptions(import_react5.BlockSelectionPlugin).selectedIds;
-    if (!selectedIds || selectedIds.size === 0) return;
-    const lastBlock = selectedBlocks.at(-1);
-    if (!lastBlock) return;
-    const nextPath = import_platejs9.PathApi.next(lastBlock[1]);
-    if (format === "none") {
-      insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
-        at: nextPath
-      });
-      return;
-    }
-    const formattedBlocks = createFormattedBlocks({
-      blocks: (0, import_cloneDeep2.default)(sourceEditor.children),
-      format,
-      sourceBlock: lastBlock
-    });
-    if (!formattedBlocks) return;
-    insertBlocksAndSelect(formattedBlocks, {
-      at: nextPath
-    });
-  } else {
-    const [, end] = import_platejs9.RangeApi.edges(editor.selection);
-    const endPath = [end.path[0]];
-    const currentBlock = editor.api.node({
-      at: endPath,
-      block: true,
-      mode: "lowest"
-    });
-    if (!currentBlock) return;
-    if (format === "none") {
-      insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
-        at: import_platejs9.PathApi.next(endPath)
-      });
-      return;
-    }
-    const formattedBlocks = createFormattedBlocks({
-      blocks: (0, import_cloneDeep2.default)(sourceEditor.children),
-      format,
-      sourceBlock: currentBlock
-    });
-    if (!formattedBlocks) return;
-    insertBlocksAndSelect(formattedBlocks, {
-      at: import_platejs9.PathApi.next(endPath)
-    });
-  }
-};
-
-// src/react/ai-chat/transforms/removeAnchorAIChat.ts
-var import_platejs10 = require("platejs");
-var removeAnchorAIChat = (editor, options) => {
-  editor.tf.withoutSaving(() => {
-    editor.tf.removeNodes({
-      at: [],
-      match: (n) => import_platejs10.ElementApi.isElement(n) && n.type === (0, import_platejs10.getPluginType)(editor, import_platejs10.KEYS.aiChat),
-      ...options
-    });
-  });
-};
-
-// src/react/ai-chat/utils/resetAIChat.ts
-var import_platejs11 = require("platejs");
-var import_react6 = require("platejs/react");
-var resetAIChat = (editor, { undo = true } = {}) => {
-  const { api, getOptions, setOptions } = (0, import_react6.getEditorPlugin)(
-    editor,
-    {
-      key: import_platejs11.KEYS.aiChat
-    }
-  );
-  api.aiChat.stop();
-  const chat = getOptions().chat;
-  if (chat.messages && chat.messages.length > 0) {
-    chat.setMessages?.([]);
-  }
-  setOptions({
-    _replaceIds: [],
-    chatNodes: [],
-    mode: "insert",
-    toolName: null
-  });
-  if (undo) {
-    editor.getTransforms(AIPlugin).ai.undo();
-  }
-};
-
-// src/react/ai-chat/utils/submitAIChat.ts
-var import_selection2 = require("@platejs/selection");
-var import_react7 = require("@platejs/selection/react");
-var import_platejs12 = require("platejs");
-var import_react8 = require("platejs/react");
-var submitAIChat = (editor, input, {
-  mode,
-  options,
-  prompt,
-  toolName: toolNameProps
-} = {}) => {
-  const { getOptions, setOption } = (0, import_react8.getEditorPlugin)(
-    editor,
-    {
-      key: import_platejs12.KEYS.aiChat
-    }
-  );
-  const { chat, toolName: toolNameOption } = getOptions();
-  const toolName = toolNameProps ?? toolNameOption ?? null;
-  if (!prompt && input?.length === 0) {
-    return;
-  }
-  if (!prompt) {
-    prompt = input;
-  }
-  if (!mode) {
-    mode = (0, import_selection2.isSelecting)(editor) ? "chat" : "insert";
-  }
-  if (mode === "insert") {
-    editor.getTransforms(AIPlugin).ai.undo();
-  }
-  setOption("mode", mode);
-  setOption("toolName", toolName);
-  const blocks = editor.getApi(import_react7.BlockSelectionPlugin).blockSelection.getNodes();
-  const blocksRange = editor.api.nodesRange(blocks);
-  const promptText = getEditorPrompt(editor, {
-    prompt
+    insertedCallback: () => {
+      withAIBatch(editor, () => {
+        acceptAISuggestions(editor);
+      });
+    }
   });
-  const selection = blocks.length > 0 ? blocksRange : editor.selection;
-  let chatNodes;
-  if (blocks.length > 0) {
-    chatNodes = blocks.map((block) => block[0]);
+  editor.getApi(AIChatPlugin).aiChat.hide({ focus: false });
+};
+var insertBelowGenerate = (editor, sourceEditor, { format = "single" } = {}) => {
+  if (!sourceEditor || sourceEditor.api.isEmpty()) return;
+  const isBlockSelecting = editor.getOption(
+    import_react11.BlockSelectionPlugin,
+    "isSelectingSome"
+  );
+  editor.getApi({ key: import_platejs12.KEYS.ai }).aiChat.hide();
+  const insertBlocksAndSelect = editor.getTransforms(import_react11.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
+  if (isBlockSelecting) {
+    const selectedBlocks = editor.getApi(import_react11.BlockSelectionPlugin).blockSelection.getNodes();
+    const selectedIds = editor.getOptions(import_react11.BlockSelectionPlugin).selectedIds;
+    if (!selectedIds || selectedIds.size === 0) return;
+    const lastBlock = selectedBlocks.at(-1);
+    if (!lastBlock) return;
+    const nextPath = import_platejs12.PathApi.next(lastBlock[1]);
+    if (format === "none") {
+      insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
+        at: nextPath
+      });
+      return;
+    }
+    const formattedBlocks = createFormattedBlocks({
+      blocks: (0, import_cloneDeep2.default)(sourceEditor.children),
+      format,
+      sourceBlock: lastBlock
+    });
+    if (!formattedBlocks) return;
+    insertBlocksAndSelect(formattedBlocks, {
+      at: nextPath
+    });
   } else {
-    const selectionBlocks = editor.api.blocks({ mode: "highest" });
-    if (selectionBlocks.length > 1) {
-      chatNodes = selectionBlocks.map((block) => block[0]);
-    } else {
-      chatNodes = editor.api.fragment();
+    const [, end] = import_platejs12.RangeApi.edges(editor.selection);
+    const endPath = [end.path[0]];
+    const currentBlock = editor.api.node({
+      at: endPath,
+      block: true,
+      mode: "lowest"
+    });
+    if (!currentBlock) return;
+    if (format === "none") {
+      insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
+        at: import_platejs12.PathApi.next(endPath)
+      });
+      return;
     }
+    const formattedBlocks = createFormattedBlocks({
+      blocks: (0, import_cloneDeep2.default)(sourceEditor.children),
+      format,
+      sourceBlock: currentBlock
+    });
+    if (!formattedBlocks) return;
+    insertBlocksAndSelect(formattedBlocks, {
+      at: import_platejs12.PathApi.next(endPath)
+    });
   }
-  setOption("chatNodes", chatNodes);
-  setOption("chatSelection", blocks.length > 0 ? null : editor.selection);
-  const ctx = {
-    children: editor.children,
-    selection: selection ?? null,
-    toolName
-  };
-  void chat.sendMessage?.(
-    {
-      text: promptText
-    },
-    {
-      body: {
-        ctx
-      },
+};
+
+// src/react/ai-chat/transforms/removeAnchorAIChat.ts
+var import_platejs13 = require("platejs");
+var removeAnchorAIChat = (editor, options) => {
+  editor.tf.withoutSaving(() => {
+    editor.tf.removeNodes({
+      at: [],
+      match: (n) => import_platejs13.ElementApi.isElement(n) && n.type === (0, import_platejs13.getPluginType)(editor, import_platejs13.KEYS.aiChat),
       ...options
-    }
-  );
+    });
+  });
 };
 
 // src/react/ai-chat/withAIChat.ts
-var import_platejs13 = require("platejs");
+var import_platejs14 = require("platejs");
 var withAIChat = ({
   api,
   editor,
@@ -739,11 +914,11 @@ var withAIChat = ({
       },
       normalizeNode(entry) {
         const [node, path] = entry;
-        if (node[import_platejs13.KEYS.ai] && !getOptions().open) {
+        if (node[import_platejs14.KEYS.ai] && !getOptions().open) {
           tf.ai.removeMarks({ at: path });
           return;
         }
-        if (import_platejs13.ElementApi.isElement(node) && node.type === type && !getOptions().open) {
+        if (import_platejs14.ElementApi.isElement(node) && node.type === type && !getOptions().open) {
           editor.getTransforms(AIChatPlugin).aiChat.removeAnchor({ at: path });
           return;
         }
@@ -754,8 +929,8 @@ var withAIChat = ({
 };
 
 // src/react/ai-chat/AIChatPlugin.ts
-var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
-  key: import_platejs14.KEYS.aiChat,
+var AIChatPlugin = (0, import_react13.createTPlatePlugin)({
+  key: import_platejs15.KEYS.aiChat,
   dependencies: ["ai"],
   node: {
     isElement: true
@@ -779,14 +954,14 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
   }
 }).overrideEditor(withAIChat).extendApi(({ editor, getOption, getOptions, setOption, type }) => {
   return {
-    reset: (0, import_platejs14.bindFirst)(resetAIChat, editor),
-    submit: (0, import_platejs14.bindFirst)(submitAIChat, editor),
+    reset: (0, import_platejs15.bindFirst)(resetAIChat, editor),
+    submit: (0, import_platejs15.bindFirst)(submitAIChat, editor),
     node: (options = {}) => {
       const { anchor = false, streaming = false, ...rest } = options;
       if (anchor) {
         return editor.api.node({
           at: [],
-          match: (n) => import_platejs14.ElementApi.isElement(n) && n.type === type,
+          match: (n) => import_platejs15.ElementApi.isElement(n) && n.type === type,
           ...rest
         });
       }
@@ -798,12 +973,12 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
           at: path,
           mode: "lowest",
           reverse: true,
-          match: (t) => !!t[(0, import_platejs14.getPluginType)(editor, import_platejs14.KEYS.ai)],
+          match: (t) => !!t[(0, import_platejs15.getPluginType)(editor, import_platejs15.KEYS.ai)],
           ...rest
         });
       }
       return editor.api.node({
-        match: (n) => n[(0, import_platejs14.getPluginType)(editor, import_platejs14.KEYS.ai)],
+        match: (n) => n[(0, import_platejs15.getPluginType)(editor, import_platejs15.KEYS.ai)],
         ...rest
       });
     },
@@ -813,9 +988,9 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
       if (chatSelection) {
         editor.tf.setSelection(chatSelection);
       } else {
-        editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.set(chatNodes.map((node) => node.id));
+        editor.getApi(import_react12.BlockSelectionPlugin).blockSelection.set(chatNodes.map((node) => node.id));
       }
-      const blocks = editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.getNodes();
+      const blocks = editor.getApi(import_react12.BlockSelectionPlugin).blockSelection.getNodes();
       const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
       const ctx = {
         children: editor.children,
@@ -841,8 +1016,8 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
     api.aiChat.reset({ undo });
     setOption("open", false);
     if (focus) {
-      if (editor.getOption(import_react9.BlockSelectionPlugin, "isSelectingSome")) {
-        editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.focus();
+      if (editor.getOption(import_react12.BlockSelectionPlugin, "isSelectingSome")) {
+        editor.getApi(import_react12.BlockSelectionPlugin).blockSelection.focus();
       } else {
         editor.tf.focus();
       }
@@ -860,21 +1035,21 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
     setOption("open", true);
   }
 })).extendTransforms(({ editor }) => ({
-  accept: (0, import_platejs14.bindFirst)(acceptAIChat, editor),
-  insertBelow: (0, import_platejs14.bindFirst)(insertBelowAIChat, editor),
-  removeAnchor: (0, import_platejs14.bindFirst)(removeAnchorAIChat, editor),
-  replaceSelection: (0, import_platejs14.bindFirst)(replaceSelectionAIChat, editor)
+  accept: (0, import_platejs15.bindFirst)(acceptAIChat, editor),
+  insertBelow: (0, import_platejs15.bindFirst)(insertBelowAIChat, editor),
+  removeAnchor: (0, import_platejs15.bindFirst)(removeAnchorAIChat, editor),
+  replaceSelection: (0, import_platejs15.bindFirst)(replaceSelectionAIChat, editor)
 }));
 
 // src/react/ai-chat/hooks/useAIChatEditor.ts
-var import_react11 = require("react");
-var import_markdown2 = require("@platejs/markdown");
-var import_react12 = require("platejs/react");
+var import_react14 = require("react");
+var import_markdown3 = require("@platejs/markdown");
+var import_react15 = require("platejs/react");
 var useAIChatEditor = (editor, content, { parser } = {}) => {
-  const { setOption } = (0, import_react12.useEditorPlugin)(AIChatPlugin);
-  editor.children = (0, import_react11.useMemo)(
+  const { setOption } = (0, import_react15.useEditorPlugin)(AIChatPlugin);
+  editor.children = (0, import_react14.useMemo)(
     () => {
-      return editor.getApi(import_markdown2.MarkdownPlugin).markdown.deserialize(content, {
+      return editor.getApi(import_markdown3.MarkdownPlugin).markdown.deserialize(content, {
         memoize: true,
         parser
       });
@@ -882,45 +1057,30 @@ var useAIChatEditor = (editor, content, { parser } = {}) => {
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [content]
   );
-  (0, import_react11.useEffect)(() => {
+  (0, import_react14.useEffect)(() => {
     setOption("aiEditor", editor);
   }, [editor, setOption]);
 };
 
 // src/react/ai-chat/hooks/useChatChunk.ts
-var import_react14 = require("react");
-var import_platejs15 = require("platejs");
-var import_react15 = require("platejs/react");
-
-// src/react/ai-chat/utils/getLastAssistantMessage.ts
-var import_react13 = require("platejs/react");
-function getLastAssistantMessage(editor) {
-  const messages = editor.getOptions(AIChatPlugin).chat.messages;
-  return messages?.findLast((message) => message.role === "assistant");
-}
-function useLastAssistantMessage() {
-  const toolName = (0, import_react13.usePluginOption)(AIChatPlugin, "toolName");
-  const chat = (0, import_react13.usePluginOption)(AIChatPlugin, "chat");
-  if (toolName === "comment") return;
-  return chat.messages?.findLast((message) => message.role === "assistant");
-}
-
-// src/react/ai-chat/hooks/useChatChunk.ts
+var import_react16 = require("react");
+var import_platejs16 = require("platejs");
+var import_react17 = require("platejs/react");
 var useChatChunk = ({
   onChunk,
   onFinish
 }) => {
-  const { status } = (0, import_react15.usePluginOption)(
-    { key: import_platejs15.KEYS.aiChat },
+  const { status } = (0, import_react17.usePluginOption)(
+    { key: import_platejs16.KEYS.aiChat },
     "chat"
   );
   const isLoading = status === "streaming" || status === "submitted";
   const content = useLastAssistantMessage()?.parts.find(
     (part) => part.type === "text"
   )?.text;
-  const insertedTextRef = (0, import_react14.useRef)("");
-  const prevIsLoadingRef = (0, import_react14.useRef)(isLoading);
-  (0, import_react14.useEffect)(() => {
+  const insertedTextRef = (0, import_react16.useRef)("");
+  const prevIsLoadingRef = (0, import_react16.useRef)(isLoading);
+  (0, import_react16.useEffect)(() => {
     if (!isLoading) {
       insertedTextRef.current = "";
     }
@@ -929,7 +1089,7 @@ var useChatChunk = ({
     }
     prevIsLoadingRef.current = isLoading;
   }, [isLoading]);
-  (0, import_react14.useEffect)(() => {
+  (0, import_react16.useEffect)(() => {
     if (!content) {
       return;
     }
@@ -950,9 +1110,9 @@ var useChatChunk = ({
 };
 
 // src/react/ai-chat/hooks/useEditorChat.ts
-var import_react16 = require("react");
-var import_react17 = require("@platejs/selection/react");
-var import_react18 = require("platejs/react");
+var import_react18 = require("react");
+var import_react19 = require("@platejs/selection/react");
+var import_react20 = require("platejs/react");
 var useEditorChat = ({
   chat,
   onOpenBlockSelection,
@@ -960,15 +1120,15 @@ var useEditorChat = ({
   onOpenCursor,
   onOpenSelection
 }) => {
-  const { editor } = (0, import_react18.useEditorPlugin)(AIChatPlugin);
-  const open = (0, import_react18.usePluginOption)(AIChatPlugin, "open");
-  (0, import_react16.useEffect)(() => {
+  const { editor } = (0, import_react20.useEditorPlugin)(AIChatPlugin);
+  const open = (0, import_react20.usePluginOption)(AIChatPlugin, "open");
+  (0, import_react18.useEffect)(() => {
     onOpenChange?.(open);
     if (open) {
       if (onOpenBlockSelection) {
-        const blockSelectionApi = editor.getApi(import_react17.BlockSelectionPlugin).blockSelection;
+        const blockSelectionApi = editor.getApi(import_react19.BlockSelectionPlugin).blockSelection;
         const isBlockSelecting = editor.getOption(
-          import_react17.BlockSelectionPlugin,
+          import_react19.BlockSelectionPlugin,
           "isSelectingSome"
         );
         if (isBlockSelecting) {
@@ -989,14 +1149,14 @@ var useEditorChat = ({
 };
 
 // src/react/ai-chat/streaming/streamDeserializeInlineMd.ts
-var import_markdown3 = require("@platejs/markdown");
+var import_markdown4 = require("@platejs/markdown");
 var streamDeserializeInlineMd = (editor, text, options) => {
-  return editor.getApi(import_markdown3.MarkdownPlugin).markdown.deserializeInline(text, options);
+  return editor.getApi(import_markdown4.MarkdownPlugin).markdown.deserializeInline(text, options);
 };
 
 // src/react/ai-chat/streaming/streamDeserializeMd.ts
-var import_markdown4 = require("@platejs/markdown");
-var import_platejs18 = require("platejs");
+var import_markdown5 = require("@platejs/markdown");
+var import_platejs19 = require("platejs");
 
 // src/react/ai-chat/streaming/utils/utils.ts
 var getChunkTrimmed = (chunk, {
@@ -1051,22 +1211,22 @@ var getListNode = (editor, node) => {
 };
 
 // src/react/ai-chat/streaming/utils/isSameNode.ts
-var import_platejs16 = require("platejs");
+var import_platejs17 = require("platejs");
 var LIST_STYLE_TYPE = "listStyleType";
 var isSameNode = (editor, node1, node2) => {
-  if (node1.type !== editor.getType(import_platejs16.KEYS.p) || node2.type !== editor.getType(import_platejs16.KEYS.p))
+  if (node1.type !== editor.getType(import_platejs17.KEYS.p) || node2.type !== editor.getType(import_platejs17.KEYS.p))
     return node1.type === node2.type;
-  if ((0, import_platejs16.isDefined)(node1[LIST_STYLE_TYPE]) || (0, import_platejs16.isDefined)(node2[LIST_STYLE_TYPE])) {
+  if ((0, import_platejs17.isDefined)(node1[LIST_STYLE_TYPE]) || (0, import_platejs17.isDefined)(node2[LIST_STYLE_TYPE])) {
     return node1[LIST_STYLE_TYPE] === node2[LIST_STYLE_TYPE];
   }
   return node1.type === node2.type;
 };
 
 // src/react/ai-chat/streaming/utils/nodesWithProps.ts
-var import_platejs17 = require("platejs");
+var import_platejs18 = require("platejs");
 var nodesWithProps = (editor, nodes, options) => {
   return nodes.map((node) => {
-    if (import_platejs17.ElementApi.isElement(node)) {
+    if (import_platejs18.ElementApi.isElement(node)) {
       return {
         ...getListNode(editor, node),
         ...options.elementProps,
@@ -1089,7 +1249,7 @@ var streamDeserializeMd = (editor, data, options) => {
   const value = withoutDeserializeInMdx(editor, input);
   if (Array.isArray(value)) return value;
   let blocks = [];
-  blocks = editor.getApi(import_markdown4.MarkdownPlugin).markdown.deserialize(input, {
+  blocks = editor.getApi(import_markdown5.MarkdownPlugin).markdown.deserialize(input, {
     ...options,
     preserveEmptyParagraphs: false
   });
@@ -1106,7 +1266,7 @@ var streamDeserializeMd = (editor, data, options) => {
       }
     ];
     const lastChild = lastBlock.children.at(-1);
-    if (lastChild && import_platejs18.TextApi.isText(lastChild) && Object.keys(lastChild).length === 1) {
+    if (lastChild && import_platejs19.TextApi.isText(lastChild) && Object.keys(lastChild).length === 1) {
       lastBlock.children.pop();
       const textNode2 = [
         {
@@ -1122,13 +1282,13 @@ var streamDeserializeMd = (editor, data, options) => {
   if (addNewLine && !isCodeBlockOrTable) {
     result.push({
       children: [{ text: "" }],
-      type: import_platejs18.KEYS.p
+      type: import_platejs19.KEYS.p
     });
   }
   if (unshiftNewLine && !isCodeBlockOrTable) {
     result.unshift({
       children: [{ text: "" }],
-      type: import_platejs18.KEYS.p
+      type: import_platejs19.KEYS.p
     });
   }
   return result;
@@ -1148,7 +1308,7 @@ var withoutDeserializeInMdx = (editor, input) => {
               text: input
             }
           ],
-          type: (0, import_platejs18.getPluginType)(editor, import_platejs18.KEYS.p)
+          type: (0, import_platejs19.getPluginType)(editor, import_platejs19.KEYS.p)
         }
       ];
     }
@@ -1161,26 +1321,26 @@ var withoutDeserializeInMdx = (editor, input) => {
 };
 
 // src/react/ai-chat/streaming/streamInsertChunk.ts
-var import_platejs20 = require("platejs");
+var import_platejs21 = require("platejs");
 
 // src/react/ai-chat/streaming/streamSerializeMd.ts
-var import_markdown5 = require("@platejs/markdown");
-var import_platejs19 = require("platejs");
+var import_markdown6 = require("@platejs/markdown");
+var import_platejs20 = require("platejs");
 var trimEndHeading = (editor, value) => {
   const headingKeys = /* @__PURE__ */ new Set([
-    import_platejs19.KEYS.h1,
-    import_platejs19.KEYS.h2,
-    import_platejs19.KEYS.h3,
-    import_platejs19.KEYS.h4,
-    import_platejs19.KEYS.h5,
-    import_platejs19.KEYS.h6
+    import_platejs20.KEYS.h1,
+    import_platejs20.KEYS.h2,
+    import_platejs20.KEYS.h3,
+    import_platejs20.KEYS.h4,
+    import_platejs20.KEYS.h5,
+    import_platejs20.KEYS.h6
   ]);
   const lastBlock = value.at(-1);
   if (lastBlock && headingKeys.has(
-    (0, import_platejs19.getPluginKey)(editor, lastBlock.type) ?? lastBlock.type
-  ) && import_platejs19.ElementApi.isElement(lastBlock)) {
+    (0, import_platejs20.getPluginKey)(editor, lastBlock.type) ?? lastBlock.type
+  ) && import_platejs20.ElementApi.isElement(lastBlock)) {
     const lastTextNode = lastBlock.children.at(-1);
-    if (import_platejs19.TextApi.isText(lastTextNode)) {
+    if (import_platejs20.TextApi.isText(lastTextNode)) {
       const trimmedText = getChunkTrimmed(lastTextNode?.text);
       const newChildren = [
         ...lastBlock.children.slice(0, -1),
@@ -1202,7 +1362,7 @@ var streamSerializeMd = (editor, options, chunk) => {
   const { value: optionsValue, ...restOptions } = options;
   const { value } = trimEndHeading(editor, optionsValue ?? editor.children);
   let result = "";
-  result = editor.getApi(import_markdown5.MarkdownPlugin).markdown.serialize({
+  result = editor.getApi(import_markdown6.MarkdownPlugin).markdown.serialize({
     value,
     ...restOptions
   });
@@ -1234,7 +1394,7 @@ var streamSerializeMd = (editor, options, chunk) => {
 var getNextPath = (path, length) => {
   let result = path;
   for (let i = 0; i < length; i++) {
-    result = import_platejs20.PathApi.next(result);
+    result = import_platejs21.PathApi.next(result);
   }
   return result;
 };
@@ -1244,7 +1404,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
     const blocks = streamDeserializeMd(editor, chunk);
     const path = getCurrentBlockPath(editor);
     const startBlock = editor.api.node(path)[0];
-    const startInEmptyParagraph = import_platejs20.NodeApi.string(startBlock).length === 0 && startBlock.type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.p);
+    const startInEmptyParagraph = import_platejs21.NodeApi.string(startBlock).length === 0 && startBlock.type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.p);
     if (startInEmptyParagraph) {
       editor.tf.removeNodes({ at: path });
     }
@@ -1302,7 +1462,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
           },
           tempBlockChunks
         );
-        const blockText = import_platejs20.NodeApi.string(tempBlocks[0]);
+        const blockText = import_platejs21.NodeApi.string(tempBlocks[0]);
         if (serializedBlock === tempBlockChunks && blockText === serializedBlock) {
           editor.setOption(AIChatPlugin, "_blockChunks", tempBlockChunks);
         } else {
@@ -1324,7 +1484,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
             AIChatPlugin,
             "_blockChunks",
             // one block includes multiple children
-            tempBlocks[0].type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.codeBlock) || tempBlocks[0].type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.table) || tempBlocks[0].type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.equation) ? tempBlockChunks : serializedBlock2
+            tempBlocks[0].type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.codeBlock) || tempBlocks[0].type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.table) || tempBlocks[0].type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.equation) ? tempBlockChunks : serializedBlock2
           );
         }
       } else {
@@ -1354,7 +1514,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
         editor.tf.insertNodes(
           nodesWithProps(editor, tempBlocks.slice(1), options),
           {
-            at: import_platejs20.PathApi.next(_blockPath),
+            at: import_platejs21.PathApi.next(_blockPath),
             select: true
           }
         );
@@ -1376,7 +1536,7 @@ var getCurrentBlockPath = (editor) => {
   const getAnchorPreviousPath = (editor2) => {
     const anchorNode = editor2.getApi(AIChatPlugin).aiChat.node({ anchor: true });
     if (anchorNode) {
-      return import_platejs20.PathApi.previous(anchorNode[1]);
+      return import_platejs21.PathApi.previous(anchorNode[1]);
     }
   };
   const getFocusPath = (editor2) => {
@@ -1384,163 +1544,32 @@ var getCurrentBlockPath = (editor) => {
   };
   const path = getAnchorPreviousPath(editor) ?? getFocusPath(editor) ?? [0];
   const entry = editor.api.node(path);
-  if (entry && (entry[0].type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.columnGroup) || entry[0].type === (0, import_platejs20.getPluginType)(editor, import_platejs20.KEYS.table))) {
+  if (entry && (entry[0].type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.columnGroup) || entry[0].type === (0, import_platejs21.getPluginType)(editor, import_platejs21.KEYS.table))) {
     return editor.api.above()?.[1] ?? path;
   }
   return path;
 };
 
-// src/react/ai-chat/utils/applyAISuggestions.ts
-var import_markdown6 = require("@platejs/markdown");
-var import_react19 = require("@platejs/selection/react");
-var import_suggestion3 = require("@platejs/suggestion");
-var import_platejs21 = require("platejs");
-var applyAISuggestions = (editor, content) => {
-  editor.getApi({ key: import_platejs21.KEYS.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
-  const isBlockSelecting = editor.getOption(
-    import_react19.BlockSelectionPlugin,
-    "isSelectingSome"
-  );
-  const { chatNodes } = editor.getOptions(AIChatPlugin);
-  if (isBlockSelecting || editor.api.blocks({ mode: "highest" }).length > 1) {
-    const setReplaceIds = (ids) => {
-      editor.setOption(AIChatPlugin, "_replaceIds", ids);
-    };
-    if (editor.getOption(AIChatPlugin, "_replaceIds").length === 0) {
-      setReplaceIds(chatNodes.map((node) => node.id));
-    }
-    const diffNodes = getDiffNodes(editor, content);
-    const replaceNodes = Array.from(
-      editor.api.nodes({
-        at: [],
-        match: (n) => import_platejs21.ElementApi.isElement(n) && editor.getOption(AIChatPlugin, "_replaceIds").includes(n.id)
-      })
-    );
-    replaceNodes.forEach(([node, path], index) => {
-      const replaceNode = node;
-      const diffNode = diffNodes[index];
-      const isSameString = (0, import_suggestion3.SkipSuggestionDeletes)(editor, replaceNode) === (0, import_suggestion3.SkipSuggestionDeletes)(editor, diffNode);
-      const isSameSuggestion = replaceNode.suggestion?.type === diffNode.suggestion?.type;
-      if (index === replaceNodes.length - 1 && diffNodes.length > replaceNodes.length) {
-        editor.tf.replaceNodes(diffNodes.slice(index), {
-          at: path
-        });
-      }
-      if (isSameString && isSameSuggestion && node.id == diffNode.id) {
-        return;
-      }
-      editor.tf.replaceNodes(diffNode, {
-        at: path
-      });
-    });
-    editor.getApi(import_react19.BlockSelectionPlugin).blockSelection.set(diffNodes.map((node) => node.id));
-    setReplaceIds(diffNodes.map((node) => node.id));
-  } else {
-    const diffNodes = getDiffNodes(editor, content);
-    editor.tf.insertFragment(diffNodes);
-    const nodes = Array.from(
-      editor.api.nodes({
-        at: [],
-        mode: "lowest",
-        match: (n) => import_platejs21.TextApi.isText(n) && !!n[(0, import_suggestion3.getTransientSuggestionKey)()]
-      })
-    );
-    const range = editor.api.nodesRange(nodes);
-    editor.tf.setSelection(range);
-    return;
-  }
-};
-var withProps = (diffNodes, chatNodes) => {
-  return diffNodes.map((node, index) => {
-    if (!import_platejs21.ElementApi.isElement(node)) return node;
-    const originalNode = chatNodes?.[index];
-    return {
-      ...node,
-      ...originalNode ?? { id: (0, import_platejs21.nanoid)() },
-      children: node.children
-    };
-  });
-};
-var withTransient = (diffNodes) => {
-  return diffNodes.map((node) => {
-    if (import_platejs21.TextApi.isText(node)) {
-      return {
-        ...node,
-        [(0, import_suggestion3.getTransientSuggestionKey)()]: true
-      };
-    } else {
-      return {
-        ...node,
-        children: withTransient(node.children),
-        [(0, import_suggestion3.getTransientSuggestionKey)()]: true
-      };
-    }
-  });
-};
-var withoutSuggestionAndComments = (nodes) => {
-  return nodes.map((node) => {
-    if (import_platejs21.TextApi.isText(node)) {
-      if (node[import_platejs21.KEYS.suggestion] || node[import_platejs21.KEYS.comment]) {
-        return {
-          text: node.text
-        };
-      }
-      return node;
-    }
-    if (import_platejs21.ElementApi.isElement(node)) {
-      if (node[import_platejs21.KEYS.suggestion]) {
-        const nodeWithoutSuggestion = {};
-        Object.keys(node).forEach((key) => {
-          if (key !== import_platejs21.KEYS.suggestion && !key.startsWith(import_platejs21.KEYS.suggestion)) {
-            nodeWithoutSuggestion[key] = node[key];
-          }
-        });
-        return {
-          ...nodeWithoutSuggestion,
-          children: withoutSuggestionAndComments(node.children)
-        };
-      }
-      return {
-        ...node,
-        children: withoutSuggestionAndComments(node.children)
-      };
-    }
-    return node;
-  });
-};
-var getDiffNodes = (editor, aiContent) => {
-  const chatNodes = withoutSuggestionAndComments(
-    editor.getOption(AIChatPlugin, "chatNodes")
-  );
-  const aiNodes = withProps((0, import_markdown6.deserializeMd)(editor, aiContent), chatNodes);
-  const diffNodes = withTransient(
-    (0, import_suggestion3.diffToSuggestions)(editor, chatNodes, aiNodes, {
-      ignoreProps: ["id", "listStart"]
-    })
-  );
-  return diffNodes;
-};
-
 // src/react/copilot/CopilotPlugin.tsx
 var import_markdown10 = require("@platejs/markdown");
 var import_debounce = __toESM(require("lodash/debounce.js"));
 var import_platejs27 = require("platejs");
-var import_react24 = require("platejs/react");
+var import_react25 = require("platejs/react");
 
 // src/react/copilot/renderCopilotBelowNodes.tsx
-var import_react20 = __toESM(require("react"));
+var import_react21 = __toESM(require("react"));
 var import_platejs22 = require("platejs");
-var import_react21 = require("platejs/react");
+var import_react22 = require("platejs/react");
 var renderCopilotBelowNodes = ({
   editor
 }) => {
-  const copilot = (0, import_react21.getEditorPlugin)(editor, {
+  const copilot = (0, import_react22.getEditorPlugin)(editor, {
     key: import_platejs22.KEYS.copilot
   });
   const { renderGhostText: GhostText } = copilot.getOptions();
   if (!GhostText) return;
   return ({ children }) => {
-    return /* @__PURE__ */ import_react20.default.createElement(import_react20.default.Fragment, null, children, /* @__PURE__ */ import_react20.default.createElement(GhostText, null));
+    return /* @__PURE__ */ import_react21.default.createElement(import_react21.default.Fragment, null, children, /* @__PURE__ */ import_react21.default.createElement(GhostText, null));
   };
 };
 
@@ -1558,7 +1587,7 @@ var acceptCopilot = (editor) => {
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
 var import_markdown8 = require("@platejs/markdown");
 var import_platejs25 = require("platejs");
-var import_react23 = require("platejs/react");
+var import_react24 = require("platejs/react");
 
 // src/react/copilot/utils/callCompletionApi.ts
 var getOriginalFetch = () => fetch;
@@ -1676,9 +1705,9 @@ var getNextWord = ({ text }) => {
 
 // src/react/copilot/utils/triggerCopilotSuggestion.ts
 var import_platejs24 = require("platejs");
-var import_react22 = require("platejs/react");
+var import_react23 = require("platejs/react");
 var triggerCopilotSuggestion = async (editor) => {
-  const { api, getOptions, setOption } = (0, import_react22.getEditorPlugin)(
+  const { api, getOptions, setOption } = (0, import_react23.getEditorPlugin)(
     editor,
     {
       key: import_platejs24.KEYS.copilot
@@ -1718,7 +1747,7 @@ var withoutAbort = (editor, fn) => {
 
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
 var acceptCopilotNextWord = (editor) => {
-  const { api, getOptions } = (0, import_react23.getEditorPlugin)(editor, {
+  const { api, getOptions } = (0, import_react24.getEditorPlugin)(editor, {
     key: import_platejs25.KEYS.copilot
   });
   const { getNextWord: getNextWord2, suggestionText } = getOptions();
@@ -1832,7 +1861,7 @@ var withCopilot = ({
 };
 
 // src/react/copilot/CopilotPlugin.tsx
-var CopilotPlugin = (0, import_react24.createTPlatePlugin)({
+var CopilotPlugin = (0, import_react25.createTPlatePlugin)({
   key: import_platejs27.KEYS.copilot,
   handlers: {
     onBlur: ({ api }) => {
@@ -1944,9 +1973,9 @@ var CopilotPlugin = (0, import_react24.createTPlatePlugin)({
   AIPlugin,
   CopilotPlugin,
   acceptAIChat,
+  acceptAISuggestions,
   acceptCopilot,
   acceptCopilotNextWord,
-  acceptSuggestions,
   aiCommentToRange,
   applyAISuggestions,
   callCompletionApi,
@@ -1964,6 +1993,7 @@ var CopilotPlugin = (0, import_react24.createTPlatePlugin)({
   isCompleteMath,
   isSameNode,
   nodesWithProps,
+  rejectAISuggestions,
   removeAnchorAIChat,
   renderCopilotBelowNodes,
   replaceSelectionAIChat,