@platejs/ai 54.0.0-beta.0 → 54.0.0-beta.1

package/dist/react/index.d.ts

@@ -2,7 +2,7 @@ import { c as AIMode, d as BaseAIPluginConfig, i as EditorPrompt, l as AIToolNam
 import * as platejs0 from "platejs";
 import { Descendant, EditorNodesOptions, ExtendConfig, NodeEntry, OmitFirst, Path, PluginConfig, Range, RemoveNodesOptions, SlateEditor, TElement, TIdElement, TRange, TTableElement, TText } from "platejs";
 import { DeserializeMdOptions, SerializeMdOptions } from "@platejs/markdown";
-import * as platejs_react0 from "platejs/react";
+import * as platejs_react1 from "platejs/react";
 import { OverrideEditor, PlateEditor, RenderNodeWrapperProps } from "platejs/react";
 import React from "react";
 import { ChatRequestOptions, UIMessage } from "ai";
@@ -11,7 +11,7 @@ import { TriggerComboboxPluginOptions } from "@platejs/combobox";
 
 //#region src/react/ai/AIPlugin.d.ts
 type AIPluginConfig = ExtendConfig<BaseAIPluginConfig>;
-declare const AIPlugin: platejs_react0.PlatePlugin<platejs0.PluginConfig<"ai", {}, {}, {
+declare const AIPlugin: platejs_react1.PlatePlugin<platejs0.PluginConfig<"ai", {}, {}, {
   ai: {
     acceptPreview: ((_value?: platejs0.Value | undefined) => boolean) & ((_value?: platejs0.Value | undefined) => boolean);
     beginPreview: ((args_0?: {
@@ -168,7 +168,7 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
     removeAnchor: (options?: EditorNodesOptions) => void;
   };
 }>;
-declare const AIChatPlugin: platejs_react0.PlatePlugin<PluginConfig<"aiChat", {
+declare const AIChatPlugin: platejs_react1.PlatePlugin<PluginConfig<"aiChat", {
   _blockChunks: string;
   _blockPath: Path | null;
   /** @private Using For streamInsertChunk */
@@ -510,7 +510,7 @@ type CompletionState = {
   error?: Error | null;
   isLoading?: boolean;
 };
-declare const CopilotPlugin: platejs_react0.PlatePlugin<PluginConfig<"copilot", CompletionState & {
+declare const CopilotPlugin: platejs_react1.PlatePlugin<PluginConfig<"copilot", CompletionState & {
   /**
    * AI completion options. See:
    * {@link https://sdk.vercel.ai/docs/reference/ai-sdk-ui/use-completion#parameters | AI SDK UI useCompletion Parameters}

package/dist/react/index.js

@@ -947,6 +947,7 @@ const withoutDeserializeInMdx = (editor, input) => {
 
 //#endregion
 //#region src/react/ai-chat/streaming/streamSerializeMd.ts
+const STREAM_LINE_BREAK_PLACEHOLDER = "platejs-stream-line-break";
 const trimEndHeading = (editor, value) => {
 	const headingKeys = new Set([
 		KEYS.h1,
@@ -977,9 +978,40 @@ const trimEndHeading = (editor, value) => {
 		value
 	};
 };
+const escapeEmbeddedTextLineBreaks = (nodes) => {
+	let hasLineBreaks = false;
+	const value = nodes.map((node) => {
+		if (TextApi.isText(node)) {
+			if (node.text !== "\n" && node.text.includes("\n")) {
+				hasLineBreaks = true;
+				return {
+					...node,
+					text: node.text.replaceAll("\n", STREAM_LINE_BREAK_PLACEHOLDER)
+				};
+			}
+			return node;
+		}
+		if (ElementApi.isElement(node)) {
+			const escapedChildren = escapeEmbeddedTextLineBreaks(node.children);
+			if (escapedChildren.hasLineBreaks) {
+				hasLineBreaks = true;
+				return {
+					...node,
+					children: escapedChildren.value
+				};
+			}
+		}
+		return node;
+	});
+	return {
+		hasLineBreaks,
+		value
+	};
+};
 const streamSerializeMd = (editor, options, chunk) => {
 	const { value: optionsValue, ...restOptions } = options;
-	const { value } = trimEndHeading(editor, optionsValue ?? editor.children);
+	const { value: trimmedValue } = trimEndHeading(editor, optionsValue ?? editor.children);
+	const { hasLineBreaks, value } = escapeEmbeddedTextLineBreaks(trimmedValue);
 	let result = "";
 	result = editor.getApi(MarkdownPlugin).markdown.serialize({
 		value,
@@ -991,6 +1023,15 @@ const streamSerializeMd = (editor, options, chunk) => {
 	result = result.replace(/ /g, " ");
 	result = result.replace(/​/g, " ");
 	result = result.replace(/\u200B/g, "");
+	if (hasLineBreaks) result = result.replaceAll(STREAM_LINE_BREAK_PLACEHOLDER, "\n");
+	if (trimmedChunk.includes("\n") && trimmedChunk !== "\n\n") {
+		const trimmedResult = result.trimEnd();
+		if (trimmedResult.endsWith("\n<br />")) result = trimmedResult.slice(0, -6);
+		else {
+			const trailingWhitespaceBreakSuffix = `${trimmedChunk}\n`;
+			if (result.endsWith(trailingWhitespaceBreakSuffix)) result = result.slice(0, -trailingWhitespaceBreakSuffix.length);
+		}
+	}
 	if (trimmedChunk !== "\n\n") result = result.trimEnd() + trimmedChunk;
 	if (chunk.endsWith("\n\n")) {
 		if (result === "\n") result = "";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@platejs/ai",
-  "version": "54.0.0-beta.0",
+  "version": "54.0.0-beta.1",
   "description": "Text AI plugin for Plate",
   "keywords": [
     "plate",
@@ -32,7 +32,7 @@
     "fastest-levenshtein": "1.0.16",
     "lodash": "^4.17.21",
     "react-compiler-runtime": "^1.0.0",
-    "@platejs/markdown": "^54.0.0-beta.0",
+    "@platejs/markdown": "^54.0.0-beta.1",
     "@platejs/selection": "^54.0.0-beta.0",
     "@platejs/suggestion": "^54.0.0-beta.0",
     "@platejs/table": "^54.0.0-beta.0"
@@ -41,10 +41,10 @@
     "@ai-sdk/react": "^3.0.0",
     "ai": "^6.0.0",
     "@plate/scripts": "1.0.0",
-    "platejs": "^54.0.0-beta.0"
+    "platejs": "^54.0.0-beta.1"
   },
   "peerDependencies": {
-    "platejs": ">=54.0.0-beta.0",
+    "platejs": ">=54.0.0-beta.1",
     "react": ">=18.0.0",
     "react-dom": ">=18.0.0"
   },