@jupyterlite/ai 0.16.0 → 0.18.0

package/src/tokens.ts

@@ -1,14 +1,13 @@
-import { ActiveCellManager, IMessage, IMessageContent } from '@jupyter/chat';
+import { ActiveCellManager, IChatModel, IMessage } from '@jupyter/chat';
 import { VDomRenderer } from '@jupyterlab/apputils';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { Token } from '@lumino/coreutils';
 import type { IDisposable } from '@lumino/disposable';
 import { ISignal } from '@lumino/signaling';
-import type { Tool, LanguageModel } from 'ai';
+import type { Tool, LanguageModel, UserContent, ModelMessage } from 'ai';
 import { ISecretsManager } from 'jupyter-secrets-manager';
 
 import type { IModelOptions } from './providers/models';
-import { AIChatModel } from './chat-model';
 import type {
   ISkillDefinition,
   ISkillRegistration,
@@ -30,6 +29,7 @@ export namespace CommandIds {
   export const openSettings = '@jupyterlite/ai:open-settings';
   export const reposition = '@jupyterlite/ai:reposition';
   export const openChat = '@jupyterlite/ai:open-chat';
+  export const openOrRevealChat = '@jupyterlite/ai:open-or-reveal-chat';
   export const moveChat = '@jupyterlite/ai:move-chat';
   export const refreshSkills = '@jupyterlite/ai:refresh-skills';
   export const saveChat = '@jupyterlite/ai:save-chat';
@@ -209,6 +209,18 @@ export interface IProviderModelInfo {
    * Default context window for the model in tokens.
    */
   contextWindow?: number;
+  /**
+   * Whether the model supports image inputs.
+   */
+  supportsImages?: boolean;
+  /**
+   * Whether the model supports PDF inputs.
+   */
+  supportsPdf?: boolean;
+  /**
+   * Whether the model supports audio inputs.
+   */
+  supportsAudio?: boolean;
 }
 
 export interface IProviderInfo {
@@ -397,6 +409,8 @@ export interface IAIConfig {
   skillsPaths: string[];
   // Directory where chat backups are saved
   chatBackupDirectory: string;
+  // Automatically request a title from the model for every message until there are 5 messages
+  autoTitle: boolean;
 }
 
 export interface IAISettingsModel extends VDomRenderer.IModel {
@@ -513,13 +527,12 @@ export namespace IAgentManager {
       isError: boolean;
     };
     tool_approval_request: {
-      approvalId: string;
       toolCallId: string;
       toolName: string;
       args: unknown;
     };
     tool_approval_resolved: {
-      approvalId: string;
+      toolCallId: string;
       approved: boolean;
     };
     error: {
@@ -584,34 +597,39 @@ export interface IAgentManager {
   /**
    * Clears conversation history and resets agent state.
    */
-  clearHistory(): void;
+  clearHistory(): Promise<void>;
   /**
-   * Sets the conversation history with a list of messages from the chat.
-   * @param messages The chat messages to set as history
+   * Sets the history from already-processed model messages.
+   * @param messages Pre-built model messages (may include binary content)
    */
-  setHistory(messages: IMessageContent[]): void;
+  setHistory(messages: ModelMessage[]): void;
   /**
    * Stops the current streaming response by aborting the request.
    */
   stopStreaming(): void;
   /**
    * Approves a pending tool call.
-   * @param approvalId The approval ID to approve
+   * @param toolCallId The tool call ID to approve
    * @param reason Optional reason for approval
    */
-  approveToolCall(approvalId: string, reason?: string): void;
+  approveToolCall(toolCallId: string, reason?: string): void;
   /**
    * Rejects a pending tool call.
-   * @param approvalId The approval ID to reject
+   * @param toolCallId The tool call ID to reject
    * @param reason Optional reason for rejection
    */
-  rejectToolCall(approvalId: string, reason?: string): void;
+  rejectToolCall(toolCallId: string, reason?: string): void;
   /**
    * Generates AI response to user message using the agent.
    * Handles the complete execution cycle including tool calls.
    * @param message The user message to respond to (may include processed attachment content)
    */
-  generateResponse(message: string): Promise<void>;
+  generateResponse(message: UserContent): Promise<void>;
+  /**
+   * Create a transient language model to request a text response, which won't be added to history.
+   * @param messages - the messages sequence to send to the model.
+   */
+  textResponse(messages: ModelMessage[]): Promise<string>;
   /**
    * Initializes the AI agent with current settings and tools.
    * Sets up the agent with model configuration, tools, and MCP tools.
@@ -660,6 +678,69 @@ export const IAgentManagerFactory = new Token<IAgentManagerFactory>(
 
 /* THE CHAT MODELS HANDLER */
 
+export interface IAIChatModel extends IChatModel {
+  /**
+   * A signal emitting when the chat name has changed.
+   */
+  readonly nameChanged: ISignal<IAIChatModel, string>;
+  /**
+   * The title of the chat.
+   */
+  title: string | null;
+  /**
+   * A signal emitting when the chat title has changed.
+   */
+  readonly titleChanged: ISignal<IAIChatModel, string | null>;
+  /**
+   * Whether to save the chat automatically.
+   */
+  autosave: boolean;
+  /**
+   * A signal emitting when the autosave flag changed.
+   */
+  readonly autosaveChanged: ISignal<IAIChatModel, boolean>;
+  /**
+   * Whether save/restore is available.
+   */
+  readonly saveAvailable: boolean;
+  /**
+   * A signal emitting when the token usage changed.
+   */
+  readonly tokenUsageChanged: ISignal<IAgentManager, ITokenUsage>;
+  /**
+   * The agent manager used in the model.
+   */
+  readonly agentManager: IAgentManager;
+  /**
+   * Save the chat as json file.
+   */
+  save(): Promise<void>;
+  /**
+   * Restore the chat from a json file.
+   *
+   * @param silent - Whether a log should be displayed in the console if the
+   * restoration is not possible.
+   */
+  restore(filepath: string, silent?: boolean): Promise<boolean>;
+  /**
+   * Request a title to this chat, regarding the message history.
+   */
+  requestTitle(): Promise<string>;
+  /**
+   * Removes a queued message by its ID.
+   * @param messageId The ID of the queued message to remove
+   */
+  removeQueuedMessage(messageId: string): void;
+  /**
+   * The current message queue
+   */
+  messageQueue: any[];
+  /**
+   * Whether the chat is currently busy processing a message
+   */
+  isBusy: boolean;
+}
+
 /**
  * The interface for the chat model handler.
  */
@@ -667,7 +748,7 @@ export interface IChatModelHandler {
   /**
    * The function to create a new model.
    */
-  createModel(options: ICreateChatOptions): AIChatModel;
+  createModel(options: ICreateChatOptions): IAIChatModel;
   /**
    * The active cell manager (to copy code from chat to cell).
    */
@@ -695,6 +776,10 @@ export interface ICreateChatOptions {
    * Whether the chat is autosaved or not.
    */
   autosave?: boolean;
+  /**
+   * An optional title to the chat.
+   */
+  title?: string | null;
 }
 /**
  * Token for the chat model handler.

package/src/widgets/ai-settings.tsx

@@ -925,6 +925,32 @@ const AISettingsComponent: React.FC<IAISettingsComponentProps> = ({
                   }
                 />
 
+                <FormControlLabel
+                  control={
+                    <Switch
+                      checked={config.autoTitle}
+                      onChange={e =>
+                        handleConfigUpdate({
+                          autoTitle: e.target.checked
+                        })
+                      }
+                      color="primary"
+                    />
+                  }
+                  label={
+                    <Box>
+                      <Typography variant="body1">
+                        {trans.__('Auto Title')}
+                      </Typography>
+                      <Typography variant="caption" color="text.secondary">
+                        {trans.__(
+                          'Automatically generate a chat title from the model for every message until there are 5 messages'
+                        )}
+                      </Typography>
+                    </Box>
+                  }
+                />
+
                 <FormControlLabel
                   control={
                     <Switch

package/src/widgets/main-area-chat.ts

@@ -4,11 +4,10 @@ import { launchIcon } from '@jupyterlab/ui-components';
 import type { TranslationBundle } from '@jupyterlab/translation';
 import { CommandRegistry } from '@lumino/commands';
 
-import { AIChatModel } from '../chat-model';
 import { SaveComponentWidget } from '../components/save-button';
 import { UsageWidget } from '../components/usage-display';
 import { RenderedMessageOutputAreaCompat } from '../rendered-message-outputarea';
-import { CommandIds, type IAISettingsModel } from '../tokens';
+import { CommandIds, IAIChatModel, type IAISettingsModel } from '../tokens';
 
 export namespace MainAreaChat {
   export interface IOptions extends MainAreaWidget.IOptions<ChatWidget> {
@@ -24,7 +23,8 @@ export namespace MainAreaChat {
 export class MainAreaChat extends MainAreaWidget<ChatWidget> {
   constructor(options: MainAreaChat.IOptions) {
     super(options);
-    this.title.label = this.content.model.name;
+    this.title.label = this.model.name;
+    this.title.caption = this.model.title ?? this.model.name;
 
     const { trans } = options;
 
@@ -68,21 +68,24 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
       chatPanel: this.content
     });
 
-    this.model.writersChanged.connect(this._writersChanged);
+    this.model.writersChanged?.connect(this._writersChanged);
+
+    this.model.titleChanged.connect(this._titleChanged);
   }
 
   dispose(): void {
     super.dispose();
     // Dispose of the approval buttons widget when the chat is disposed.
     this._outputAreaCompat.dispose();
-    this.model.writersChanged.disconnect(this._writersChanged);
+    this.model.writersChanged?.disconnect(this._writersChanged);
+    this.model.titleChanged.disconnect(this._titleChanged);
   }
 
   /**
    * Get the model of the chat.
    */
-  get model(): AIChatModel {
-    return this.content.model as AIChatModel;
+  get model(): IAIChatModel {
+    return this.content.model as IAIChatModel;
   }
 
   /**
@@ -99,13 +102,15 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
     );
 
     if (aiWriting) {
-      this.content.inputToolbarRegistry?.hide('send');
       this.content.inputToolbarRegistry?.show('stop');
     } else {
       this.content.inputToolbarRegistry?.hide('stop');
-      this.content.inputToolbarRegistry?.show('send');
     }
   };
 
+  private _titleChanged = () => {
+    this.title.caption = this.model.title ?? this.model.name;
+  };
+
   private _outputAreaCompat: RenderedMessageOutputAreaCompat;
 }

package/lib/providers/generated-context-windows.js

@@ -1,96 +0,0 @@
-/**
- * This file is generated by `jlpm sync:model-context-windows`.
- * Source: https://models.dev/api.json
- * Backed by: https://github.com/anomalyco/models.dev
- * Generated: 2026-04-08T16:23:34.080Z
- */
-export const BUILT_IN_PROVIDER_MODEL_INFO = {
-    anthropic: {
-        'claude-opus-4-6': { contextWindow: 1000000 },
-        'claude-sonnet-4-6': { contextWindow: 1000000 },
-        'claude-opus-4-5': { contextWindow: 200000 },
-        'claude-opus-4-5-20251101': { contextWindow: 200000 },
-        'claude-sonnet-4-5': { contextWindow: 200000 },
-        'claude-sonnet-4-5-20250929': { contextWindow: 200000 },
-        'claude-haiku-4-5': { contextWindow: 200000 },
-        'claude-haiku-4-5-20251001': { contextWindow: 200000 },
-        'claude-opus-4-1': { contextWindow: 200000 },
-        'claude-opus-4-1-20250805': { contextWindow: 200000 },
-        'claude-opus-4-0': { contextWindow: 200000 },
-        'claude-opus-4-20250514': { contextWindow: 200000 },
-        'claude-sonnet-4-0': { contextWindow: 200000 },
-        'claude-sonnet-4-20250514': { contextWindow: 200000 }
-    },
-    google: {
-        'gemini-3.1-pro-preview': { contextWindow: 1048576 },
-        'gemini-3.1-pro-preview-customtools': { contextWindow: 1048576 },
-        'gemini-3.1-flash-image-preview': { contextWindow: 131072 },
-        'gemini-3.1-flash-lite-preview': { contextWindow: 1048576 },
-        'gemini-3-flash-preview': { contextWindow: 1048576 },
-        'gemini-2.5-pro': { contextWindow: 1048576 },
-        'gemini-2.5-flash': { contextWindow: 1048576 },
-        'gemini-2.5-flash-image': { contextWindow: 32768 },
-        'gemini-2.5-flash-lite': { contextWindow: 1048576 },
-        'gemini-flash-latest': { contextWindow: 1048576 },
-        'gemini-flash-lite-latest': { contextWindow: 1048576 }
-    },
-    mistral: {
-        'mistral-large-latest': { contextWindow: 262144 },
-        'mistral-medium-latest': { contextWindow: 128000 },
-        'mistral-medium-2508': { contextWindow: 262144 },
-        'mistral-small-latest': { contextWindow: 256000 },
-        'mistral-small-2506': { contextWindow: 128000 },
-        'ministral-3b-latest': { contextWindow: 128000 },
-        'ministral-8b-latest': { contextWindow: 128000 },
-        'magistral-small-latest': { contextWindow: 128000 },
-        'magistral-medium-latest': { contextWindow: 128000 },
-        'pixtral-large-latest': { contextWindow: 128000 },
-        'codestral-latest': { contextWindow: 256000 },
-        'devstral-latest': { contextWindow: 262144 },
-        'devstral-2512': { contextWindow: 262144 }
-    },
-    openai: {
-        'gpt-5.4': { contextWindow: 1050000 },
-        'gpt-5.4-mini': { contextWindow: 400000 },
-        'gpt-5.4-nano': { contextWindow: 400000 },
-        'gpt-5.2': { contextWindow: 400000 },
-        'gpt-5.2-2025-12-11': { contextWindow: 400000 },
-        'gpt-5.2-chat-latest': { contextWindow: 128000 },
-        'gpt-5.2-pro': { contextWindow: 400000 },
-        'gpt-5.2-pro-2025-12-11': { contextWindow: 400000 },
-        'gpt-5.2-codex': { contextWindow: 400000 },
-        'gpt-5.1': { contextWindow: 400000 },
-        'gpt-5.1-2025-11-13': { contextWindow: 400000 },
-        'gpt-5.1-chat-latest': { contextWindow: 128000 },
-        'gpt-5': { contextWindow: 400000 },
-        'gpt-5-2025-08-07': { contextWindow: 400000 },
-        'gpt-5-chat-latest': { contextWindow: 400000 },
-        'gpt-5-mini': { contextWindow: 400000 },
-        'gpt-5-mini-2025-08-07': { contextWindow: 400000 },
-        'gpt-5-nano': { contextWindow: 400000 },
-        'gpt-5-nano-2025-08-07': { contextWindow: 400000 },
-        'o4-mini': { contextWindow: 200000 },
-        'o4-mini-2025-04-16': { contextWindow: 200000 },
-        'o3-pro': { contextWindow: 200000 },
-        o3: { contextWindow: 200000 },
-        'o3-2025-04-16': { contextWindow: 200000 },
-        'o3-mini': { contextWindow: 200000 },
-        'o3-mini-2025-01-31': { contextWindow: 200000 },
-        o1: { contextWindow: 200000 },
-        'o1-2024-12-17': { contextWindow: 200000 },
-        'gpt-4.1': { contextWindow: 1047576 },
-        'gpt-4.1-2025-04-14': { contextWindow: 1047576 },
-        'gpt-4.1-mini': { contextWindow: 1047576 },
-        'gpt-4.1-mini-2025-04-14': { contextWindow: 1047576 },
-        'gpt-4.1-nano': { contextWindow: 1047576 },
-        'gpt-4.1-nano-2025-04-14': { contextWindow: 1047576 },
-        'gpt-4o': { contextWindow: 128000 },
-        'gpt-4o-2024-05-13': { contextWindow: 128000 },
-        'gpt-4o-2024-08-06': { contextWindow: 128000 },
-        'gpt-4o-2024-11-20': { contextWindow: 128000 },
-        'gpt-4o-mini': { contextWindow: 128000 },
-        'gpt-4o-mini-2024-07-18': { contextWindow: 128000 },
-        'gpt-3.5-turbo': { contextWindow: 16385 },
-        'gpt-3.5-turbo-0125': { contextWindow: 16385 }
-    }
-};

package/src/providers/generated-context-windows.ts

@@ -1,102 +0,0 @@
-/**
- * This file is generated by `jlpm sync:model-context-windows`.
- * Source: https://models.dev/api.json
- * Backed by: https://github.com/anomalyco/models.dev
- * Generated: 2026-04-08T16:23:34.080Z
- */
-
-import type { IProviderModelInfo } from '../tokens';
-
-export const BUILT_IN_PROVIDER_MODEL_INFO: Record<
-  string,
-  Record<string, IProviderModelInfo>
-> = {
-  anthropic: {
-    'claude-opus-4-6': { contextWindow: 1000000 },
-    'claude-sonnet-4-6': { contextWindow: 1000000 },
-    'claude-opus-4-5': { contextWindow: 200000 },
-    'claude-opus-4-5-20251101': { contextWindow: 200000 },
-    'claude-sonnet-4-5': { contextWindow: 200000 },
-    'claude-sonnet-4-5-20250929': { contextWindow: 200000 },
-    'claude-haiku-4-5': { contextWindow: 200000 },
-    'claude-haiku-4-5-20251001': { contextWindow: 200000 },
-    'claude-opus-4-1': { contextWindow: 200000 },
-    'claude-opus-4-1-20250805': { contextWindow: 200000 },
-    'claude-opus-4-0': { contextWindow: 200000 },
-    'claude-opus-4-20250514': { contextWindow: 200000 },
-    'claude-sonnet-4-0': { contextWindow: 200000 },
-    'claude-sonnet-4-20250514': { contextWindow: 200000 }
-  },
-  google: {
-    'gemini-3.1-pro-preview': { contextWindow: 1048576 },
-    'gemini-3.1-pro-preview-customtools': { contextWindow: 1048576 },
-    'gemini-3.1-flash-image-preview': { contextWindow: 131072 },
-    'gemini-3.1-flash-lite-preview': { contextWindow: 1048576 },
-    'gemini-3-flash-preview': { contextWindow: 1048576 },
-    'gemini-2.5-pro': { contextWindow: 1048576 },
-    'gemini-2.5-flash': { contextWindow: 1048576 },
-    'gemini-2.5-flash-image': { contextWindow: 32768 },
-    'gemini-2.5-flash-lite': { contextWindow: 1048576 },
-    'gemini-flash-latest': { contextWindow: 1048576 },
-    'gemini-flash-lite-latest': { contextWindow: 1048576 }
-  },
-  mistral: {
-    'mistral-large-latest': { contextWindow: 262144 },
-    'mistral-medium-latest': { contextWindow: 128000 },
-    'mistral-medium-2508': { contextWindow: 262144 },
-    'mistral-small-latest': { contextWindow: 256000 },
-    'mistral-small-2506': { contextWindow: 128000 },
-    'ministral-3b-latest': { contextWindow: 128000 },
-    'ministral-8b-latest': { contextWindow: 128000 },
-    'magistral-small-latest': { contextWindow: 128000 },
-    'magistral-medium-latest': { contextWindow: 128000 },
-    'pixtral-large-latest': { contextWindow: 128000 },
-    'codestral-latest': { contextWindow: 256000 },
-    'devstral-latest': { contextWindow: 262144 },
-    'devstral-2512': { contextWindow: 262144 }
-  },
-  openai: {
-    'gpt-5.4': { contextWindow: 1050000 },
-    'gpt-5.4-mini': { contextWindow: 400000 },
-    'gpt-5.4-nano': { contextWindow: 400000 },
-    'gpt-5.2': { contextWindow: 400000 },
-    'gpt-5.2-2025-12-11': { contextWindow: 400000 },
-    'gpt-5.2-chat-latest': { contextWindow: 128000 },
-    'gpt-5.2-pro': { contextWindow: 400000 },
-    'gpt-5.2-pro-2025-12-11': { contextWindow: 400000 },
-    'gpt-5.2-codex': { contextWindow: 400000 },
-    'gpt-5.1': { contextWindow: 400000 },
-    'gpt-5.1-2025-11-13': { contextWindow: 400000 },
-    'gpt-5.1-chat-latest': { contextWindow: 128000 },
-    'gpt-5': { contextWindow: 400000 },
-    'gpt-5-2025-08-07': { contextWindow: 400000 },
-    'gpt-5-chat-latest': { contextWindow: 400000 },
-    'gpt-5-mini': { contextWindow: 400000 },
-    'gpt-5-mini-2025-08-07': { contextWindow: 400000 },
-    'gpt-5-nano': { contextWindow: 400000 },
-    'gpt-5-nano-2025-08-07': { contextWindow: 400000 },
-    'o4-mini': { contextWindow: 200000 },
-    'o4-mini-2025-04-16': { contextWindow: 200000 },
-    'o3-pro': { contextWindow: 200000 },
-    o3: { contextWindow: 200000 },
-    'o3-2025-04-16': { contextWindow: 200000 },
-    'o3-mini': { contextWindow: 200000 },
-    'o3-mini-2025-01-31': { contextWindow: 200000 },
-    o1: { contextWindow: 200000 },
-    'o1-2024-12-17': { contextWindow: 200000 },
-    'gpt-4.1': { contextWindow: 1047576 },
-    'gpt-4.1-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4.1-mini': { contextWindow: 1047576 },
-    'gpt-4.1-mini-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4.1-nano': { contextWindow: 1047576 },
-    'gpt-4.1-nano-2025-04-14': { contextWindow: 1047576 },
-    'gpt-4o': { contextWindow: 128000 },
-    'gpt-4o-2024-05-13': { contextWindow: 128000 },
-    'gpt-4o-2024-08-06': { contextWindow: 128000 },
-    'gpt-4o-2024-11-20': { contextWindow: 128000 },
-    'gpt-4o-mini': { contextWindow: 128000 },
-    'gpt-4o-mini-2024-07-18': { contextWindow: 128000 },
-    'gpt-3.5-turbo': { contextWindow: 16385 },
-    'gpt-3.5-turbo-0125': { contextWindow: 16385 }
-  }
-};