@platejs/ai 49.1.11 → 49.2.1

package/dist/index.js

@@ -52,7 +52,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/lib/BaseAIPlugin.ts
-var import_platejs2 = require("platejs");
+var import_platejs3 = require("platejs");
 
 // src/lib/transforms/insertAINodes.ts
 var insertAINodes = (editor, nodes, {
@@ -73,19 +73,21 @@ var insertAINodes = (editor, nodes, {
 };
 
 // src/lib/transforms/removeAIMarks.ts
+var import_platejs = require("platejs");
 var removeAIMarks = (editor, { at = [] } = {}) => {
-  editor.tf.unsetNodes("ai", {
+  const nodeType = (0, import_platejs.getPluginType)(editor, import_platejs.KEYS.ai);
+  editor.tf.unsetNodes(nodeType, {
     at,
-    match: (n) => n.ai
+    match: (n) => n[nodeType]
   });
 };
 
 // src/lib/transforms/removeAINodes.ts
-var import_platejs = require("platejs");
+var import_platejs2 = require("platejs");
 var removeAINodes = (editor, { at = [] } = {}) => {
   editor.tf.removeNodes({
     at,
-    match: (n) => import_platejs.TextApi.isText(n) && !!n.ai
+    match: (n) => import_platejs2.TextApi.isText(n) && !!n.ai
   });
 };
 
@@ -116,14 +118,14 @@ var withAIBatch = (editor, fn, {
 };
 
 // src/lib/BaseAIPlugin.ts
-var BaseAIPlugin = (0, import_platejs2.createTSlatePlugin)({
-  key: import_platejs2.KEYS.ai,
+var BaseAIPlugin = (0, import_platejs3.createTSlatePlugin)({
+  key: import_platejs3.KEYS.ai,
   node: { isDecoration: false, isLeaf: true }
 }).extendTransforms(({ editor }) => ({
-  insertNodes: (0, import_platejs2.bindFirst)(insertAINodes, editor),
-  removeMarks: (0, import_platejs2.bindFirst)(removeAIMarks, editor),
-  removeNodes: (0, import_platejs2.bindFirst)(removeAINodes, editor),
-  undo: (0, import_platejs2.bindFirst)(undoAI, editor)
+  insertNodes: (0, import_platejs3.bindFirst)(insertAINodes, editor),
+  removeMarks: (0, import_platejs3.bindFirst)(removeAIMarks, editor),
+  removeNodes: (0, import_platejs3.bindFirst)(removeAINodes, editor),
+  undo: (0, import_platejs3.bindFirst)(undoAI, editor)
 }));
 
 // src/lib/streaming/streamDeserializeInlineMd.ts
@@ -133,144 +135,8 @@ var streamDeserializeInlineMd = (editor, text, options) => {
 };
 
 // src/lib/streaming/streamDeserializeMd.ts
-var import_markdown2 = require("@platejs/markdown");
-var import_platejs5 = require("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
-var import_platejs3 = require("platejs");
-var LIST_STYLE_TYPE = "listStyleType";
-var isSameNode = (editor, node1, node2) => {
-  if (node1.type !== editor.getType(import_platejs3.KEYS.p) || node2.type !== editor.getType(import_platejs3.KEYS.p))
-    return node1.type === node2.type;
-  if ((0, import_platejs3.isDefined)(node1[LIST_STYLE_TYPE]) || (0, import_platejs3.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
-var import_platejs4 = require("platejs");
-var nodesWithProps = (editor, nodes, options) => {
-  return nodes.map((node) => {
-    if (import_platejs4.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(import_markdown2.MarkdownPlugin).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 && import_platejs5.TextApi.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: import_platejs5.KEYS.p
-    });
-  }
-  if (unshiftNewLine && !isCodeBlockOrTable) {
-    result.unshift({
-      children: [{ text: "" }],
-      type: import_platejs5.KEYS.p
-    });
-  }
-  return result;
-};
-
-// src/lib/streaming/streamInsertChunk.ts
-var import_platejs16 = require("platejs");
+var import_markdown3 = require("@platejs/markdown");
+var import_platejs15 = require("platejs");
 
 // src/react/ai/AIPlugin.ts
 var import_react = require("platejs/react");
@@ -278,15 +144,15 @@ var AIPlugin = (0, import_react.toPlatePlugin)(BaseAIPlugin);
 
 // src/react/ai-chat/AIChatPlugin.ts
 var import_react9 = require("@platejs/selection/react");
-var import_platejs14 = require("platejs");
+var import_platejs12 = require("platejs");
 var import_react10 = require("platejs/react");
 
 // src/react/ai-chat/transforms/acceptAIChat.ts
-var import_platejs6 = require("platejs");
+var import_platejs4 = require("platejs");
 var import_react2 = require("platejs/react");
 var acceptAIChat = (editor) => {
   const { tf } = (0, import_react2.getEditorPlugin)(editor, AIPlugin);
-  const api = editor.getApi({ key: import_platejs6.KEYS.ai });
+  const api = editor.getApi({ key: import_platejs4.KEYS.ai });
   const lastAINodePath = api.aiChat.node({ at: [], reverse: true })[1];
   withAIBatch(editor, () => {
     tf.ai.removeMarks();
@@ -304,12 +170,12 @@ var acceptAIChat = (editor) => {
 // src/react/ai-chat/transforms/insertBelowAIChat.ts
 var import_react4 = require("@platejs/selection/react");
 var import_cloneDeep2 = __toESM(require("lodash/cloneDeep.js"));
-var import_platejs8 = require("platejs");
+var import_platejs6 = require("platejs");
 
 // src/react/ai-chat/transforms/replaceSelectionAIChat.ts
 var import_react3 = require("@platejs/selection/react");
 var import_cloneDeep = __toESM(require("lodash/cloneDeep.js"));
-var import_platejs7 = require("platejs");
+var import_platejs5 = require("platejs");
 var createFormattedBlocks = ({
   blocks,
   format,
@@ -317,12 +183,12 @@ var createFormattedBlocks = ({
 }) => {
   if (format === "none") return (0, import_cloneDeep.default)(blocks);
   const [sourceNode] = sourceBlock;
-  const firstTextEntry = import_platejs7.NodeApi.firstText(sourceNode);
+  const firstTextEntry = import_platejs5.NodeApi.firstText(sourceNode);
   if (!firstTextEntry) return null;
-  const blockProps = import_platejs7.NodeApi.extractProps(sourceNode);
-  const textProps = import_platejs7.NodeApi.extractProps(firstTextEntry[0]);
+  const blockProps = import_platejs5.NodeApi.extractProps(sourceNode);
+  const textProps = import_platejs5.NodeApi.extractProps(firstTextEntry[0]);
   const applyTextFormatting = (node) => {
-    if (import_platejs7.TextApi.isText(node)) {
+    if (import_platejs5.TextApi.isText(node)) {
       return { ...textProps, ...node };
     }
     if (node.children) {
@@ -349,7 +215,7 @@ var replaceSelectionAIChat = (editor, sourceEditor, { format = "single" } = {})
     import_react3.BlockSelectionPlugin,
     "isSelectingSome"
   );
-  editor.getApi({ key: import_platejs7.KEYS.ai }).aiChat.hide();
+  editor.getApi({ key: import_platejs5.KEYS.ai }).aiChat.hide();
   if (!isBlockSelecting) {
     const firstBlock = editor.api.node({
       block: true,
@@ -413,7 +279,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     import_react4.BlockSelectionPlugin,
     "isSelectingSome"
   );
-  editor.getApi({ key: import_platejs8.KEYS.ai }).aiChat.hide();
+  editor.getApi({ key: import_platejs6.KEYS.ai }).aiChat.hide();
   const insertBlocksAndSelect = editor.getTransforms(import_react4.BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
   if (isBlockSelecting) {
     const selectedBlocks = editor.getApi(import_react4.BlockSelectionPlugin).blockSelection.getNodes();
@@ -421,7 +287,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     if (!selectedIds || selectedIds.size === 0) return;
     const lastBlock = selectedBlocks.at(-1);
     if (!lastBlock) return;
-    const nextPath = import_platejs8.PathApi.next(lastBlock[1]);
+    const nextPath = import_platejs6.PathApi.next(lastBlock[1]);
     if (format === "none") {
       insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
         at: nextPath
@@ -438,7 +304,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
       at: nextPath
     });
   } else {
-    const [, end] = import_platejs8.RangeApi.edges(editor.selection);
+    const [, end] = import_platejs6.RangeApi.edges(editor.selection);
     const endPath = [end.path[0]];
     const currentBlock = editor.api.node({
       at: endPath,
@@ -448,7 +314,7 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     if (!currentBlock) return;
     if (format === "none") {
       insertBlocksAndSelect((0, import_cloneDeep2.default)(sourceEditor.children), {
-        at: import_platejs8.PathApi.next(endPath)
+        at: import_platejs6.PathApi.next(endPath)
       });
       return;
     }
@@ -459,18 +325,18 @@ var insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) => {
     });
     if (!formattedBlocks) return;
     insertBlocksAndSelect(formattedBlocks, {
-      at: import_platejs8.PathApi.next(endPath)
+      at: import_platejs6.PathApi.next(endPath)
     });
   }
 };
 
 // src/react/ai-chat/transforms/removeAnchorAIChat.ts
-var import_platejs9 = require("platejs");
+var import_platejs7 = require("platejs");
 var removeAnchorAIChat = (editor, options) => {
   editor.tf.withoutSaving(() => {
     editor.tf.removeNodes({
       at: [],
-      match: (n) => import_platejs9.ElementApi.isElement(n) && n.type === import_platejs9.KEYS.aiChat,
+      match: (n) => import_platejs7.ElementApi.isElement(n) && n.type === (0, import_platejs7.getPluginType)(editor, import_platejs7.KEYS.aiChat),
       ...options
     });
   });
@@ -481,12 +347,12 @@ var import_selection = require("@platejs/selection");
 var import_react6 = require("@platejs/selection/react");
 
 // src/react/ai-chat/utils/getMarkdown.ts
-var import_markdown3 = require("@platejs/markdown");
+var import_markdown2 = require("@platejs/markdown");
 var import_react5 = require("@platejs/selection/react");
-var import_platejs10 = require("platejs");
+var import_platejs8 = require("platejs");
 var getMarkdown = (editor, type) => {
   if (type === "editor") {
-    return (0, import_markdown3.serializeMd)(editor);
+    return (0, import_markdown2.serializeMd)(editor);
   }
   if (type === "block") {
     const blocks = editor.getOption(import_react5.BlockSelectionPlugin, "isSelectingSome") ? editor.getApi(import_react5.BlockSelectionPlugin).blockSelection.getNodes() : editor.api.nodes({
@@ -494,7 +360,7 @@ var getMarkdown = (editor, type) => {
       match: (n) => editor.api.isBlock(n)
     });
     const nodes = Array.from(blocks, (entry) => entry[0]);
-    return (0, import_markdown3.serializeMd)(editor, { value: nodes });
+    return (0, import_markdown2.serializeMd)(editor, { value: nodes });
   }
   if (type === "selection") {
     const fragment = editor.api.fragment();
@@ -502,12 +368,12 @@ var getMarkdown = (editor, type) => {
       const modifiedFragment = [
         {
           children: fragment[0].children,
-          type: import_platejs10.KEYS.p
+          type: import_platejs8.KEYS.p
         }
       ];
-      return (0, import_markdown3.serializeMd)(editor, { value: modifiedFragment });
+      return (0, import_markdown2.serializeMd)(editor, { value: modifiedFragment });
     }
-    return (0, import_markdown3.serializeMd)(editor, { value: fragment });
+    return (0, import_markdown2.serializeMd)(editor, { value: fragment });
   }
   return "";
 };
@@ -564,11 +430,11 @@ var getEditorPrompt = (editor, {
 };
 
 // src/react/ai-chat/utils/resetAIChat.ts
-var import_platejs11 = require("platejs");
+var import_platejs9 = require("platejs");
 var import_react7 = require("platejs/react");
 var resetAIChat = (editor) => {
   const { api, getOptions } = (0, import_react7.getEditorPlugin)(editor, {
-    key: import_platejs11.KEYS.aiChat
+    key: import_platejs9.KEYS.aiChat
   });
   api.aiChat.stop();
   const chat = getOptions().chat;
@@ -580,7 +446,7 @@ var resetAIChat = (editor) => {
 
 // src/react/ai-chat/utils/submitAIChat.ts
 var import_selection2 = require("@platejs/selection");
-var import_platejs12 = require("platejs");
+var import_platejs10 = require("platejs");
 var import_react8 = require("platejs/react");
 var submitAIChat = (editor, {
   mode,
@@ -591,7 +457,7 @@ var submitAIChat = (editor, {
   const { getOptions, setOption } = (0, import_react8.getEditorPlugin)(
     editor,
     {
-      key: import_platejs12.KEYS.aiChat
+      key: import_platejs10.KEYS.aiChat
     }
   );
   const { chat, promptTemplate, systemTemplate } = getOptions();
@@ -630,7 +496,7 @@ var submitAIChat = (editor, {
 };
 
 // src/react/ai-chat/withAIChat.ts
-var import_platejs13 = require("platejs");
+var import_platejs11 = require("platejs");
 var withAIChat = ({
   api,
   editor,
@@ -672,11 +538,11 @@ var withAIChat = ({
       },
       normalizeNode(entry) {
         const [node, path] = entry;
-        if (node[import_platejs13.KEYS.ai] && !getOptions().open) {
+        if (node[import_platejs11.KEYS.ai] && !getOptions().open) {
           tf.ai.removeMarks({ at: path });
           return;
         }
-        if (import_platejs13.ElementApi.isElement(node) && node.type === type && !getOptions().open) {
+        if (import_platejs11.ElementApi.isElement(node) && node.type === type && !getOptions().open) {
           editor.getTransforms(AIChatPlugin).aiChat.removeAnchor({ at: path });
           return;
         }
@@ -688,7 +554,7 @@ var withAIChat = ({
 
 // src/react/ai-chat/AIChatPlugin.ts
 var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
-  key: import_platejs14.KEYS.aiChat,
+  key: import_platejs12.KEYS.aiChat,
   dependencies: ["ai"],
   node: {
     isElement: true
@@ -696,6 +562,7 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
   options: {
     _blockChunks: "",
     _blockPath: null,
+    _mdxName: null,
     aiEditor: null,
     chat: { messages: [] },
     experimental_lastTextId: null,
@@ -710,14 +577,14 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
   }
 }).overrideEditor(withAIChat).extendApi(({ editor, getOption, getOptions, setOption, type }) => {
   return {
-    reset: (0, import_platejs14.bindFirst)(resetAIChat, editor),
-    submit: (0, import_platejs14.bindFirst)(submitAIChat, editor),
+    reset: (0, import_platejs12.bindFirst)(resetAIChat, editor),
+    submit: (0, import_platejs12.bindFirst)(submitAIChat, editor),
     node: (options = {}) => {
       const { anchor = false, streaming = false, ...rest } = options;
       if (anchor) {
         return editor.api.node({
           at: [],
-          match: (n) => import_platejs14.ElementApi.isElement(n) && n.type === type,
+          match: (n) => import_platejs12.ElementApi.isElement(n) && n.type === type,
           ...rest
         });
       }
@@ -729,12 +596,12 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
           at: path,
           mode: "lowest",
           reverse: true,
-          match: (t) => !!t.ai,
+          match: (t) => !!t[(0, import_platejs12.getPluginType)(editor, import_platejs12.KEYS.ai)],
           ...rest
         });
       }
       return editor.api.node({
-        match: (n) => n.ai,
+        match: (n) => n[(0, import_platejs12.getPluginType)(editor, import_platejs12.KEYS.ai)],
         ...rest
       });
     },
@@ -776,21 +643,195 @@ var AIChatPlugin = (0, import_react10.createTPlatePlugin)({
     setOption("open", true);
   }
 })).extendTransforms(({ editor }) => ({
-  accept: (0, import_platejs14.bindFirst)(acceptAIChat, editor),
-  insertBelow: (0, import_platejs14.bindFirst)(insertBelowAIChat, editor),
-  removeAnchor: (0, import_platejs14.bindFirst)(removeAnchorAIChat, editor),
-  replaceSelection: (0, import_platejs14.bindFirst)(replaceSelectionAIChat, editor)
+  accept: (0, import_platejs12.bindFirst)(acceptAIChat, editor),
+  insertBelow: (0, import_platejs12.bindFirst)(insertBelowAIChat, editor),
+  removeAnchor: (0, import_platejs12.bindFirst)(removeAnchorAIChat, editor),
+  replaceSelection: (0, import_platejs12.bindFirst)(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
+var import_platejs13 = require("platejs");
+var LIST_STYLE_TYPE = "listStyleType";
+var isSameNode = (editor, node1, node2) => {
+  if (node1.type !== editor.getType(import_platejs13.KEYS.p) || node2.type !== editor.getType(import_platejs13.KEYS.p))
+    return node1.type === node2.type;
+  if ((0, import_platejs13.isDefined)(node1[LIST_STYLE_TYPE]) || (0, import_platejs13.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
+var import_platejs14 = require("platejs");
+var nodesWithProps = (editor, nodes, options) => {
+  return nodes.map((node) => {
+    if (import_platejs14.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 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(import_markdown3.MarkdownPlugin).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 && import_platejs15.TextApi.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: import_platejs15.KEYS.p
+    });
+  }
+  if (unshiftNewLine && !isCodeBlockOrTable) {
+    result.unshift({
+      children: [{ text: "" }],
+      type: import_platejs15.KEYS.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: (0, import_platejs15.getPluginType)(editor, import_platejs15.KEYS.p)
+        }
+      ];
+    }
+  } else {
+    const newMdxName = statMdxTagRegex.exec(input)?.[1];
+    if (input.startsWith(`<${newMdxName}`)) {
+      editor.setOption(AIChatPlugin, "_mdxName", newMdxName ?? null);
+    }
+  }
+};
+
+// src/lib/streaming/streamInsertChunk.ts
+var import_platejs17 = require("platejs");
+
 // src/lib/streaming/streamSerializeMd.ts
 var import_markdown4 = require("@platejs/markdown");
-var import_platejs15 = require("platejs");
-var trimEndHeading = (value) => {
-  const headingKeys = /* @__PURE__ */ new Set(["h1", "h2", "h3", "h4", "h5", "h6"]);
+var import_platejs16 = require("platejs");
+var trimEndHeading = (editor, value) => {
+  const headingKeys = /* @__PURE__ */ new Set([
+    import_platejs16.KEYS.h1,
+    import_platejs16.KEYS.h2,
+    import_platejs16.KEYS.h3,
+    import_platejs16.KEYS.h4,
+    import_platejs16.KEYS.h5,
+    import_platejs16.KEYS.h6
+  ]);
   const lastBlock = value.at(-1);
-  if (lastBlock && headingKeys.has(lastBlock.type) && import_platejs15.ElementApi.isElement(lastBlock)) {
+  if (lastBlock && headingKeys.has(
+    (0, import_platejs16.getPluginKey)(editor, lastBlock.type) ?? lastBlock.type
+  ) && import_platejs16.ElementApi.isElement(lastBlock)) {
     const lastTextNode = lastBlock.children.at(-1);
-    if (import_platejs15.TextApi.isText(lastTextNode)) {
+    if (import_platejs16.TextApi.isText(lastTextNode)) {
       const trimmedText = getChunkTrimmed(lastTextNode?.text);
       const newChildren = [
         ...lastBlock.children.slice(0, -1),
@@ -810,7 +851,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(import_markdown4.MarkdownPlugin).markdown.serialize({
     preserveEmptyParagraphs: false,
@@ -842,12 +883,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 = import_platejs17.PathApi.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 = import_platejs16.NodeApi.string(editor.api.node(path)[0]).length === 0;
+    const startBlock = editor.api.node(path)[0];
+    const startInEmptyParagraph = import_platejs17.NodeApi.string(startBlock).length === 0 && startBlock.type === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.p);
     if (startInEmptyParagraph) {
       editor.tf.removeNodes({ at: path });
     }
@@ -867,7 +916,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,
@@ -903,7 +952,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
           },
           tempBlockChunks
         );
-        const blockText = import_platejs16.NodeApi.string(tempBlocks[0]);
+        const blockText = import_platejs17.NodeApi.string(tempBlocks[0]);
         if (serializedBlock === tempBlockChunks && blockText === serializedBlock) {
           editor.setOption(AIChatPlugin, "_blockChunks", tempBlockChunks);
         } else {
@@ -925,7 +974,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 === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.codeBlock) || tempBlocks[0].type === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.table) || tempBlocks[0].type === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.equation) ? tempBlockChunks : serializedBlock2
           );
         }
       } else {
@@ -951,11 +1000,11 @@ 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),
           {
-            at: import_platejs16.PathApi.next(_blockPath),
+            at: import_platejs17.PathApi.next(_blockPath),
             select: true
           }
         );
@@ -974,11 +1023,21 @@ function streamInsertChunk(editor, chunk, options = {}) {
   }
 }
 var getCurrentBlockPath = (editor) => {
-  const anchorNode = editor.getApi(AIChatPlugin).aiChat.node({ anchor: true });
-  if (anchorNode) {
-    return import_platejs16.PathApi.previous(anchorNode[1]);
+  const getAnchorPreviousPath = (editor2) => {
+    const anchorNode = editor2.getApi(AIChatPlugin).aiChat.node({ anchor: true });
+    if (anchorNode) {
+      return import_platejs17.PathApi.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 === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.columnGroup) || entry[0].type === (0, import_platejs17.getPluginType)(editor, import_platejs17.KEYS.table))) {
+    return editor.api.above()?.[1] ?? path;
   }
-  return editor.selection?.focus.path.slice(0, 1) ?? [0];
+  return path;
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {