@platejs/ai 51.0.0 → 51.0.1

package/dist/react/index.mjs

@@ -271,24 +271,46 @@ var aiCommentToRange = (editor, aiComment) => {
 };
 
 // src/react/ai-chat/AIChatPlugin.ts
-import { BlockSelectionPlugin as BlockSelectionPlugin4 } from "@platejs/selection/react";
+import { BlockSelectionPlugin as BlockSelectionPlugin5 } from "@platejs/selection/react";
 import {
   bindFirst as bindFirst2,
-  ElementApi as ElementApi3,
+  ElementApi as ElementApi4,
   getPluginType as getPluginType3,
-  KEYS as KEYS11
+  KEYS as KEYS12
 } from "platejs";
 import { createTPlatePlugin } from "platejs/react";
 
 // src/react/ai-chat/transforms/acceptAIChat.ts
-import {
-  acceptSuggestion,
-  getSuggestionKey,
-  getTransientSuggestionKey as getTransientSuggestionKey2
-} from "@platejs/suggestion";
-import { SuggestionPlugin } from "@platejs/suggestion/react";
 import { KEYS as KEYS4 } from "platejs";
 import { getEditorPlugin } from "platejs/react";
+
+// src/react/ai-chat/utils/acceptAISuggestions.ts
+import { acceptSuggestion, getSuggestionKey, getTransientSuggestionKey as getTransientSuggestionKey2 } from "@platejs/suggestion";
+import { SuggestionPlugin } from "@platejs/suggestion/react";
+var acceptAISuggestions = (editor) => {
+  const suggestions = editor.getApi(SuggestionPlugin).suggestion.nodes({
+    transient: true
+  });
+  suggestions.forEach(([suggestionNode]) => {
+    const suggestionData = editor.getApi(SuggestionPlugin).suggestion.suggestionData(suggestionNode);
+    if (!suggestionData) return;
+    const description = {
+      createdAt: new Date(suggestionData.createdAt),
+      keyId: getSuggestionKey(suggestionData.id),
+      suggestionId: suggestionData.id,
+      type: suggestionData.type,
+      userId: suggestionData.userId
+    };
+    acceptSuggestion(editor, description);
+  });
+  editor.tf.unsetNodes([getTransientSuggestionKey2()], {
+    at: [],
+    mode: "all",
+    match: (n) => !!n[getTransientSuggestionKey2()]
+  });
+};
+
+// src/react/ai-chat/transforms/acceptAIChat.ts
 var acceptAIChat = (editor) => {
   const mode = editor.getOption(AIChatPlugin, "mode");
   if (mode === "insert") {
@@ -309,49 +331,302 @@ var acceptAIChat = (editor) => {
   }
   if (mode === "chat") {
     withAIBatch(editor, () => {
-      acceptSuggestions(editor);
+      acceptAISuggestions(editor);
     });
     editor.getApi(AIChatPlugin).aiChat.hide();
   }
 };
-var acceptSuggestions = (editor) => {
-  const suggestions = editor.getApi(SuggestionPlugin).suggestion.nodes({
+
+// src/react/ai-chat/transforms/insertBelowAIChat.ts
+import { BlockSelectionPlugin as BlockSelectionPlugin4 } from "@platejs/selection/react";
+import cloneDeep2 from "lodash/cloneDeep.js";
+import { KEYS as KEYS9, PathApi, RangeApi } from "platejs";
+
+// src/react/ai-chat/utils/applyAISuggestions.ts
+import { deserializeMd as deserializeMd2 } from "@platejs/markdown";
+import { BlockSelectionPlugin } from "@platejs/selection/react";
+import {
+  diffToSuggestions,
+  getTransientSuggestionKey as getTransientSuggestionKey3,
+  SkipSuggestionDeletes
+} from "@platejs/suggestion";
+import {
+  ElementApi,
+  KEYS as KEYS5,
+  nanoid,
+  TextApi as TextApi2
+} from "platejs";
+var applyAISuggestions = (editor, content) => {
+  editor.getApi({ key: KEYS5.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
+  const isBlockSelecting = editor.getOption(
+    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) => ElementApi.isElement(n) && editor.getOption(AIChatPlugin, "_replaceIds").includes(n.id)
+      })
+    );
+    replaceNodes.forEach(([node, path], index) => {
+      const replaceNode = node;
+      const diffNode = diffNodes[index];
+      const isSameString = SkipSuggestionDeletes(editor, replaceNode) === 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(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) => TextApi2.isText(n) && !!n[getTransientSuggestionKey3()]
+      })
+    );
+    const range = editor.api.nodesRange(nodes);
+    editor.tf.setSelection(range);
+    return;
+  }
+};
+var withProps = (diffNodes, chatNodes) => {
+  return diffNodes.map((node, index) => {
+    if (!ElementApi.isElement(node)) return node;
+    const originalNode = chatNodes?.[index];
+    return {
+      ...node,
+      ...originalNode ?? { id: nanoid() },
+      children: node.children
+    };
+  });
+};
+var withTransient = (diffNodes) => {
+  return diffNodes.map((node) => {
+    if (TextApi2.isText(node)) {
+      return {
+        ...node,
+        [getTransientSuggestionKey3()]: true
+      };
+    } else {
+      return {
+        ...node,
+        children: withTransient(node.children),
+        [getTransientSuggestionKey3()]: true
+      };
+    }
+  });
+};
+var withoutSuggestionAndComments = (nodes) => {
+  return nodes.map((node) => {
+    if (TextApi2.isText(node)) {
+      if (node[KEYS5.suggestion] || node[KEYS5.comment]) {
+        return {
+          text: node.text
+        };
+      }
+      return node;
+    }
+    if (ElementApi.isElement(node)) {
+      if (node[KEYS5.suggestion]) {
+        const nodeWithoutSuggestion = {};
+        Object.keys(node).forEach((key) => {
+          if (key !== KEYS5.suggestion && !key.startsWith(KEYS5.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(deserializeMd2(editor, aiContent), chatNodes);
+  const diffNodes = withTransient(
+    diffToSuggestions(editor, chatNodes, aiNodes, {
+      ignoreProps: ["id", "listStart"]
+    })
+  );
+  return diffNodes;
+};
+
+// src/react/ai-chat/utils/getLastAssistantMessage.ts
+import { usePluginOption } from "platejs/react";
+function getLastAssistantMessage(editor) {
+  const messages = editor.getOptions(AIChatPlugin).chat.messages;
+  return messages?.findLast((message) => message.role === "assistant");
+}
+function useLastAssistantMessage() {
+  const toolName = usePluginOption(AIChatPlugin, "toolName");
+  const chat = usePluginOption(AIChatPlugin, "chat");
+  if (toolName === "comment") return;
+  return chat.messages?.findLast((message) => message.role === "assistant");
+}
+
+// src/react/ai-chat/utils/rejectAISuggestions.ts
+import { getSuggestionKey as getSuggestionKey2, getTransientSuggestionKey as getTransientSuggestionKey4, rejectSuggestion } from "@platejs/suggestion";
+import { SuggestionPlugin as SuggestionPlugin2 } from "@platejs/suggestion/react";
+var rejectAISuggestions = (editor) => {
+  const suggestions = editor.getApi(SuggestionPlugin2).suggestion.nodes({
     transient: true
   });
   suggestions.forEach(([suggestionNode]) => {
-    const suggestionData = editor.getApi(SuggestionPlugin).suggestion.suggestionData(suggestionNode);
+    const suggestionData = editor.getApi(SuggestionPlugin2).suggestion.suggestionData(suggestionNode);
     if (!suggestionData) return;
     const description = {
       createdAt: new Date(suggestionData.createdAt),
-      keyId: getSuggestionKey(suggestionData.id),
+      keyId: getSuggestionKey2(suggestionData.id),
       suggestionId: suggestionData.id,
       type: suggestionData.type,
       userId: suggestionData.userId
     };
-    acceptSuggestion(editor, description);
+    rejectSuggestion(editor, description);
   });
-  editor.tf.unsetNodes([getTransientSuggestionKey2()], {
+  editor.tf.unsetNodes([getTransientSuggestionKey4()], {
     at: [],
     mode: "all",
-    match: (n) => !!n[getTransientSuggestionKey2()]
+    match: (n) => !!n[getTransientSuggestionKey4()]
+  });
+};
+
+// src/react/ai-chat/utils/resetAIChat.ts
+import { KEYS as KEYS6 } from "platejs";
+import { getEditorPlugin as getEditorPlugin2 } from "platejs/react";
+var resetAIChat = (editor, { undo = true } = {}) => {
+  const { api, getOptions, setOptions } = getEditorPlugin2(
+    editor,
+    {
+      key: KEYS6.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
+import { isSelecting as isSelecting2 } from "@platejs/selection";
+import { BlockSelectionPlugin as BlockSelectionPlugin2 } from "@platejs/selection/react";
+import { KEYS as KEYS7 } from "platejs";
+import { getEditorPlugin as getEditorPlugin3 } from "platejs/react";
+var submitAIChat = (editor, input, {
+  mode,
+  options,
+  prompt,
+  toolName: toolNameProps
+} = {}) => {
+  const { getOptions, setOption } = getEditorPlugin3(
+    editor,
+    {
+      key: KEYS7.aiChat
+    }
+  );
+  const { chat, toolName: toolNameOption } = getOptions();
+  const toolName = toolNameProps ?? toolNameOption ?? null;
+  if (!prompt && input?.length === 0) {
+    return;
+  }
+  if (!prompt) {
+    prompt = input;
+  }
+  if (!mode) {
+    mode = isSelecting2(editor) ? "chat" : "insert";
+  }
+  if (mode === "insert") {
+    editor.getTransforms(AIPlugin).ai.undo();
+  }
+  setOption("mode", mode);
+  setOption("toolName", toolName);
+  const blocks = editor.getApi(BlockSelectionPlugin2).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/insertBelowAIChat.ts
-import { BlockSelectionPlugin as BlockSelectionPlugin2 } from "@platejs/selection/react";
-import cloneDeep2 from "lodash/cloneDeep.js";
-import { KEYS as KEYS6, PathApi, RangeApi } from "platejs";
-
 // src/react/ai-chat/transforms/replaceSelectionAIChat.ts
 import {
-  BlockSelectionPlugin,
+  BlockSelectionPlugin as BlockSelectionPlugin3,
   removeBlockSelectionNodes
 } from "@platejs/selection/react";
 import cloneDeep from "lodash/cloneDeep.js";
 import {
-  KEYS as KEYS5,
+  KEYS as KEYS8,
   NodeApi as NodeApi3,
-  TextApi as TextApi2
+  TextApi as TextApi3
 } from "platejs";
 var createFormattedBlocks = ({
   blocks,
@@ -365,7 +640,7 @@ var createFormattedBlocks = ({
   const blockProps = NodeApi3.extractProps(sourceNode);
   const textProps = NodeApi3.extractProps(firstTextEntry[0]);
   const applyTextFormatting = (node) => {
-    if (TextApi2.isText(node)) {
+    if (TextApi3.isText(node)) {
       return { ...textProps, ...node };
     }
     if (node.children) {
@@ -389,10 +664,10 @@ var createFormattedBlocks = ({
 var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
   if (!sourceEditor || sourceEditor.api.isEmpty()) return;
   const isBlockSelecting = editor.getOption(
-    BlockSelectionPlugin,
+    BlockSelectionPlugin3,
     "isSelectingSome"
   );
-  editor.getApi({ key: KEYS5.ai }).aiChat.hide();
+  editor.getApi({ key: KEYS8.ai }).aiChat.hide();
   if (!isBlockSelecting) {
     const firstBlock = editor.api.node({
       block: true,
@@ -405,7 +680,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
         sourceBlock: firstBlock
       });
       if (!formattedBlocks2) return;
-      if (firstBlock[0].type === KEYS5.codeLine && sourceEditor.children[0].type === KEYS5.codeBlock && sourceEditor.children.length === 1) {
+      if (firstBlock[0].type === KEYS8.codeLine && sourceEditor.children[0].type === KEYS8.codeBlock && sourceEditor.children.length === 1) {
         editor.tf.insertFragment(formattedBlocks2[0].children);
       } else {
         editor.tf.insertFragment(formattedBlocks2);
@@ -417,14 +692,14 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
     editor.tf.focus();
     return;
   }
-  const blockSelectionApi = editor.getApi(BlockSelectionPlugin).blockSelection;
+  const blockSelectionApi = editor.getApi(BlockSelectionPlugin3).blockSelection;
   const selectedBlocks = blockSelectionApi.getNodes();
   if (selectedBlocks.length === 0) return;
   if (format === "none" || format === "single" && selectedBlocks.length > 1) {
     editor.tf.withoutNormalizing(() => {
       removeBlockSelectionNodes(editor);
       editor.tf.withNewBatch(() => {
-        editor.getTransforms(BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(
+        editor.getTransforms(BlockSelectionPlugin3).blockSelection.insertBlocksAndSelect(
           cloneDeep(sourceEditor.children),
           {
             at: selectedBlocks[0][1]
@@ -432,7 +707,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
         );
       });
     });
-    editor.getApi(BlockSelectionPlugin).blockSelection.focus();
+    editor.getApi(BlockSelectionPlugin3).blockSelection.focus();
     return;
   }
   const [, firstBlockPath] = selectedBlocks[0];
@@ -445,12 +720,12 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
   editor.tf.withoutNormalizing(() => {
     removeBlockSelectionNodes(editor);
     editor.tf.withNewBatch(() => {
-      editor.getTransforms(BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(formattedBlocks, {
+      editor.getTransforms(BlockSelectionPlugin3).blockSelection.insertBlocksAndSelect(formattedBlocks, {
         at: firstBlockPath
       });
     });
   });
-  editor.getApi(BlockSelectionPlugin).blockSelection.focus();
+  editor.getApi(BlockSelectionPlugin3).blockSelection.focus();
 };
 
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
@@ -458,10 +733,10 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
   const { toolName } = editor.getOptions(AIChatPlugin);
   if (toolName === "generate")
     return insertBelowGenerate(editor, sourceEditor, { format });
-  const selectedBlocks = editor.getApi(BlockSelectionPlugin2).blockSelection.getNodes();
-  const selectedIds = editor.getOptions(BlockSelectionPlugin2).selectedIds;
+  const selectedBlocks = editor.getApi(BlockSelectionPlugin4).blockSelection.getNodes();
+  const selectedIds = editor.getOptions(BlockSelectionPlugin4).selectedIds;
   editor.getTransforms(AIPlugin).ai.undo();
-  const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin2).blockSelection.insertBlocksAndSelect;
+  const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin4).blockSelection.insertBlocksAndSelect;
   if (!selectedIds || selectedIds.size === 0) return;
   const lastBlock = selectedBlocks.at(-1);
   if (!lastBlock) return;
@@ -471,7 +746,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     at: nextPath,
     insertedCallback: () => {
       withAIBatch(editor, () => {
-        acceptSuggestions(editor);
+        acceptAISuggestions(editor);
       });
     }
   });
@@ -480,14 +755,14 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
 var insertBelowGenerate = (editor, sourceEditor, { format = "single" } = {}) => {
   if (!sourceEditor || sourceEditor.api.isEmpty()) return;
   const isBlockSelecting = editor.getOption(
-    BlockSelectionPlugin2,
+    BlockSelectionPlugin4,
     "isSelectingSome"
   );
-  editor.getApi({ key: KEYS6.ai }).aiChat.hide();
-  const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin2).blockSelection.insertBlocksAndSelect;
+  editor.getApi({ key: KEYS9.ai }).aiChat.hide();
+  const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin4).blockSelection.insertBlocksAndSelect;
   if (isBlockSelecting) {
-    const selectedBlocks = editor.getApi(BlockSelectionPlugin2).blockSelection.getNodes();
-    const selectedIds = editor.getOptions(BlockSelectionPlugin2).selectedIds;
+    const selectedBlocks = editor.getApi(BlockSelectionPlugin4).blockSelection.getNodes();
+    const selectedIds = editor.getOptions(BlockSelectionPlugin4).selectedIds;
     if (!selectedIds || selectedIds.size === 0) return;
     const lastBlock = selectedBlocks.at(-1);
     if (!lastBlock) return;
@@ -536,118 +811,22 @@ var insertBelowGenerate = (editor, sourceEditor, { format = "single" } = {}) =>
 
 // src/react/ai-chat/transforms/removeAnchorAIChat.ts
 import {
-  ElementApi,
+  ElementApi as ElementApi2,
   getPluginType as getPluginType2,
-  KEYS as KEYS7
+  KEYS as KEYS10
 } from "platejs";
 var removeAnchorAIChat = (editor, options) => {
   editor.tf.withoutSaving(() => {
     editor.tf.removeNodes({
       at: [],
-      match: (n) => ElementApi.isElement(n) && n.type === getPluginType2(editor, KEYS7.aiChat),
+      match: (n) => ElementApi2.isElement(n) && n.type === getPluginType2(editor, KEYS10.aiChat),
       ...options
     });
   });
 };
 
-// src/react/ai-chat/utils/resetAIChat.ts
-import { KEYS as KEYS8 } from "platejs";
-import { getEditorPlugin as getEditorPlugin2 } from "platejs/react";
-var resetAIChat = (editor, { undo = true } = {}) => {
-  const { api, getOptions, setOptions } = getEditorPlugin2(
-    editor,
-    {
-      key: KEYS8.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
-import { isSelecting as isSelecting2 } from "@platejs/selection";
-import { BlockSelectionPlugin as BlockSelectionPlugin3 } from "@platejs/selection/react";
-import { KEYS as KEYS9 } from "platejs";
-import { getEditorPlugin as getEditorPlugin3 } from "platejs/react";
-var submitAIChat = (editor, input, {
-  mode,
-  options,
-  prompt,
-  toolName: toolNameProps
-} = {}) => {
-  const { getOptions, setOption } = getEditorPlugin3(
-    editor,
-    {
-      key: KEYS9.aiChat
-    }
-  );
-  const { chat, toolName: toolNameOption } = getOptions();
-  const toolName = toolNameProps ?? toolNameOption ?? null;
-  if (!prompt && input?.length === 0) {
-    return;
-  }
-  if (!prompt) {
-    prompt = input;
-  }
-  if (!mode) {
-    mode = isSelecting2(editor) ? "chat" : "insert";
-  }
-  if (mode === "insert") {
-    editor.getTransforms(AIPlugin).ai.undo();
-  }
-  setOption("mode", mode);
-  setOption("toolName", toolName);
-  const blocks = editor.getApi(BlockSelectionPlugin3).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/withAIChat.ts
-import { ElementApi as ElementApi2, KEYS as KEYS10 } from "platejs";
+import { ElementApi as ElementApi3, KEYS as KEYS11 } from "platejs";
 var withAIChat = ({
   api,
   editor,
@@ -689,11 +868,11 @@ var withAIChat = ({
       },
       normalizeNode(entry) {
         const [node, path] = entry;
-        if (node[KEYS10.ai] && !getOptions().open) {
+        if (node[KEYS11.ai] && !getOptions().open) {
           tf.ai.removeMarks({ at: path });
           return;
         }
-        if (ElementApi2.isElement(node) && node.type === type && !getOptions().open) {
+        if (ElementApi3.isElement(node) && node.type === type && !getOptions().open) {
           editor.getTransforms(AIChatPlugin).aiChat.removeAnchor({ at: path });
           return;
         }
@@ -705,7 +884,7 @@ var withAIChat = ({
 
 // src/react/ai-chat/AIChatPlugin.ts
 var AIChatPlugin = createTPlatePlugin({
-  key: KEYS11.aiChat,
+  key: KEYS12.aiChat,
   dependencies: ["ai"],
   node: {
     isElement: true
@@ -736,7 +915,7 @@ var AIChatPlugin = createTPlatePlugin({
       if (anchor) {
         return editor.api.node({
           at: [],
-          match: (n) => ElementApi3.isElement(n) && n.type === type,
+          match: (n) => ElementApi4.isElement(n) && n.type === type,
           ...rest
         });
       }
@@ -748,12 +927,12 @@ var AIChatPlugin = createTPlatePlugin({
           at: path,
           mode: "lowest",
           reverse: true,
-          match: (t) => !!t[getPluginType3(editor, KEYS11.ai)],
+          match: (t) => !!t[getPluginType3(editor, KEYS12.ai)],
           ...rest
         });
       }
       return editor.api.node({
-        match: (n) => n[getPluginType3(editor, KEYS11.ai)],
+        match: (n) => n[getPluginType3(editor, KEYS12.ai)],
         ...rest
       });
     },
@@ -763,9 +942,9 @@ var AIChatPlugin = createTPlatePlugin({
       if (chatSelection) {
         editor.tf.setSelection(chatSelection);
       } else {
-        editor.getApi(BlockSelectionPlugin4).blockSelection.set(chatNodes.map((node) => node.id));
+        editor.getApi(BlockSelectionPlugin5).blockSelection.set(chatNodes.map((node) => node.id));
       }
-      const blocks = editor.getApi(BlockSelectionPlugin4).blockSelection.getNodes();
+      const blocks = editor.getApi(BlockSelectionPlugin5).blockSelection.getNodes();
       const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
       const ctx = {
         children: editor.children,
@@ -791,8 +970,8 @@ var AIChatPlugin = createTPlatePlugin({
     api.aiChat.reset({ undo });
     setOption("open", false);
     if (focus) {
-      if (editor.getOption(BlockSelectionPlugin4, "isSelectingSome")) {
-        editor.getApi(BlockSelectionPlugin4).blockSelection.focus();
+      if (editor.getOption(BlockSelectionPlugin5, "isSelectingSome")) {
+        editor.getApi(BlockSelectionPlugin5).blockSelection.focus();
       } else {
         editor.tf.focus();
       }
@@ -839,29 +1018,14 @@ var useAIChatEditor = (editor, content, { parser } = {}) => {
 
 // src/react/ai-chat/hooks/useChatChunk.ts
 import { useEffect as useEffect2, useRef } from "react";
-import { KEYS as KEYS12 } from "platejs";
+import { KEYS as KEYS13 } from "platejs";
 import { usePluginOption as usePluginOption2 } from "platejs/react";
-
-// src/react/ai-chat/utils/getLastAssistantMessage.ts
-import { usePluginOption } from "platejs/react";
-function getLastAssistantMessage(editor) {
-  const messages = editor.getOptions(AIChatPlugin).chat.messages;
-  return messages?.findLast((message) => message.role === "assistant");
-}
-function useLastAssistantMessage() {
-  const toolName = usePluginOption(AIChatPlugin, "toolName");
-  const chat = usePluginOption(AIChatPlugin, "chat");
-  if (toolName === "comment") return;
-  return chat.messages?.findLast((message) => message.role === "assistant");
-}
-
-// src/react/ai-chat/hooks/useChatChunk.ts
 var useChatChunk = ({
   onChunk,
   onFinish
 }) => {
   const { status } = usePluginOption2(
-    { key: KEYS12.aiChat },
+    { key: KEYS13.aiChat },
     "chat"
   );
   const isLoading = status === "streaming" || status === "submitted";
@@ -901,7 +1065,7 @@ var useChatChunk = ({
 
 // src/react/ai-chat/hooks/useEditorChat.ts
 import { useEffect as useEffect3 } from "react";
-import { BlockSelectionPlugin as BlockSelectionPlugin5 } from "@platejs/selection/react";
+import { BlockSelectionPlugin as BlockSelectionPlugin6 } from "@platejs/selection/react";
 import { useEditorPlugin as useEditorPlugin2, usePluginOption as usePluginOption3 } from "platejs/react";
 var useEditorChat = ({
   chat,
@@ -916,9 +1080,9 @@ var useEditorChat = ({
     onOpenChange?.(open);
     if (open) {
       if (onOpenBlockSelection) {
-        const blockSelectionApi = editor.getApi(BlockSelectionPlugin5).blockSelection;
+        const blockSelectionApi = editor.getApi(BlockSelectionPlugin6).blockSelection;
         const isBlockSelecting = editor.getOption(
-          BlockSelectionPlugin5,
+          BlockSelectionPlugin6,
           "isSelectingSome"
         );
         if (isBlockSelecting) {
@@ -946,7 +1110,7 @@ var streamDeserializeInlineMd = (editor, text, options) => {
 
 // src/react/ai-chat/streaming/streamDeserializeMd.ts
 import { MarkdownPlugin as MarkdownPlugin3 } from "@platejs/markdown";
-import { getPluginType as getPluginType4, KEYS as KEYS14, TextApi as TextApi3 } from "platejs";
+import { getPluginType as getPluginType4, KEYS as KEYS15, TextApi as TextApi4 } from "platejs";
 
 // src/react/ai-chat/streaming/utils/utils.ts
 var getChunkTrimmed = (chunk, {
@@ -1001,10 +1165,10 @@ var getListNode = (editor, node) => {
 };
 
 // src/react/ai-chat/streaming/utils/isSameNode.ts
-import { isDefined, KEYS as KEYS13 } from "platejs";
+import { isDefined, KEYS as KEYS14 } from "platejs";
 var LIST_STYLE_TYPE = "listStyleType";
 var isSameNode = (editor, node1, node2) => {
-  if (node1.type !== editor.getType(KEYS13.p) || node2.type !== editor.getType(KEYS13.p))
+  if (node1.type !== editor.getType(KEYS14.p) || node2.type !== editor.getType(KEYS14.p))
     return node1.type === node2.type;
   if (isDefined(node1[LIST_STYLE_TYPE]) || isDefined(node2[LIST_STYLE_TYPE])) {
     return node1[LIST_STYLE_TYPE] === node2[LIST_STYLE_TYPE];
@@ -1013,10 +1177,10 @@ var isSameNode = (editor, node1, node2) => {
 };
 
 // src/react/ai-chat/streaming/utils/nodesWithProps.ts
-import { ElementApi as ElementApi4 } from "platejs";
+import { ElementApi as ElementApi5 } from "platejs";
 var nodesWithProps = (editor, nodes, options) => {
   return nodes.map((node) => {
-    if (ElementApi4.isElement(node)) {
+    if (ElementApi5.isElement(node)) {
       return {
         ...getListNode(editor, node),
         ...options.elementProps,
@@ -1056,7 +1220,7 @@ var streamDeserializeMd = (editor, data, options) => {
       }
     ];
     const lastChild = lastBlock.children.at(-1);
-    if (lastChild && TextApi3.isText(lastChild) && Object.keys(lastChild).length === 1) {
+    if (lastChild && TextApi4.isText(lastChild) && Object.keys(lastChild).length === 1) {
       lastBlock.children.pop();
       const textNode2 = [
         {
@@ -1072,13 +1236,13 @@ var streamDeserializeMd = (editor, data, options) => {
   if (addNewLine && !isCodeBlockOrTable) {
     result.push({
       children: [{ text: "" }],
-      type: KEYS14.p
+      type: KEYS15.p
     });
   }
   if (unshiftNewLine && !isCodeBlockOrTable) {
     result.unshift({
       children: [{ text: "" }],
-      type: KEYS14.p
+      type: KEYS15.p
     });
   }
   return result;
@@ -1098,7 +1262,7 @@ var withoutDeserializeInMdx = (editor, input) => {
               text: input
             }
           ],
-          type: getPluginType4(editor, KEYS14.p)
+          type: getPluginType4(editor, KEYS15.p)
         }
       ];
     }
@@ -1113,7 +1277,7 @@ var withoutDeserializeInMdx = (editor, input) => {
 // src/react/ai-chat/streaming/streamInsertChunk.ts
 import {
   getPluginType as getPluginType5,
-  KEYS as KEYS16,
+  KEYS as KEYS17,
   NodeApi as NodeApi4,
   PathApi as PathApi2
 } from "platejs";
@@ -1121,26 +1285,26 @@ import {
 // src/react/ai-chat/streaming/streamSerializeMd.ts
 import { MarkdownPlugin as MarkdownPlugin4 } from "@platejs/markdown";
 import {
-  ElementApi as ElementApi5,
+  ElementApi as ElementApi6,
   getPluginKey,
-  KEYS as KEYS15,
-  TextApi as TextApi4
+  KEYS as KEYS16,
+  TextApi as TextApi5
 } from "platejs";
 var trimEndHeading = (editor, value) => {
   const headingKeys = /* @__PURE__ */ new Set([
-    KEYS15.h1,
-    KEYS15.h2,
-    KEYS15.h3,
-    KEYS15.h4,
-    KEYS15.h5,
-    KEYS15.h6
+    KEYS16.h1,
+    KEYS16.h2,
+    KEYS16.h3,
+    KEYS16.h4,
+    KEYS16.h5,
+    KEYS16.h6
   ]);
   const lastBlock = value.at(-1);
   if (lastBlock && headingKeys.has(
     getPluginKey(editor, lastBlock.type) ?? lastBlock.type
-  ) && ElementApi5.isElement(lastBlock)) {
+  ) && ElementApi6.isElement(lastBlock)) {
     const lastTextNode = lastBlock.children.at(-1);
-    if (TextApi4.isText(lastTextNode)) {
+    if (TextApi5.isText(lastTextNode)) {
       const trimmedText = getChunkTrimmed(lastTextNode?.text);
       const newChildren = [
         ...lastBlock.children.slice(0, -1),
@@ -1204,7 +1368,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
     const blocks = streamDeserializeMd(editor, chunk);
     const path = getCurrentBlockPath(editor);
     const startBlock = editor.api.node(path)[0];
-    const startInEmptyParagraph = NodeApi4.string(startBlock).length === 0 && startBlock.type === getPluginType5(editor, KEYS16.p);
+    const startInEmptyParagraph = NodeApi4.string(startBlock).length === 0 && startBlock.type === getPluginType5(editor, KEYS17.p);
     if (startInEmptyParagraph) {
       editor.tf.removeNodes({ at: path });
     }
@@ -1284,7 +1448,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
             AIChatPlugin,
             "_blockChunks",
             // one block includes multiple children
-            tempBlocks[0].type === getPluginType5(editor, KEYS16.codeBlock) || tempBlocks[0].type === getPluginType5(editor, KEYS16.table) || tempBlocks[0].type === getPluginType5(editor, KEYS16.equation) ? tempBlockChunks : serializedBlock2
+            tempBlocks[0].type === getPluginType5(editor, KEYS17.codeBlock) || tempBlocks[0].type === getPluginType5(editor, KEYS17.table) || tempBlocks[0].type === getPluginType5(editor, KEYS17.equation) ? tempBlockChunks : serializedBlock2
           );
         }
       } else {
@@ -1344,152 +1508,12 @@ var getCurrentBlockPath = (editor) => {
   };
   const path = getAnchorPreviousPath(editor) ?? getFocusPath(editor) ?? [0];
   const entry = editor.api.node(path);
-  if (entry && (entry[0].type === getPluginType5(editor, KEYS16.columnGroup) || entry[0].type === getPluginType5(editor, KEYS16.table))) {
+  if (entry && (entry[0].type === getPluginType5(editor, KEYS17.columnGroup) || entry[0].type === getPluginType5(editor, KEYS17.table))) {
     return editor.api.above()?.[1] ?? path;
   }
   return path;
 };
 
-// src/react/ai-chat/utils/applyAISuggestions.ts
-import { deserializeMd as deserializeMd2 } from "@platejs/markdown";
-import { BlockSelectionPlugin as BlockSelectionPlugin6 } from "@platejs/selection/react";
-import {
-  diffToSuggestions,
-  getTransientSuggestionKey as getTransientSuggestionKey3,
-  SkipSuggestionDeletes
-} from "@platejs/suggestion";
-import {
-  ElementApi as ElementApi6,
-  KEYS as KEYS17,
-  nanoid,
-  TextApi as TextApi5
-} from "platejs";
-var applyAISuggestions = (editor, content) => {
-  editor.getApi({ key: KEYS17.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
-  const isBlockSelecting = editor.getOption(
-    BlockSelectionPlugin6,
-    "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) => ElementApi6.isElement(n) && editor.getOption(AIChatPlugin, "_replaceIds").includes(n.id)
-      })
-    );
-    replaceNodes.forEach(([node, path], index) => {
-      const replaceNode = node;
-      const diffNode = diffNodes[index];
-      const isSameString = SkipSuggestionDeletes(editor, replaceNode) === 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(BlockSelectionPlugin6).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) => TextApi5.isText(n) && !!n[getTransientSuggestionKey3()]
-      })
-    );
-    const range = editor.api.nodesRange(nodes);
-    editor.tf.setSelection(range);
-    return;
-  }
-};
-var withProps = (diffNodes, chatNodes) => {
-  return diffNodes.map((node, index) => {
-    if (!ElementApi6.isElement(node)) return node;
-    const originalNode = chatNodes?.[index];
-    return {
-      ...node,
-      ...originalNode ?? { id: nanoid() },
-      children: node.children
-    };
-  });
-};
-var withTransient = (diffNodes) => {
-  return diffNodes.map((node) => {
-    if (TextApi5.isText(node)) {
-      return {
-        ...node,
-        [getTransientSuggestionKey3()]: true
-      };
-    } else {
-      return {
-        ...node,
-        children: withTransient(node.children),
-        [getTransientSuggestionKey3()]: true
-      };
-    }
-  });
-};
-var withoutSuggestionAndComments = (nodes) => {
-  return nodes.map((node) => {
-    if (TextApi5.isText(node)) {
-      if (node[KEYS17.suggestion] || node[KEYS17.comment]) {
-        return {
-          text: node.text
-        };
-      }
-      return node;
-    }
-    if (ElementApi6.isElement(node)) {
-      if (node[KEYS17.suggestion]) {
-        const nodeWithoutSuggestion = {};
-        Object.keys(node).forEach((key) => {
-          if (key !== KEYS17.suggestion && !key.startsWith(KEYS17.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(deserializeMd2(editor, aiContent), chatNodes);
-  const diffNodes = withTransient(
-    diffToSuggestions(editor, chatNodes, aiNodes, {
-      ignoreProps: ["id", "listStart"]
-    })
-  );
-  return diffNodes;
-};
-
 // src/react/copilot/CopilotPlugin.tsx
 import { serializeMd } from "@platejs/markdown";
 import debounce from "lodash/debounce.js";
@@ -1918,9 +1942,9 @@ export {
   AIPlugin,
   CopilotPlugin,
   acceptAIChat,
+  acceptAISuggestions,
   acceptCopilot,
   acceptCopilotNextWord,
-  acceptSuggestions,
   aiCommentToRange,
   applyAISuggestions,
   callCompletionApi,
@@ -1938,6 +1962,7 @@ export {
   isCompleteMath,
   isSameNode,
   nodesWithProps,
+  rejectAISuggestions,
   removeAnchorAIChat,
   renderCopilotBelowNodes,
   replaceSelectionAIChat,