@platejs/ai 49.1.11 → 49.2.1

package/dist/index.mjs

@@ -2,7 +2,7 @@
 import {
   bindFirst,
   createTSlatePlugin,
-  KEYS
+  KEYS as KEYS2
 } from "platejs";
 
 // src/lib/transforms/insertAINodes.ts
@@ -24,10 +24,12 @@ var insertAINodes = (editor, nodes, {
 };
 
 // src/lib/transforms/removeAIMarks.ts
+import { getPluginType, KEYS } from "platejs";
 var removeAIMarks = (editor, { at = [] } = {}) => {
-  editor.tf.unsetNodes("ai", {
+  const nodeType = getPluginType(editor, KEYS.ai);
+  editor.tf.unsetNodes(nodeType, {
     at,
-    match: (n) => n.ai
+    match: (n) => n[nodeType]
   });
 };
 
@@ -68,7 +70,7 @@ var withAIBatch = (editor, fn, {
 
 // src/lib/BaseAIPlugin.ts
 var BaseAIPlugin = createTSlatePlugin({
-  key: KEYS.ai,
+  key: KEYS2.ai,
   node: { isDecoration: false, isLeaf: true }
 }).extendTransforms(({ editor }) => ({
   insertNodes: bindFirst(insertAINodes, editor),
@@ -85,143 +87,7 @@ var streamDeserializeInlineMd = (editor, text, options) => {
 
 // src/lib/streaming/streamDeserializeMd.ts
 import { MarkdownPlugin as MarkdownPlugin2 } from "@platejs/markdown";
-import { KEYS as KEYS3, TextApi as TextApi2 } from "platejs";
-
-// src/lib/streaming/utils/escapeInput.ts
-var escapeInput = (data) => {
-  let res = data;
-  if (data.startsWith("$$") && !data.startsWith("$$\n") && !isCompleteMath(data)) {
-    res = data.replace("$$", String.raw`\$\$`);
-  }
-  return res;
-};
-
-// src/lib/streaming/utils/getListNode.ts
-var getListNode = (editor, node) => {
-  if (node.listStyleType && node.listStart) {
-    const previousNode = editor.api.previous({
-      at: editor.selection?.focus
-    })?.[0];
-    if (previousNode?.listStyleType && previousNode?.listStart) {
-      return node;
-    } else {
-      if (node.listStart === 1) return node;
-      return {
-        ...node,
-        listRestartPolite: node.listStart
-      };
-    }
-  }
-  return node;
-};
-
-// src/lib/streaming/utils/isSameNode.ts
-import { isDefined, KEYS as KEYS2 } from "platejs";
-var LIST_STYLE_TYPE = "listStyleType";
-var isSameNode = (editor, node1, node2) => {
-  if (node1.type !== editor.getType(KEYS2.p) || node2.type !== editor.getType(KEYS2.p))
-    return node1.type === node2.type;
-  if (isDefined(node1[LIST_STYLE_TYPE]) || isDefined(node2[LIST_STYLE_TYPE])) {
-    return node1[LIST_STYLE_TYPE] === node2[LIST_STYLE_TYPE];
-  }
-  return node1.type === node2.type;
-};
-
-// src/lib/streaming/utils/nodesWithProps.ts
-import { ElementApi } from "platejs";
-var nodesWithProps = (editor, nodes, options) => {
-  return nodes.map((node) => {
-    if (ElementApi.isElement(node)) {
-      return {
-        ...getListNode(editor, node),
-        ...options.elementProps,
-        children: nodesWithProps(editor, node.children, options)
-      };
-    } else {
-      return {
-        ...options.textProps,
-        ...node,
-        text: node.text
-      };
-    }
-  });
-};
-
-// src/lib/streaming/utils/utils.ts
-var getChunkTrimmed = (chunk, {
-  direction = "right"
-} = {}) => {
-  const str = direction === "right" ? chunk.trimEnd() : chunk.trimStart();
-  if (direction === "right") {
-    return chunk.slice(str.length);
-  } else {
-    return chunk.slice(0, chunk.length - str.length);
-  }
-};
-function isCompleteCodeBlock(str) {
-  const trimmed = str.trim();
-  const startsWithCodeBlock = trimmed.startsWith("```");
-  const endsWithCodeBlock = trimmed.endsWith("```");
-  return startsWithCodeBlock && endsWithCodeBlock;
-}
-function isCompleteMath(str) {
-  const trimmed = str.trim();
-  const startsWithMath = trimmed.startsWith("$$");
-  const endsWithMath = trimmed.endsWith("$$");
-  return startsWithMath && endsWithMath;
-}
-
-// src/lib/streaming/streamDeserializeMd.ts
-var streamDeserializeMd = (editor, data, options) => {
-  const input = escapeInput(data);
-  let blocks = [];
-  blocks = editor.getApi(MarkdownPlugin2).markdown.deserialize(input, {
-    ...options,
-    preserveEmptyParagraphs: false
-  });
-  const trimmedData = getChunkTrimmed(data);
-  const lastBlock = blocks.at(-1);
-  const addNewLine = trimmedData === "\n\n";
-  const unshiftNewLine = getChunkTrimmed(data, { direction: "left" }) === "\n\n";
-  const isCodeBlockOrTable = lastBlock?.type === "code_block" || lastBlock?.type === "table";
-  let result = blocks;
-  if (lastBlock && !isCodeBlockOrTable && trimmedData.length > 0 && !addNewLine) {
-    const textNode = [
-      {
-        text: trimmedData
-      }
-    ];
-    const lastChild = lastBlock.children.at(-1);
-    if (lastChild && TextApi2.isText(lastChild) && Object.keys(lastChild).length === 1) {
-      lastBlock.children.pop();
-      const textNode2 = [
-        {
-          text: lastChild.text + trimmedData
-        }
-      ];
-      lastBlock.children.push(...textNode2);
-    } else {
-      lastBlock.children.push(...textNode);
-    }
-    result = [...blocks.slice(0, -1), lastBlock];
-  }
-  if (addNewLine && !isCodeBlockOrTable) {
-    result.push({
-      children: [{ text: "" }],
-      type: KEYS3.p
-    });
-  }
-  if (unshiftNewLine && !isCodeBlockOrTable) {
-    result.unshift({
-      children: [{ text: "" }],
-      type: KEYS3.p
-    });
-  }
-  return result;
-};
-
-// src/lib/streaming/streamInsertChunk.ts
-import { NodeApi as NodeApi2, PathApi as PathApi2 } from "platejs";
+import { getPluginType as getPluginType4, KEYS as KEYS13, TextApi as TextApi3 } from "platejs";
 
 // src/react/ai/AIPlugin.ts
 import { toPlatePlugin } from "platejs/react";
@@ -231,17 +97,18 @@ var AIPlugin = toPlatePlugin(BaseAIPlugin);
 import { BlockSelectionPlugin as BlockSelectionPlugin5 } from "@platejs/selection/react";
 import {
   bindFirst as bindFirst2,
-  ElementApi as ElementApi4,
-  KEYS as KEYS12
+  ElementApi as ElementApi3,
+  getPluginType as getPluginType3,
+  KEYS as KEYS11
 } from "platejs";
 import { createTPlatePlugin } from "platejs/react";
 
 // src/react/ai-chat/transforms/acceptAIChat.ts
-import { KEYS as KEYS4 } from "platejs";
+import { KEYS as KEYS3 } from "platejs";
 import { getEditorPlugin } from "platejs/react";
 var acceptAIChat = (editor) => {
   const { tf } = getEditorPlugin(editor, AIPlugin);
-  const api = editor.getApi({ key: KEYS4.ai });
+  const api = editor.getApi({ key: KEYS3.ai });
   const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
   withAIBatch(editor, () => {
     tf.ai.removeMarks();
@@ -259,7 +126,7 @@ var acceptAIChat = (editor) => {
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
 import { BlockSelectionPlugin as BlockSelectionPlugin2 } from "@platejs/selection/react";
 import cloneDeep2 from "lodash/cloneDeep.js";
-import { KEYS as KEYS6, PathApi, RangeApi } from "platejs";
+import { KEYS as KEYS5, PathApi, RangeApi } from "platejs";
 
 // src/react/ai-chat/transforms/replaceSelectionAIChat.ts
 import {
@@ -268,9 +135,9 @@ import {
 } from "@platejs/selection/react";
 import cloneDeep from "lodash/cloneDeep.js";
 import {
-  KEYS as KEYS5,
+  KEYS as KEYS4,
   NodeApi,
-  TextApi as TextApi3
+  TextApi as TextApi2
 } from "platejs";
 var createFormattedBlocks = ({
   blocks,
@@ -284,7 +151,7 @@ var createFormattedBlocks = ({
   const blockProps = NodeApi.extractProps(sourceNode);
   const textProps = NodeApi.extractProps(firstTextEntry[0]);
   const applyTextFormatting = (node) => {
-    if (TextApi3.isText(node)) {
+    if (TextApi2.isText(node)) {
       return { ...textProps, ...node };
     }
     if (node.children) {
@@ -311,7 +178,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
     BlockSelectionPlugin,
     "isSelectingSome"
   );
-  editor.getApi({ key: KEYS5.ai }).aiChat.hide();
+  editor.getApi({ key: KEYS4.ai }).aiChat.hide();
   if (!isBlockSelecting) {
     const firstBlock = editor.api.node({
       block: true,
@@ -375,7 +242,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     BlockSelectionPlugin2,
     "isSelectingSome"
   );
-  editor.getApi({ key: KEYS6.ai }).aiChat.hide();
+  editor.getApi({ key: KEYS5.ai }).aiChat.hide();
   const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin2).blockSelection.insertBlocksAndSelect;
   if (isBlockSelecting) {
     const selectedBlocks = editor.getApi(BlockSelectionPlugin2).blockSelection.getNodes();
@@ -427,12 +294,12 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
 };
 
 // src/react/ai-chat/transforms/removeAnchorAIChat.ts
-import { ElementApi as ElementApi2, KEYS as KEYS7 } from "platejs";
+import { ElementApi, getPluginType as getPluginType2, KEYS as KEYS6 } from "platejs";
 var removeAnchorAIChat = (editor, options) => {
   editor.tf.withoutSaving(() => {
     editor.tf.removeNodes({
       at: [],
-      match: (n) => ElementApi2.isElement(n) && n.type === KEYS7.aiChat,
+      match: (n) => ElementApi.isElement(n) && n.type === getPluginType2(editor, KEYS6.aiChat),
       ...options
     });
   });
@@ -445,7 +312,7 @@ import { BlockSelectionPlugin as BlockSelectionPlugin4 } from "@platejs/selectio
 // src/react/ai-chat/utils/getMarkdown.ts
 import { serializeMd } from "@platejs/markdown";
 import { BlockSelectionPlugin as BlockSelectionPlugin3 } from "@platejs/selection/react";
-import { KEYS as KEYS8 } from "platejs";
+import { KEYS as KEYS7 } from "platejs";
 var getMarkdown = (editor, type) => {
   if (type === "editor") {
     return serializeMd(editor);
@@ -464,7 +331,7 @@ var getMarkdown = (editor, type) => {
       const modifiedFragment = [
         {
           children: fragment[0].children,
-          type: KEYS8.p
+          type: KEYS7.p
         }
       ];
       return serializeMd(editor, { value: modifiedFragment });
@@ -526,11 +393,11 @@ var getEditorPrompt = (editor, {
 };
 
 // src/react/ai-chat/utils/resetAIChat.ts
-import { KEYS as KEYS9 } from "platejs";
+import { KEYS as KEYS8 } from "platejs";
 import { getEditorPlugin as getEditorPlugin2 } from "platejs/react";
 var resetAIChat = (editor) => {
   const { api, getOptions } = getEditorPlugin2(editor, {
-    key: KEYS9.aiChat
+    key: KEYS8.aiChat
   });
   api.aiChat.stop();
   const chat = getOptions().chat;
@@ -542,7 +409,7 @@ var resetAIChat = (editor) => {
 
 // src/react/ai-chat/utils/submitAIChat.ts
 import { isSelecting as isSelecting2 } from "@platejs/selection";
-import { KEYS as KEYS10 } from "platejs";
+import { KEYS as KEYS9 } from "platejs";
 import { getEditorPlugin as getEditorPlugin3 } from "platejs/react";
 var submitAIChat = (editor, {
   mode,
@@ -553,7 +420,7 @@ var submitAIChat = (editor, {
   const { getOptions, setOption } = getEditorPlugin3(
     editor,
     {
-      key: KEYS10.aiChat
+      key: KEYS9.aiChat
     }
   );
   const { chat, promptTemplate, systemTemplate } = getOptions();
@@ -592,7 +459,7 @@ var submitAIChat = (editor, {
 };
 
 // src/react/ai-chat/withAIChat.ts
-import { ElementApi as ElementApi3, KEYS as KEYS11 } from "platejs";
+import { ElementApi as ElementApi2, KEYS as KEYS10 } from "platejs";
 var withAIChat = ({
   api,
   editor,
@@ -634,11 +501,11 @@ var withAIChat = ({
       },
       normalizeNode(entry) {
         const [node, path] = entry;
-        if (node[KEYS11.ai] && !getOptions().open) {
+        if (node[KEYS10.ai] && !getOptions().open) {
           tf.ai.removeMarks({ at: path });
           return;
         }
-        if (ElementApi3.isElement(node) && node.type === type && !getOptions().open) {
+        if (ElementApi2.isElement(node) && node.type === type && !getOptions().open) {
           editor.getTransforms(AIChatPlugin).aiChat.removeAnchor({ at: path });
           return;
         }
@@ -650,7 +517,7 @@ var withAIChat = ({
 
 // src/react/ai-chat/AIChatPlugin.ts
 var AIChatPlugin = createTPlatePlugin({
-  key: KEYS12.aiChat,
+  key: KEYS11.aiChat,
   dependencies: ["ai"],
   node: {
     isElement: true
@@ -658,6 +525,7 @@ var AIChatPlugin = createTPlatePlugin({
   options: {
     _blockChunks: "",
     _blockPath: null,
+    _mdxName: null,
     aiEditor: null,
     chat: { messages: [] },
     experimental_lastTextId: null,
@@ -679,7 +547,7 @@ var AIChatPlugin = createTPlatePlugin({
       if (anchor) {
         return editor.api.node({
           at: [],
-          match: (n) => ElementApi4.isElement(n) && n.type === type,
+          match: (n) => ElementApi3.isElement(n) && n.type === type,
           ...rest
         });
       }
@@ -691,12 +559,12 @@ var AIChatPlugin = createTPlatePlugin({
           at: path,
           mode: "lowest",
           reverse: true,
-          match: (t) => !!t.ai,
+          match: (t) => !!t[getPluginType3(editor, KEYS11.ai)],
           ...rest
         });
       }
       return editor.api.node({
-        match: (n) => n.ai,
+        match: (n) => n[getPluginType3(editor, KEYS11.ai)],
         ...rest
       });
     },
@@ -744,13 +612,197 @@ var AIChatPlugin = createTPlatePlugin({
   replaceSelection: bindFirst2(replaceSelectionAIChat, editor)
 }));
 
+// src/lib/streaming/utils/escapeInput.ts
+var escapeInput = (data) => {
+  let res = data;
+  if (data.startsWith("$$") && !data.startsWith("$$\n") && !isCompleteMath(data)) {
+    res = data.replace("$$", String.raw`\$\$`);
+  }
+  return res;
+};
+
+// src/lib/streaming/utils/getListNode.ts
+var getListNode = (editor, node) => {
+  if (node.listStyleType && node.listStart) {
+    const previousNode = editor.api.previous({
+      at: editor.selection?.focus
+    })?.[0];
+    if (previousNode?.listStyleType && previousNode?.listStart) {
+      return node;
+    } else {
+      if (node.listStart === 1) return node;
+      return {
+        ...node,
+        listRestartPolite: node.listStart
+      };
+    }
+  }
+  return node;
+};
+
+// src/lib/streaming/utils/isSameNode.ts
+import { isDefined, KEYS as KEYS12 } from "platejs";
+var LIST_STYLE_TYPE = "listStyleType";
+var isSameNode = (editor, node1, node2) => {
+  if (node1.type !== editor.getType(KEYS12.p) || node2.type !== editor.getType(KEYS12.p))
+    return node1.type === node2.type;
+  if (isDefined(node1[LIST_STYLE_TYPE]) || isDefined(node2[LIST_STYLE_TYPE])) {
+    return node1[LIST_STYLE_TYPE] === node2[LIST_STYLE_TYPE];
+  }
+  return node1.type === node2.type;
+};
+
+// src/lib/streaming/utils/nodesWithProps.ts
+import { ElementApi as ElementApi4 } from "platejs";
+var nodesWithProps = (editor, nodes, options) => {
+  return nodes.map((node) => {
+    if (ElementApi4.isElement(node)) {
+      return {
+        ...getListNode(editor, node),
+        ...options.elementProps,
+        children: nodesWithProps(editor, node.children, options)
+      };
+    } else {
+      return {
+        ...options.textProps,
+        ...node,
+        text: node.text
+      };
+    }
+  });
+};
+
+// src/lib/streaming/utils/utils.ts
+var getChunkTrimmed = (chunk, {
+  direction = "right"
+} = {}) => {
+  const str = direction === "right" ? chunk.trimEnd() : chunk.trimStart();
+  if (direction === "right") {
+    return chunk.slice(str.length);
+  } else {
+    return chunk.slice(0, chunk.length - str.length);
+  }
+};
+function isCompleteCodeBlock(str) {
+  const trimmed = str.trim();
+  const startsWithCodeBlock = trimmed.startsWith("```");
+  const endsWithCodeBlock = trimmed.endsWith("```");
+  return startsWithCodeBlock && endsWithCodeBlock;
+}
+function isCompleteMath(str) {
+  const trimmed = str.trim();
+  const startsWithMath = trimmed.startsWith("$$");
+  const endsWithMath = trimmed.endsWith("$$");
+  return startsWithMath && endsWithMath;
+}
+
+// src/lib/streaming/streamDeserializeMd.ts
+var statMdxTagRegex = /<([A-Za-z][A-Za-z0-9\-_]*)>/;
+var streamDeserializeMd = (editor, data, options) => {
+  const input = escapeInput(data);
+  const value = withoutDeserializeInMdx(editor, input);
+  if (Array.isArray(value)) return value;
+  let blocks = [];
+  blocks = editor.getApi(MarkdownPlugin2).markdown.deserialize(input, {
+    ...options,
+    preserveEmptyParagraphs: false
+  });
+  const trimmedData = getChunkTrimmed(data);
+  const lastBlock = blocks.at(-1);
+  const addNewLine = trimmedData === "\n\n";
+  const unshiftNewLine = getChunkTrimmed(data, { direction: "left" }) === "\n\n";
+  const isCodeBlockOrTable = lastBlock?.type === "code_block" || lastBlock?.type === "table";
+  let result = blocks;
+  if (lastBlock && !isCodeBlockOrTable && trimmedData.length > 0 && !addNewLine) {
+    const textNode = [
+      {
+        text: trimmedData
+      }
+    ];
+    const lastChild = lastBlock.children.at(-1);
+    if (lastChild && TextApi3.isText(lastChild) && Object.keys(lastChild).length === 1) {
+      lastBlock.children.pop();
+      const textNode2 = [
+        {
+          text: lastChild.text + trimmedData
+        }
+      ];
+      lastBlock.children.push(...textNode2);
+    } else {
+      lastBlock.children.push(...textNode);
+    }
+    result = [...blocks.slice(0, -1), lastBlock];
+  }
+  if (addNewLine && !isCodeBlockOrTable) {
+    result.push({
+      children: [{ text: "" }],
+      type: KEYS13.p
+    });
+  }
+  if (unshiftNewLine && !isCodeBlockOrTable) {
+    result.unshift({
+      children: [{ text: "" }],
+      type: KEYS13.p
+    });
+  }
+  return result;
+};
+var withoutDeserializeInMdx = (editor, input) => {
+  const mdxName = editor.getOption(AIChatPlugin, "_mdxName");
+  if (mdxName) {
+    const isMdxEnd = input.includes(`</${mdxName}>`);
+    if (isMdxEnd) {
+      editor.setOption(AIChatPlugin, "_mdxName", null);
+      return false;
+    } else {
+      return [
+        {
+          children: [
+            {
+              text: input
+            }
+          ],
+          type: getPluginType4(editor, KEYS13.p)
+        }
+      ];
+    }
+  } else {
+    const newMdxName = statMdxTagRegex.exec(input)?.[1];
+    if (input.startsWith(`<${newMdxName}`)) {
+      editor.setOption(AIChatPlugin, "_mdxName", newMdxName ?? null);
+    }
+  }
+};
+
+// src/lib/streaming/streamInsertChunk.ts
+import {
+  getPluginType as getPluginType5,
+  KEYS as KEYS15,
+  NodeApi as NodeApi2,
+  PathApi as PathApi2
+} from "platejs";
+
 // src/lib/streaming/streamSerializeMd.ts
 import { MarkdownPlugin as MarkdownPlugin3 } from "@platejs/markdown";
-import { ElementApi as ElementApi5, TextApi as TextApi4 } from "platejs";
-var trimEndHeading = (value) => {
-  const headingKeys = /* @__PURE__ */ new Set(["h1", "h2", "h3", "h4", "h5", "h6"]);
+import {
+  ElementApi as ElementApi5,
+  getPluginKey,
+  KEYS as KEYS14,
+  TextApi as TextApi4
+} from "platejs";
+var trimEndHeading = (editor, value) => {
+  const headingKeys = /* @__PURE__ */ new Set([
+    KEYS14.h1,
+    KEYS14.h2,
+    KEYS14.h3,
+    KEYS14.h4,
+    KEYS14.h5,
+    KEYS14.h6
+  ]);
   const lastBlock = value.at(-1);
-  if (lastBlock && headingKeys.has(lastBlock.type) && ElementApi5.isElement(lastBlock)) {
+  if (lastBlock && headingKeys.has(
+    getPluginKey(editor, lastBlock.type) ?? lastBlock.type
+  ) && ElementApi5.isElement(lastBlock)) {
     const lastTextNode = lastBlock.children.at(-1);
     if (TextApi4.isText(lastTextNode)) {
       const trimmedText = getChunkTrimmed(lastTextNode?.text);
@@ -772,7 +824,7 @@ var trimEndHeading = (value) => {
 };
 var streamSerializeMd = (editor, options, chunk) => {
   const { value: optionsValue, ...restOptions } = options;
-  const { value } = trimEndHeading(optionsValue ?? editor.children);
+  const { value } = trimEndHeading(editor, optionsValue ?? editor.children);
   let result = "";
   result = editor.getApi(MarkdownPlugin3).markdown.serialize({
     preserveEmptyParagraphs: false,
@@ -804,12 +856,20 @@ var streamSerializeMd = (editor, options, chunk) => {
 };
 
 // src/lib/streaming/streamInsertChunk.ts
+var getNextPath = (path, length) => {
+  let result = path;
+  for (let i = 0; i < length; i++) {
+    result = PathApi2.next(result);
+  }
+  return result;
+};
 function streamInsertChunk(editor, chunk, options = {}) {
   const { _blockChunks, _blockPath } = editor.getOptions(AIChatPlugin);
   if (_blockPath === null) {
     const blocks = streamDeserializeMd(editor, chunk);
     const path = getCurrentBlockPath(editor);
-    const startInEmptyParagraph = NodeApi2.string(editor.api.node(path)[0]).length === 0;
+    const startBlock = editor.api.node(path)[0];
+    const startInEmptyParagraph = NodeApi2.string(startBlock).length === 0 && startBlock.type === getPluginType5(editor, KEYS15.p);
     if (startInEmptyParagraph) {
       editor.tf.removeNodes({ at: path });
     }
@@ -829,7 +889,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
           nextBlock: true,
           select: true
         });
-        const lastBlock = editor.api.node(nextBlocks.at(-1)) ?? editor.api.node([nextPath[0] + nextBlocks.length]);
+        const lastBlock = editor.api.node(getNextPath(nextPath, nextBlocks.length));
         editor.setOption(AIChatPlugin, "_blockPath", lastBlock[1]);
         const lastBlockChunks = streamSerializeMd(
           editor,
@@ -887,7 +947,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
             AIChatPlugin,
             "_blockChunks",
             // one block includes multiple children
-            tempBlocks[0].type === "code_block" || tempBlocks[0].type === "table" || tempBlocks[0].type === "equation" ? tempBlockChunks : serializedBlock2
+            tempBlocks[0].type === getPluginType5(editor, KEYS15.codeBlock) || tempBlocks[0].type === getPluginType5(editor, KEYS15.table) || tempBlocks[0].type === getPluginType5(editor, KEYS15.equation) ? tempBlockChunks : serializedBlock2
           );
         }
       } else {
@@ -913,7 +973,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
         select: true
       });
       if (tempBlocks.length > 1) {
-        const newEndBlockPath = [_blockPath[0] + tempBlocks.length - 1];
+        const newEndBlockPath = getNextPath(_blockPath, tempBlocks.length - 1);
         editor.tf.insertNodes(
           nodesWithProps(editor, tempBlocks.slice(1), options),
           {
@@ -936,11 +996,21 @@ function streamInsertChunk(editor, chunk, options = {}) {
   }
 }
 var getCurrentBlockPath = (editor) => {
-  const anchorNode = editor.getApi(AIChatPlugin).aiChat.node({ anchor: true });
-  if (anchorNode) {
-    return PathApi2.previous(anchorNode[1]);
+  const getAnchorPreviousPath = (editor2) => {
+    const anchorNode = editor2.getApi(AIChatPlugin).aiChat.node({ anchor: true });
+    if (anchorNode) {
+      return PathApi2.previous(anchorNode[1]);
+    }
+  };
+  const getFocusPath = (editor2) => {
+    return editor2.selection?.focus.path.slice(0, 1);
+  };
+  const path = getAnchorPreviousPath(editor) ?? getFocusPath(editor) ?? [0];
+  const entry = editor.api.node(path);
+  if (entry && (entry[0].type === getPluginType5(editor, KEYS15.columnGroup) || entry[0].type === getPluginType5(editor, KEYS15.table))) {
+    return editor.api.above()?.[1] ?? path;
   }
-  return editor.selection?.focus.path.slice(0, 1) ?? [0];
+  return path;
 };
 export {
   BaseAIPlugin,