@platejs/ai 50.3.4 → 50.3.7

package/dist/react/index.mjs

@@ -569,7 +569,7 @@ var resetAIChat = (editor, { undo = true } = {}) => {
     _replaceIds: [],
     chatNodes: [],
     mode: "insert",
-    toolName: "generate"
+    toolName: null
   });
   if (undo) {
     editor.getTransforms(AIPlugin).ai.undo();
@@ -585,7 +585,7 @@ var submitAIChat = (editor, input, {
   mode,
   options,
   prompt,
-  toolName
+  toolName: toolNameProps
 } = {}) => {
   const { getOptions, setOption } = getEditorPlugin3(
     editor,
@@ -593,7 +593,8 @@ var submitAIChat = (editor, input, {
       key: KEYS9.aiChat
     }
   );
-  const { chat } = getOptions();
+  const { chat, toolName: toolNameOption } = getOptions();
+  const toolName = toolNameProps ?? toolNameOption ?? null;
   if (!prompt && input?.length === 0) {
     return;
   }
@@ -607,20 +608,30 @@ var submitAIChat = (editor, input, {
     editor.getTransforms(AIPlugin).ai.undo();
   }
   setOption("mode", mode);
-  setOption("toolName", toolName ?? null);
+  setOption("toolName", toolName);
   const blocks = editor.getApi(BlockSelectionPlugin3).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?.(
     {
@@ -1354,12 +1365,13 @@ import {
   TextApi as TextApi5
 } from "platejs";
 var applyAISuggestions = (editor, content) => {
+  editor.getApi({ key: KEYS17.cursorOverlay })?.cursorOverlay?.removeCursor("selection");
   const isBlockSelecting = editor.getOption(
     BlockSelectionPlugin6,
     "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);
     };