@platejs/ai 50.2.0 → 50.2.2

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';
@@ -63,7 +63,9 @@ declare const replaceSelectionAIChat: (editor: PlateEditor, sourceEditor: SlateE
     format?: "all" | "none" | "single";
 }) => void;
 
-declare const resetAIChat: (editor: PlateEditor) => void;
+declare const resetAIChat: (editor: PlateEditor, { undo }?: {
+    undo?: boolean;
+}) => void;
 
 declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName, }?: {
     mode?: AIMode;
@@ -82,6 +84,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;
     /**
@@ -102,6 +105,7 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
         submit: OmitFirst<typeof submitAIChat>;
         hide: (options?: {
             focus?: boolean;
+            undo?: boolean;
         }) => void;
         node: (options?: EditorNodesOptions & {
             anchor?: boolean;
@@ -129,6 +133,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;
     /**
@@ -149,6 +154,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
         submit: OmitFirst<typeof submitAIChat>;
         hide: (options?: {
             focus?: boolean;
+            undo?: boolean;
         }) => void;
         node: (options?: EditorNodesOptions & {
             anchor?: boolean;
@@ -163,6 +169,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     submit: OmitFirst<typeof submitAIChat>;
     hide: (options?: {
         focus?: boolean;
+        undo?: boolean;
     }) => void;
     node: (options?: EditorNodesOptions & {
         anchor?: boolean;
@@ -172,8 +179,9 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     show: () => void;
     stop: () => void;
 }, "node" | "stop" | "reset" | "submit">> & Record<"aiChat", {
-    hide: ({ focus }?: {
+    hide: ({ focus, undo, }?: {
         focus?: boolean;
+        undo?: boolean;
     }) => void;
     show: () => void;
 }>, {

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';
@@ -63,7 +63,9 @@ declare const replaceSelectionAIChat: (editor: PlateEditor, sourceEditor: SlateE
     format?: "all" | "none" | "single";
 }) => void;
 
-declare const resetAIChat: (editor: PlateEditor) => void;
+declare const resetAIChat: (editor: PlateEditor, { undo }?: {
+    undo?: boolean;
+}) => void;
 
 declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName, }?: {
     mode?: AIMode;
@@ -82,6 +84,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;
     /**
@@ -102,6 +105,7 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
         submit: OmitFirst<typeof submitAIChat>;
         hide: (options?: {
             focus?: boolean;
+            undo?: boolean;
         }) => void;
         node: (options?: EditorNodesOptions & {
             anchor?: boolean;
@@ -129,6 +133,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;
     /**
@@ -149,6 +154,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
         submit: OmitFirst<typeof submitAIChat>;
         hide: (options?: {
             focus?: boolean;
+            undo?: boolean;
         }) => void;
         node: (options?: EditorNodesOptions & {
             anchor?: boolean;
@@ -163,6 +169,7 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     submit: OmitFirst<typeof submitAIChat>;
     hide: (options?: {
         focus?: boolean;
+        undo?: boolean;
     }) => void;
     node: (options?: EditorNodesOptions & {
         anchor?: boolean;
@@ -172,8 +179,9 @@ declare const AIChatPlugin: platejs_react.PlatePlugin<PluginConfig<"aiChat", {
     show: () => void;
     stop: () => void;
 }, "node" | "stop" | "reset" | "submit">> & Record<"aiChat", {
-    hide: ({ focus }?: {
+    hide: ({ focus, undo, }?: {
         focus?: boolean;
+        undo?: boolean;
     }) => void;
     show: () => void;
 }>, {

package/dist/react/index.js

@@ -3252,7 +3252,7 @@ var removeAnchorAIChat = (editor, options) => {
 // src/react/ai-chat/utils/resetAIChat.ts
 var import_platejs10 = require("platejs");
 var import_react6 = require("platejs/react");
-var resetAIChat = (editor) => {
+var resetAIChat = (editor, { undo = true } = {}) => {
   const { api, getOptions, setOptions } = (0, import_react6.getEditorPlugin)(
     editor,
     {
@@ -3270,7 +3270,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
@@ -3313,6 +3315,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 +3406,7 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
     aiEditor: null,
     chat: { messages: [] },
     chatNodes: [],
+    chatSelection: null,
     experimental_lastTextId: null,
     mode: "insert",
     open: false,
@@ -3442,8 +3446,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 = {
@@ -3463,8 +3472,11 @@ var AIChatPlugin = (0, import_react10.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(import_react9.BlockSelectionPlugin, "isSelectingSome")) {
@@ -4059,6 +4071,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);