@platejs/ai 50.1.2 → 50.2.1

package/dist/react/index.js

@@ -36,7 +36,9 @@ __export(react_exports, {
   acceptAIChat: () => acceptAIChat,
   acceptCopilot: () => acceptCopilot,
   acceptCopilotNextWord: () => acceptCopilotNextWord,
+  acceptSuggestions: () => acceptSuggestions,
   aiCommentToRange: () => aiCommentToRange,
+  applyAISuggestions: () => applyAISuggestions,
   callCompletionApi: () => callCompletionApi,
   createFormattedBlocks: () => createFormattedBlocks,
   escapeInput: () => escapeInput,
@@ -47,6 +49,7 @@ __export(react_exports, {
   getListNode: () => getListNode,
   getNextWord: () => getNextWord,
   insertBelowAIChat: () => insertBelowAIChat,
+  insertBelowGenerate: () => insertBelowGenerate,
   isCompleteCodeBlock: () => isCompleteCodeBlock,
   isCompleteMath: () => isCompleteMath,
   isSameNode: () => isSameNode,
@@ -115,11 +118,16 @@ var removeAINodes = (editor, { at = [] } = {}) => {
 };
 
 // src/lib/transforms/undoAI.ts
+var import_suggestion = require("@platejs/suggestion");
 var undoAI = (editor) => {
-  if (editor.history.undos.at(-1)?.ai && editor.api.some({
+  const hasAINodeOrAISuggestion = editor.api.some({
     at: [],
     match: (n) => !!n.ai
-  })) {
+  }) || editor.api.some({
+    at: [],
+    match: (n) => !!n[(0, import_suggestion.getTransientSuggestionKey)()]
+  });
+  if (editor.history.undos.at(-1)?.ai && hasAINodeOrAISuggestion) {
     editor.undo();
     editor.history.redos.pop();
   }
@@ -2982,37 +2990,70 @@ var aiCommentToRange = (editor, aiComment) => {
 };
 
 // src/react/ai-chat/AIChatPlugin.ts
-var import_react8 = require("@platejs/selection/react");
+var import_react9 = require("@platejs/selection/react");
 var import_platejs13 = require("platejs");
-var import_react9 = require("platejs/react");
+var import_react10 = 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_platejs6 = require("platejs");
-var import_react2 = require("platejs/react");
+var import_react3 = require("platejs/react");
 var acceptAIChat = (editor) => {
-  const { tf } = (0, import_react2.getEditorPlugin)(editor, AIPlugin);
-  const api = editor.getApi({ key: import_platejs6.KEYS.ai });
-  const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
-  withAIBatch(editor, () => {
-    tf.ai.removeMarks();
-    editor.getTransforms(AIChatPlugin).aiChat.removeAnchor();
+  const mode = editor.getOption(AIChatPlugin, "mode");
+  if (mode === "insert") {
+    const { tf } = (0, import_react3.getEditorPlugin)(editor, AIPlugin);
+    const api = editor.getApi({ key: import_platejs6.KEYS.ai });
+    const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
+    withAIBatch(editor, () => {
+      tf.ai.removeMarks();
+      editor.getTransforms(AIChatPlugin).aiChat.removeAnchor();
+    });
+    api.aiChat.hide();
+    editor.tf.focus();
+    const focusPoint = editor.api.end(lastAINodePath);
+    editor.tf.setSelection({
+      anchor: focusPoint,
+      focus: focusPoint
+    });
+  }
+  if (mode === "chat") {
+    withAIBatch(editor, () => {
+      acceptSuggestions(editor);
+    });
+    editor.getApi(AIChatPlugin).aiChat.hide();
+  }
+};
+var acceptSuggestions = (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);
   });
-  api.aiChat.hide();
-  editor.tf.focus();
-  const focusPoint = editor.api.end(lastAINodePath);
-  editor.tf.setSelection({
-    anchor: focusPoint,
-    focus: focusPoint
+  editor.tf.unsetNodes([(0, import_suggestion2.getTransientSuggestionKey)()], {
+    at: [],
+    mode: "all",
+    match: (n) => !!n[(0, import_suggestion2.getTransientSuggestionKey)()]
   });
 };
 
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
-var import_react4 = require("@platejs/selection/react");
+var import_react5 = require("@platejs/selection/react");
 var import_cloneDeep2 = __toESM(require("lodash/cloneDeep.js"));
 var import_platejs8 = require("platejs");
 
 // src/react/ai-chat/transforms/replaceSelectionAIChat.ts
-var import_react3 = require("@platejs/selection/react");
+var import_react4 = require("@platejs/selection/react");
 var import_cloneDeep = __toESM(require("lodash/cloneDeep.js"));
 var import_platejs7 = require("platejs");
 var createFormattedBlocks = ({
@@ -3051,7 +3092,7 @@ var createFormattedBlocks = ({
 var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
   if (!sourceEditor || sourceEditor.api.isEmpty()) return;
   const isBlockSelecting = editor.getOption(
-    import_react3.BlockSelectionPlugin,
+    import_react4.BlockSelectionPlugin,
     "isSelectingSome"
   );
   editor.getApi({ key: import_platejs7.KEYS.ai }).aiChat.hide();
@@ -3079,14 +3120,14 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
     editor.tf.focus();
     return;
   }
-  const blockSelectionApi = editor.getApi(import_react3.BlockSelectionPlugin).blockSelection;
+  const blockSelectionApi = editor.getApi(import_react4.BlockSelectionPlugin).blockSelection;
   const selectedBlocks = blockSelectionApi.getNodes();
   if (selectedBlocks.length === 0) return;
   if (format === "none" || format === "single" && selectedBlocks.length > 1) {
     editor.tf.withoutNormalizing(() => {
-      (0, import_react3.removeBlockSelectionNodes)(editor);
+      (0, import_react4.removeBlockSelectionNodes)(editor);
       editor.tf.withNewBatch(() => {
-        editor.getTransforms(import_react3.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(
+        editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(
           (0, import_cloneDeep.default)(sourceEditor.children),
           {
             at: selectedBlocks[0][1]
@@ -3094,7 +3135,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
         );
       });
     });
-    editor.getApi(import_react3.BlockSelectionPlugin).blockSelection.focus();
+    editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.focus();
     return;
   }
   const [, firstBlockPath] = selectedBlocks[0];
@@ -3105,28 +3146,51 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
   });
   if (!formattedBlocks) return;
   editor.tf.withoutNormalizing(() => {
-    (0, import_react3.removeBlockSelectionNodes)(editor);
+    (0, import_react4.removeBlockSelectionNodes)(editor);
     editor.tf.withNewBatch(() => {
-      editor.getTransforms(import_react3.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(formattedBlocks, {
+      editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect(formattedBlocks, {
         at: firstBlockPath
       });
     });
   });
-  editor.getApi(import_react3.BlockSelectionPlugin).blockSelection.focus();
+  editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.focus();
 };
 
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
 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;
+  editor.getTransforms(AIPlugin).ai.undo();
+  const insertBlocksAndSelect = editor.getTransforms(import_react5.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
+  if (!selectedIds || selectedIds.size === 0) return;
+  const lastBlock = selectedBlocks.at(-1);
+  if (!lastBlock) return;
+  const nextPath = import_platejs8.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_react4.BlockSelectionPlugin,
+    import_react5.BlockSelectionPlugin,
     "isSelectingSome"
   );
   editor.getApi({ key: import_platejs8.KEYS.ai }).aiChat.hide();
-  const insertBlocksAndSelect = editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
+  const insertBlocksAndSelect = editor.getTransforms(import_react5.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
   if (isBlockSelecting) {
-    const selectedBlocks = editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.getNodes();
-    const selectedIds = editor.getOptions(import_react4.BlockSelectionPlugin).selectedIds;
+    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;
@@ -3187,31 +3251,40 @@ var removeAnchorAIChat = (editor, options) => {
 
 // src/react/ai-chat/utils/resetAIChat.ts
 var import_platejs10 = require("platejs");
-var import_react5 = require("platejs/react");
+var import_react6 = require("platejs/react");
 var resetAIChat = (editor) => {
-  const { api, getOptions } = (0, import_react5.getEditorPlugin)(editor, {
-    key: import_platejs10.KEYS.aiChat
-  });
+  const { api, getOptions, setOptions } = (0, import_react6.getEditorPlugin)(
+    editor,
+    {
+      key: import_platejs10.KEYS.aiChat
+    }
+  );
   api.aiChat.stop();
   const chat = getOptions().chat;
   if (chat.messages && chat.messages.length > 0) {
     chat.setMessages?.([]);
   }
+  setOptions({
+    _replaceIds: [],
+    chatNodes: [],
+    mode: "insert",
+    toolName: "generate"
+  });
   editor.getTransforms(AIPlugin).ai.undo();
 };
 
 // src/react/ai-chat/utils/submitAIChat.ts
 var import_selection2 = require("@platejs/selection");
-var import_react6 = require("@platejs/selection/react");
+var import_react7 = require("@platejs/selection/react");
 var import_platejs11 = require("platejs");
-var import_react7 = require("platejs/react");
+var import_react8 = require("platejs/react");
 var submitAIChat = (editor, input, {
   mode,
   options,
   prompt,
   toolName
 } = {}) => {
-  const { getOptions, setOption } = (0, import_react7.getEditorPlugin)(
+  const { getOptions, setOption } = (0, import_react8.getEditorPlugin)(
     editor,
     {
       key: import_platejs11.KEYS.aiChat
@@ -3232,11 +3305,15 @@ var submitAIChat = (editor, input, {
   }
   setOption("mode", mode);
   setOption("toolName", toolName ?? null);
-  const blocks = editor.getApi(import_react6.BlockSelectionPlugin).blockSelection.getNodes();
+  const blocks = editor.getApi(import_react7.BlockSelectionPlugin).blockSelection.getNodes();
+  const blocksRange = editor.api.nodesRange(blocks);
   const promptText = getEditorPrompt(editor, {
     prompt
   });
-  const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
+  const selection = blocks.length > 0 ? blocksRange : editor.selection;
+  const chatNodes = blocks.length > 0 ? blocks.map((block) => block[0]) : editor.api.fragment();
+  setOption("chatNodes", chatNodes);
+  setOption("chatSelection", blocks.length > 0 ? null : editor.selection);
   const ctx = {
     children: editor.children,
     selection: selection ?? null,
@@ -3261,7 +3338,7 @@ var withAIChat = ({
   api,
   editor,
   getOptions,
-  tf: { insertText, normalizeNode, setSelection },
+  tf: { apply, insertText, normalizeNode },
   type
 }) => {
   const tf = editor.getTransforms(AIPlugin);
@@ -3313,7 +3390,7 @@ var withAIChat = ({
 };
 
 // src/react/ai-chat/AIChatPlugin.ts
-var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
+var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
   key: import_platejs13.KEYS.aiChat,
   dependencies: ["ai"],
   node: {
@@ -3323,8 +3400,11 @@ var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
     _blockChunks: "",
     _blockPath: null,
     _mdxName: null,
+    _replaceIds: [],
     aiEditor: null,
     chat: { messages: [] },
+    chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3364,11 +3444,25 @@ var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
       });
     },
     reload: () => {
-      const { chat, mode } = getOptions();
-      if (mode === "insert") {
-        editor.getTransforms(AIPlugin).ai.undo();
+      const { chat, chatNodes, chatSelection } = getOptions();
+      editor.getTransforms(AIPlugin).ai.undo();
+      if (chatSelection) {
+        editor.tf.setSelection(chatSelection);
+      } else {
+        editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.set(chatNodes.map((node) => node.id));
       }
-      void chat.regenerate?.();
+      const blocks = editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.getNodes();
+      const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
+      const ctx = {
+        children: editor.children,
+        selection: selection ?? null,
+        toolName: getOption("toolName")
+      };
+      void chat.regenerate?.({
+        body: {
+          ctx
+        }
+      });
     },
     stop: () => {
       setOption("streaming", false);
@@ -3376,12 +3470,15 @@ var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
     }
   };
 }).extendApi(({ api, editor, getOptions, setOption, tf, type }) => ({
-  hide: () => {
+  hide: ({ focus = true } = {}) => {
     api.aiChat.reset();
     setOption("open", false);
-    if (editor.getOption(import_react8.BlockSelectionPlugin, "isSelectingSome")) {
-    } else {
-      editor.tf.focus();
+    if (focus) {
+      if (editor.getOption(import_react9.BlockSelectionPlugin, "isSelectingSome")) {
+        editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.focus();
+      } else {
+        editor.tf.focus();
+      }
     }
     const lastBatch = editor.history.undos.at(-1);
     if (lastBatch?.ai) {
@@ -3391,6 +3488,7 @@ var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
   },
   show: () => {
     api.aiChat.reset();
+    setOption("toolName", null);
     getOptions().chat.setMessages?.([]);
     setOption("open", true);
   }
@@ -3402,12 +3500,12 @@ var AIChatPlugin = (0, import_react9.createTPlatePlugin)({
 }));
 
 // src/react/ai-chat/hooks/useAIChatEditor.ts
-var import_react10 = require("react");
+var import_react11 = require("react");
 var import_markdown2 = require("@platejs/markdown");
-var import_react11 = require("platejs/react");
+var import_react12 = require("platejs/react");
 var useAIChatEditor = (editor, content, { parser } = {}) => {
-  const { setOption } = (0, import_react11.useEditorPlugin)(AIChatPlugin);
-  editor.children = (0, import_react10.useMemo)(
+  const { setOption } = (0, import_react12.useEditorPlugin)(AIChatPlugin);
+  editor.children = (0, import_react11.useMemo)(
     () => {
       return editor.getApi(import_markdown2.MarkdownPlugin).markdown.deserialize(content, {
         memoize: true,
@@ -3417,25 +3515,25 @@ var useAIChatEditor = (editor, content, { parser } = {}) => {
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [content]
   );
-  (0, import_react10.useEffect)(() => {
+  (0, import_react11.useEffect)(() => {
     setOption("aiEditor", editor);
   }, [editor, setOption]);
 };
 
 // src/react/ai-chat/hooks/useChatChunk.ts
-var import_react13 = require("react");
+var import_react14 = require("react");
 var import_platejs14 = require("platejs");
-var import_react14 = require("platejs/react");
+var import_react15 = require("platejs/react");
 
 // src/react/ai-chat/utils/getLastAssistantMessage.ts
-var import_react12 = require("platejs/react");
+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_react12.usePluginOption)(AIChatPlugin, "toolName");
-  const chat = (0, import_react12.usePluginOption)(AIChatPlugin, "chat");
+  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");
 }
@@ -3445,7 +3543,7 @@ var useChatChunk = ({
   onChunk,
   onFinish
 }) => {
-  const { status } = (0, import_react14.usePluginOption)(
+  const { status } = (0, import_react15.usePluginOption)(
     { key: import_platejs14.KEYS.aiChat },
     "chat"
   );
@@ -3453,9 +3551,9 @@ var useChatChunk = ({
   const content = useLastAssistantMessage()?.parts.find(
     (part) => part.type === "text"
   )?.text;
-  const insertedTextRef = (0, import_react13.useRef)("");
-  const prevIsLoadingRef = (0, import_react13.useRef)(isLoading);
-  (0, import_react13.useEffect)(() => {
+  const insertedTextRef = (0, import_react14.useRef)("");
+  const prevIsLoadingRef = (0, import_react14.useRef)(isLoading);
+  (0, import_react14.useEffect)(() => {
     if (!isLoading) {
       insertedTextRef.current = "";
     }
@@ -3464,7 +3562,7 @@ var useChatChunk = ({
     }
     prevIsLoadingRef.current = isLoading;
   }, [isLoading]);
-  (0, import_react13.useEffect)(() => {
+  (0, import_react14.useEffect)(() => {
     if (!content) {
       return;
     }
@@ -3485,9 +3583,9 @@ var useChatChunk = ({
 };
 
 // src/react/ai-chat/hooks/useEditorChat.ts
-var import_react15 = require("react");
-var import_react16 = require("@platejs/selection/react");
-var import_react17 = require("platejs/react");
+var import_react16 = require("react");
+var import_react17 = require("@platejs/selection/react");
+var import_react18 = require("platejs/react");
 var useEditorChat = ({
   chat,
   onOpenBlockSelection,
@@ -3495,15 +3593,15 @@ var useEditorChat = ({
   onOpenCursor,
   onOpenSelection
 }) => {
-  const { editor } = (0, import_react17.useEditorPlugin)(AIChatPlugin);
-  const open = (0, import_react17.usePluginOption)(AIChatPlugin, "open");
-  (0, import_react15.useEffect)(() => {
+  const { editor } = (0, import_react18.useEditorPlugin)(AIChatPlugin);
+  const open = (0, import_react18.usePluginOption)(AIChatPlugin, "open");
+  (0, import_react16.useEffect)(() => {
     onOpenChange?.(open);
     if (open) {
       if (onOpenBlockSelection) {
-        const blockSelectionApi = editor.getApi(import_react16.BlockSelectionPlugin).blockSelection;
+        const blockSelectionApi = editor.getApi(import_react17.BlockSelectionPlugin).blockSelection;
         const isBlockSelecting = editor.getOption(
-          import_react16.BlockSelectionPlugin,
+          import_react17.BlockSelectionPlugin,
           "isSelectingSome"
         );
         if (isBlockSelecting) {
@@ -3925,44 +4023,171 @@ var getCurrentBlockPath = (editor) => {
   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_platejs20 = require("platejs");
+var applyAISuggestions = (editor, content) => {
+  const isBlockSelecting = editor.getOption(
+    import_react19.BlockSelectionPlugin,
+    "isSelectingSome"
+  );
+  const { chatNodes } = editor.getOptions(AIChatPlugin);
+  if (isBlockSelecting) {
+    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_platejs20.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 (isSameString && isSameSuggestion && node.id == diffNode.id) {
+        return;
+      }
+      if (index === replaceNodes.length - 1 && diffNodes.length > replaceNodes.length) {
+        editor.tf.replaceNodes(diffNodes.slice(index), {
+          at: path
+        });
+      } else {
+        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 = editor.api.nodes({
+      at: [],
+      mode: "lowest",
+      match: (n) => import_platejs20.TextApi.isText(n) && !!n[(0, import_suggestion3.getTransientSuggestionKey)()]
+    });
+    const range = editor.api.nodesRange(Array.from(nodes));
+    editor.tf.setSelection(range);
+    return;
+  }
+};
+var withProps = (diffNodes, chatNodes) => {
+  return diffNodes.map((node, index) => {
+    if (!import_platejs20.ElementApi.isElement(node)) return node;
+    const originalNode = chatNodes?.[index];
+    return {
+      ...node,
+      ...originalNode ?? { id: (0, import_platejs20.nanoid)() },
+      children: node.children
+    };
+  });
+};
+var withTransient = (diffNodes) => {
+  return diffNodes.map((node) => {
+    if (import_platejs20.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_platejs20.TextApi.isText(node)) {
+      if (node[import_platejs20.KEYS.suggestion] || node[import_platejs20.KEYS.comment]) {
+        return {
+          text: node.text
+        };
+      }
+      return node;
+    }
+    if (import_platejs20.ElementApi.isElement(node)) {
+      if (node[import_platejs20.KEYS.suggestion]) {
+        const nodeWithoutSuggestion = {};
+        Object.keys(node).forEach((key) => {
+          if (key !== import_platejs20.KEYS.suggestion && !key.startsWith(import_platejs20.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"] })
+  );
+  return diffNodes;
+};
+
 // src/react/copilot/CopilotPlugin.tsx
-var import_markdown9 = require("@platejs/markdown");
+var import_markdown10 = require("@platejs/markdown");
 var import_debounce = __toESM(require("lodash/debounce.js"));
-var import_platejs25 = require("platejs");
-var import_react22 = require("platejs/react");
+var import_platejs26 = require("platejs");
+var import_react24 = require("platejs/react");
 
 // src/react/copilot/renderCopilotBelowNodes.tsx
-var import_react18 = __toESM(require("react"));
-var import_platejs20 = require("platejs");
-var import_react19 = require("platejs/react");
+var import_react20 = __toESM(require("react"));
+var import_platejs21 = require("platejs");
+var import_react21 = require("platejs/react");
 var renderCopilotBelowNodes = ({
   editor
 }) => {
-  const copilot = (0, import_react19.getEditorPlugin)(editor, {
-    key: import_platejs20.KEYS.copilot
+  const copilot = (0, import_react21.getEditorPlugin)(editor, {
+    key: import_platejs21.KEYS.copilot
   });
   const { renderGhostText: GhostText } = copilot.getOptions();
   if (!GhostText) return;
   return ({ children }) => {
-    return /* @__PURE__ */ import_react18.default.createElement(import_react18.default.Fragment, null, children, /* @__PURE__ */ import_react18.default.createElement(GhostText, null));
+    return /* @__PURE__ */ import_react20.default.createElement(import_react20.default.Fragment, null, children, /* @__PURE__ */ import_react20.default.createElement(GhostText, null));
   };
 };
 
 // src/react/copilot/transforms/acceptCopilot.ts
-var import_markdown6 = require("@platejs/markdown");
-var import_platejs21 = require("platejs");
+var import_markdown7 = require("@platejs/markdown");
+var import_platejs22 = require("platejs");
 var acceptCopilot = (editor) => {
   const { suggestionText } = editor.getOptions({
-    key: import_platejs21.KEYS.copilot
+    key: import_platejs22.KEYS.copilot
   });
   if (!suggestionText?.length) return false;
-  editor.tf.insertFragment((0, import_markdown6.deserializeInlineMd)(editor, suggestionText));
+  editor.tf.insertFragment((0, import_markdown7.deserializeInlineMd)(editor, suggestionText));
 };
 
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
-var import_markdown7 = require("@platejs/markdown");
-var import_platejs23 = require("platejs");
-var import_react21 = require("platejs/react");
+var import_markdown8 = require("@platejs/markdown");
+var import_platejs24 = require("platejs");
+var import_react23 = require("platejs/react");
 
 // src/react/copilot/utils/callCompletionApi.ts
 var getOriginalFetch = () => fetch;
@@ -4079,17 +4304,17 @@ var getNextWord = ({ text }) => {
 };
 
 // src/react/copilot/utils/triggerCopilotSuggestion.ts
-var import_platejs22 = require("platejs");
-var import_react20 = require("platejs/react");
+var import_platejs23 = require("platejs");
+var import_react22 = require("platejs/react");
 var triggerCopilotSuggestion = async (editor) => {
-  const { api, getOptions, setOption } = (0, import_react20.getEditorPlugin)(
+  const { api, getOptions, setOption } = (0, import_react22.getEditorPlugin)(
     editor,
     {
-      key: import_platejs22.KEYS.copilot
+      key: import_platejs23.KEYS.copilot
     }
   );
   const { completeOptions, getPrompt, isLoading, triggerQuery } = getOptions();
-  if (isLoading || editor.getOptions({ key: import_platejs22.KEYS.aiChat }).chat?.isLoading) {
+  if (isLoading || editor.getOptions({ key: import_platejs23.KEYS.aiChat }).chat?.isLoading) {
     return false;
   }
   if (!triggerQuery({ editor })) return false;
@@ -4122,8 +4347,8 @@ var withoutAbort = (editor, fn) => {
 
 // src/react/copilot/transforms/acceptCopilotNextWord.ts
 var acceptCopilotNextWord = (editor) => {
-  const { api, getOptions } = (0, import_react21.getEditorPlugin)(editor, {
-    key: import_platejs23.KEYS.copilot
+  const { api, getOptions } = (0, import_react23.getEditorPlugin)(editor, {
+    key: import_platejs24.KEYS.copilot
   });
   const { getNextWord: getNextWord2, suggestionText } = getOptions();
   if (!suggestionText?.length) {
@@ -4134,19 +4359,19 @@ var acceptCopilotNextWord = (editor) => {
     text: remainingText
   });
   withoutAbort(editor, () => {
-    editor.tf.insertFragment((0, import_markdown7.deserializeInlineMd)(editor, firstWord));
+    editor.tf.insertFragment((0, import_markdown8.deserializeInlineMd)(editor, firstWord));
   });
 };
 
 // src/react/copilot/withCopilot.ts
-var import_markdown8 = require("@platejs/markdown");
-var import_platejs24 = require("platejs");
+var import_markdown9 = require("@platejs/markdown");
+var import_platejs25 = require("platejs");
 var getPatchString = (editor, operations) => {
   let string = "";
   for (const operation of operations) {
     if (operation.type === "insert_node") {
       const node = operation.node;
-      const text = (0, import_markdown8.serializeInlineMd)(editor, { value: [node] });
+      const text = (0, import_markdown9.serializeInlineMd)(editor, { value: [node] });
       string += text;
     } else if (operation.type === "insert_text") {
       string += operation.text;
@@ -4202,7 +4427,7 @@ var withCopilot = ({
       },
       setSelection(props) {
         setSelection(props);
-        if (editor.selection && (!prevSelection || !import_platejs24.RangeApi.equals(prevSelection, editor.selection)) && getOptions().autoTriggerQuery({ editor }) && editor.api.isFocused()) {
+        if (editor.selection && (!prevSelection || !import_platejs25.RangeApi.equals(prevSelection, editor.selection)) && getOptions().autoTriggerQuery({ editor }) && editor.api.isFocused()) {
           void api.copilot.triggerSuggestion();
         }
         prevSelection = editor.selection;
@@ -4236,8 +4461,8 @@ var withCopilot = ({
 };
 
 // src/react/copilot/CopilotPlugin.tsx
-var CopilotPlugin = (0, import_react22.createTPlatePlugin)({
-  key: import_platejs25.KEYS.copilot,
+var CopilotPlugin = (0, import_react24.createTPlatePlugin)({
+  key: import_platejs26.KEYS.copilot,
   handlers: {
     onBlur: ({ api }) => {
       api.copilot.reject();
@@ -4259,20 +4484,20 @@ var CopilotPlugin = (0, import_react22.createTPlatePlugin)({
     suggestionNodeId: null,
     suggestionText: null,
     autoTriggerQuery: ({ editor }) => {
-      if (editor.getOptions({ key: import_platejs25.KEYS.copilot }).suggestionText) {
+      if (editor.getOptions({ key: import_platejs26.KEYS.copilot }).suggestionText) {
         return false;
       }
       const isEmpty = editor.api.isEmpty(editor.selection, { block: true });
       if (isEmpty) return false;
       const blockAbove = editor.api.block();
       if (!blockAbove) return false;
-      const blockString = import_platejs25.NodeApi.string(blockAbove[0]);
+      const blockString = import_platejs26.NodeApi.string(blockAbove[0]);
       return blockString.at(-1) === " ";
     },
     getPrompt: ({ editor }) => {
       const contextEntry = editor.api.block({ highest: true });
       if (!contextEntry) return "";
-      return (0, import_markdown9.serializeMd)(editor, {
+      return (0, import_markdown10.serializeMd)(editor, {
         value: [contextEntry[0]]
       });
     },
@@ -4286,15 +4511,15 @@ var CopilotPlugin = (0, import_react22.createTPlatePlugin)({
 }).overrideEditor(withCopilot).extendSelectors(({ getOptions }) => ({
   isSuggested: (id) => getOptions().suggestionNodeId === id
 })).extendTransforms(({ editor }) => ({
-  accept: (0, import_platejs25.bindFirst)(acceptCopilot, editor),
-  acceptNextWord: (0, import_platejs25.bindFirst)(acceptCopilotNextWord, editor)
+  accept: (0, import_platejs26.bindFirst)(acceptCopilot, editor),
+  acceptNextWord: (0, import_platejs26.bindFirst)(acceptCopilotNextWord, editor)
 })).extendApi(
   ({ api, editor, getOptions, setOption, setOptions }) => {
     const debounceDelay = getOptions().debounceDelay;
-    let triggerSuggestion = (0, import_platejs25.bindFirst)(triggerCopilotSuggestion, editor);
+    let triggerSuggestion = (0, import_platejs26.bindFirst)(triggerCopilotSuggestion, editor);
     if (debounceDelay) {
       triggerSuggestion = (0, import_debounce.default)(
-        (0, import_platejs25.bindFirst)(triggerCopilotSuggestion, editor),
+        (0, import_platejs26.bindFirst)(triggerCopilotSuggestion, editor),
         debounceDelay
       );
     }
@@ -4350,7 +4575,9 @@ var CopilotPlugin = (0, import_react22.createTPlatePlugin)({
   acceptAIChat,
   acceptCopilot,
   acceptCopilotNextWord,
+  acceptSuggestions,
   aiCommentToRange,
+  applyAISuggestions,
   callCompletionApi,
   createFormattedBlocks,
   escapeInput,
@@ -4361,6 +4588,7 @@ var CopilotPlugin = (0, import_react22.createTPlatePlugin)({
   getListNode,
   getNextWord,
   insertBelowAIChat,
+  insertBelowGenerate,
   isCompleteCodeBlock,
   isCompleteMath,
   isSameNode,