@platejs/ai 50.2.0 → 50.2.2

package/dist/react/index.mjs

@@ -3202,7 +3202,7 @@ var removeAnchorAIChat = (editor, options) => {
 // src/react/ai-chat/utils/resetAIChat.ts
 import { KEYS as KEYS8 } from "platejs";
 import { getEditorPlugin as getEditorPlugin2 } from "platejs/react";
-var resetAIChat = (editor) => {
+var resetAIChat = (editor, { undo = true } = {}) => {
   const { api, getOptions, setOptions } = getEditorPlugin2(
     editor,
     {
@@ -3220,7 +3220,9 @@ var resetAIChat = (editor) => {
     mode: "insert",
     toolName: "generate"
   });
-  editor.getTransforms(AIPlugin).ai.undo();
+  if (undo) {
+    editor.getTransforms(AIPlugin).ai.undo();
+  }
 };
 
 // src/react/ai-chat/utils/submitAIChat.ts
@@ -3263,6 +3265,7 @@ var submitAIChat = (editor, input, {
   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,
@@ -3353,6 +3356,7 @@ var AIChatPlugin = createTPlatePlugin({
     aiEditor: null,
     chat: { messages: [] },
     chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3392,8 +3396,13 @@ var AIChatPlugin = createTPlatePlugin({
       });
     },
     reload: () => {
-      const { chat } = getOptions();
+      const { chat, chatNodes, chatSelection } = getOptions();
       editor.getTransforms(AIPlugin).ai.undo();
+      if (chatSelection) {
+        editor.tf.setSelection(chatSelection);
+      } else {
+        editor.getApi(BlockSelectionPlugin4).blockSelection.set(chatNodes.map((node) => node.id));
+      }
       const blocks = editor.getApi(BlockSelectionPlugin4).blockSelection.getNodes();
       const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
       const ctx = {
@@ -3413,8 +3422,11 @@ var AIChatPlugin = createTPlatePlugin({
     }
   };
 }).extendApi(({ api, editor, getOptions, setOption, tf, type }) => ({
-  hide: ({ focus = true } = {}) => {
-    api.aiChat.reset();
+  hide: ({
+    focus = true,
+    undo = true
+  } = {}) => {
+    api.aiChat.reset({ undo });
     setOption("open", false);
     if (focus) {
       if (editor.getOption(BlockSelectionPlugin4, "isSelectingSome")) {
@@ -4028,6 +4040,7 @@ var applyAISuggestions = (editor, content) => {
         });
       }
     });
+    editor.getApi(BlockSelectionPlugin6).blockSelection.set(diffNodes.map((node) => node.id));
     setReplaceIds(diffNodes.map((node) => node.id));
   } else {
     const diffNodes = getDiffNodes(editor, content);