fix

.changeset/ai.md

@@ -0,0 +1,10 @@
+---
+'@udecode/plate-ai': minor
+---
+
+- `api.aiChat.replaceSelection()` – new option `format: 'none' | 'single' | 'all'`
+  - `'single'` (default):
+    - Single block: Applies block's formatting to inserted content
+    - Multiple blocks: Preserves source formatting
+  - `'all'`: Forces first block's formatting on all inserted blocks
+  - `'none'`: Preserves source formatting completely

.changeset/list.md

@@ -0,0 +1,5 @@
+---
+'@udecode/plate-indent-list': patch
+---
+
+Fix todo list: insert break should inherit format

apps/www/content/docs/ai.mdx

@@ -280,15 +280,15 @@ Template for system messages. Supports same placeholders as `promptTemplate`.
 
 ## API
 
-### editor.aiChat.accept()
+### api.aiChat.accept()
 
 Accepts the current AI suggestion:
 
 - Removes AI marks from the content
 - Hides the AI chat interface
 - Focuses the editor
 
-### editor.aiChat.insertBelow()
+### api.aiChat.insertBelow()
 
 Inserts AI content below the current block.
 
@@ -303,30 +303,38 @@ Handles both block selection and normal selection modes:
 - In block selection: Inserts after the last selected block
 - In normal selection: Inserts after the current block
 
-### editor.aiChat.replaceSelection()
+### api.aiChat.replaceSelection()
 
 Replaces the current selection with AI content.
 
 <APIParameters>
   <APIItem name="sourceEditor" type="PlateEditor">
     Editor containing the content to replace with.
   </APIItem>
+  <APIItem name="options" type="object" optional>
+    <APISubList>
+      <APISubListItem parent="options" name="forceUniformFormatting" type="boolean" optional>
+        When true, applies the first block's formatting to all inserted blocks. Defaults to false.
+      </APISubListItem>
+    </APISubList>
+  </APIItem>
 </APIParameters>
 
-Handles both block selection and normal selection modes:
+Handles different selection modes:
 
-- In block selection: Replaces all selected blocks
-- In normal selection: Replaces the current selection
+- Single block selection: Replaces the selected block, applying its formatting to all inserted content
+- Multiple block selection: Replaces all selected blocks, preserving the original formatting unless `forceUniformFormatting` is enabled
+- Normal selection: Replaces the current selection while maintaining surrounding context
 
-### editor.aiChat.reset()
+### api.aiChat.reset()
 
 Resets the chat state:
 
 - Stops any ongoing generation
 - Clears chat messages
 - Removes all AI nodes from the editor
 
-### editor.aiChat.submit()
+### api.aiChat.submit()
 
 Submits a prompt to generate AI content.
 
@@ -355,7 +363,7 @@ In insert mode, undoes previous AI changes before submitting.
 
 ## Transforms
 
-### editor.ai.insertNodes()
+### tf.ai.insertNodes()
 
 Inserts AI-generated nodes with the AI mark.
 
@@ -372,7 +380,7 @@ Inserts AI-generated nodes with the AI mark.
   </APIItem>
 </APIParameters>
 
-### editor.ai.removeMarks()
+### tf.ai.removeMarks()
 
 Removes AI marks from nodes in the specified location.
 
@@ -386,7 +394,7 @@ Removes AI marks from nodes in the specified location.
   </APIItem>
 </APIParameters>
 
-### editor.ai.removeNodes()
+### tf.ai.removeNodes()
 
 Removes nodes that have the AI mark.
 
@@ -400,7 +408,7 @@ Removes nodes that have the AI mark.
   </APIItem>
 </APIParameters>
 
-### editor.ai.undo()
+### tf.ai.undo()
 
 Special undo operation for AI changes:
 

apps/www/src/registry/default/components/editor/transforms.ts

@@ -25,6 +25,7 @@ import { TocPlugin } from '@udecode/plate-heading/react';
 import { INDENT_LIST_KEYS, ListStyleType } from '@udecode/plate-indent-list';
 import { IndentListPlugin } from '@udecode/plate-indent-list/react';
 import { insertColumnGroup, toggleColumnGroup } from '@udecode/plate-layout';
+import { ColumnItemPlugin, ColumnPlugin } from '@udecode/plate-layout/react';
 import { LinkPlugin, triggerFloatingLink } from '@udecode/plate-link/react';
 import { insertEquation, insertInlineEquation } from '@udecode/plate-math';
 import {
@@ -44,9 +45,22 @@ import {
   MediaEmbedPlugin,
   VideoPlugin,
 } from '@udecode/plate-media/react';
-import { TablePlugin, insertTable } from '@udecode/plate-table/react';
+import {
+  TableCellPlugin,
+  TablePlugin,
+  TableRowPlugin,
+  insertTable,
+} from '@udecode/plate-table/react';
 import { Path } from 'slate';
 
+export const STRUCTURAL_TYPES = [
+  ColumnPlugin.key,
+  ColumnItemPlugin.key,
+  TablePlugin.key,
+  TableRowPlugin.key,
+  TableCellPlugin.key,
+];
+
 const ACTION_THREE_COLUMNS = 'action_three_columns';
 
 const insertList = (editor: PlateEditor, type: string) => {

apps/www/src/registry/default/plate-ui/turn-into-dropdown-menu.tsx

@@ -30,6 +30,7 @@ import {
 } from 'lucide-react';
 
 import {
+  STRUCTURAL_TYPES,
   getBlockType,
   setBlockType,
 } from '@/registry/default/components/editor/transforms';
@@ -119,6 +120,7 @@ export function TurnIntoDropdownMenu(props: DropdownMenuProps) {
   const value = useSelectionFragmentProp({
     defaultValue: ParagraphPlugin.key,
     getProp: (node) => getBlockType(node as any),
+    structuralTypes: STRUCTURAL_TYPES,
   });
   const selectedItem = React.useMemo(
     () =>

packages/ai/src/react/ai-chat/transforms/replaceSelectionAIChat.ts

@@ -1,7 +1,12 @@
-import type { PlateEditor } from '@udecode/plate-common/react';
-
-import { isEditorEmpty, withNewBatch } from '@udecode/plate-common';
-import { focusEditor } from '@udecode/plate-common/react';
+import {
+  type TElement,
+  getFirstNodeText,
+  getNodeProps,
+  isEditorEmpty,
+  isText,
+  withNewBatch,
+} from '@udecode/plate-common';
+import { type PlateEditor, focusEditor } from '@udecode/plate-common/react';
 import {
   BlockSelectionPlugin,
   removeBlockSelectionNodes,
@@ -12,7 +17,8 @@ import type { AIChatPluginConfig } from '../AIChatPlugin';
 
 export const replaceSelectionAIChat = (
   editor: PlateEditor,
-  sourceEditor: PlateEditor
+  sourceEditor: PlateEditor,
+  { format = 'single' }: { format?: 'all' | 'none' | 'single' } = {}
 ) => {
   if (!sourceEditor || isEditorEmpty(sourceEditor)) return;
 
@@ -23,11 +29,21 @@ export const replaceSelectionAIChat = (
 
   editor.getApi<AIChatPluginConfig>({ key: 'ai' }).aiChat.hide();
 
-  if (isBlockSelecting) {
-    const firstBlockPath = editor
-      .getApi(BlockSelectionPlugin)
-      .blockSelection.getNodes()[0][1];
+  // If no blocks selected, treat it like a normal selection replacement
+  if (!isBlockSelecting) {
+    editor.insertFragment(sourceEditor.children);
+    focusEditor(editor);
+
+    return;
+  }
 
+  const blockSelectionApi = editor.getApi(BlockSelectionPlugin).blockSelection;
+  const selectedBlocks = blockSelectionApi.getNodes();
+
+  if (selectedBlocks.length === 0) return;
+  // If format is 'none' or multiple blocks with 'single',
+  // just insert the content as is
+  if (format === 'none' || (format === 'single' && selectedBlocks.length > 1)) {
     editor.withoutNormalizing(() => {
       removeBlockSelectionNodes(editor);
 
@@ -37,13 +53,63 @@ export const replaceSelectionAIChat = (
           .blockSelection.insertBlocksAndSelect(
             cloneDeep(sourceEditor.children),
             {
-              at: firstBlockPath,
+              at: selectedBlocks[0][1],
             }
           );
       });
     });
-  } else {
-    editor.insertFragment(sourceEditor.children);
+
     focusEditor(editor);
+
+    return;
   }
+
+  // Apply formatting from first block when:
+  // - formatting is 'all', or
+  // - only one block is selected
+  const [firstBlockNode, firstBlockPath] = selectedBlocks[0];
+  const firstBlockProps = getNodeProps(firstBlockNode);
+
+  // Get formatting from first text node
+  const firstTextEntry = getFirstNodeText(firstBlockNode as TElement);
+
+  if (!firstTextEntry) return;
+
+  const textProps = getNodeProps(firstTextEntry[0]);
+
+  // Apply text props recursively to text nodes
+  const applyTextProps = (node: any): any => {
+    if (isText(node)) {
+      return { ...textProps, ...node };
+    }
+    if (node.children) {
+      return {
+        ...node,
+        children: node.children.map(applyTextProps),
+      };
+    }
+
+    return node;
+  };
+
+  editor.withoutNormalizing(() => {
+    removeBlockSelectionNodes(editor);
+
+    withNewBatch(editor, () => {
+      // Create new blocks with first block's formatting
+      const newBlocks = cloneDeep(sourceEditor.children).map((block) => ({
+        ...block,
+        ...firstBlockProps,
+        children: block.children.map(applyTextProps),
+      }));
+
+      editor
+        .getTransforms(BlockSelectionPlugin)
+        .blockSelection.insertBlocksAndSelect(newBlocks, {
+          at: firstBlockPath,
+        });
+    });
+  });
+
+  focusEditor(editor);
 };