Passes the input model to the getChatCommand() method

Author: brichet

packages/jupyter-chat/src/chat-commands/types.ts

@@ -50,21 +50,15 @@ export interface IChatCommandProvider {
   id: string;
 
   /**
-   * Async function which accepts the current word and returns a list of
+   * Async function which accepts the input model and returns a list of
    * valid chat commands that match the current word. The current word is
    * space-separated word at the user's cursor.
-   *
-   * TODO: Pass a ChatModel/InputModel instance here to give the command access
-   * to more than the current word.
    */
-  getChatCommands(currentWord: string): Promise<ChatCommand[]>;
+  getChatCommands(inputModel: IInputModel): Promise<ChatCommand[]>;
 
   /**
    * Function called when a chat command is run by the user through the chat
    * commands menu.
-   *
-   * TODO: Pass a ChatModel/InputModel instance here to provide a function to
-   * replace the current word.
    */
   handleChatCommand(
     command: ChatCommand,

packages/jupyter-chat/src/components/input/use-chat-commands.tsx

@@ -63,7 +63,7 @@ export function useChatCommands(
         // TODO: optimize performance when this method is truly async
         try {
           newCommands = newCommands.concat(
-            await provider.getChatCommands(currentWord)
+            await provider.getChatCommands(inputModel)
           );
         } catch (e) {
           console.error(

python/jupyterlab-chat/src/chat-commands/providers/emoji.ts

@@ -47,8 +47,8 @@ export class EmojiCommandProvider implements IChatCommandProvider {
   // regex used to test the current word
   private _regex: RegExp = /^:\w*:?/;
 
-  async getChatCommands(currentWord: string) {
-    const match = currentWord.match(this._regex)?.[0];
+  async getChatCommands(inputModel: IInputModel) {
+    const match = inputModel.currentWord?.match(this._regex)?.[0];
     if (!match) {
       return [];
     }