@jupyterlite/ai 0.11.1 → 0.13.0

package/src/chat-commands/clear.ts

@@ -0,0 +1,46 @@
+import { ChatCommand, IChatCommandProvider, IInputModel } from '@jupyter/chat';
+
+import { AIChatModel } from '../chat-model';
+
+export class ClearCommandProvider implements IChatCommandProvider {
+  public id: string = '@jupyterlite/ai:clear-command';
+
+  async listCommandCompletions(
+    inputModel: IInputModel
+  ): Promise<ChatCommand[]> {
+    const match = inputModel.currentWord?.match(this._regex)?.[0];
+    if (!match) {
+      return [];
+    }
+
+    if (this._command.name.startsWith(match)) {
+      return [this._command];
+    }
+
+    return [];
+  }
+
+  async onSubmit(inputModel: IInputModel): Promise<void> {
+    const trimmed = inputModel.value.trim();
+    if (trimmed !== this._command.name) {
+      return;
+    }
+
+    const context = inputModel.chatContext as
+      | AIChatModel.IAIChatContext
+      | undefined;
+    context?.clearMessages?.();
+
+    inputModel.value = '';
+    inputModel.clearAttachments();
+    inputModel.clearMentions();
+  }
+
+  private _command: ChatCommand = {
+    name: '/clear',
+    providerId: this.id,
+    description: 'Clear the current chat history'
+  };
+
+  private _regex: RegExp = /^\/\w*$/;
+}

package/src/chat-commands/index.ts

@@ -0,0 +1,2 @@
+export * from './clear';
+export * from './skills';

package/src/chat-commands/skills.ts

@@ -0,0 +1,87 @@
+import { ChatCommand, IChatCommandProvider, IInputModel } from '@jupyter/chat';
+
+import { CommandRegistry } from '@lumino/commands';
+
+import { AIChatModel } from '../chat-model';
+import { CommandIds, ISkillRegistry } from '../tokens';
+
+export class SkillsCommandProvider implements IChatCommandProvider {
+  constructor(options: SkillsCommandProvider.IOptions) {
+    this._skillRegistry = options.skillRegistry;
+    this._commands = options.commands;
+  }
+
+  public id: string = '@jupyterlite/ai:skills-command';
+
+  async listCommandCompletions(
+    inputModel: IInputModel
+  ): Promise<ChatCommand[]> {
+    const match = inputModel.currentWord?.match(this._regex)?.[0];
+    if (!match) {
+      return [];
+    }
+
+    if (this._command.name.startsWith(match)) {
+      return [this._command];
+    }
+
+    return [];
+  }
+
+  async onSubmit(inputModel: IInputModel): Promise<void> {
+    const trimmed = inputModel.value.trim();
+    const match = trimmed.match(/^\/skills(?:\s+(.+))?$/);
+    if (!match) {
+      return;
+    }
+
+    // Refresh skills from disk before listing
+    if (this._commands.hasCommand(CommandIds.refreshSkills)) {
+      await this._commands.execute(CommandIds.refreshSkills);
+    }
+
+    const query = match[1]?.trim();
+    const filtered = this._skillRegistry.listSkills(query);
+
+    let body = '';
+    if (filtered.length === 0) {
+      body = query
+        ? `No skills found matching "${query}".`
+        : 'No skills are currently registered.';
+    } else {
+      const heading = query
+        ? `Skills matching "${query}" (${filtered.length}):`
+        : `Available skills (${filtered.length}):`;
+      const lines = filtered.map(
+        skill => `- \`${skill.name}\` — ${skill.description}`
+      );
+      body = [heading, '', ...lines].join('\n');
+    }
+
+    const context = inputModel.chatContext as
+      | AIChatModel.IAIChatContext
+      | undefined;
+    context?.addSystemMessage?.(body);
+
+    inputModel.value = '';
+    inputModel.clearAttachments();
+    inputModel.clearMentions();
+  }
+
+  private _command: ChatCommand = {
+    name: '/skills',
+    providerId: this.id,
+    description: 'List available skills'
+  };
+
+  private _regex: RegExp = /^\/\w*$/;
+  private _commands: CommandRegistry;
+  private _skillRegistry: ISkillRegistry;
+}
+
+export namespace SkillsCommandProvider {
+  export interface IOptions {
+    skillRegistry: ISkillRegistry;
+    commands: CommandRegistry;
+  }
+}

package/src/{chat-model-registry.ts → chat-model-handler.ts}

@@ -4,31 +4,32 @@ import { AgentManagerFactory } from './agent';
 import { AIChatModel } from './chat-model';
 import { AISettingsModel } from './models/settings-model';
 import {
-  IChatModelRegistry,
+  IChatModelHandler,
   IProviderRegistry,
   ITokenUsage,
   IToolRegistry
 } from './tokens';
 import { IDocumentManager } from '@jupyterlab/docmanager';
-import { UUID } from '@lumino/coreutils';
+import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 
 /**
- * The chat model registry.
+ * The chat model handler.
  */
-export class ChatModelRegistry implements IChatModelRegistry {
-  constructor(options: ChatModelRegistry.IOptions) {
+export class ChatModelHandler implements IChatModelHandler {
+  constructor(options: ChatModelHandler.IOptions) {
     this._docManager = options.docManager;
     this._agentManagerFactory = options.agentManagerFactory;
     this._settingsModel = options.settingsModel;
     this._toolRegistry = options.toolRegistry;
     this._providerRegistry = options.providerRegistry;
+    this._rmRegistry = options.rmRegistry;
     this._activeCellManager = options.activeCellManager;
     this._trans = options.trans;
   }
 
   createModel(
-    name?: string,
-    activeProvider?: string,
+    name: string,
+    activeProvider: string,
     tokenUsage?: ITokenUsage
   ): AIChatModel {
     // Create Agent Manager first so it can be shared
@@ -37,7 +38,8 @@ export class ChatModelRegistry implements IChatModelRegistry {
       toolRegistry: this._toolRegistry,
       providerRegistry: this._providerRegistry,
       activeProvider,
-      tokenUsage
+      tokenUsage,
+      renderMimeRegistry: this._rmRegistry
     });
 
     // Create AI chat model
@@ -50,52 +52,11 @@ export class ChatModelRegistry implements IChatModelRegistry {
       trans: this._trans
     });
 
-    // Set the name of the chat if not provided.
-    // The name will be the name set by the user to the model if not already used by
-    // another chat.
-    if (!name || this._models.findIndex(m => m.name === name) !== -1) {
-      const existingName = this.getAll().map(model => model.name);
-
-      const modelName =
-        this._settingsModel.getProvider(agentManager.activeProvider)?.name ||
-        UUID.uuid4();
-      name = modelName;
-      let i = 1;
-      while (existingName.includes(name)) {
-        name = `${modelName}-${i}`;
-        i += 1;
-      }
-    }
     model.name = name;
-    this.add(model);
 
     return model;
   }
 
-  add(model: AIChatModel): void {
-    if (!this._models.find(m => m.name === model.name)) {
-      this._models.push(model);
-      model.disposed.connect(() => {
-        this.remove(model.name);
-      });
-    }
-  }
-
-  get(name: string): AIChatModel | undefined {
-    return this._models.find(m => m.name === name);
-  }
-
-  getAll(): AIChatModel[] {
-    return this._models;
-  }
-
-  remove(name: string): void {
-    const index = this._models.findIndex(m => m.name === name);
-    if (index !== -1) {
-      this._models.splice(index, 1);
-    }
-  }
-
   /**
    * Getter/setter for the active cell manager.
    */
@@ -106,17 +67,17 @@ export class ChatModelRegistry implements IChatModelRegistry {
     this._activeCellManager = manager;
   }
 
-  private _models: AIChatModel[] = [];
   private _docManager: IDocumentManager;
   private _agentManagerFactory: AgentManagerFactory;
   private _settingsModel: AISettingsModel;
   private _toolRegistry?: IToolRegistry;
   private _providerRegistry?: IProviderRegistry;
+  private _rmRegistry: IRenderMimeRegistry;
   private _activeCellManager?: ActiveCellManager;
   private _trans: TranslationBundle;
 }
 
-export namespace ChatModelRegistry {
+export namespace ChatModelHandler {
   export interface IOptions {
     /**
      * The document manager.
@@ -138,6 +99,10 @@ export namespace ChatModelRegistry {
      * Optional provider registry for model creation
      */
     providerRegistry?: IProviderRegistry;
+    /**
+     * Render mime registry.
+     */
+    rmRegistry: IRenderMimeRegistry;
     /**
      * The active cell manager.
      */