Chat tracker token

packages/jupyter-chat/src/index.ts

@@ -11,5 +11,6 @@ export * from './markdown-renderer';
 export * from './model';
 export * from './registers';
 export * from './selection-watcher';
+export * from './tokens';
 export * from './types';
 export * from './widgets';

packages/jupyter-chat/src/tokens.ts

@@ -0,0 +1,24 @@
+/*
+ * Copyright (c) Jupyter Development Team.
+ * Distributed under the terms of the Modified BSD License.
+ */
+
+import { IWidgetTracker, MainAreaWidget } from '@jupyterlab/apputils';
+import { Token } from '@lumino/coreutils';
+
+import { ChatWidget } from './widgets';
+
+/**
+ * the chat tracker type.
+ */
+export type IChatTracker = IWidgetTracker<
+  ChatWidget | MainAreaWidget<ChatWidget>
+>;
+
+/**
+ * A chat tracker token.
+ */
+export const IChatTracker = new Token<IChatTracker>(
+  '@jupyter/chat:IChatTracker',
+  'The chat widget tracker'
+);

packages/jupyter-chat/src/widgets/chat-widget.tsx

@@ -15,6 +15,7 @@ import { Chat, IInputToolbarRegistry, MESSAGE_CLASS } from '../components';
 import { chatIcon } from '../icons';
 import { IChatModel } from '../model';
 import {
+  ChatArea,
   IFileAttachment,
   INotebookAttachment,
   INotebookAttachmentCell
@@ -71,6 +72,13 @@ export class ChatWidget extends ReactWidget {
     return this._chatOptions.model;
   }
 
+  /**
+   * The area where the chat is opened.
+   */
+  get area(): ChatArea | undefined {
+    return this._chatOptions.area;
+  }
+
   /**
    * Get the input toolbar registry (if it has been provided when creating the widget).
    */