@platejs/ai 50.2.0 → 50.2.1

package/dist/react/index.d.mts

@@ -1,7 +1,7 @@
 import * as platejs_react from 'platejs/react';
 import { PlateEditor, OverrideEditor, RenderNodeWrapperProps } from 'platejs/react';
 import * as platejs from 'platejs';
-import { ExtendConfig, SlateEditor, Range, NodeEntry, TElement, PluginConfig, Path, TIdElement, OmitFirst, EditorNodesOptions, TText, Descendant, RemoveNodesOptions } from 'platejs';
+import { ExtendConfig, SlateEditor, Range, NodeEntry, TElement, PluginConfig, Path, TIdElement, TRange, OmitFirst, EditorNodesOptions, TText, Descendant, RemoveNodesOptions } from 'platejs';
 import { B as BaseAIPluginConfig, b as AIMode, E as EditorPrompt, A as AIToolName } from '../getEditorPrompt-C7EEuMw8.mjs';
 import { UIMessage, ChatRequestOptions } from 'ai';
 import { Range as Range$1 } from 'slate';
@@ -82,6 +82,7 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
     aiEditor: SlateEditor | null;
     chat: UseChatHelpers<ChatMessage>;
     chatNodes: TIdElement[];
+    chatSelection: TRange | null;
     /** @deprecated Use api.aiChat.node({streaming:true}) instead */
     experimental_lastTextId: string | null;
     /**
@@ -129,6 +130,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     aiEditor: SlateEditor | null;
     chat: UseChatHelpers<ChatMessage>;
     chatNodes: TIdElement[];
+    chatSelection: TRange | null;
     /** @deprecated Use api.aiChat.node({streaming:true}) instead */
     experimental_lastTextId: string | null;
     /**

package/dist/react/index.d.ts

@@ -1,7 +1,7 @@
 import * as platejs_react from 'platejs/react';
 import { PlateEditor, OverrideEditor, RenderNodeWrapperProps } from 'platejs/react';
 import * as platejs from 'platejs';
-import { ExtendConfig, SlateEditor, Range, NodeEntry, TElement, PluginConfig, Path, TIdElement, OmitFirst, EditorNodesOptions, TText, Descendant, RemoveNodesOptions } from 'platejs';
+import { ExtendConfig, SlateEditor, Range, NodeEntry, TElement, PluginConfig, Path, TIdElement, TRange, OmitFirst, EditorNodesOptions, TText, Descendant, RemoveNodesOptions } from 'platejs';
 import { B as BaseAIPluginConfig, b as AIMode, E as EditorPrompt, A as AIToolName } from '../getEditorPrompt-C7EEuMw8.js';
 import { UIMessage, ChatRequestOptions } from 'ai';
 import { Range as Range$1 } from 'slate';
@@ -82,6 +82,7 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
     aiEditor: SlateEditor | null;
     chat: UseChatHelpers<ChatMessage>;
     chatNodes: TIdElement[];
+    chatSelection: TRange | null;
     /** @deprecated Use api.aiChat.node({streaming:true}) instead */
     experimental_lastTextId: string | null;
     /**
@@ -129,6 +130,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     aiEditor: SlateEditor | null;
     chat: UseChatHelpers<ChatMessage>;
     chatNodes: TIdElement[];
+    chatSelection: TRange | null;
     /** @deprecated Use api.aiChat.node({streaming:true}) instead */
     experimental_lastTextId: string | null;
     /**

package/dist/react/index.js

@@ -3313,6 +3313,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,
@@ -3403,6 +3404,7 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
     aiEditor: null,
     chat: { messages: [] },
     chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3442,8 +3444,13 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
       });
     },
     reload: () => {
-      const { chat } = getOptions();
+      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));
+      }
       const blocks = editor.getApi(import_react9.BlockSelectionPlugin).blockSelection.getNodes();
       const selection = blocks.length > 0 ? editor.api.nodesRange(blocks) : editor.selection;
       const ctx = {
@@ -4059,6 +4066,7 @@ var applyAISuggestions = (editor, content) => {
         });
       }
     });
+    editor.getApi(import_react19.BlockSelectionPlugin).blockSelection.set(diffNodes.map((node) => node.id));
     setReplaceIds(diffNodes.map((node) => node.id));
   } else {
     const diffNodes = getDiffNodes(editor, content);