@platejs/ai 52.3.8 → 52.3.12

package/dist/getEditorPrompt-axDm9Fg4.js

@@ -0,0 +1,235 @@
+import { ElementApi, KEYS, TextApi, bindFirst, createTSlatePlugin, getPluginType } from "platejs";
+import cloneDeep from "lodash/cloneDeep.js";
+import { getTransientSuggestionKey } from "@platejs/suggestion";
+import { isSelecting } from "@platejs/selection";
+
+//#region src/lib/transforms/aiStreamSnapshot.ts
+const AI_PREVIEW_KEY = "aiPreview";
+const AI_STREAM_SNAPSHOT = /* @__PURE__ */ new WeakMap();
+const clearAIPreview = (editor) => {
+	AI_STREAM_SNAPSHOT.delete(editor);
+};
+const getAIPreview = (editor) => AI_STREAM_SNAPSHOT.get(editor);
+const getAIPreviewRange = (editor) => {
+	let closed = false;
+	let end = -1;
+	let invalid = false;
+	let start = -1;
+	editor.children.forEach((node, index) => {
+		if (!node?.[AI_PREVIEW_KEY]) {
+			if (start !== -1) closed = true;
+			return;
+		}
+		if (closed) {
+			invalid = true;
+			return;
+		}
+		if (start === -1) start = index;
+		end = index;
+	});
+	if (invalid) return { kind: "invalid" };
+	if (start === -1 && end === -1) return { kind: "none" };
+	return {
+		end,
+		kind: "range",
+		start
+	};
+};
+const removeAIPreviewAnchor = (editor) => {
+	const aiChatType = getPluginType(editor, KEYS.aiChat);
+	editor.tf.removeNodes({
+		at: [],
+		match: (node) => ElementApi.isElement(node) && node.type === aiChatType
+	});
+};
+const restoreAIPreviewSelection = (editor, selection) => {
+	if (selection) {
+		editor.tf.select(cloneDeep(selection));
+		return;
+	}
+	editor.tf.deselect();
+};
+const removePreviewRange = (editor, range) => {
+	for (let index = range.end; index >= range.start; index--) editor.tf.removeNodes({ at: [index] });
+};
+const replacePreviewRange = (editor, range, blocks) => {
+	removePreviewRange(editor, range);
+	if (blocks.length === 0) return;
+	editor.tf.insertNodes(cloneDeep(blocks), { at: [range.start] });
+};
+const cloneAcceptedPreviewBlocks = (editor, range) => {
+	const aiType = getPluginType(editor, KEYS.ai);
+	const blocks = cloneDeep(editor.children.slice(range.start, range.end + 1));
+	const stripNode = (node) => {
+		if (ElementApi.isElement(node)) {
+			const { [AI_PREVIEW_KEY]: _preview, children, ...rest$1 } = node;
+			return {
+				...rest$1,
+				children: children.map(stripNode)
+			};
+		}
+		const { [aiType]: _ai, ...rest } = node;
+		return rest;
+	};
+	return blocks.map(stripNode);
+};
+const beginAIPreview = (editor, { originalBlocks = [] } = {}) => {
+	if (getAIPreview(editor)) return false;
+	AI_STREAM_SNAPSHOT.set(editor, {
+		originalBlocks: cloneDeep(originalBlocks),
+		selectionBefore: cloneDeep(editor.selection)
+	});
+	return true;
+};
+const hasAIPreview = (editor) => !!getAIPreview(editor);
+const cancelAIPreview = (editor) => {
+	const preview = getAIPreview(editor);
+	if (!preview) return false;
+	const range = getAIPreviewRange(editor);
+	if (range.kind === "invalid") return false;
+	editor.tf.withoutSaving(() => {
+		if (range.kind === "range") replacePreviewRange(editor, range, preview.originalBlocks);
+		removeAIPreviewAnchor(editor);
+		restoreAIPreviewSelection(editor, preview.selectionBefore);
+	});
+	clearAIPreview(editor);
+	return true;
+};
+const discardAIPreview = (editor) => {
+	if (!getAIPreview(editor)) return false;
+	clearAIPreview(editor);
+	return true;
+};
+const acceptAIPreview = (editor, _value) => {
+	const preview = getAIPreview(editor);
+	if (!preview) return false;
+	const range = getAIPreviewRange(editor);
+	if (range.kind === "invalid") return false;
+	if (range.kind === "range") {
+		const acceptedBlocks = cloneAcceptedPreviewBlocks(editor, range);
+		editor.tf.withoutSaving(() => {
+			replacePreviewRange(editor, range, preview.originalBlocks);
+			removeAIPreviewAnchor(editor);
+			restoreAIPreviewSelection(editor, preview.selectionBefore);
+		});
+		editor.tf.withNewBatch(() => {
+			if (preview.originalBlocks.length > 0) for (let index = preview.originalBlocks.length - 1; index >= 0; index--) editor.tf.removeNodes({ at: [range.start + index] });
+			if (acceptedBlocks.length > 0) editor.tf.insertNodes(acceptedBlocks, { at: [range.start] });
+		});
+		const lastBatch = editor.history?.undos.at(-1);
+		if (lastBatch) lastBatch.selectionBefore = cloneDeep(preview.selectionBefore);
+	} else editor.tf.withoutSaving(() => {
+		removeAIPreviewAnchor(editor);
+	});
+	clearAIPreview(editor);
+	return true;
+};
+
+//#endregion
+//#region src/lib/transforms/insertAINodes.ts
+const insertAINodes = (editor, nodes, { target } = {}) => {
+	if (!target && !editor.selection?.focus.path) return;
+	const aiNodes = nodes.map((node) => ({
+		...node,
+		ai: true
+	}));
+	editor.tf.withoutNormalizing(() => {
+		editor.tf.insertNodes(aiNodes, {
+			at: editor.api.end(target || editor.selection.focus.path),
+			select: true
+		});
+		editor.tf.collapse({ edge: "end" });
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/removeAIMarks.ts
+const removeAIMarks = (editor, { at = [] } = {}) => {
+	const nodeType = getPluginType(editor, KEYS.ai);
+	editor.tf.unsetNodes(nodeType, {
+		at,
+		match: (n) => n[nodeType]
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/removeAINodes.ts
+const removeAINodes = (editor, { at = [] } = {}) => {
+	editor.tf.removeNodes({
+		at,
+		match: (n) => TextApi.isText(n) && !!n.ai
+	});
+};
+
+//#endregion
+//#region src/lib/transforms/undoAI.ts
+const undoAI = (editor) => {
+	if (hasAIPreview(editor) && cancelAIPreview(editor)) return;
+	const hasAINodeOrAISuggestion = editor.api.some({
+		at: [],
+		match: (n) => !!n.ai
+	}) || editor.api.some({
+		at: [],
+		match: (n) => !!n[getTransientSuggestionKey()]
+	});
+	if (editor.history.undos.at(-1)?.ai && hasAINodeOrAISuggestion) {
+		editor.undo();
+		editor.history.redos.pop();
+		return;
+	}
+	if (hasAINodeOrAISuggestion) cancelAIPreview(editor);
+};
+
+//#endregion
+//#region src/lib/transforms/withAIBatch.ts
+const withAIBatch = (editor, fn, { split } = {}) => {
+	if (split) editor.tf.withNewBatch(fn);
+	else editor.tf.withMerging(fn);
+	const lastBatch = editor.history.undos?.at(-1);
+	if (lastBatch) lastBatch.ai = true;
+};
+
+//#endregion
+//#region src/lib/BaseAIPlugin.ts
+const getAITransforms = (editor) => ({
+	acceptPreview: bindFirst(acceptAIPreview, editor),
+	beginPreview: bindFirst(beginAIPreview, editor),
+	cancelPreview: bindFirst(cancelAIPreview, editor),
+	discardPreview: bindFirst(discardAIPreview, editor),
+	hasPreview: bindFirst(hasAIPreview, editor),
+	insertNodes: bindFirst(insertAINodes, editor),
+	removeMarks: bindFirst(removeAIMarks, editor),
+	removeNodes: bindFirst(removeAINodes, editor),
+	undo: bindFirst(undoAI, editor)
+});
+const BaseAIPlugin = createTSlatePlugin({
+	key: KEYS.ai,
+	node: {
+		isDecoration: false,
+		isLeaf: true
+	}
+}).extendTransforms(({ editor }) => getAITransforms(editor)).extendEditorTransforms(({ editor }) => ({ ai: getAITransforms(editor) }));
+
+//#endregion
+//#region src/lib/utils/getEditorPrompt.ts
+const createPromptFromConfig = (config, params) => {
+	const { isBlockSelecting, isSelecting: isSelecting$1 } = params;
+	if (isBlockSelecting && config.blockSelecting) return config.blockSelecting ?? config.default;
+	if (isSelecting$1 && config.selecting) return config.selecting ?? config.default;
+	return config.default;
+};
+const getEditorPrompt = (editor, { prompt = "" }) => {
+	const params = {
+		editor,
+		isBlockSelecting: editor.getOption({ key: KEYS.blockSelection }, "isSelectingSome"),
+		isSelecting: isSelecting(editor)
+	};
+	let promptText = "";
+	if (typeof prompt === "function") promptText = prompt(params);
+	else if (typeof prompt === "object") promptText = createPromptFromConfig(prompt, params);
+	else promptText = prompt;
+	return promptText;
+};
+
+//#endregion
+export { removeAINodes as a, AI_PREVIEW_KEY as c, cancelAIPreview as d, discardAIPreview as f, undoAI as i, acceptAIPreview as l, BaseAIPlugin as n, removeAIMarks as o, hasAIPreview as p, withAIBatch as r, insertAINodes as s, getEditorPrompt as t, beginAIPreview as u };

package/dist/index-D2hJDomP.d.ts

@@ -0,0 +1,137 @@
+import * as platejs0 from "platejs";
+import { Descendant, History, OmitFirst, Path, PluginConfig, SlateEditor, TLocation, Value } from "platejs";
+
+//#region src/lib/transforms/aiStreamSnapshot.d.ts
+type BeginAIPreviewOptions = {
+  originalBlocks?: Value;
+};
+declare const AI_PREVIEW_KEY = "aiPreview";
+declare const beginAIPreview: (editor: SlateEditor, {
+  originalBlocks
+}?: BeginAIPreviewOptions) => boolean;
+declare const hasAIPreview: (editor: SlateEditor) => boolean;
+declare const cancelAIPreview: (editor: SlateEditor) => boolean;
+declare const discardAIPreview: (editor: SlateEditor) => boolean;
+declare const acceptAIPreview: (editor: SlateEditor, _value?: Value) => boolean;
+//#endregion
+//#region src/lib/transforms/insertAINodes.d.ts
+declare const insertAINodes: (editor: SlateEditor, nodes: Descendant[], {
+  target
+}?: {
+  target?: Path;
+}) => void;
+//#endregion
+//#region src/lib/transforms/removeAIMarks.d.ts
+declare const removeAIMarks: (editor: SlateEditor, {
+  at
+}?: {
+  at?: TLocation;
+}) => void;
+//#endregion
+//#region src/lib/transforms/removeAINodes.d.ts
+declare const removeAINodes: (editor: SlateEditor, {
+  at
+}?: {
+  at?: Path;
+}) => void;
+//#endregion
+//#region src/lib/transforms/undoAI.d.ts
+declare const undoAI: (editor: SlateEditor) => void;
+//#endregion
+//#region src/lib/transforms/withAIBatch.d.ts
+type AIBatch = History['undos'][number] & {
+  ai?: boolean;
+};
+declare const withAIBatch: (editor: SlateEditor, fn: () => void, {
+  split
+}?: {
+  split?: boolean;
+}) => void;
+//#endregion
+//#region src/lib/BaseAIPlugin.d.ts
+type BaseAIPluginConfig = PluginConfig<'ai', {}, {}, {
+  ai: {
+    /** Commit the active preview as one fresh undoable batch. */
+    acceptPreview: OmitFirst<typeof acceptAIPreview>;
+    /** Capture the rollback slice and selection for AI preview. */
+    beginPreview: OmitFirst<typeof beginAIPreview>;
+    /** Restore the rollback point and clear active preview state. */
+    cancelPreview: OmitFirst<typeof cancelAIPreview>;
+    /** Clear active preview bookkeeping without restoring content. */
+    discardPreview: OmitFirst<typeof discardAIPreview>;
+    /** Report whether an AI preview rollback point is active. */
+    hasPreview: OmitFirst<typeof hasAIPreview>;
+    insertNodes: OmitFirst<typeof insertAINodes>;
+    removeMarks: OmitFirst<typeof removeAIMarks>;
+    removeNodes: OmitFirst<typeof removeAINodes>;
+    undo: OmitFirst<typeof undoAI>;
+  };
+}>;
+declare const BaseAIPlugin: platejs0.SlatePlugin<PluginConfig<"ai", {}, {}, {
+  ai: {
+    acceptPreview: ((_value?: platejs0.Value | undefined) => boolean) & ((_value?: platejs0.Value | undefined) => boolean);
+    beginPreview: ((args_0?: {
+      originalBlocks?: platejs0.Value;
+    } | undefined) => boolean) & ((args_0?: {
+      originalBlocks?: platejs0.Value;
+    } | undefined) => boolean);
+    cancelPreview: (() => boolean) & (() => boolean);
+    discardPreview: (() => boolean) & (() => boolean);
+    hasPreview: (() => boolean) & (() => boolean);
+    insertNodes: ((nodes: platejs0.Descendant[], args_1?: {
+      target?: platejs0.Path;
+    } | undefined) => void) & ((nodes: platejs0.Descendant[], args_1?: {
+      target?: platejs0.Path;
+    } | undefined) => void);
+    removeMarks: ((args_0?: {
+      at?: platejs0.TLocation;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs0.TLocation;
+    } | undefined) => void);
+    removeNodes: ((args_0?: {
+      at?: platejs0.Path;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs0.Path;
+    } | undefined) => void);
+    undo: (() => void) & (() => void);
+  };
+}, {}>>;
+//#endregion
+//#region src/lib/types.d.ts
+type AIToolName = 'comment' | 'edit' | 'generate' | null;
+type AIMode = 'chat' | 'insert';
+//#endregion
+//#region src/lib/utils/getEditorPrompt.d.ts
+type EditorPrompt = ((params: EditorPromptParams) => string) | PromptConfig | string;
+type EditorPromptParams = {
+  editor: SlateEditor;
+  isBlockSelecting: boolean;
+  isSelecting: boolean;
+};
+type PromptConfig = {
+  default: string;
+  blockSelecting?: string;
+  selecting?: string;
+};
+declare const getEditorPrompt: (editor: SlateEditor, {
+  prompt
+}: {
+  prompt?: EditorPrompt;
+}) => string;
+//#endregion
+//#region src/lib/utils/replacePlaceholders.d.ts
+type MarkdownType = 'block' | 'blockSelection' | 'blockSelectionWithBlockId' | 'blockWithBlockId' | 'editor' | 'editorWithBlockId' | 'tableCellWithId';
+declare const replacePlaceholders: (editor: SlateEditor, text: string, {
+  prompt
+}?: {
+  prompt?: string;
+}) => string;
+//#endregion
+//#region src/lib/utils/getMarkdown.d.ts
+declare const getMarkdown: (editor: SlateEditor, {
+  type
+}: {
+  type: MarkdownType;
+}) => string;
+//#endregion
+export { hasAIPreview as C, discardAIPreview as S, insertAINodes as _, EditorPromptParams as a, beginAIPreview as b, AIMode as c, BaseAIPluginConfig as d, AIBatch as f, removeAIMarks as g, removeAINodes as h, EditorPrompt as i, AIToolName as l, undoAI as m, MarkdownType as n, PromptConfig as o, withAIBatch as p, replacePlaceholders as r, getEditorPrompt as s, getMarkdown as t, BaseAIPlugin as u, AI_PREVIEW_KEY as v, cancelAIPreview as x, acceptAIPreview as y };

package/dist/index.d.ts

@@ -1,2 +1,2 @@
-import { _ as insertAINodes, a as EditorPromptParams, c as AIMode, d as BaseAIPluginConfig, f as AIBatch, g as removeAIMarks, h as removeAINodes, i as EditorPrompt, l as AIToolName, m as undoAI, n as MarkdownType, o as PromptConfig, p as withAIBatch, r as replacePlaceholders, s as getEditorPrompt, t as getMarkdown, u as BaseAIPlugin } from "./index-B_bqJoKL";
-export { AIBatch, AIMode, AIToolName, BaseAIPlugin, BaseAIPluginConfig, EditorPrompt, EditorPromptParams, MarkdownType, PromptConfig, getEditorPrompt, getMarkdown, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };
+import { C as hasAIPreview, S as discardAIPreview, _ as insertAINodes, a as EditorPromptParams, b as beginAIPreview, c as AIMode, d as BaseAIPluginConfig, f as AIBatch, g as removeAIMarks, h as removeAINodes, i as EditorPrompt, l as AIToolName, m as undoAI, n as MarkdownType, o as PromptConfig, p as withAIBatch, r as replacePlaceholders, s as getEditorPrompt, t as getMarkdown, u as BaseAIPlugin, v as AI_PREVIEW_KEY, x as cancelAIPreview, y as acceptAIPreview } from "./index-D2hJDomP";
+export { AIBatch, AIMode, AIToolName, AI_PREVIEW_KEY, BaseAIPlugin, BaseAIPluginConfig, EditorPrompt, EditorPromptParams, MarkdownType, PromptConfig, acceptAIPreview, beginAIPreview, cancelAIPreview, discardAIPreview, getEditorPrompt, getMarkdown, hasAIPreview, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { a as removeAINodes, i as undoAI, n as BaseAIPlugin, o as removeAIMarks, r as withAIBatch, s as insertAINodes, t as getEditorPrompt } from "./getEditorPrompt-BnVrgUl3.js";
+import { a as removeAINodes, c as AI_PREVIEW_KEY, d as cancelAIPreview, f as discardAIPreview, i as undoAI, l as acceptAIPreview, n as BaseAIPlugin, o as removeAIMarks, p as hasAIPreview, r as withAIBatch, s as insertAINodes, t as getEditorPrompt, u as beginAIPreview } from "./getEditorPrompt-axDm9Fg4.js";
 import { KEYS } from "platejs";
 import { serializeMd } from "@platejs/markdown";
 import { getTableGridAbove } from "@platejs/table";
@@ -121,4 +121,4 @@ const replacePlaceholders = (editor, text, { prompt } = {}) => {
 };
 
 //#endregion
-export { BaseAIPlugin, getEditorPrompt, getMarkdown, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };
+export { AI_PREVIEW_KEY, BaseAIPlugin, acceptAIPreview, beginAIPreview, cancelAIPreview, discardAIPreview, getEditorPrompt, getMarkdown, hasAIPreview, insertAINodes, removeAIMarks, removeAINodes, replacePlaceholders, undoAI, withAIBatch };

package/dist/react/index.d.ts

@@ -1,6 +1,8 @@
-import { c as AIMode, d as BaseAIPluginConfig, i as EditorPrompt, l as AIToolName } from "../index-B_bqJoKL";
-import { Descendant, EditorNodesOptions, ExtendConfig, NodeEntry, OmitFirst, Path, PluginConfig, Range, RemoveNodesOptions, SlateEditor, TElement as TElement$1, TIdElement, TRange, TTableElement, TText } from "platejs";
+import { c as AIMode, d as BaseAIPluginConfig, i as EditorPrompt, l as AIToolName } from "../index-D2hJDomP";
+import * as platejs12 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 { OverrideEditor, PlateEditor, RenderNodeWrapperProps } from "platejs/react";
 import React from "react";
 import { ChatRequestOptions, UIMessage } from "ai";
@@ -9,7 +11,35 @@ import { TriggerComboboxPluginOptions } from "@platejs/combobox";
 
 //#region src/react/ai/AIPlugin.d.ts
 type AIPluginConfig = ExtendConfig<BaseAIPluginConfig>;
-declare const AIPlugin: any;
+declare const AIPlugin: platejs_react0.PlatePlugin<platejs12.PluginConfig<"ai", {}, {}, {
+  ai: {
+    acceptPreview: ((_value?: platejs12.Value | undefined) => boolean) & ((_value?: platejs12.Value | undefined) => boolean);
+    beginPreview: ((args_0?: {
+      originalBlocks?: platejs12.Value;
+    } | undefined) => boolean) & ((args_0?: {
+      originalBlocks?: platejs12.Value;
+    } | undefined) => boolean);
+    cancelPreview: (() => boolean) & (() => boolean);
+    discardPreview: (() => boolean) & (() => boolean);
+    hasPreview: (() => boolean) & (() => boolean);
+    insertNodes: ((nodes: platejs12.Descendant[], args_1?: {
+      target?: platejs12.Path;
+    } | undefined) => void) & ((nodes: platejs12.Descendant[], args_1?: {
+      target?: platejs12.Path;
+    } | undefined) => void);
+    removeMarks: ((args_0?: {
+      at?: platejs12.TLocation;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs12.TLocation;
+    } | undefined) => void);
+    removeNodes: ((args_0?: {
+      at?: platejs12.Path;
+    } | undefined) => void) & ((args_0?: {
+      at?: platejs12.Path;
+    } | undefined) => void);
+    undo: (() => void) & (() => void);
+  };
+}, {}>>;
 //#endregion
 //#region src/react/ai-chat/internal/types.d.ts
 type ToolName = 'comment' | 'edit' | 'generate';
@@ -58,10 +88,10 @@ declare const createFormattedBlocks: ({
   format,
   sourceBlock
 }: {
-  blocks: TElement$1[];
+  blocks: TElement[];
   format: "all" | "none" | "single";
   sourceBlock: NodeEntry;
-}) => TElement$1[] | null;
+}) => any[] | null;
 declare const replaceSelectionAIChat: (editor: PlateEditor, sourceEditor: SlateEditor, {
   format
 }?: {
@@ -138,7 +168,87 @@ type AIChatPluginConfig = PluginConfig<'aiChat', {
     removeAnchor: (options?: EditorNodesOptions) => void;
   };
 }>;
-declare const AIChatPlugin: any;
+declare const AIChatPlugin: platejs_react0.PlatePlugin<PluginConfig<"aiChat", {
+  _blockChunks: string;
+  _blockPath: Path | null;
+  /** @private Using For streamInsertChunk */
+  _mdxName: string | null;
+  _replaceIds: string[];
+  /** @private The Editor used to generate the AI response. */
+  aiEditor: SlateEditor | null;
+  chat: UseChatHelpers<ChatMessage>;
+  chatNodes: TIdElement[];
+  chatSelection: TRange | null;
+  /** @deprecated Use api.aiChat.node({streaming:true}) instead */
+  experimental_lastTextId: string | null;
+  /**
+   * Specifies how the assistant message is handled:
+   *
+   * - 'insert': Directly inserts content into the editor without preview.
+   * - 'chat': Initiates an interactive session to review and refine content
+   *   before insertion.
+   */
+  mode: AIMode;
+  open: boolean;
+  /** Whether the AI response is currently streaming. Cursor mode only. */
+  streaming: boolean;
+  toolName: AIToolName;
+} & TriggerComboboxPluginOptions, {
+  aiChat: {
+    reset: OmitFirst<typeof resetAIChat>;
+    submit: OmitFirst<typeof submitAIChat>;
+    hide: (options?: {
+      focus?: boolean;
+      undo?: boolean;
+    }) => void;
+    node: (options?: EditorNodesOptions & {
+      anchor?: boolean;
+      streaming?: boolean;
+    }) => NodeEntry | undefined;
+    reload: () => void;
+    show: () => void;
+    stop: () => void;
+  };
+} & Record<"aiChat", Pick<{
+  reset: OmitFirst<typeof resetAIChat>;
+  submit: OmitFirst<typeof submitAIChat>;
+  hide: (options?: {
+    focus?: boolean;
+    undo?: boolean;
+  }) => void;
+  node: (options?: EditorNodesOptions & {
+    anchor?: boolean;
+    streaming?: boolean;
+  }) => NodeEntry | undefined;
+  reload: () => void;
+  show: () => void;
+  stop: () => void;
+}, "node" | "reset" | "stop" | "submit">> & Record<"aiChat", {
+  hide: ({
+    focus,
+    undo
+  }?: {
+    focus?: boolean;
+    undo?: boolean;
+  }) => void;
+  show: () => void;
+}>, {
+  aiChat: {
+    accept: OmitFirst<typeof acceptAIChat>;
+    insertBelow: OmitFirst<typeof insertBelowAIChat>;
+    replaceSelection: OmitFirst<typeof replaceSelectionAIChat>;
+    removeAnchor: (options?: EditorNodesOptions) => void;
+  };
+} & Record<"aiChat", {
+  accept: () => void;
+  insertBelow: (sourceEditor: SlateEditor, args_1?: {
+    format?: "all" | "none" | "single";
+  } | undefined) => void;
+  removeAnchor: (options?: platejs12.RemoveNodesOptions | undefined) => void;
+  replaceSelection: (sourceEditor: SlateEditor, args_1?: {
+    format?: "all" | "none" | "single";
+  } | undefined) => void;
+}>, {}>>;
 //#endregion
 //#region src/react/ai-chat/withAIChat.d.ts
 declare const withAIChat: OverrideEditor<AIChatPluginConfig>;
@@ -152,7 +262,7 @@ declare const withAIChat: OverrideEditor<AIChatPluginConfig>;
  */
 declare const useAIChatEditor: (editor: SlateEditor, content: string, {
   parser
-}?: DeserializeMdOptions) => any;
+}?: DeserializeMdOptions) => platejs12.Value;
 //#endregion
 //#region src/react/ai-chat/hooks/useChatChunk.d.ts
 declare const useChatChunk: ({
@@ -188,19 +298,31 @@ declare const useEditorChat: ({
 }: UseEditorChatOptions) => void;
 //#endregion
 //#region src/react/ai-chat/streaming/streamDeserializeInlineMd.d.ts
-declare const streamDeserializeInlineMd: (editor: PlateEditor, text: string, options?: DeserializeMdOptions) => any;
+declare const streamDeserializeInlineMd: (editor: PlateEditor, text: string, options?: DeserializeMdOptions) => platejs12.Descendant[] | {
+  text: string;
+}[];
 //#endregion
 //#region src/react/ai-chat/streaming/streamDeserializeMd.d.ts
-declare const streamDeserializeMd: (editor: PlateEditor, data: string, options?: DeserializeMdOptions) => TElement[];
+declare const streamDeserializeMd: (editor: PlateEditor, data: string, options?: DeserializeMdOptions) => TElement[] | {
+  children: {
+    text: string;
+  }[];
+  type: string;
+}[];
 //#endregion
 //#region src/react/ai-chat/streaming/streamInsertChunk.d.ts
 type SteamInsertChunkOptions = {
   elementProps?: any;
   textProps?: any;
 };
+declare const getInsertPreviewStart: (editor: SlateEditor) => {
+  path: Path;
+  startBlock: platejs12.TElement | platejs12.TText | undefined;
+  startInEmptyParagraph: boolean;
+};
 /** @experimental */
 declare function streamInsertChunk(editor: PlateEditor, chunk: string, options?: SteamInsertChunkOptions): void;
-declare const getCurrentBlockPath: (editor: SlateEditor) => any;
+declare const getCurrentBlockPath: (editor: SlateEditor) => Path;
 //#endregion
 //#region src/react/ai-chat/streaming/streamSerializeMd.d.ts
 declare const streamSerializeMd: (editor: PlateEditor, options: SerializeMdOptions, chunk: string) => string;
@@ -209,10 +331,10 @@ declare const streamSerializeMd: (editor: PlateEditor, options: SerializeMdOptio
 declare const escapeInput: (data: string) => string;
 //#endregion
 //#region src/react/ai-chat/streaming/utils/getListNode.d.ts
-declare const getListNode: (editor: PlateEditor, node: TElement$1) => TElement$1;
+declare const getListNode: (editor: PlateEditor, node: TElement) => TElement;
 //#endregion
 //#region src/react/ai-chat/streaming/utils/isSameNode.d.ts
-declare const isSameNode: (editor: PlateEditor, node1: TElement$1 | TText, node2: TElement$1 | TText) => boolean;
+declare const isSameNode: (editor: PlateEditor, node1: TElement | TText, node2: TElement | TText) => boolean;
 //#endregion
 //#region src/react/ai-chat/streaming/utils/nodesWithProps.d.ts
 declare const nodesWithProps: (editor: PlateEditor, nodes: Descendant[], options: SteamInsertChunkOptions) => Descendant[];
@@ -250,8 +372,8 @@ type TableCellUpdate = {
 declare const applyTableCellSuggestion: (editor: SlateEditor, cellUpdate: TableCellUpdate) => void;
 //#endregion
 //#region src/react/ai-chat/utils/getLastAssistantMessage.d.ts
-declare function getLastAssistantMessage(editor: PlateEditor): any;
-declare function useLastAssistantMessage(): any;
+declare function getLastAssistantMessage(editor: PlateEditor): ChatMessage | undefined;
+declare function useLastAssistantMessage(): ChatMessage | undefined;
 //#endregion
 //#region src/react/ai-chat/utils/nestedContainerUtils.d.ts
 /** Check if nodes is a single table with single cell */
@@ -388,7 +510,86 @@ type CompletionState = {
   error?: Error | null;
   isLoading?: boolean;
 };
-declare const CopilotPlugin: any;
+declare const CopilotPlugin: platejs_react0.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}
+   */
+  completeOptions?: Partial<CompleteOptions>;
+  /**
+   * Debounce delay for auto triggering AI completion.
+   *
+   * @default 0
+   */
+  debounceDelay?: number;
+  /** Get the next word to be inserted. */
+  getNextWord?: GetNextWord;
+  /** Render the ghost text. */
+  renderGhostText?: (() => React.ReactNode) | null;
+  shouldAbort?: boolean;
+  /** The node id where the suggestion is located. */
+  suggestionNodeId?: string | null;
+  /** The text of the suggestion. */
+  suggestionText?: string | null;
+  /**
+   * Conditions to auto trigger copilot, used in addition to triggerQuery.
+   * Disabling defaults to:
+   *
+   * - Block above is empty
+   * - Block above ends with a space
+   * - There is already a suggestion
+   */
+  autoTriggerQuery?: (options: {
+    editor: PlateEditor;
+  }) => boolean;
+  /**
+   * Get the prompt for AI completion.
+   *
+   * @default serializeMdNodes(editor.api.block({ highest: true }))
+   */
+  getPrompt?: (options: {
+    editor: PlateEditor;
+  }) => string;
+  /**
+   * Conditions to trigger copilot. Disabling defaults to:
+   *
+   * - Selection is expanded
+   * - Selection is not at the end of block
+   */
+  triggerQuery?: (options: {
+    editor: PlateEditor;
+  }) => boolean;
+}, {
+  copilot: {
+    triggerSuggestion: OmitFirst<typeof triggerCopilotSuggestion>;
+    reject: () => false | undefined;
+    setBlockSuggestion: (options: {
+      text: string;
+      id?: string;
+    }) => void;
+    stop: () => void;
+  };
+} & Record<"copilot", Omit<{
+  triggerSuggestion: OmitFirst<typeof triggerCopilotSuggestion>;
+  reject: () => false | undefined;
+  setBlockSuggestion: (options: {
+    text: string;
+    id?: string;
+  }) => void;
+  stop: () => void;
+}, "reject">> & Record<"copilot", {
+  reject: () => false | undefined;
+}>, {
+  copilot: {
+    accept: OmitFirst<typeof acceptCopilot>;
+    acceptNextWord: OmitFirst<typeof acceptCopilotNextWord>;
+  };
+} & Record<"copilot", {
+  accept: () => false | undefined;
+  acceptNextWord: () => false | undefined;
+}>, {
+  isSuggested?: (id: string) => boolean;
+}>>;
 //#endregion
 //#region src/react/copilot/renderCopilotBelowNodes.d.ts
 declare const renderCopilotBelowNodes: ({
@@ -405,4 +606,4 @@ declare const withCopilot: OverrideEditor<CopilotPluginConfig>;
 //#region src/react/copilot/utils/withoutAbort.d.ts
 declare const withoutAbort: (editor: PlateEditor, fn: () => void) => void;
 //#endregion
-export { AIChatPlugin, AIChatPluginConfig, AIPlugin, AIPluginConfig, CallCompletionApiOptions, CompleteOptions, CopilotPlugin, CopilotPluginConfig, GetNextWord, SteamInsertChunkOptions, TableCellUpdate, UseEditorChatOptions, acceptAIChat, acceptAISuggestions, acceptCopilot, acceptCopilotNextWord, aiCommentToRange, applyAISuggestions, applyTableCellSuggestion, callCompletionApi, createFormattedBlocks, escapeInput, findTextRangeInBlock, getChunkTrimmed, getCurrentBlockPath, getLastAssistantMessage, getListNode, getNextWord, getTableCellChildren, insertBelowAIChat, insertBelowGenerate, isCompleteCodeBlock, isCompleteMath, isSameNode, isSingleCellTable, nodesWithProps, rejectAISuggestions, removeAnchorAIChat, renderCopilotBelowNodes, replaceSelectionAIChat, resetAIChat, streamDeserializeInlineMd, streamDeserializeMd, streamInsertChunk, streamSerializeMd, submitAIChat, triggerCopilotSuggestion, useAIChatEditor, useChatChunk, useEditorChat, useLastAssistantMessage, withAIChat, withCopilot, withTransient, withoutAbort, withoutSuggestionAndComments };
+export { AIChatPlugin, AIChatPluginConfig, AIPlugin, AIPluginConfig, CallCompletionApiOptions, CompleteOptions, CopilotPlugin, CopilotPluginConfig, GetNextWord, SteamInsertChunkOptions, TableCellUpdate, UseEditorChatOptions, acceptAIChat, acceptAISuggestions, acceptCopilot, acceptCopilotNextWord, aiCommentToRange, applyAISuggestions, applyTableCellSuggestion, callCompletionApi, createFormattedBlocks, escapeInput, findTextRangeInBlock, getChunkTrimmed, getCurrentBlockPath, getInsertPreviewStart, getLastAssistantMessage, getListNode, getNextWord, getTableCellChildren, insertBelowAIChat, insertBelowGenerate, isCompleteCodeBlock, isCompleteMath, isSameNode, isSingleCellTable, nodesWithProps, rejectAISuggestions, removeAnchorAIChat, renderCopilotBelowNodes, replaceSelectionAIChat, resetAIChat, streamDeserializeInlineMd, streamDeserializeMd, streamInsertChunk, streamSerializeMd, submitAIChat, triggerCopilotSuggestion, useAIChatEditor, useChatChunk, useEditorChat, useLastAssistantMessage, withAIChat, withCopilot, withTransient, withoutAbort, withoutSuggestionAndComments };

package/dist/react/index.js

@@ -1,5 +1,6 @@
-import { n as BaseAIPlugin, r as withAIBatch, t as getEditorPrompt } from "../getEditorPrompt-BnVrgUl3.js";
+import { c as AI_PREVIEW_KEY, n as BaseAIPlugin, p as hasAIPreview, r as withAIBatch, t as getEditorPrompt } from "../getEditorPrompt-axDm9Fg4.js";
 import { ElementApi, KEYS, NodeApi, PathApi, RangeApi, TextApi, bindFirst, getPluginKey, getPluginType, isDefined, nanoid } from "platejs";
+import cloneDeep from "lodash/cloneDeep.js";
 import { SkipSuggestionDeletes, acceptSuggestion, diffToSuggestions, getSuggestionKey, getTransientSuggestionKey, rejectSuggestion } from "@platejs/suggestion";
 import { isSelecting } from "@platejs/selection";
 import { MarkdownPlugin, deserializeInlineMd, deserializeMd, serializeInlineMd, serializeMd } from "@platejs/markdown";
@@ -7,7 +8,6 @@ import { createTPlatePlugin, getEditorPlugin, toPlatePlugin, useEditorPlugin, us
 import { distance } from "fastest-levenshtein";
 import { BlockSelectionPlugin, removeBlockSelectionNodes } from "@platejs/selection/react";
 import { SuggestionPlugin } from "@platejs/suggestion/react";
-import cloneDeep from "lodash/cloneDeep.js";
 import { c } from "react-compiler-runtime";
 import React, { useEffect, useMemo, useRef } from "react";
 import debounce from "lodash/debounce.js";
@@ -168,23 +168,31 @@ const acceptAISuggestions = (editor) => {
 
 //#endregion
 //#region src/react/ai-chat/transforms/acceptAIChat.ts
+const getAcceptedInsertFocusPoint = (editor) => {
+	let endIndex = null;
+	editor.children.forEach((node, index) => {
+		if (ElementApi.isElement(node) && node[AI_PREVIEW_KEY]) endIndex = index;
+	});
+	if (endIndex === null) return null;
+	return editor.api.end([endIndex]) ?? null;
+};
 const acceptAIChat = (editor) => {
 	const mode = editor.getOption(AIChatPlugin, "mode");
 	if (mode === "insert") {
-		const { tf } = getEditorPlugin(editor, AIPlugin);
+		const ai = editor.getTransforms(BaseAIPlugin).ai;
 		const api = editor.getApi({ key: KEYS.ai });
-		const lastAINodePath = api.aiChat.node({
-			at: [],
-			reverse: true
-		})[1];
-		withAIBatch(editor, () => {
-			tf.ai.removeMarks();
+		const focusPoint = getAcceptedInsertFocusPoint(editor);
+		if (!ai.acceptPreview()) withAIBatch(editor, () => {
+			editor.tf.unsetNodes(AI_PREVIEW_KEY, {
+				at: [],
+				match: (node) => ElementApi.isElement(node) && !!node[AI_PREVIEW_KEY]
+			});
+			ai.removeMarks();
 			editor.getTransforms(AIChatPlugin).aiChat.removeAnchor();
 		});
 		api.aiChat.hide();
 		editor.tf.focus();
-		const focusPoint = editor.api.end(lastAINodePath);
-		editor.tf.setSelection({
+		if (focusPoint) editor.tf.select({
 			anchor: focusPoint,
 			focus: focusPoint
 		});
@@ -389,7 +397,8 @@ const resetAIChat = (editor, { undo = true } = {}) => {
 		mode: "insert",
 		toolName: null
 	});
-	if (undo) editor.getTransforms(AIPlugin).ai.undo();
+	if (undo) editor.getTransforms(BaseAIPlugin).ai.undo();
+	else editor.getTransforms(BaseAIPlugin).ai.discardPreview();
 };
 
 //#endregion
@@ -401,7 +410,7 @@ const submitAIChat = (editor, input, { mode, options, prompt, toolName: toolName
 	if (!prompt && input?.length === 0) return;
 	if (!prompt) prompt = input;
 	if (!mode) mode = isSelecting(editor) ? "chat" : "insert";
-	if (mode === "insert") editor.getTransforms(AIPlugin).ai.undo();
+	if (mode === "insert") editor.getTransforms(BaseAIPlugin).ai.undo();
 	setOption("mode", mode);
 	setOption("toolName", toolName);
 	const blocks = editor.getApi(BlockSelectionPlugin).blockSelection.getNodes();
@@ -517,7 +526,7 @@ const insertBelowAIChat = (editor, sourceEditor, { format = "single" } = {}) =>
 	if (toolName === "generate") return insertBelowGenerate(editor, sourceEditor, { format });
 	const selectedBlocks = editor.getApi(BlockSelectionPlugin).blockSelection.getNodes();
 	const selectedIds = editor.getOptions(BlockSelectionPlugin).selectedIds;
-	editor.getTransforms(AIPlugin).ai.undo();
+	editor.getTransforms(BaseAIPlugin).ai.undo();
 	const insertBlocksAndSelect = editor.getTransforms(BlockSelectionPlugin).blockSelection.insertBlocksAndSelect;
 	if (!selectedIds || selectedIds.size === 0) return;
 	const lastBlock = selectedBlocks.at(-1);
@@ -594,7 +603,7 @@ const removeAnchorAIChat = (editor, options) => {
 //#endregion
 //#region src/react/ai-chat/withAIChat.ts
 const withAIChat = ({ api, editor, getOptions, tf: { insertText, normalizeNode }, type }) => {
-	const tf = editor.getTransforms(AIPlugin);
+	const ai = editor.getTransforms(BaseAIPlugin).ai;
 	const matchesTrigger = (text) => {
 		const { trigger } = getOptions();
 		if (trigger instanceof RegExp) return trigger.test(text);
@@ -619,7 +628,7 @@ const withAIChat = ({ api, editor, getOptions, tf: { insertText, normalizeNode }
 		normalizeNode(entry) {
 			const [node, path] = entry;
 			if (node[KEYS.ai] && !getOptions().open) {
-				tf.ai.removeMarks({ at: path });
+				ai.removeMarks({ at: path });
 				return;
 			}
 			if (ElementApi.isElement(node) && node.type === type && !getOptions().open) {
@@ -683,7 +692,7 @@ const AIChatPlugin = createTPlatePlugin({
 	},
 	reload: () => {
 		const { chat, chatNodes, chatSelection } = getOptions();
-		editor.getTransforms(AIPlugin).ai.undo();
+		editor.getTransforms(BaseAIPlugin).ai.undo();
 		if (chatSelection) editor.tf.setSelection(chatSelection);
 		else editor.getApi(BlockSelectionPlugin).blockSelection.set(chatNodes.map((node) => node.id));
 		const blocks = editor.getApi(BlockSelectionPlugin).blockSelection.getNodes();
@@ -995,17 +1004,35 @@ const getNextPath = (path, length) => {
 	for (let i = 0; i < length; i++) result = PathApi.next(result);
 	return result;
 };
+const withPreviewElementProps = (editor, options) => {
+	if (!hasAIPreview(editor)) return options;
+	return {
+		...options,
+		elementProps: {
+			...options.elementProps,
+			[AI_PREVIEW_KEY]: true
+		}
+	};
+};
+const getInsertPreviewStart = (editor) => {
+	const path = getCurrentBlockPath(editor);
+	const startBlock = editor.api.node(path)?.[0];
+	return {
+		path,
+		startBlock,
+		startInEmptyParagraph: !!startBlock && NodeApi.string(startBlock).length === 0 && startBlock.type === getPluginType(editor, KEYS.p)
+	};
+};
 /** @experimental */
 function streamInsertChunk(editor, chunk, options = {}) {
+	const insertOptions = withPreviewElementProps(editor, options);
 	const { _blockChunks, _blockPath } = editor.getOptions(AIChatPlugin);
 	if (_blockPath === null) {
 		const blocks = streamDeserializeMd(editor, chunk);
-		const path = getCurrentBlockPath(editor);
-		const startBlock = editor.api.node(path)[0];
-		const startInEmptyParagraph = NodeApi.string(startBlock).length === 0 && startBlock.type === getPluginType(editor, KEYS.p);
+		const { path, startInEmptyParagraph } = getInsertPreviewStart(editor);
 		if (startInEmptyParagraph) editor.tf.removeNodes({ at: path });
 		if (blocks.length > 0) {
-			editor.tf.insertNodes(nodesWithProps(editor, [blocks[0]], options), {
+			editor.tf.insertNodes(nodesWithProps(editor, [blocks[0]], insertOptions), {
 				at: path,
 				nextBlock: !startInEmptyParagraph,
 				select: true
@@ -1015,7 +1042,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
 			if (blocks.length > 1) {
 				const nextBlocks = blocks.slice(1);
 				const nextPath = getCurrentBlockPath(editor);
-				editor.tf.insertNodes(nodesWithProps(editor, nextBlocks, options), {
+				editor.tf.insertNodes(nodesWithProps(editor, nextBlocks, insertOptions), {
 					at: nextPath,
 					nextBlock: true,
 					select: true
@@ -1034,7 +1061,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
 			const currentBlock = editor.api.node(_blockPath)[0];
 			if (isSameNode(editor, currentBlock, tempBlocks[0])) {
 				const chunkNodes = streamDeserializeInlineMd(editor, chunk);
-				editor.tf.insertNodes(nodesWithProps(editor, chunkNodes, options), {
+				editor.tf.insertNodes(nodesWithProps(editor, chunkNodes, insertOptions), {
 					at: editor.api.end(_blockPath),
 					select: true
 				});
@@ -1042,7 +1069,7 @@ function streamInsertChunk(editor, chunk, options = {}) {
 				const blockText = NodeApi.string(tempBlocks[0]);
 				if (serializedBlock === tempBlockChunks && blockText === serializedBlock) editor.setOption(AIChatPlugin, "_blockChunks", tempBlockChunks);
 				else {
-					editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], options), {
+					editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], insertOptions), {
 						at: _blockPath,
 						select: true
 					});
@@ -1051,20 +1078,20 @@ function streamInsertChunk(editor, chunk, options = {}) {
 				}
 			} else {
 				const serializedBlock = streamSerializeMd(editor, { value: [tempBlocks[0]] }, tempBlockChunks);
-				editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], options), {
+				editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], insertOptions), {
 					at: _blockPath,
 					select: true
 				});
 				editor.setOption(AIChatPlugin, "_blockChunks", serializedBlock);
 			}
 		} else {
-			editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], options), {
+			editor.tf.replaceNodes(nodesWithProps(editor, [tempBlocks[0]], insertOptions), {
 				at: _blockPath,
 				select: true
 			});
 			if (tempBlocks.length > 1) {
 				const newEndBlockPath = getNextPath(_blockPath, tempBlocks.length - 1);
-				editor.tf.insertNodes(nodesWithProps(editor, tempBlocks.slice(1), options), {
+				editor.tf.insertNodes(nodesWithProps(editor, tempBlocks.slice(1), insertOptions), {
 					at: PathApi.next(_blockPath),
 					select: true
 				});
@@ -1400,4 +1427,4 @@ const CopilotPlugin = createTPlatePlugin({
 });
 
 //#endregion
-export { AIChatPlugin, AIPlugin, CopilotPlugin, acceptAIChat, acceptAISuggestions, acceptCopilot, acceptCopilotNextWord, aiCommentToRange, applyAISuggestions, applyTableCellSuggestion, callCompletionApi, createFormattedBlocks, escapeInput, findTextRangeInBlock, getChunkTrimmed, getCurrentBlockPath, getLastAssistantMessage, getListNode, getNextWord, getTableCellChildren, insertBelowAIChat, insertBelowGenerate, isCompleteCodeBlock, isCompleteMath, isSameNode, isSingleCellTable, nodesWithProps, rejectAISuggestions, removeAnchorAIChat, renderCopilotBelowNodes, replaceSelectionAIChat, resetAIChat, streamDeserializeInlineMd, streamDeserializeMd, streamInsertChunk, streamSerializeMd, submitAIChat, triggerCopilotSuggestion, useAIChatEditor, useChatChunk, useEditorChat, useLastAssistantMessage, withAIChat, withCopilot, withTransient, withoutAbort, withoutSuggestionAndComments };
+export { AIChatPlugin, AIPlugin, CopilotPlugin, acceptAIChat, acceptAISuggestions, acceptCopilot, acceptCopilotNextWord, aiCommentToRange, applyAISuggestions, applyTableCellSuggestion, callCompletionApi, createFormattedBlocks, escapeInput, findTextRangeInBlock, getChunkTrimmed, getCurrentBlockPath, getInsertPreviewStart, getLastAssistantMessage, getListNode, getNextWord, getTableCellChildren, insertBelowAIChat, insertBelowGenerate, isCompleteCodeBlock, isCompleteMath, isSameNode, isSingleCellTable, nodesWithProps, rejectAISuggestions, removeAnchorAIChat, renderCopilotBelowNodes, replaceSelectionAIChat, resetAIChat, streamDeserializeInlineMd, streamDeserializeMd, streamInsertChunk, streamSerializeMd, submitAIChat, triggerCopilotSuggestion, useAIChatEditor, useChatChunk, useEditorChat, useLastAssistantMessage, withAIChat, withCopilot, withTransient, withoutAbort, withoutSuggestionAndComments };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@platejs/ai",
-  "version": "52.3.8",
+  "version": "52.3.12",
   "description": "Text AI plugin for Plate",
   "keywords": [
     "plate",
@@ -32,15 +32,16 @@
     "fastest-levenshtein": "1.0.16",
     "lodash": "^4.17.21",
     "react-compiler-runtime": "^1.0.0",
-    "@platejs/markdown": "^52.3.7",
-    "@platejs/selection": "^52.3.6",
-    "@platejs/suggestion": "^52.3.8",
-    "@platejs/table": "^52.3.6"
+    "@platejs/markdown": "^52.3.10",
+    "@platejs/selection": "^52.3.10",
+    "@platejs/suggestion": "^52.3.10",
+    "@platejs/table": "^52.3.10"
   },
   "devDependencies": {
     "@ai-sdk/react": "^3.0.0",
     "ai": "^6.0.0",
-    "@plate/scripts": "1.0.0"
+    "@plate/scripts": "1.0.0",
+    "platejs": "^52.3.11"
   },
   "peerDependencies": {
     "platejs": ">=52.0.15",

package/dist/getEditorPrompt-BnVrgUl3.js

@@ -1,102 +0,0 @@
-import { KEYS, TextApi, bindFirst, createTSlatePlugin, getPluginType } from "platejs";
-import { getTransientSuggestionKey } from "@platejs/suggestion";
-import { isSelecting } from "@platejs/selection";
-
-//#region src/lib/transforms/insertAINodes.ts
-const insertAINodes = (editor, nodes, { target } = {}) => {
-	if (!target && !editor.selection?.focus.path) return;
-	const aiNodes = nodes.map((node) => ({
-		...node,
-		ai: true
-	}));
-	editor.tf.withoutNormalizing(() => {
-		editor.tf.insertNodes(aiNodes, {
-			at: editor.api.end(target || editor.selection.focus.path),
-			select: true
-		});
-		editor.tf.collapse({ edge: "end" });
-	});
-};
-
-//#endregion
-//#region src/lib/transforms/removeAIMarks.ts
-const removeAIMarks = (editor, { at = [] } = {}) => {
-	const nodeType = getPluginType(editor, KEYS.ai);
-	editor.tf.unsetNodes(nodeType, {
-		at,
-		match: (n) => n[nodeType]
-	});
-};
-
-//#endregion
-//#region src/lib/transforms/removeAINodes.ts
-const removeAINodes = (editor, { at = [] } = {}) => {
-	editor.tf.removeNodes({
-		at,
-		match: (n) => TextApi.isText(n) && !!n.ai
-	});
-};
-
-//#endregion
-//#region src/lib/transforms/undoAI.ts
-const undoAI = (editor) => {
-	const hasAINodeOrAISuggestion = editor.api.some({
-		at: [],
-		match: (n) => !!n.ai
-	}) || editor.api.some({
-		at: [],
-		match: (n) => !!n[getTransientSuggestionKey()]
-	});
-	if (editor.history.undos.at(-1)?.ai && hasAINodeOrAISuggestion) {
-		editor.undo();
-		editor.history.redos.pop();
-	}
-};
-
-//#endregion
-//#region src/lib/transforms/withAIBatch.ts
-const withAIBatch = (editor, fn, { split } = {}) => {
-	if (split) editor.tf.withNewBatch(fn);
-	else editor.tf.withMerging(fn);
-	const lastBatch = editor.history.undos?.at(-1);
-	if (lastBatch) lastBatch.ai = true;
-};
-
-//#endregion
-//#region src/lib/BaseAIPlugin.ts
-const BaseAIPlugin = createTSlatePlugin({
-	key: KEYS.ai,
-	node: {
-		isDecoration: false,
-		isLeaf: true
-	}
-}).extendTransforms(({ editor }) => ({
-	insertNodes: bindFirst(insertAINodes, editor),
-	removeMarks: bindFirst(removeAIMarks, editor),
-	removeNodes: bindFirst(removeAINodes, editor),
-	undo: bindFirst(undoAI, editor)
-}));
-
-//#endregion
-//#region src/lib/utils/getEditorPrompt.ts
-const createPromptFromConfig = (config, params) => {
-	const { isBlockSelecting, isSelecting: isSelecting$1 } = params;
-	if (isBlockSelecting && config.blockSelecting) return config.blockSelecting ?? config.default;
-	if (isSelecting$1 && config.selecting) return config.selecting ?? config.default;
-	return config.default;
-};
-const getEditorPrompt = (editor, { prompt = "" }) => {
-	const params = {
-		editor,
-		isBlockSelecting: editor.getOption({ key: KEYS.blockSelection }, "isSelectingSome"),
-		isSelecting: isSelecting(editor)
-	};
-	let promptText = "";
-	if (typeof prompt === "function") promptText = prompt(params);
-	else if (typeof prompt === "object") promptText = createPromptFromConfig(prompt, params);
-	else promptText = prompt;
-	return promptText;
-};
-
-//#endregion
-export { removeAINodes as a, undoAI as i, BaseAIPlugin as n, removeAIMarks as o, withAIBatch as r, insertAINodes as s, getEditorPrompt as t };

package/dist/index-B_bqJoKL.d.ts

@@ -1,84 +0,0 @@
-import { Descendant, History, OmitFirst, Path, PluginConfig, SlateEditor, TLocation } from "platejs";
-
-//#region src/lib/transforms/insertAINodes.d.ts
-declare const insertAINodes: (editor: SlateEditor, nodes: Descendant[], {
-  target
-}?: {
-  target?: Path;
-}) => void;
-//#endregion
-//#region src/lib/transforms/removeAIMarks.d.ts
-declare const removeAIMarks: (editor: SlateEditor, {
-  at
-}?: {
-  at?: TLocation;
-}) => void;
-//#endregion
-//#region src/lib/transforms/removeAINodes.d.ts
-declare const removeAINodes: (editor: SlateEditor, {
-  at
-}?: {
-  at?: Path;
-}) => void;
-//#endregion
-//#region src/lib/transforms/undoAI.d.ts
-declare const undoAI: (editor: SlateEditor) => void;
-//#endregion
-//#region src/lib/transforms/withAIBatch.d.ts
-type AIBatch = History['undos'][number] & {
-  ai?: boolean;
-};
-declare const withAIBatch: (editor: SlateEditor, fn: () => void, {
-  split
-}?: {
-  split?: boolean;
-}) => void;
-//#endregion
-//#region src/lib/BaseAIPlugin.d.ts
-type BaseAIPluginConfig = PluginConfig<'ai', {}, {}, {
-  ai: {
-    insertNodes: OmitFirst<typeof insertAINodes>;
-    removeMarks: OmitFirst<typeof removeAIMarks>;
-    removeNodes: OmitFirst<typeof removeAINodes>;
-  };
-}>;
-declare const BaseAIPlugin: any;
-//#endregion
-//#region src/lib/types.d.ts
-type AIToolName = 'comment' | 'edit' | 'generate' | null;
-type AIMode = 'chat' | 'insert';
-//#endregion
-//#region src/lib/utils/getEditorPrompt.d.ts
-type EditorPrompt = ((params: EditorPromptParams) => string) | PromptConfig | string;
-type EditorPromptParams = {
-  editor: SlateEditor;
-  isBlockSelecting: boolean;
-  isSelecting: boolean;
-};
-type PromptConfig = {
-  default: string;
-  blockSelecting?: string;
-  selecting?: string;
-};
-declare const getEditorPrompt: (editor: SlateEditor, {
-  prompt
-}: {
-  prompt?: EditorPrompt;
-}) => string;
-//#endregion
-//#region src/lib/utils/replacePlaceholders.d.ts
-type MarkdownType = 'block' | 'blockSelection' | 'blockSelectionWithBlockId' | 'blockWithBlockId' | 'editor' | 'editorWithBlockId' | 'tableCellWithId';
-declare const replacePlaceholders: (editor: SlateEditor, text: string, {
-  prompt
-}?: {
-  prompt?: string;
-}) => string;
-//#endregion
-//#region src/lib/utils/getMarkdown.d.ts
-declare const getMarkdown: (editor: SlateEditor, {
-  type
-}: {
-  type: MarkdownType;
-}) => string;
-//#endregion
-export { insertAINodes as _, EditorPromptParams as a, AIMode as c, BaseAIPluginConfig as d, AIBatch as f, removeAIMarks as g, removeAINodes as h, EditorPrompt as i, AIToolName as l, undoAI as m, MarkdownType as n, PromptConfig as o, withAIBatch as p, replacePlaceholders as r, getEditorPrompt as s, getMarkdown as t, BaseAIPlugin as u };