@platejs/ai 50.3.4 → 50.3.5

package/dist/react/index.d.mts

@@ -66,7 +66,7 @@ declare const resetAIChat: (editor: PlateEditor, { undo }?: {
     undo?: boolean;
 }) => void;
 
-declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName, }?: {
+declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName: toolNameProps, }?: {
     mode?: AIMode;
     options?: ChatRequestOptions;
     prompt?: EditorPrompt;

package/dist/react/index.d.ts

@@ -66,7 +66,7 @@ declare const resetAIChat: (editor: PlateEditor, { undo }?: {
     undo?: boolean;
 }) => void;
 
-declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName, }?: {
+declare const submitAIChat: (editor: PlateEditor, input: string, { mode, options, prompt, toolName: toolNameProps, }?: {
     mode?: AIMode;
     options?: ChatRequestOptions;
     prompt?: EditorPrompt;

package/dist/react/index.js

@@ -619,7 +619,7 @@ var resetAIChat = (editor, { undo = true } = {}) => {
     _replaceIds: [],
     chatNodes: [],
     mode: "insert",
-    toolName: "generate"
+    toolName: null
   });
   if (undo) {
     editor.getTransforms(AIPlugin).ai.undo();
@@ -635,7 +635,7 @@ var submitAIChat = (editor, input, {
   mode,
   options,
   prompt,
-  toolName
+  toolName: toolNameProps
 } = {}) => {
   const { getOptions, setOption } = (0, import_react8.getEditorPlugin)(
     editor,
@@ -643,7 +643,8 @@ var submitAIChat = (editor, input, {
       key: import_platejs12.KEYS.aiChat
     }
   );
-  const { chat } = getOptions();
+  const { chat, toolName: toolNameOption } = getOptions();
+  const toolName = toolNameProps ?? toolNameOption ?? null;
   if (!prompt && input?.length === 0) {
     return;
   }
@@ -657,20 +658,30 @@ var submitAIChat = (editor, input, {
     editor.getTransforms(AIPlugin).ai.undo();
   }
   setOption("mode", mode);
-  setOption("toolName", toolName ?? null);
+  setOption("toolName", toolName);
   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 ? blocksRange : editor.selection;
-  const chatNodes = blocks.length > 0 ? blocks.map((block) => block[0]) : editor.api.fragment();
+  let chatNodes;
+  if (blocks.length > 0) {
+    chatNodes = blocks.map((block) => block[0]);
+  } else {
+    const selectionBlocks = editor.api.blocks({ mode: "highest" });
+    if (selectionBlocks.length > 1) {
+      chatNodes = selectionBlocks.map((block) => block[0]);
+    } else {
+      chatNodes = editor.api.fragment();
+    }
+  }
   setOption("chatNodes", chatNodes);
   setOption("chatSelection", blocks.length > 0 ? null : editor.selection);
   const ctx = {
     children: editor.children,
     selection: selection ?? null,
-    toolName: toolName ?? null
+    toolName
   };
   void chat.sendMessage?.(
     {
@@ -1385,12 +1396,13 @@ var import_react19 = require("@platejs/selection/react");
 var import_suggestion3 = require("@platejs/suggestion");
 var import_platejs21 = require("platejs");
 var applyAISuggestions = (editor, content) => {
+  editor.getApi({ key: import_platejs21.KEYS.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
   const isBlockSelecting = editor.getOption(
     import_react19.BlockSelectionPlugin,
     "isSelectingSome"
   );
   const { chatNodes } = editor.getOptions(AIChatPlugin);
-  if (isBlockSelecting) {
+  if (isBlockSelecting || editor.api.blocks({ mode: "highest" }).length > 1) {
     const setReplaceIds = (ids) => {
       editor.setOption(AIChatPlugin, "_replaceIds", ids);
     };