@jupyterlite/ai 0.16.0 → 0.18.0

package/lib/tokens.d.ts

@@ -1,13 +1,12 @@
-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, ISkillResourceResult, ISkillSummary } from './skills/types';
 export type { ISkillDefinition, ISkillRegistration, ISkillResourceResult, ISkillSummary } from './skills/types';
 /**
@@ -17,6 +16,7 @@ export declare namespace CommandIds {
     const openSettings = "@jupyterlite/ai:open-settings";
     const reposition = "@jupyterlite/ai:reposition";
     const openChat = "@jupyterlite/ai:open-chat";
+    const openOrRevealChat = "@jupyterlite/ai:open-or-reveal-chat";
     const moveChat = "@jupyterlite/ai:move-chat";
     const refreshSkills = "@jupyterlite/ai:refresh-skills";
     const saveChat = "@jupyterlite/ai:save-chat";
@@ -158,6 +158,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 {
     /**
@@ -302,6 +314,7 @@ export interface IAIConfig {
     diffDisplayMode: 'split' | 'unified';
     skillsPaths: string[];
     chatBackupDirectory: string;
+    autoTitle: boolean;
 }
 export interface IAISettingsModel extends VDomRenderer.IModel {
     readonly config: IAIConfig;
@@ -400,13 +413,12 @@ export declare 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: {
@@ -465,34 +477,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.
@@ -527,6 +544,68 @@ export interface IAgentManagerFactory {
     getMCPTools(): Promise<ToolMap>;
 }
 export declare const IAgentManagerFactory: Token<IAgentManagerFactory>;
+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.
  */
@@ -534,7 +613,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).
      */
@@ -561,6 +640,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/lib/tokens.js

@@ -7,6 +7,7 @@ export var CommandIds;
     CommandIds.openSettings = '@jupyterlite/ai:open-settings';
     CommandIds.reposition = '@jupyterlite/ai:reposition';
     CommandIds.openChat = '@jupyterlite/ai:open-chat';
+    CommandIds.openOrRevealChat = '@jupyterlite/ai:open-or-reveal-chat';
     CommandIds.moveChat = '@jupyterlite/ai:move-chat';
     CommandIds.refreshSkills = '@jupyterlite/ai:refresh-skills';
     CommandIds.saveChat = '@jupyterlite/ai:save-chat';

package/lib/widgets/ai-settings.js

@@ -482,6 +482,11 @@ const AISettingsComponent = ({ model, agentManagerFactory, themeManager, provide
                                 }), color: "primary" }), label: React.createElement(Box, null,
                                 React.createElement(Typography, { variant: "body1" }, trans.__('Send with Shift+Enter')),
                                 React.createElement(Typography, { variant: "caption", color: "text.secondary" }, trans.__('Use Shift+Enter to send messages (Enter creates new line)'))) }),
+                        React.createElement(FormControlLabel, { control: React.createElement(Switch, { checked: config.autoTitle, onChange: e => handleConfigUpdate({
+                                    autoTitle: e.target.checked
+                                }), color: "primary" }), label: React.createElement(Box, null,
+                                React.createElement(Typography, { variant: "body1" }, trans.__('Auto Title')),
+                                React.createElement(Typography, { variant: "caption", color: "text.secondary" }, trans.__('Automatically generate a chat title from the model for every message until there are 5 messages'))) }),
                         React.createElement(FormControlLabel, { control: React.createElement(Switch, { checked: config.showTokenUsage, onChange: e => handleConfigUpdate({
                                     showTokenUsage: e.target.checked
                                 }), color: "primary" }), label: React.createElement(Box, null,

package/lib/widgets/main-area-chat.d.ts

@@ -2,8 +2,7 @@ import { ChatWidget } from '@jupyter/chat';
 import { MainAreaWidget } from '@jupyterlab/apputils';
 import type { TranslationBundle } from '@jupyterlab/translation';
 import { CommandRegistry } from '@lumino/commands';
-import { AIChatModel } from '../chat-model';
-import { type IAISettingsModel } from '../tokens';
+import { IAIChatModel, type IAISettingsModel } from '../tokens';
 export declare namespace MainAreaChat {
     interface IOptions extends MainAreaWidget.IOptions<ChatWidget> {
         commands: CommandRegistry;
@@ -20,11 +19,12 @@ export declare class MainAreaChat extends MainAreaWidget<ChatWidget> {
     /**
      * Get the model of the chat.
      */
-    get model(): AIChatModel;
+    get model(): IAIChatModel;
     /**
      * Get the area of the chat.
      */
     get area(): string | undefined;
     private _writersChanged;
+    private _titleChanged;
     private _outputAreaCompat;
 }

package/lib/widgets/main-area-chat.js

@@ -10,7 +10,8 @@ import { CommandIds } from '../tokens';
 export class MainAreaChat extends MainAreaWidget {
     constructor(options) {
         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;
         // Move to side button.
         this.toolbar.addItem('moveToSide', new CommandToolbarButton({
@@ -42,13 +43,15 @@ export class MainAreaChat extends MainAreaWidget {
         this._outputAreaCompat = new RenderedMessageOutputAreaCompat({
             chatPanel: this.content
         });
-        this.model.writersChanged.connect(this._writersChanged);
+        this.model.writersChanged?.connect(this._writersChanged);
+        this.model.titleChanged.connect(this._titleChanged);
     }
     dispose() {
         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.
@@ -66,13 +69,14 @@ export class MainAreaChat extends MainAreaWidget {
         // Check if AI is currently writing (streaming)
         const aiWriting = writers.some(writer => writer.user.username === 'ai-assistant');
         if (aiWriting) {
-            this.content.inputToolbarRegistry?.hide('send');
             this.content.inputToolbarRegistry?.show('stop');
         }
         else {
             this.content.inputToolbarRegistry?.hide('stop');
-            this.content.inputToolbarRegistry?.show('send');
         }
     };
+    _titleChanged = () => {
+        this.title.caption = this.model.title ?? this.model.name;
+    };
     _outputAreaCompat;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@jupyterlite/ai",
-    "version": "0.16.0",
+    "version": "0.18.0",
     "description": "AI code completions and chat for JupyterLite",
     "keywords": [
         "jupyter",
@@ -54,7 +54,7 @@
         "watch:labextension": "jupyter labextension watch .",
         "docs": "jupyter book start",
         "docs:build": "sed -e 's/\\[@/[/g' -e 's/@/\\@/g' CHANGELOG.md > docs/_changelog_content.md && jupyter book build --html",
-        "sync:model-context-windows": "node scripts/sync-model-context-windows.mjs && prettier --write src/providers/generated-context-windows.ts && eslint --fix src/providers/generated-context-windows.ts"
+        "sync:model-info": "node scripts/sync-model-info.mjs && prettier --write src/providers/generated-model-info.ts && eslint --fix src/providers/generated-model-info.ts"
     },
     "dependencies": {
         "@ai-sdk/anthropic": "^3.0.58",
@@ -89,7 +89,7 @@
         "@mui/icons-material": "^7",
         "@mui/material": "^7",
         "ai": "^6.0.116",
-        "jupyter-chat-components": "^0.2.0",
+        "jupyter-chat-components": "^0.5.0",
         "jupyter-secrets-manager": "^0.5.0",
         "yaml": "^2.8.1",
         "zod": "^4.3.6"

package/schema/settings-model.json

@@ -210,6 +210,12 @@
       "type": "boolean",
       "default": false
     },
+    "autoTitle": {
+      "title": "Auto Title",
+      "description": "Automatically request a title from the model for every message until there are 5 messages",
+      "type": "boolean",
+      "default": false
+    },
     "showTokenUsage": {
       "title": "Show Token Usage",
       "description": "Display token usage information in the chat toolbar",

package/src/agent.ts

@@ -1,9 +1,9 @@
 import { createMCPClient, type MCPClient } from '@ai-sdk/mcp';
-import type { IMessageContent } from '@jupyter/chat';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { PromiseDelegate } from '@lumino/coreutils';
 import { ISignal, Signal } from '@lumino/signaling';
 import {
+  generateText,
   ToolLoopAgent,
   type ModelMessage,
   type LanguageModel,
@@ -13,7 +13,9 @@ import {
   type TypedToolError,
   type TypedToolOutputDenied,
   type TypedToolResult,
-  type AssistantModelMessage
+  type UserContent,
+  type AssistantModelMessage,
+  APICallError
 } from 'ai';
 import { ISecretsManager } from 'jupyter-secrets-manager';
 
@@ -487,31 +489,12 @@ export class AgentManager implements IAgentManager {
   }
 
   /**
-   * Sets the 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 {
-    // Stop any ongoing streaming and reject awaiting approvals
-    this.stopStreaming();
-
-    for (const [approvalId, pending] of this._pendingApprovals) {
-      pending.resolve(false, 'Chat history changed');
-      this._agentEvent.emit({
-        type: 'tool_approval_resolved',
-        data: { approvalId, approved: false }
-      });
-    }
-    this._pendingApprovals.clear();
-
-    // Convert chat messages to model messages
-    const modelMessages = messages.map(msg => {
-      const isAIMessage = msg.sender.username === 'ai-assistant';
-      return {
-        role: isAIMessage ? 'assistant' : 'user',
-        content: msg.body
-      } as ModelMessage;
-    });
-    this._history = Private.sanitizeModelMessages(modelMessages);
+  setHistory(messages: ModelMessage[]): void {
+    this.stopStreaming('Chat history changed');
+    this._history = Private.sanitizeModelMessages(messages);
   }
 
   /**
@@ -522,11 +505,11 @@ export class AgentManager implements IAgentManager {
     this._controller?.abort();
 
     // Reject any pending approvals
-    for (const [approvalId, pending] of this._pendingApprovals) {
+    for (const [toolCallId, pending] of this._pendingApprovals) {
       pending.resolve(false, reason ?? 'Stream ended by user');
       this._agentEvent.emit({
         type: 'tool_approval_resolved',
-        data: { approvalId, approved: false }
+        data: { toolCallId, approved: false }
       });
     }
     this._pendingApprovals.clear();
@@ -534,34 +517,34 @@ export class AgentManager implements IAgentManager {
 
   /**
    * 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 {
-    const pending = this._pendingApprovals.get(approvalId);
+  approveToolCall(toolCallId: string, reason?: string): void {
+    const pending = this._pendingApprovals.get(toolCallId);
     if (pending) {
       pending.resolve(true, reason);
-      this._pendingApprovals.delete(approvalId);
+      this._pendingApprovals.delete(toolCallId);
       this._agentEvent.emit({
         type: 'tool_approval_resolved',
-        data: { approvalId, approved: true }
+        data: { toolCallId, approved: true }
       });
     }
   }
 
   /**
    * 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 {
-    const pending = this._pendingApprovals.get(approvalId);
+  rejectToolCall(toolCallId: string, reason?: string): void {
+    const pending = this._pendingApprovals.get(toolCallId);
     if (pending) {
       pending.resolve(false, reason);
-      this._pendingApprovals.delete(approvalId);
+      this._pendingApprovals.delete(toolCallId);
       this._agentEvent.emit({
         type: 'tool_approval_resolved',
-        data: { approvalId, approved: false }
+        data: { toolCallId, approved: false }
       });
     }
   }
@@ -571,10 +554,17 @@ export class AgentManager implements IAgentManager {
    * Handles the complete execution cycle including tool calls.
    * @param message The user message to respond to (may include processed attachment content)
    */
-  async generateResponse(message: string): Promise<void> {
+  async generateResponse(message: UserContent): Promise<void> {
     this._streaming = new PromiseDelegate();
     this._controller = new AbortController();
     const responseHistory: ModelMessage[] = [];
+
+    // Add user message to history
+    responseHistory.push({
+      role: 'user',
+      content: message
+    });
+
     try {
       // Ensure we have an agent
       if (!this._agent) {
@@ -585,12 +575,6 @@ export class AgentManager implements IAgentManager {
         throw new Error('Failed to initialize agent');
       }
 
-      // Add user message to history
-      responseHistory.push({
-        role: 'user',
-        content: message
-      });
-
       let continueLoop = true;
       while (continueLoop) {
         const result = await this._agent.stream({
@@ -647,9 +631,32 @@ export class AgentManager implements IAgentManager {
       this._history.push(...Private.sanitizeModelMessages(responseHistory));
     } catch (error) {
       if ((error as Error).name !== 'AbortError') {
+        let helpMessage = `${(error as Error).message}`;
+
+        // Remove attachments from history on payload rejection errors
+        if (
+          APICallError.isInstance(error) &&
+          (error.statusCode === 400 ||
+            error.statusCode === 404 ||
+            error.statusCode === 413 ||
+            error.statusCode === 415 ||
+            error.statusCode === 422)
+        ) {
+          this._stripAttachments(
+            [...this._history, ...responseHistory],
+            '_Attachment removed due to error_'
+          );
+          helpMessage +=
+            '\n\nAttachments have been removed from history. Please send your prompt again.';
+        }
         this._agentEvent.emit({
           type: 'error',
-          data: { error: error as Error }
+          data: { error: new Error(helpMessage) }
+        });
+        this._history.push(...Private.sanitizeModelMessages(responseHistory));
+        this._history.push({
+          role: 'assistant',
+          content: helpMessage
         });
       }
     } finally {
@@ -658,6 +665,24 @@ export class AgentManager implements IAgentManager {
     }
   }
 
+  /**
+   * 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.
+   */
+  async textResponse(messages: ModelMessage[]): Promise<string> {
+    try {
+      const model = await this._createModel();
+      const result = await generateText({
+        model,
+        messages
+      });
+      this._updateTokenUsage(result.totalUsage, result.totalUsage.inputTokens);
+      return result.text;
+    } catch (e) {
+      throw `Error while getting the topic of the chat\n${e}`;
+    }
+  }
+
   /**
    * Updates cumulative token usage statistics from a completed model step.
    */
@@ -680,6 +705,27 @@ export class AgentManager implements IAgentManager {
     this._tokenUsageChanged.emit(this._tokenUsage);
   }
 
+  /**
+   * Removes image and file parts from all user messages in the given list.
+   */
+  private _stripAttachments(
+    messages: ModelMessage[],
+    placeholder: string
+  ): void {
+    for (const msg of messages) {
+      if (msg.role === 'user' && Array.isArray(msg.content)) {
+        const hasMedia = msg.content.some(p => p.type !== 'text');
+        if (hasMedia) {
+          const textContent = msg.content
+            .filter(p => p.type === 'text')
+            .map(p => (p as { text: string }).text)
+            .join('\n');
+          msg.content = textContent || placeholder;
+        }
+      }
+    }
+  }
+
   /**
    * Gets the configured context window for the active provider.
    */
@@ -932,6 +978,9 @@ ${richOutputWorkflowInstruction}`;
           await this._handleApprovalRequest(part, processResult);
           break;
 
+        case 'error':
+          throw part.error;
+
         case 'finish-step':
           this._updateTokenUsage(part.usage, part.usage.inputTokens);
           break;
@@ -940,7 +989,7 @@ ${richOutputWorkflowInstruction}`;
           processResult.aborted = true;
           break;
 
-        // Ignore: text-start, text-end, finish, error, and others
+        // Ignore: text-start, text-end, finish, and others
         default:
           break;
       }
@@ -1034,14 +1083,13 @@ ${richOutputWorkflowInstruction}`;
     this._agentEvent.emit({
       type: 'tool_approval_request',
       data: {
-        approvalId,
         toolCallId: toolCall.toolCallId,
         toolName: toolCall.toolName,
         args: toolCall.input
       }
     });
 
-    const approved = await this._waitForApproval(approvalId);
+    const approved = await this._waitForApproval(toolCall.toolCallId);
 
     result.approvalProcessed = true;
     result.approvalResponse = {
@@ -1058,12 +1106,12 @@ ${richOutputWorkflowInstruction}`;
 
   /**
    * Waits for user approval of a tool call.
-   * @param approvalId The approval ID to wait for
+   * @param toolCallId The tool call ID to wait for approval
    * @returns Promise that resolves to true if approved, false if rejected
    */
-  private _waitForApproval(approvalId: string): Promise<boolean> {
+  private _waitForApproval(toolCallId: string): Promise<boolean> {
     return new Promise(resolve => {
-      this._pendingApprovals.set(approvalId, {
+      this._pendingApprovals.set(toolCallId, {
         resolve: (approved: boolean) => {
           resolve(approved);
         }

package/src/chat-commands/clear.ts

@@ -29,7 +29,7 @@ export class ClearCommandProvider implements IChatCommandProvider {
     const context = inputModel.chatContext as
       | AIChatModel.IAIChatContext
       | undefined;
-    context?.clearMessages?.();
+    await context?.clearMessages?.();
 
     inputModel.value = '';
     inputModel.clearAttachments();

package/src/chat-model-handler.ts

@@ -6,6 +6,7 @@ import { Contents } from '@jupyterlab/services';
 import { AIChatModel } from './chat-model';
 import type {
   IAgentManagerFactory,
+  IAIChatModel,
   IAISettingsModel,
   IChatModelHandler,
   ICreateChatOptions,
@@ -28,8 +29,9 @@ export class ChatModelHandler implements IChatModelHandler {
     this._contentsManager = options.contentsManager;
   }
 
-  createModel(options: ICreateChatOptions): AIChatModel {
-    const { name, activeProvider, tokenUsage, messages, autosave } = options;
+  createModel(options: ICreateChatOptions): IAIChatModel {
+    const { name, activeProvider, tokenUsage, messages, autosave, title } =
+      options;
 
     // Create Agent Manager first so it can be shared
     const agentManager = this._agentManagerFactory.createAgent({
@@ -48,7 +50,8 @@ export class ChatModelHandler implements IChatModelHandler {
       agentManager,
       activeCellManager: this._activeCellManager,
       documentManager: this._docManager,
-      contentsManager: this._contentsManager
+      contentsManager: this._contentsManager,
+      providerRegistry: this._providerRegistry
     });
 
     messages?.forEach(message => {
@@ -58,6 +61,10 @@ export class ChatModelHandler implements IChatModelHandler {
 
     model.name = name;
 
+    if (title) {
+      model.title = title;
+    }
+
     return model;
   }