move registry to jupyter-chat

Author: dlqqq

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

@@ -4,4 +4,4 @@
  */
 
 export * from './types';
-export * from './tokens';
+export * from './registry';

packages/jupyter-chat/src/chat-commands/tokens.ts renamed to packages/jupyter-chat/src/chat-commands/registry.ts

@@ -26,6 +26,42 @@ export interface IChatCommandRegistry {
   ): void;
 }
 
+/**
+ * Default chat command registry implementation.
+ */
+export class ChatCommandRegistry implements IChatCommandRegistry {
+  constructor() {
+    this._providers = new Map<string, IChatCommandProvider>();
+  }
+
+  addProvider(provider: IChatCommandProvider): void {
+    this._providers.set(provider.id, provider);
+  }
+
+  getProviders(): IChatCommandProvider[] {
+    return Array.from(this._providers.values());
+  }
+
+  handleChatCommand(
+    command: ChatCommand,
+    currentWord: string,
+    replaceCurrentWord: (newWord: string) => void
+  ) {
+    const provider = this._providers.get(command.providerId);
+    if (!provider) {
+      console.error(
+        'Error in handling chat command: No command provider has an ID of ' +
+          command.providerId
+      );
+      return;
+    }
+
+    provider.handleChatCommand(command, currentWord, replaceCurrentWord);
+  }
+
+  private _providers: Map<string, IChatCommandProvider>;
+}
+
 export const IChatCommandRegistry = new Token<IChatCommandRegistry>(
   '@jupyter/chat:IChatCommandRegistry'
 );

python/jupyterlab-chat/src/chat-commands/plugins.ts

@@ -5,8 +5,7 @@
 
 import { JupyterFrontEndPlugin } from '@jupyterlab/application';
 
-import { IChatCommandRegistry } from '@jupyter/chat';
-import { ChatCommandRegistry } from './registry';
+import { IChatCommandRegistry, ChatCommandRegistry } from '@jupyter/chat';
 
 export const chatCommandRegistryPlugin: JupyterFrontEndPlugin<IChatCommandRegistry> =
   {

python/jupyterlab-chat/src/chat-commands/registry.ts

@@ -2,42 +2,3 @@
  * Copyright (c) Jupyter Development Team.
  * Distributed under the terms of the Modified BSD License.
  */
-
-import {
-  ChatCommand,
-  IChatCommandRegistry,
-  IChatCommandProvider
-} from '@jupyter/chat';
-
-export class ChatCommandRegistry implements IChatCommandRegistry {
-  constructor() {
-    this._providers = new Map<string, IChatCommandProvider>();
-  }
-
-  addProvider(provider: IChatCommandProvider): void {
-    this._providers.set(provider.id, provider);
-  }
-
-  getProviders(): IChatCommandProvider[] {
-    return Array.from(this._providers.values());
-  }
-
-  handleChatCommand(
-    command: ChatCommand,
-    currentWord: string,
-    replaceCurrentWord: (newWord: string) => void
-  ) {
-    const provider = this._providers.get(command.providerId);
-    if (!provider) {
-      console.error(
-        'Error in handling chat command: No command provider has an ID of ' +
-          command.providerId
-      );
-      return;
-    }
-
-    provider.handleChatCommand(command, currentWord, replaceCurrentWord);
-  }
-
-  private _providers: Map<string, IChatCommandProvider>;
-}