jupyterlab-chat 0.8.1 → 0.10.0

package/lib/factory.d.ts

@@ -1,4 +1,4 @@
-import { IAttachmentOpenerRegistry, IChatCommandRegistry } from '@jupyter/chat';
+import { IAttachmentOpenerRegistry, IChatCommandRegistry, IInputToolbarRegistry } from '@jupyter/chat';
 import { IThemeManager } from '@jupyterlab/apputils';
 import { IDocumentManager } from '@jupyterlab/docmanager';
 import { ABCWidgetFactory, DocumentRegistry } from '@jupyterlab/docregistry';
@@ -8,7 +8,7 @@ import { ISignal } from '@lumino/signaling';
 import { LabChatModel } from './model';
 import { LabChatPanel } from './widget';
 import { YChat } from './ychat';
-import { ILabChatConfig, IWidgetConfig } from './token';
+import { IInputToolbarRegistryFactory, ILabChatConfig, IWidgetConfig } from './token';
 /**
  * The object provided by the chatDocument extension.
  * It is used to set the current config (from settings) to newly created chat widget,
@@ -50,9 +50,9 @@ export declare class ChatWidgetFactory extends ABCWidgetFactory<LabChatPanel, La
     protected createNewWidget(context: ChatWidgetFactory.IContext): LabChatPanel;
     private _themeManager;
     private _rmRegistry;
-    private _documentManager?;
     private _chatCommandRegistry?;
     private _attachmentOpenerRegistry?;
+    private _inputToolbarFactory?;
 }
 export declare namespace ChatWidgetFactory {
     interface IContext extends DocumentRegistry.IContext<LabChatModel> {
@@ -61,13 +61,14 @@ export declare namespace ChatWidgetFactory {
         documentManager?: IDocumentManager;
         chatCommandRegistry?: IChatCommandRegistry;
         attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+        inputToolbarRegistry?: IInputToolbarRegistry;
     }
     interface IOptions<T extends LabChatPanel> extends DocumentRegistry.IWidgetFactoryOptions<T> {
         themeManager: IThemeManager | null;
         rmRegistry: IRenderMimeRegistry;
-        documentManager?: IDocumentManager;
         chatCommandRegistry?: IChatCommandRegistry;
         attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+        inputToolbarFactory?: IInputToolbarRegistryFactory;
     }
 }
 export declare class LabChatModelFactory implements DocumentRegistry.IModelFactory<LabChatModel> {
@@ -122,4 +123,5 @@ export declare class LabChatModelFactory implements DocumentRegistry.IModelFacto
     private _commands?;
     private _activeCellManager;
     private _selectionWatcher;
+    private _documentManager;
 }

package/lib/factory.js

@@ -50,9 +50,9 @@ export class ChatWidgetFactory extends ABCWidgetFactory {
         super(options);
         this._themeManager = options.themeManager;
         this._rmRegistry = options.rmRegistry;
-        this._documentManager = options.documentManager;
         this._chatCommandRegistry = options.chatCommandRegistry;
         this._attachmentOpenerRegistry = options.attachmentOpenerRegistry;
+        this._inputToolbarFactory = options.inputToolbarFactory;
     }
     /**
      * Create a new widget given a context.
@@ -63,9 +63,11 @@ export class ChatWidgetFactory extends ABCWidgetFactory {
     createNewWidget(context) {
         context.rmRegistry = this._rmRegistry;
         context.themeManager = this._themeManager;
-        context.documentManager = this._documentManager;
         context.chatCommandRegistry = this._chatCommandRegistry;
         context.attachmentOpenerRegistry = this._attachmentOpenerRegistry;
+        if (this._inputToolbarFactory) {
+            context.inputToolbarRegistry = this._inputToolbarFactory.create();
+        }
         return new LabChatPanel({
             context,
             content: new ChatWidget(context)
@@ -74,7 +76,7 @@ export class ChatWidgetFactory extends ABCWidgetFactory {
 }
 export class LabChatModelFactory {
     constructor(options) {
-        var _a, _b;
+        var _a, _b, _c;
         this.collaborative = true;
         this._disposed = false;
         this._user = options.user;
@@ -82,6 +84,7 @@ export class LabChatModelFactory {
         this._commands = options.commands;
         this._activeCellManager = (_a = options.activeCellManager) !== null && _a !== void 0 ? _a : null;
         this._selectionWatcher = (_b = options.selectionWatcher) !== null && _b !== void 0 ? _b : null;
+        this._documentManager = (_c = options.documentManager) !== null && _c !== void 0 ? _c : null;
     }
     /**
      * The name of the model.
@@ -144,7 +147,8 @@ export class LabChatModelFactory {
             widgetConfig: this._widgetConfig,
             commands: this._commands,
             activeCellManager: this._activeCellManager,
-            selectionWatcher: this._selectionWatcher
+            selectionWatcher: this._selectionWatcher,
+            documentManager: this._documentManager
         });
     }
 }

package/lib/model.d.ts

@@ -1,4 +1,4 @@
-import { ChatModel, IChatMessage, IInputModel, INewMessage, IUser } from '@jupyter/chat';
+import { AbstractChatModel, AbstractChatContext, IChatContext, IChatMessage, IChatModel, IInputModel, INewMessage, IUser } from '@jupyter/chat';
 import { IChangedArgs } from '@jupyterlab/coreutils';
 import { DocumentRegistry } from '@jupyterlab/docregistry';
 import { User } from '@jupyterlab/services';
@@ -10,7 +10,7 @@ import { YChat } from './ychat';
  * Chat model namespace.
  */
 export declare namespace LabChatModel {
-    interface IOptions extends ChatModel.IOptions {
+    interface IOptions extends IChatModel.IOptions {
         widgetConfig: IWidgetConfig;
         user: User.IIdentity | null;
         sharedModel?: YChat;
@@ -20,7 +20,7 @@ export declare namespace LabChatModel {
 /**
  * The chat model.
  */
-export declare class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
+export declare class LabChatModel extends AbstractChatModel implements DocumentRegistry.IModel {
     constructor(options: LabChatModel.IOptions);
     readonly collaborative = true;
     get user(): IUser;
@@ -38,8 +38,13 @@ export declare class LabChatModel extends ChatModel implements DocumentRegistry.
     fromString(data: string): void;
     toJSON(): PartialJSONObject;
     fromJSON(data: PartialJSONObject): void;
-    messagesInserted(index: number, messages: IChatMessage[]): void;
+    createChatContext(): IChatContext;
+    messagesInserted(index: number, messages: IChatMessage[]): Promise<void>;
     sendMessage(message: INewMessage): Promise<boolean | void> | boolean | void;
+    /**
+     * Override the clear messages method.
+     */
+    clearMessages(): void;
     updateMessage(id: string, updatedMessage: IChatMessage): Promise<boolean | void> | boolean | void;
     deleteMessage(id: string): Promise<boolean | void> | boolean | void;
     /**
@@ -51,6 +56,7 @@ export declare class LabChatModel extends ChatModel implements DocumentRegistry.
      * Used to populate the writers list.
      */
     onAwarenessChange: () => void;
+    private _buildMentionList;
     private _resetWritingStatus;
     private _onchange;
     readonly defaultKernelName: string;
@@ -65,3 +71,12 @@ export declare class LabChatModel extends ChatModel implements DocumentRegistry.
     private _timeoutWriting;
     private _user;
 }
+/**
+ * The chat context to be sent to the input model.
+ */
+export declare class LabChatContext extends AbstractChatContext {
+    /**
+     * The list of users who have connected to this chat.
+     */
+    get users(): IUser[];
+}

package/lib/model.js

@@ -2,7 +2,7 @@
  * Copyright (c) Jupyter Development Team.
  * Distributed under the terms of the Modified BSD License.
  */
-import { ChatModel } from '@jupyter/chat';
+import { AbstractChatModel, AbstractChatContext } from '@jupyter/chat';
 import { PromiseDelegate, UUID } from '@lumino/coreutils';
 import { Signal } from '@lumino/signaling';
 import { YChat } from './ychat';
@@ -10,7 +10,7 @@ const WRITING_DELAY = 1000;
 /**
  * The chat model.
  */
-export class LabChatModel extends ChatModel {
+export class LabChatModel extends AbstractChatModel {
     constructor(options) {
         super(options);
         this.collaborative = true;
@@ -48,17 +48,17 @@ export class LabChatModel extends ChatModel {
             }
             this.updateWriters(writers);
         };
-        this._onchange = (_, changes) => {
+        this._onchange = async (_, changes) => {
             if (changes.messageChanges) {
                 const msgDelta = changes.messageChanges;
                 let index = 0;
-                msgDelta.forEach(delta => {
+                for (const delta of msgDelta) {
                     if (delta.retain) {
                         index += delta.retain;
                     }
                     else if (delta.insert) {
                         const messages = delta.insert.map(ymessage => {
-                            const { sender, attachments: attachmentIds, ...baseMessage } = ymessage;
+                            const { sender, attachments: attachmentIds, mentions: mentionsIds, ...baseMessage } = ymessage;
                             // Build the base message with sender.
                             const msg = {
                                 ...baseMessage,
@@ -79,15 +79,21 @@ export class LabChatModel extends ChatModel {
                                     msg.attachments = attachments;
                                 }
                             }
+                            const mentions = mentionsIds === null || mentionsIds === void 0 ? void 0 : mentionsIds.map(user => this.sharedModel.getUser(user) || {
+                                username: 'User undefined'
+                            });
+                            if (mentions === null || mentions === void 0 ? void 0 : mentions.length) {
+                                msg.mentions = mentions;
+                            }
                             return msg;
                         });
-                        this.messagesInserted(index, messages);
+                        await this.messagesInserted(index, messages);
                         index += messages.length;
                     }
                     else if (delta.delete) {
                         this.messagesDeleted(index, delta.delete);
                     }
-                });
+                }
             }
             if (changes.metadataChanges) {
                 changes.metadataChanges.forEach(change => {
@@ -98,6 +104,14 @@ export class LabChatModel extends ChatModel {
                     }
                 });
             }
+            if (changes.userChanges) {
+                // Update the current user if it changes (if it has been mentioned for example).
+                changes.userChanges.forEach(change => {
+                    if (change.key === this._user.username && change.newValue) {
+                        this._user = change.newValue;
+                    }
+                });
+            }
         };
         this.defaultKernelName = '';
         this.defaultKernelLanguage = '';
@@ -178,14 +192,18 @@ export class LabChatModel extends ChatModel {
     fromJSON(data) {
         // nothing to do
     }
-    messagesInserted(index, messages) {
+    createChatContext() {
+        return new LabChatContext({ model: this });
+    }
+    async messagesInserted(index, messages) {
         // Ensure the chat has an ID before inserting the messages, to properly catch the
         // unread messages (the last read message is saved using the chat ID).
-        this._ready.promise.then(() => {
+        return this._ready.promise.then(() => {
             super.messagesInserted(index, messages);
         });
     }
     sendMessage(message) {
+        var _a;
         this._resetWritingStatus();
         if (this._timeoutWriting !== null) {
             window.clearTimeout(this._timeoutWriting);
@@ -203,13 +221,25 @@ export class LabChatModel extends ChatModel {
             this.sharedModel.setUser(this._user);
         }
         // Add the attachments to the message.
-        if (this.input.attachments.length) {
-            const attachmentIds = this.input.attachments.map(attachment => this.sharedModel.setAttachment(attachment));
+        const attachmentIds = (_a = this.input.attachments) === null || _a === void 0 ? void 0 : _a.map(attachment => this.sharedModel.setAttachment(attachment));
+        if (attachmentIds === null || attachmentIds === void 0 ? void 0 : attachmentIds.length) {
             msg.attachments = attachmentIds;
-            this.input.clearAttachments();
         }
+        this.input.clearAttachments();
+        // Add the mentioned users.
+        const mentions = this._buildMentionList(this.input.mentions, message.body);
+        if (mentions.length) {
+            msg.mentions = mentions;
+        }
+        this.input.clearMentions();
         this.sharedModel.addMessage(msg);
     }
+    /**
+     * Override the clear messages method.
+     */
+    clearMessages() {
+        // No-op as we may not need to clear the messages in file based chat.
+    }
     updateMessage(id, updatedMessage) {
         var _a;
         const index = this.sharedModel.getMessageIndex(id);
@@ -229,10 +259,22 @@ export class LabChatModel extends ChatModel {
                 edited: true
             };
         }
+        // Update the attachments.
         const attachmentIds = (_a = updatedMessage.attachments) === null || _a === void 0 ? void 0 : _a.map(attachment => this.sharedModel.setAttachment(attachment));
-        if (attachmentIds) {
+        if (attachmentIds === null || attachmentIds === void 0 ? void 0 : attachmentIds.length) {
             message.attachments = attachmentIds;
         }
+        else {
+            delete message.attachments;
+        }
+        // Update the mentioned users.
+        const mentions = this._buildMentionList(updatedMessage.mentions, updatedMessage.body);
+        if (mentions.length) {
+            message.mentions = mentions;
+        }
+        else {
+            delete message.mentions;
+        }
         this.sharedModel.updateMessage(index, message);
     }
     deleteMessage(id) {
@@ -246,6 +288,28 @@ export class LabChatModel extends ChatModel {
         message.deleted = true;
         this.sharedModel.updateMessage(index, message);
     }
+    _buildMentionList(userMentions, body) {
+        if (!userMentions) {
+            return [];
+        }
+        const mentions = [];
+        userMentions.forEach(user => {
+            // Make sure the user is still mentioned.
+            if (!user.mention_name) {
+                return;
+            }
+            const regex = new RegExp(user.mention_name);
+            if (!regex.exec(body)) {
+                return;
+            }
+            // Save the mention name if necessary.
+            if (!(this.sharedModel.getUser(user.username) === user)) {
+                this.sharedModel.setUser(user);
+            }
+            mentions.push(user.username);
+        });
+        return mentions;
+    }
     _resetWritingStatus() {
         const awareness = this.sharedModel.awareness;
         const states = awareness.getLocalState();
@@ -254,3 +318,28 @@ export class LabChatModel extends ChatModel {
         this._timeoutWriting = null;
     }
 }
+/**
+ * The chat context to be sent to the input model.
+ */
+export class LabChatContext extends AbstractChatContext {
+    /**
+     * The list of users who have connected to this chat.
+     */
+    get users() {
+        const model = this._model;
+        const users = new Set();
+        // Get the user list from the shared YChat
+        Object.values(model.sharedModel.users).forEach(userObject => {
+            users.add(userObject);
+        });
+        // Add the users connected to the chat (even if they never sent a message).
+        model.sharedModel.awareness.getStates().forEach(value => {
+            const user = value.user;
+            if (!user) {
+                return;
+            }
+            users.add(user);
+        });
+        return Array.from(users);
+    }
+}

package/lib/token.d.ts

@@ -1,4 +1,4 @@
-import { IConfig, IActiveCellManager, ISelectionWatcher, ChatWidget } from '@jupyter/chat';
+import { IConfig, IActiveCellManager, ISelectionWatcher, ChatWidget, IInputToolbarRegistry } from '@jupyter/chat';
 import { WidgetTracker } from '@jupyterlab/apputils';
 import { DocumentRegistry } from '@jupyterlab/docregistry';
 import { Token } from '@lumino/coreutils';
@@ -89,3 +89,16 @@ export declare const IActiveCellManagerToken: Token<IActiveCellManager>;
  * The selection watcher plugin.
  */
 export declare const ISelectionWatcherToken: Token<ISelectionWatcher>;
+/**
+ * The input toolbar registry factory.
+ */
+export interface IInputToolbarRegistryFactory {
+    /**
+     * Create an input toolbar registry.
+     */
+    create: () => IInputToolbarRegistry;
+}
+/**
+ * The token of the factory to create an input toolbar registry.
+ */
+export declare const IInputToolbarRegistryFactory: Token<IInputToolbarRegistryFactory>;

package/lib/token.js

@@ -57,3 +57,7 @@ export const IActiveCellManagerToken = new Token('jupyterlab-chat:IActiveCellMan
  * The selection watcher plugin.
  */
 export const ISelectionWatcherToken = new Token('jupyterlab-chat:ISelectionWatcher');
+/**
+ * The token of the factory to create an input toolbar registry.
+ */
+export const IInputToolbarRegistryFactory = new Token('jupyterlab-chat:IInputToolbarRegistryFactory');

package/lib/widget.d.ts

@@ -8,7 +8,7 @@ import { CommandRegistry } from '@lumino/commands';
 import { Message } from '@lumino/messaging';
 import { Panel } from '@lumino/widgets';
 import { LabChatModel } from './model';
-import { IDocumentManager } from '@jupyterlab/docmanager';
+import { IInputToolbarRegistryFactory } from './token';
 /**
  * DocumentWidget: widget that represents the view or editor for a file type.
  */
@@ -88,9 +88,9 @@ export declare class ChatPanel extends SidePanel {
     private _openChat;
     private _rmRegistry;
     private _themeManager;
-    private _documentManager?;
     private _chatCommandRegistry?;
     private _attachmentOpenerRegistry?;
+    private _inputToolbarFactory?;
 }
 /**
  * The chat panel namespace.
@@ -105,9 +105,9 @@ export declare namespace ChatPanel {
         rmRegistry: IRenderMimeRegistry;
         themeManager: IThemeManager | null;
         defaultDirectory: string;
-        documentManager?: IDocumentManager;
         chatCommandRegistry?: IChatCommandRegistry;
         attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+        inputToolbarFactory?: IInputToolbarRegistryFactory;
     }
 }
 /**

package/lib/widget.js

@@ -105,9 +105,9 @@ export class ChatPanel extends SidePanel {
         this._rmRegistry = options.rmRegistry;
         this._themeManager = options.themeManager;
         this._defaultDirectory = options.defaultDirectory;
-        this._documentManager = options.documentManager;
         this._chatCommandRegistry = options.chatCommandRegistry;
         this._attachmentOpenerRegistry = options.attachmentOpenerRegistry;
+        this._inputToolbarFactory = options.inputToolbarFactory;
         const addChat = new CommandToolbarButton({
             commands: this._commands,
             id: CommandIDs.createChat,
@@ -152,14 +152,19 @@ export class ChatPanel extends SidePanel {
         for (let i = 0; i < this.widgets.length; i++) {
             content.collapse(i);
         }
+        // Create the toolbar registry.
+        let inputToolbarRegistry;
+        if (this._inputToolbarFactory) {
+            inputToolbarRegistry = this._inputToolbarFactory.create();
+        }
         // Create a new widget.
         const widget = new ChatWidget({
             model: model,
             rmRegistry: this._rmRegistry,
             themeManager: this._themeManager,
-            documentManager: this._documentManager,
             chatCommandRegistry: this._chatCommandRegistry,
-            attachmentOpenerRegistry: this._attachmentOpenerRegistry
+            attachmentOpenerRegistry: this._attachmentOpenerRegistry,
+            inputToolbarRegistry
         });
         this.addWidget(new ChatSection({
             widget,

package/lib/ychat.d.ts

@@ -65,7 +65,7 @@ export declare class YChat extends YDocument<IChatChanges> {
      */
     static create(options?: YDocument.IOptions): YChat;
     get id(): string;
-    get users(): JSONObject;
+    get users(): Record<string, IUser>;
     get messages(): string[];
     get attachments(): JSONObject;
     getSource(): JSONObject;

package/lib/ychat.js

@@ -140,7 +140,8 @@ export class YChat extends YDocument {
         return this._metadata.get('id') || '';
     }
     get users() {
-        return JSONExt.deepCopy(this._users.toJSON());
+        const userDict = JSONExt.deepCopy(this._users.toJSON());
+        return userDict;
     }
     get messages() {
         return JSONExt.deepCopy(this._messages.toJSON());

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jupyterlab-chat",
-  "version": "0.8.1",
+  "version": "0.10.0",
   "description": "The library to build a chat based on shared document",
   "keywords": [
     "jupyter",
@@ -42,7 +42,7 @@
     "watch:src": "tsc -w --sourceMap"
   },
   "dependencies": {
-    "@jupyter/chat": "^0.8.1",
+    "@jupyter/chat": "^0.10.0",
     "@jupyter/collaborative-drive": "^3.0.0",
     "@jupyter/ydoc": "^2.0.0 || ^3.0.0",
     "@jupyterlab/application": "^4.2.0",

package/src/factory.ts

@@ -8,6 +8,7 @@ import {
   IActiveCellManager,
   IAttachmentOpenerRegistry,
   IChatCommandRegistry,
+  IInputToolbarRegistry,
   ISelectionWatcher
 } from '@jupyter/chat';
 import { IThemeManager } from '@jupyterlab/apputils';
@@ -21,7 +22,11 @@ import { ISignal, Signal } from '@lumino/signaling';
 import { LabChatModel } from './model';
 import { LabChatPanel } from './widget';
 import { YChat } from './ychat';
-import { ILabChatConfig, IWidgetConfig } from './token';
+import {
+  IInputToolbarRegistryFactory,
+  ILabChatConfig,
+  IWidgetConfig
+} from './token';
 
 /**
  * The object provided by the chatDocument extension.
@@ -76,9 +81,9 @@ export class ChatWidgetFactory extends ABCWidgetFactory<
     super(options);
     this._themeManager = options.themeManager;
     this._rmRegistry = options.rmRegistry;
-    this._documentManager = options.documentManager;
     this._chatCommandRegistry = options.chatCommandRegistry;
     this._attachmentOpenerRegistry = options.attachmentOpenerRegistry;
+    this._inputToolbarFactory = options.inputToolbarFactory;
   }
 
   /**
@@ -90,9 +95,11 @@ export class ChatWidgetFactory extends ABCWidgetFactory<
   protected createNewWidget(context: ChatWidgetFactory.IContext): LabChatPanel {
     context.rmRegistry = this._rmRegistry;
     context.themeManager = this._themeManager;
-    context.documentManager = this._documentManager;
     context.chatCommandRegistry = this._chatCommandRegistry;
     context.attachmentOpenerRegistry = this._attachmentOpenerRegistry;
+    if (this._inputToolbarFactory) {
+      context.inputToolbarRegistry = this._inputToolbarFactory.create();
+    }
     return new LabChatPanel({
       context,
       content: new ChatWidget(context)
@@ -101,9 +108,9 @@ export class ChatWidgetFactory extends ABCWidgetFactory<
 
   private _themeManager: IThemeManager | null;
   private _rmRegistry: IRenderMimeRegistry;
-  private _documentManager?: IDocumentManager;
   private _chatCommandRegistry?: IChatCommandRegistry;
   private _attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+  private _inputToolbarFactory?: IInputToolbarRegistryFactory;
 }
 
 export namespace ChatWidgetFactory {
@@ -113,15 +120,16 @@ export namespace ChatWidgetFactory {
     documentManager?: IDocumentManager;
     chatCommandRegistry?: IChatCommandRegistry;
     attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+    inputToolbarRegistry?: IInputToolbarRegistry;
   }
 
   export interface IOptions<T extends LabChatPanel>
     extends DocumentRegistry.IWidgetFactoryOptions<T> {
     themeManager: IThemeManager | null;
     rmRegistry: IRenderMimeRegistry;
-    documentManager?: IDocumentManager;
     chatCommandRegistry?: IChatCommandRegistry;
     attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+    inputToolbarFactory?: IInputToolbarRegistryFactory;
   }
 }
 
@@ -134,6 +142,7 @@ export class LabChatModelFactory
     this._commands = options.commands;
     this._activeCellManager = options.activeCellManager ?? null;
     this._selectionWatcher = options.selectionWatcher ?? null;
+    this._documentManager = options.documentManager ?? null;
   }
 
   collaborative = true;
@@ -206,7 +215,8 @@ export class LabChatModelFactory
       widgetConfig: this._widgetConfig,
       commands: this._commands,
       activeCellManager: this._activeCellManager,
-      selectionWatcher: this._selectionWatcher
+      selectionWatcher: this._selectionWatcher,
+      documentManager: this._documentManager
     });
   }
 
@@ -216,4 +226,5 @@ export class LabChatModelFactory
   private _commands?: CommandRegistry;
   private _activeCellManager: IActiveCellManager | null;
   private _selectionWatcher: ISelectionWatcher | null;
+  private _documentManager: IDocumentManager | null;
 }

package/src/model.ts

@@ -4,9 +4,12 @@
  */
 
 import {
-  ChatModel,
+  AbstractChatModel,
+  AbstractChatContext,
   IAttachment,
+  IChatContext,
   IChatMessage,
+  IChatModel,
   IInputModel,
   INewMessage,
   IUser
@@ -26,7 +29,7 @@ const WRITING_DELAY = 1000;
  * Chat model namespace.
  */
 export namespace LabChatModel {
-  export interface IOptions extends ChatModel.IOptions {
+  export interface IOptions extends IChatModel.IOptions {
     widgetConfig: IWidgetConfig;
     user: User.IIdentity | null;
     sharedModel?: YChat;
@@ -37,7 +40,10 @@ export namespace LabChatModel {
 /**
  * The chat model.
  */
-export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
+export class LabChatModel
+  extends AbstractChatModel
+  implements DocumentRegistry.IModel
+{
   constructor(options: LabChatModel.IOptions) {
     super(options);
 
@@ -133,10 +139,17 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
     // nothing to do
   }
 
-  messagesInserted(index: number, messages: IChatMessage[]): void {
+  createChatContext(): IChatContext {
+    return new LabChatContext({ model: this });
+  }
+
+  async messagesInserted(
+    index: number,
+    messages: IChatMessage[]
+  ): Promise<void> {
     // Ensure the chat has an ID before inserting the messages, to properly catch the
     // unread messages (the last read message is saved using the chat ID).
-    this._ready.promise.then(() => {
+    return this._ready.promise.then(() => {
       super.messagesInserted(index, messages);
     });
   }
@@ -162,17 +175,31 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
     }
 
     // Add the attachments to the message.
-    if (this.input.attachments.length) {
-      const attachmentIds = this.input.attachments.map(attachment =>
-        this.sharedModel.setAttachment(attachment)
-      );
+    const attachmentIds = this.input.attachments?.map(attachment =>
+      this.sharedModel.setAttachment(attachment)
+    );
+    if (attachmentIds?.length) {
       msg.attachments = attachmentIds;
-      this.input.clearAttachments();
     }
+    this.input.clearAttachments();
+
+    // Add the mentioned users.
+    const mentions = this._buildMentionList(this.input.mentions, message.body);
+    if (mentions.length) {
+      msg.mentions = mentions;
+    }
+    this.input.clearMentions();
 
     this.sharedModel.addMessage(msg);
   }
 
+  /**
+   * Override the clear messages method.
+   */
+  clearMessages(): void {
+    // No-op as we may not need to clear the messages in file based chat.
+  }
+
   updateMessage(
     id: string,
     updatedMessage: IChatMessage
@@ -194,12 +221,28 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
         edited: true
       };
     }
+
+    // Update the attachments.
     const attachmentIds = updatedMessage.attachments?.map(attachment =>
       this.sharedModel.setAttachment(attachment)
     );
-    if (attachmentIds) {
+    if (attachmentIds?.length) {
       message.attachments = attachmentIds;
+    } else {
+      delete message.attachments;
     }
+
+    // Update the mentioned users.
+    const mentions = this._buildMentionList(
+      updatedMessage.mentions,
+      updatedMessage.body
+    );
+    if (mentions.length) {
+      message.mentions = mentions;
+    } else {
+      delete message.mentions;
+    }
+
     this.sharedModel.updateMessage(index, message as IYmessage);
   }
 
@@ -251,6 +294,33 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
     this.updateWriters(writers);
   };
 
+  private _buildMentionList(
+    userMentions: IUser[] | undefined,
+    body: string
+  ): string[] {
+    if (!userMentions) {
+      return [];
+    }
+    const mentions: string[] = [];
+    userMentions.forEach(user => {
+      // Make sure the user is still mentioned.
+      if (!user.mention_name) {
+        return;
+      }
+      const regex = new RegExp(user.mention_name);
+      if (!regex.exec(body)) {
+        return;
+      }
+
+      // Save the mention name if necessary.
+      if (!(this.sharedModel.getUser(user.username) === user)) {
+        this.sharedModel.setUser(user);
+      }
+      mentions.push(user.username);
+    });
+    return mentions;
+  }
+
   private _resetWritingStatus() {
     const awareness = this.sharedModel.awareness;
     const states = awareness.getLocalState();
@@ -259,11 +329,11 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
     this._timeoutWriting = null;
   }
 
-  private _onchange = (_: YChat, changes: IChatChanges) => {
+  private _onchange = async (_: YChat, changes: IChatChanges) => {
     if (changes.messageChanges) {
       const msgDelta = changes.messageChanges;
       let index = 0;
-      msgDelta.forEach(delta => {
+      for (const delta of msgDelta) {
         if (delta.retain) {
           index += delta.retain;
         } else if (delta.insert) {
@@ -271,6 +341,7 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
             const {
               sender,
               attachments: attachmentIds,
+              mentions: mentionsIds,
               ...baseMessage
             } = ymessage;
 
@@ -296,14 +367,24 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
               }
             }
 
+            const mentions = mentionsIds?.map(
+              user =>
+                this.sharedModel.getUser(user) || {
+                  username: 'User undefined'
+                }
+            );
+
+            if (mentions?.length) {
+              msg.mentions = mentions;
+            }
             return msg;
           });
-          this.messagesInserted(index, messages);
+          await this.messagesInserted(index, messages);
           index += messages.length;
         } else if (delta.delete) {
           this.messagesDeleted(index, delta.delete);
         }
-      });
+      }
     }
 
     if (changes.metadataChanges) {
@@ -315,6 +396,15 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
         }
       });
     }
+
+    if (changes.userChanges) {
+      // Update the current user if it changes (if it has been mentioned for example).
+      changes.userChanges.forEach(change => {
+        if (change.key === this._user.username && change.newValue) {
+          this._user = change.newValue;
+        }
+      });
+    }
   };
 
   readonly defaultKernelName: string = '';
@@ -332,3 +422,32 @@ export class LabChatModel extends ChatModel implements DocumentRegistry.IModel {
 
   private _user: IUser;
 }
+
+/**
+ * The chat context to be sent to the input model.
+ */
+export class LabChatContext extends AbstractChatContext {
+  /**
+   * The list of users who have connected to this chat.
+   */
+  get users(): IUser[] {
+    const model = this._model as LabChatModel;
+    const users = new Set<IUser>();
+
+    // Get the user list from the shared YChat
+    Object.values(model.sharedModel.users).forEach(userObject => {
+      users.add(userObject);
+    });
+
+    // Add the users connected to the chat (even if they never sent a message).
+    model.sharedModel.awareness.getStates().forEach(value => {
+      const user = value.user as IUser;
+      if (!user) {
+        return;
+      }
+      users.add(user);
+    });
+
+    return Array.from(users);
+  }
+}

package/src/token.ts

@@ -8,7 +8,8 @@ import {
   chatIcon,
   IActiveCellManager,
   ISelectionWatcher,
-  ChatWidget
+  ChatWidget,
+  IInputToolbarRegistry
 } from '@jupyter/chat';
 import { WidgetTracker } from '@jupyterlab/apputils';
 import { DocumentRegistry } from '@jupyterlab/docregistry';
@@ -125,3 +126,21 @@ export const IActiveCellManagerToken = new Token<IActiveCellManager>(
 export const ISelectionWatcherToken = new Token<ISelectionWatcher>(
   'jupyterlab-chat:ISelectionWatcher'
 );
+
+/**
+ * The input toolbar registry factory.
+ */
+export interface IInputToolbarRegistryFactory {
+  /**
+   * Create an input toolbar registry.
+   */
+  create: () => IInputToolbarRegistry;
+}
+
+/**
+ * The token of the factory to create an input toolbar registry.
+ */
+export const IInputToolbarRegistryFactory =
+  new Token<IInputToolbarRegistryFactory>(
+    'jupyterlab-chat:IInputToolbarRegistryFactory'
+  );

package/src/widget.tsx

@@ -8,6 +8,7 @@ import {
   IAttachmentOpenerRegistry,
   IChatCommandRegistry,
   IChatModel,
+  IInputToolbarRegistry,
   readIcon
 } from '@jupyter/chat';
 import { ICollaborativeDrive } from '@jupyter/collaborative-drive';
@@ -33,8 +34,11 @@ import { AccordionPanel, Panel } from '@lumino/widgets';
 import React, { useState } from 'react';
 
 import { LabChatModel } from './model';
-import { CommandIDs, chatFileType } from './token';
-import { IDocumentManager } from '@jupyterlab/docmanager';
+import {
+  CommandIDs,
+  IInputToolbarRegistryFactory,
+  chatFileType
+} from './token';
 
 const MAIN_PANEL_CLASS = 'jp-lab-chat-main-panel';
 const TITLE_UNREAD_CLASS = 'jp-lab-chat-title-unread';
@@ -104,9 +108,9 @@ export class ChatPanel extends SidePanel {
     this._rmRegistry = options.rmRegistry;
     this._themeManager = options.themeManager;
     this._defaultDirectory = options.defaultDirectory;
-    this._documentManager = options.documentManager;
     this._chatCommandRegistry = options.chatCommandRegistry;
     this._attachmentOpenerRegistry = options.attachmentOpenerRegistry;
+    this._inputToolbarFactory = options.inputToolbarFactory;
 
     const addChat = new CommandToolbarButton({
       commands: this._commands,
@@ -163,14 +167,20 @@ export class ChatPanel extends SidePanel {
       content.collapse(i);
     }
 
+    // Create the toolbar registry.
+    let inputToolbarRegistry: IInputToolbarRegistry | undefined;
+    if (this._inputToolbarFactory) {
+      inputToolbarRegistry = this._inputToolbarFactory.create();
+    }
+
     // Create a new widget.
     const widget = new ChatWidget({
       model: model,
       rmRegistry: this._rmRegistry,
       themeManager: this._themeManager,
-      documentManager: this._documentManager,
       chatCommandRegistry: this._chatCommandRegistry,
-      attachmentOpenerRegistry: this._attachmentOpenerRegistry
+      attachmentOpenerRegistry: this._attachmentOpenerRegistry,
+      inputToolbarRegistry
     });
 
     this.addWidget(
@@ -289,9 +299,9 @@ export class ChatPanel extends SidePanel {
   private _openChat: ReactWidget;
   private _rmRegistry: IRenderMimeRegistry;
   private _themeManager: IThemeManager | null;
-  private _documentManager?: IDocumentManager;
   private _chatCommandRegistry?: IChatCommandRegistry;
   private _attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+  private _inputToolbarFactory?: IInputToolbarRegistryFactory;
 }
 
 /**
@@ -307,9 +317,9 @@ export namespace ChatPanel {
     rmRegistry: IRenderMimeRegistry;
     themeManager: IThemeManager | null;
     defaultDirectory: string;
-    documentManager?: IDocumentManager;
     chatCommandRegistry?: IChatCommandRegistry;
     attachmentOpenerRegistry?: IAttachmentOpenerRegistry;
+    inputToolbarFactory?: IInputToolbarRegistryFactory;
   }
 }
 

package/src/ychat.ts

@@ -100,8 +100,13 @@ export class YChat extends YDocument<IChatChanges> {
     return (this._metadata.get('id') as string) || '';
   }
 
-  get users(): JSONObject {
-    return JSONExt.deepCopy(this._users.toJSON());
+  get users(): Record<string, IUser> {
+    const userDict = JSONExt.deepCopy(this._users.toJSON()) as Record<
+      string,
+      IUser
+    >;
+
+    return userDict;
   }
 
   get messages(): string[] {