@platejs/ai 50.2.0 → 50.2.1

package/dist/react/index.mjs

@@ -3263,6 +3263,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 +3354,7 @@ var AIChatPlugin = createTPlatePlugin({
     aiEditor: null,
     chat: { messages: [] },
     chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3392,8 +3394,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 = {
@@ -4028,6 +4035,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);