@jupyter/chat 0.11.0 → 0.12.0

package/lib/components/chat-messages.d.ts

@@ -30,6 +30,10 @@ type BaseMessageProps = {
      * The footer registry.
      */
     messageFooterRegistry?: IMessageFooterRegistry;
+    /**
+     * The welcome message.
+     */
+    welcomeMessage?: string;
 };
 /**
  * The messages list component.

package/lib/components/chat-messages.js

@@ -10,8 +10,9 @@ import clsx from 'clsx';
 import React, { useEffect, useState, useRef, forwardRef } from 'react';
 import { AttachmentPreviewList } from './attachments';
 import { ChatInput } from './chat-input';
-import { MarkdownRenderer } from './markdown-renderer';
 import { MessageFooter } from './messages/footer';
+import { MessageRenderer } from './messages/message-renderer';
+import { WelcomeMessage } from './messages/welcome';
 import { ScrollContainer } from './scroll-container';
 import { InputModel } from '../input-model';
 import { replaceSpanToMention } from '../utils';
@@ -62,7 +63,7 @@ export function ChatMessages(props) {
             setMessages([...model.messages]);
         }
         function handleWritersChange(_, writers) {
-            setCurrentWriters(writers);
+            setCurrentWriters(writers.map(writer => writer.user));
         }
         model.messagesUpdated.connect(handleChatEvents);
         (_a = model.writersChanged) === null || _a === void 0 ? void 0 : _a.connect(handleWritersChange);
@@ -133,6 +134,7 @@ export function ChatMessages(props) {
     }, [messages, allRendered]);
     return (React.createElement(React.Fragment, null,
         React.createElement(ScrollContainer, { sx: { flexGrow: 1 } },
+            props.welcomeMessage && (React.createElement(WelcomeMessage, { rmRegistry: props.rmRegistry, content: props.welcomeMessage })),
             React.createElement(Box, { ref: refMsgBox, className: clsx(MESSAGES_BOX_CLASS) }, messages.map((message, i) => {
                 renderedPromise.current[i] = new PromiseDelegate();
                 return (
@@ -230,10 +232,10 @@ export const ChatMessage = forwardRef((props, ref) => {
     const [deleted, setDeleted] = useState(false);
     const [canEdit, setCanEdit] = useState(false);
     const [canDelete, setCanDelete] = useState(false);
-    const [inputModel, setInputModel] = useState(null);
     // Look if the message can be deleted or edited.
     useEffect(() => {
         var _a;
+        // Init canDelete and canEdit state.
         setDeleted((_a = message.deleted) !== null && _a !== void 0 ? _a : false);
         if (model.user !== undefined && !message.deleted) {
             if (model.user.username === message.sender.username) {
@@ -251,40 +253,41 @@ export const ChatMessage = forwardRef((props, ref) => {
         }
     }, [model, message]);
     // Create an input model only if the message is edited.
-    useEffect(() => {
-        if (edit && canEdit) {
-            setInputModel(() => {
-                var _a;
-                let body = message.body;
-                (_a = message.mentions) === null || _a === void 0 ? void 0 : _a.forEach(user => {
-                    body = replaceSpanToMention(body, user);
-                });
-                return new InputModel({
-                    chatContext: model.createChatContext(),
-                    onSend: (input, model) => updateMessage(message.id, input, model),
-                    onCancel: () => cancelEdition(),
-                    value: body,
-                    activeCellManager: model.activeCellManager,
-                    selectionWatcher: model.selectionWatcher,
-                    documentManager: model.documentManager,
-                    config: {
-                        sendWithShiftEnter: model.config.sendWithShiftEnter
-                    },
-                    attachments: message.attachments,
-                    mentions: message.mentions
-                });
-            });
-        }
-        else {
-            setInputModel(null);
+    const startEdition = () => {
+        var _a;
+        if (!canEdit) {
+            return;
         }
-    }, [edit]);
+        let body = message.body;
+        (_a = message.mentions) === null || _a === void 0 ? void 0 : _a.forEach(user => {
+            body = replaceSpanToMention(body, user);
+        });
+        const inputModel = new InputModel({
+            chatContext: model.createChatContext(),
+            onSend: (input, model) => updateMessage(message.id, input, model),
+            onCancel: () => cancelEdition(),
+            value: body,
+            activeCellManager: model.activeCellManager,
+            selectionWatcher: model.selectionWatcher,
+            documentManager: model.documentManager,
+            config: {
+                sendWithShiftEnter: model.config.sendWithShiftEnter
+            },
+            attachments: message.attachments,
+            mentions: message.mentions
+        });
+        model.addEditionModel(message.id, inputModel);
+        setEdit(true);
+    };
     // Cancel the current edition of the message.
     const cancelEdition = () => {
+        var _a;
+        (_a = model.getEditionModel(message.id)) === null || _a === void 0 ? void 0 : _a.dispose();
         setEdit(false);
     };
     // Update the content of the message.
     const updateMessage = (id, input, inputModel) => {
+        var _a;
         if (!canEdit || !inputModel) {
             return;
         }
@@ -294,6 +297,7 @@ export const ChatMessage = forwardRef((props, ref) => {
         updatedMessage.attachments = inputModel.attachments;
         updatedMessage.mentions = inputModel.mentions;
         model.updateMessage(id, updatedMessage);
+        (_a = model.getEditionModel(message.id)) === null || _a === void 0 ? void 0 : _a.dispose();
         setEdit(false);
     };
     // Delete the message.
@@ -305,7 +309,7 @@ export const ChatMessage = forwardRef((props, ref) => {
     };
     // Empty if the message has been deleted.
     return deleted ? (React.createElement("div", { ref: ref, "data-index": props.index })) : (React.createElement("div", { ref: ref, "data-index": props.index },
-        edit && canEdit && inputModel ? (React.createElement(ChatInput, { onCancel: () => cancelEdition(), model: inputModel, chatCommandRegistry: props.chatCommandRegistry, toolbarRegistry: props.inputToolbarRegistry })) : (React.createElement(MarkdownRenderer, { rmRegistry: rmRegistry, markdownStr: message.body, model: model, edit: canEdit ? () => setEdit(true) : undefined, delete: canDelete ? () => deleteMessage(message.id) : undefined, rendered: props.renderedPromise })),
+        edit && canEdit && model.getEditionModel(message.id) ? (React.createElement(ChatInput, { onCancel: () => cancelEdition(), model: model.getEditionModel(message.id), chatCommandRegistry: props.chatCommandRegistry, toolbarRegistry: props.inputToolbarRegistry })) : (React.createElement(MessageRenderer, { rmRegistry: rmRegistry, markdownStr: message.body, model: model, edit: canEdit ? startEdition : undefined, delete: canDelete ? () => deleteMessage(message.id) : undefined, rendered: props.renderedPromise })),
         message.attachments && !edit && (
         // Display the attachments only if message is not edited, otherwise the
         // input component display them.
@@ -397,7 +401,8 @@ export function Navigation(props) {
     useEffect(() => {
         var _a, _b;
         const viewportChanged = (model, viewport) => {
-            setLastInViewport(viewport.includes(model.messages.length - 1));
+            setLastInViewport(model.messages.length === 0 ||
+                viewport.includes(model.messages.length - 1));
         };
         (_a = model.viewportChanged) === null || _a === void 0 ? void 0 : _a.connect(viewportChanged);
         viewportChanged(model, (_b = model.messagesInViewport) !== null && _b !== void 0 ? _b : []);

package/lib/components/chat.d.ts

@@ -40,6 +40,10 @@ export declare namespace Chat {
          * The footer registry.
          */
         messageFooterRegistry?: IMessageFooterRegistry;
+        /**
+         * The welcome message.
+         */
+        welcomeMessage?: string;
     }
     /**
      * The options to build the Chat UI.

package/lib/components/chat.js

@@ -19,7 +19,7 @@ export function ChatBody(props) {
         inputToolbarRegistry = InputToolbarRegistry.defaultToolbarRegistry();
     }
     return (React.createElement(AttachmentOpenerContext.Provider, { value: props.attachmentOpenerRegistry },
-        React.createElement(ChatMessages, { rmRegistry: props.rmRegistry, model: model, chatCommandRegistry: props.chatCommandRegistry, inputToolbarRegistry: inputToolbarRegistry, messageFooterRegistry: props.messageFooterRegistry }),
+        React.createElement(ChatMessages, { rmRegistry: props.rmRegistry, model: model, chatCommandRegistry: props.chatCommandRegistry, inputToolbarRegistry: inputToolbarRegistry, messageFooterRegistry: props.messageFooterRegistry, welcomeMessage: props.welcomeMessage }),
         React.createElement(ChatInput, { sx: {
                 paddingLeft: 4,
                 paddingRight: 4,

package/lib/components/index.d.ts

@@ -4,7 +4,6 @@ export * from './chat-messages';
 export * from './code-blocks';
 export * from './input';
 export * from './jl-theme-provider';
-export * from './markdown-renderer';
 export * from './mui-extras';
 export * from './scroll-container';
 export * from './toolbar';

package/lib/components/index.js

@@ -8,7 +8,6 @@ export * from './chat-messages';
 export * from './code-blocks';
 export * from './input';
 export * from './jl-theme-provider';
-export * from './markdown-renderer';
 export * from './mui-extras';
 export * from './scroll-container';
 export * from './toolbar';

package/lib/components/{markdown-renderer.d.ts → messages/message-renderer.d.ts}

@@ -1,8 +1,11 @@
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { PromiseDelegate } from '@lumino/coreutils';
 import React from 'react';
-import { IChatModel } from '../model';
-type MarkdownRendererProps = {
+import { IChatModel } from '../../model';
+/**
+ * The type of the props for the MessageRenderer component.
+ */
+type MessageRendererProps = {
     /**
      * The string to render.
      */
@@ -32,6 +35,9 @@ type MarkdownRendererProps = {
      */
     delete?: () => void;
 };
-declare function MarkdownRendererBase(props: MarkdownRendererProps): JSX.Element;
-export declare const MarkdownRenderer: React.MemoExoticComponent<typeof MarkdownRendererBase>;
+/**
+ * The message renderer base component.
+ */
+declare function MessageRendererBase(props: MessageRendererProps): JSX.Element;
+export declare const MessageRenderer: React.MemoExoticComponent<typeof MessageRendererBase>;
 export {};

package/lib/components/{markdown-renderer.js → messages/message-renderer.js}

@@ -4,47 +4,24 @@
  */
 import React, { useState, useEffect } from 'react';
 import { createPortal } from 'react-dom';
-import { CodeToolbar } from './code-blocks/code-toolbar';
-import { MessageToolbar } from './toolbar';
-const MD_MIME_TYPE = 'text/markdown';
-const MD_RENDERED_CLASS = 'jp-chat-rendered-markdown';
+import { CodeToolbar } from '../code-blocks/code-toolbar';
+import { MessageToolbar } from '../toolbar';
+import { MarkdownRenderer, MD_RENDERED_CLASS } from '../../markdown-renderer';
 /**
- * Escapes backslashes in LaTeX delimiters such that they appear in the DOM
- * after the initial MarkDown render. For example, this function takes '\(` and
- * returns `\\(`.
- *
- * Required for proper rendering of MarkDown + LaTeX markup in the chat by
- * `ILatexTypesetter`.
+ * The message renderer base component.
  */
-function escapeLatexDelimiters(text) {
-    return text
-        .replace(/\\\(/g, '\\\\(')
-        .replace(/\\\)/g, '\\\\)')
-        .replace(/\\\[/g, '\\\\[')
-        .replace(/\\\]/g, '\\\\]');
-}
-function MarkdownRendererBase(props) {
+function MessageRendererBase(props) {
+    const { markdownStr, rmRegistry } = props;
     const appendContent = props.appendContent || false;
     const [renderedContent, setRenderedContent] = useState(null);
     // each element is a two-tuple with the structure [codeToolbarRoot, codeToolbarProps].
     const [codeToolbarDefns, setCodeToolbarDefns] = useState([]);
     useEffect(() => {
         const renderContent = async () => {
-            var _a, _b;
-            // initialize mime model
-            const mdStr = escapeLatexDelimiters(props.markdownStr);
-            const model = props.rmRegistry.createModel({
-                data: { [MD_MIME_TYPE]: mdStr }
+            const renderer = await MarkdownRenderer.renderContent({
+                content: markdownStr,
+                rmRegistry
             });
-            const renderer = props.rmRegistry.createRenderer(MD_MIME_TYPE);
-            // step 1: render markdown
-            await renderer.renderModel(model);
-            (_a = props.rmRegistry.latexTypesetter) === null || _a === void 0 ? void 0 : _a.typeset(renderer.node);
-            if (!renderer.node) {
-                throw new Error('Rendermime was unable to render Markdown content within a chat message. Please report this upstream to Jupyter chat on GitHub.');
-            }
-            // step 2: render LaTeX via MathJax.
-            (_b = props.rmRegistry.latexTypesetter) === null || _b === void 0 ? void 0 : _b.typeset(renderer.node);
             const newCodeToolbarDefns = [];
             // Attach CodeToolbar root element to each <pre> block
             const preBlocks = renderer.node.querySelectorAll('pre');
@@ -63,7 +40,7 @@ function MarkdownRendererBase(props) {
             props.rendered.resolve();
         };
         renderContent();
-    }, [props.markdownStr, props.rmRegistry]);
+    }, [markdownStr, rmRegistry]);
     return (React.createElement("div", { className: MD_RENDERED_CLASS },
         renderedContent &&
             (appendContent ? (React.createElement("div", { ref: node => node && node.appendChild(renderedContent) })) : (React.createElement("div", { ref: node => node && node.replaceChildren(renderedContent) }))),
@@ -76,4 +53,4 @@ function MarkdownRendererBase(props) {
             return createPortal(React.createElement(CodeToolbar, { ...codeToolbarProps }), codeToolbarRoot);
         })));
 }
-export const MarkdownRenderer = React.memo(MarkdownRendererBase);
+export const MessageRenderer = React.memo(MessageRendererBase);

package/lib/components/messages/welcome.d.ts

@@ -0,0 +1,8 @@
+/// <reference types="react" />
+import { MarkdownRenderer } from '../../markdown-renderer';
+/**
+ * The welcome message component.
+ * This message is displayed on top of the chat messages, and is rendered using a
+ * markdown renderer.
+ */
+export declare function WelcomeMessage(props: MarkdownRenderer.IOptions): JSX.Element;

package/lib/components/messages/welcome.js

@@ -0,0 +1,41 @@
+/*
+ * Copyright (c) Jupyter Development Team.
+ * Distributed under the terms of the Modified BSD License.
+ */
+import { classes } from '@jupyterlab/ui-components';
+import React, { useEffect, useRef } from 'react';
+import { MarkdownRenderer, MD_RENDERED_CLASS } from '../../markdown-renderer';
+const WELCOME_MESSAGE_CLASS = 'jp-chat-welcome-message';
+/**
+ * The welcome message component.
+ * This message is displayed on top of the chat messages, and is rendered using a
+ * markdown renderer.
+ */
+export function WelcomeMessage(props) {
+    const { rmRegistry } = props;
+    const content = props.content + '\n----\n';
+    // ref that tracks the content container to store the rendermime node in
+    const renderingContainer = useRef(null);
+    // ref that tracks whether the rendermime node has already been inserted
+    const renderingInserted = useRef(false);
+    /**
+     * Effect: use Rendermime to render `props.markdownStr` into an HTML element,
+     * and insert it into `renderingContainer` if not yet inserted.
+     */
+    useEffect(() => {
+        const renderContent = async () => {
+            const renderer = await MarkdownRenderer.renderContent({
+                content,
+                rmRegistry
+            });
+            // insert the rendering into renderingContainer if not yet inserted
+            if (renderingContainer.current !== null && !renderingInserted.current) {
+                renderingContainer.current.appendChild(renderer.node);
+                renderingInserted.current = true;
+            }
+        };
+        renderContent();
+    }, [content]);
+    return (React.createElement("div", { className: classes(MD_RENDERED_CLASS, WELCOME_MESSAGE_CLASS) },
+        React.createElement("div", { ref: renderingContainer })));
+}

package/lib/components/scroll-container.js

@@ -28,6 +28,6 @@ export function ScrollContainer(props) {
             },
             ...props.sx
         } },
-        React.createElement(Box, { sx: { minHeight: '100.01%' } }, props.children),
+        React.createElement(Box, null, props.children),
         React.createElement(Box, { sx: { overflowAnchor: 'auto', height: '1px' } })));
 }

package/lib/index.d.ts

@@ -4,6 +4,7 @@ export * from './components';
 export * from './footers';
 export * from './icons';
 export * from './input-model';
+export * from './markdown-renderer';
 export * from './model';
 export * from './registry';
 export * from './selection-watcher';

package/lib/index.js

@@ -8,6 +8,7 @@ export * from './components';
 export * from './footers';
 export * from './icons';
 export * from './input-model';
+export * from './markdown-renderer';
 export * from './model';
 export * from './registry';
 export * from './selection-watcher';

package/lib/input-model.d.ts

@@ -114,6 +114,10 @@ export interface IInputModel extends IDisposable {
      * Clear mentions list.
      */
     clearMentions(): void;
+    /**
+     * A signal emitting when disposing of the model.
+     */
+    readonly onDisposed: ISignal<InputModel, void>;
 }
 /**
  * The input model.
@@ -231,6 +235,10 @@ export declare class InputModel implements IInputModel {
      * Dispose the input model.
      */
     dispose(): void;
+    /**
+     * A signal emitting when disposing of the model.
+     */
+    get onDisposed(): ISignal<InputModel, void>;
     /**
      * Whether the input model is disposed.
      */
@@ -252,6 +260,7 @@ export declare class InputModel implements IInputModel {
     private _configChanged;
     private _focusInputSignal;
     private _attachmentsChanged;
+    private _onDisposed;
     private _isDisposed;
 }
 export declare namespace InputModel {

package/lib/input-model.js

@@ -60,6 +60,7 @@ export class InputModel {
         this._configChanged = new Signal(this);
         this._focusInputSignal = new Signal(this);
         this._attachmentsChanged = new Signal(this);
+        this._onDisposed = new Signal(this);
         this._isDisposed = false;
         this._onSend = options.onSend;
         this._chatContext = options.chatContext;
@@ -235,8 +236,15 @@ export class InputModel {
         if (this.isDisposed) {
             return;
         }
+        this._onDisposed.emit();
         this._isDisposed = true;
     }
+    /**
+     * A signal emitting when disposing of the model.
+     */
+    get onDisposed() {
+        return this._onDisposed;
+    }
     /**
      * Whether the input model is disposed.
      */

package/lib/markdown-renderer.d.ts

@@ -0,0 +1,38 @@
+import { IRenderMime, IRenderMimeRegistry } from '@jupyterlab/rendermime';
+export declare const MD_RENDERED_CLASS = "jp-chat-rendered-markdown";
+export declare const MD_MIME_TYPE = "text/markdown";
+/**
+ * A namespace for the MarkdownRenderer.
+ */
+export declare namespace MarkdownRenderer {
+    /**
+     * The options for the MarkdownRenderer.
+     */
+    interface IOptions {
+        /**
+         * The rendermime registry.
+         */
+        rmRegistry: IRenderMimeRegistry;
+        /**
+         * The markdown content.
+         */
+        content: string;
+    }
+    /**
+     * A generic function to render a markdown string into a DOM element.
+     *
+     * @param content - the markdown content.
+     * @param rmRegistry - the rendermime registry.
+     * @returns a promise that resolves to the renderer.
+     */
+    function renderContent(options: IOptions): Promise<IRenderMime.IRenderer>;
+    /**
+     * Escapes backslashes in LaTeX delimiters such that they appear in the DOM
+     * after the initial MarkDown render. For example, this function takes '\(` and
+     * returns `\\(`.
+     *
+     * Required for proper rendering of MarkDown + LaTeX markup in the chat by
+     * `ILatexTypesetter`.
+     */
+    function escapeLatexDelimiters(text: string): string;
+}

package/lib/markdown-renderer.js

@@ -0,0 +1,54 @@
+/*
+ * Copyright (c) Jupyter Development Team.
+ * Distributed under the terms of the Modified BSD License.
+ */
+export const MD_RENDERED_CLASS = 'jp-chat-rendered-markdown';
+export const MD_MIME_TYPE = 'text/markdown';
+/**
+ * A namespace for the MarkdownRenderer.
+ */
+export var MarkdownRenderer;
+(function (MarkdownRenderer) {
+    /**
+     * A generic function to render a markdown string into a DOM element.
+     *
+     * @param content - the markdown content.
+     * @param rmRegistry - the rendermime registry.
+     * @returns a promise that resolves to the renderer.
+     */
+    async function renderContent(options) {
+        var _a;
+        const { rmRegistry, content } = options;
+        // initialize mime model
+        const mdStr = escapeLatexDelimiters(content);
+        const model = rmRegistry.createModel({
+            data: { [MD_MIME_TYPE]: mdStr }
+        });
+        const renderer = rmRegistry.createRenderer(MD_MIME_TYPE);
+        // step 1: render markdown
+        await renderer.renderModel(model);
+        if (!renderer.node) {
+            throw new Error('Rendermime was unable to render Markdown content within a chat message. Please report this upstream to Jupyter chat on GitHub.');
+        }
+        // step 2: render LaTeX via MathJax.
+        (_a = rmRegistry.latexTypesetter) === null || _a === void 0 ? void 0 : _a.typeset(renderer.node);
+        return renderer;
+    }
+    MarkdownRenderer.renderContent = renderContent;
+    /**
+     * Escapes backslashes in LaTeX delimiters such that they appear in the DOM
+     * after the initial MarkDown render. For example, this function takes '\(` and
+     * returns `\\(`.
+     *
+     * Required for proper rendering of MarkDown + LaTeX markup in the chat by
+     * `ILatexTypesetter`.
+     */
+    function escapeLatexDelimiters(text) {
+        return text
+            .replace(/\\\(/g, '\\\\(')
+            .replace(/\\\)/g, '\\\\)')
+            .replace(/\\\[/g, '\\\\[')
+            .replace(/\\\]/g, '\\\\]');
+    }
+    MarkdownRenderer.escapeLatexDelimiters = escapeLatexDelimiters;
+})(MarkdownRenderer || (MarkdownRenderer = {}));

package/lib/model.d.ts

@@ -38,6 +38,10 @@ export interface IChatModel extends IDisposable {
      * The input model.
      */
     readonly input: IInputModel;
+    /**
+     * The current writer list.
+     */
+    readonly writers: IChatModel.IWriter[];
     /**
      * Get the active cell manager.
      */
@@ -57,7 +61,7 @@ export interface IChatModel extends IDisposable {
     /**
      * A signal emitting when the messages list is updated.
      */
-    get configChanged(): ISignal<IChatModel, IConfig>;
+    readonly configChanged: ISignal<IChatModel, IConfig>;
     /**
      * A signal emitting when unread messages change.
      */
@@ -69,7 +73,11 @@ export interface IChatModel extends IDisposable {
     /**
      * A signal emitting when the writers change.
      */
-    readonly writersChanged?: ISignal<IChatModel, IUser[]>;
+    readonly writersChanged?: ISignal<IChatModel, IChatModel.IWriter[]>;
+    /**
+     * A signal emitting when the message edition input changed change.
+     */
+    readonly messageEditionAdded: ISignal<IChatModel, IChatModel.IMessageEdition>;
     /**
      * Send a message, to be defined depending on the chosen technology.
      * Default to no-op.
@@ -130,11 +138,19 @@ export interface IChatModel extends IDisposable {
     /**
      * Update the current writers list.
      */
-    updateWriters(writers: IUser[]): void;
+    updateWriters(writers: IChatModel.IWriter[]): void;
     /**
      * Create the chat context that will be passed to the input model.
      */
     createChatContext(): IChatContext;
+    /**
+     * Get the input model of the edited message, given its id.
+     */
+    getEditionModel(messageID: string): IInputModel | undefined;
+    /**
+     * Add an input model of the edited message.
+     */
+    addEditionModel(messageID: string, inputModel: IInputModel): void;
 }
 /**
  * An abstract implementation of IChatModel.
@@ -165,6 +181,10 @@ export declare abstract class AbstractChatModel implements IChatModel {
      * The input model.
      */
     get input(): IInputModel;
+    /**
+     * The current writer list.
+     */
+    get writers(): IChatModel.IWriter[];
     /**
      * Get the active cell manager.
      */
@@ -216,7 +236,11 @@ export declare abstract class AbstractChatModel implements IChatModel {
     /**
      * A signal emitting when the writers change.
      */
-    get writersChanged(): ISignal<IChatModel, IUser[]>;
+    get writersChanged(): ISignal<IChatModel, IChatModel.IWriter[]>;
+    /**
+     * A signal emitting when the message edition input changed change.
+     */
+    get messageEditionAdded(): ISignal<IChatModel, IChatModel.IMessageEdition>;
     /**
      * Send a message, to be defined depending on the chosen technology.
      * Default to no-op.
@@ -266,11 +290,19 @@ export declare abstract class AbstractChatModel implements IChatModel {
      * Update the current writers list.
      * This implementation only propagate the list via a signal.
      */
-    updateWriters(writers: IUser[]): void;
+    updateWriters(writers: IChatModel.IWriter[]): void;
     /**
      * Create the chat context that will be passed to the input model.
      */
     abstract createChatContext(): IChatContext;
+    /**
+     * Get the input model of the edited message, given its id.
+     */
+    getEditionModel(messageID: string): IInputModel | undefined;
+    /**
+     * Add an input model of the edited message.
+     */
+    addEditionModel(messageID: string, inputModel: IInputModel): void;
     /**
      * Add unread messages to the list.
      * @param indexes - list of new indexes.
@@ -299,11 +331,14 @@ export declare abstract class AbstractChatModel implements IChatModel {
     private _selectionWatcher;
     private _documentManager;
     private _notificationId;
+    private _writers;
+    private _messageEditions;
     private _messagesUpdated;
     private _configChanged;
     private _unreadChanged;
     private _viewportChanged;
     private _writersChanged;
+    private _messageEditionAdded;
 }
 /**
  * The chat model namespace.
@@ -338,6 +373,32 @@ export declare namespace IChatModel {
          */
         documentManager?: IDocumentManager | null;
     }
+    /**
+     * Representation of a message edition.
+     */
+    interface IMessageEdition {
+        /**
+         * The id of the edited message.
+         */
+        id: string;
+        /**
+         * The model of the input editing the message.
+         */
+        model: IInputModel;
+    }
+    /**
+     * Writer interface, including the message ID if the writer is editing a message.
+     */
+    interface IWriter {
+        /**
+         * The user currently writing.
+         */
+        user: IUser;
+        /**
+         * The message ID (optional)
+         */
+        messageID?: string;
+    }
 }
 /**
  * Interface of the chat context, a 'subset' of the model with readonly attribute,