@platejs/ai 53.1.5 → 53.2.2

package/dist/react/index.d.ts

@@ -223,7 +223,7 @@ declare const AIChatPlugin: platejs_react0.PlatePlugin<PluginConfig<"aiChat", {
   reload: () => void;
   show: () => void;
   stop: () => void;
-}, "node" | "stop" | "reset" | "submit">> & Record<"aiChat", {
+}, "node" | "reset" | "stop" | "submit">> & Record<"aiChat", {
   hide: ({
     focus,
     undo

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": "53.1.5",
+  "version": "53.2.2",
   "description": "Text AI plugin for Plate",
   "keywords": [
     "plate",
@@ -32,16 +32,16 @@
     "fastest-levenshtein": "1.0.16",
     "lodash": "^4.17.21",
     "react-compiler-runtime": "^1.0.0",
-    "@platejs/markdown": "^53.1.5",
+    "@platejs/markdown": "^53.2.2",
+    "@platejs/selection": "^53.1.6",
     "@platejs/suggestion": "^53.0.3",
-    "@platejs/selection": "^53.0.0",
     "@platejs/table": "^53.0.9"
   },
   "devDependencies": {
     "@ai-sdk/react": "^3.0.0",
     "ai": "^6.0.0",
     "@plate/scripts": "1.0.0",
-    "platejs": "^53.1.2"
+    "platejs": "^53.2.1"
   },
   "peerDependencies": {
     "platejs": ">=53.0.7",