@jupyterlite/ai 0.9.1 → 0.10.0

package/src/index.ts

@@ -38,6 +38,12 @@ import { ISettingRegistry } from '@jupyterlab/settingregistry';
 
 import { IStatusBar } from '@jupyterlab/statusbar';
 
+import {
+  ITranslator,
+  nullTranslator,
+  TranslationBundle
+} from '@jupyterlab/translation';
+
 import {
   settingsIcon,
   Toolbar,
@@ -212,7 +218,7 @@ const chatModelRegistry: JupyterFrontEndPlugin<IChatModelRegistry> = {
   description: 'Registry for the current chat model',
   autoStart: true,
   requires: [IAISettingsModel, IAgentManagerFactory, IDocumentManager],
-  optional: [IProviderRegistry, INotebookTracker, IToolRegistry],
+  optional: [IProviderRegistry, INotebookTracker, IToolRegistry, ITranslator],
   provides: IChatModelRegistry,
   activate: (
     app: JupyterFrontEnd,
@@ -221,8 +227,11 @@ const chatModelRegistry: JupyterFrontEndPlugin<IChatModelRegistry> = {
     docManager: IDocumentManager,
     providerRegistry?: IProviderRegistry,
     notebookTracker?: INotebookTracker,
-    toolRegistry?: IToolRegistry
+    toolRegistry?: IToolRegistry,
+    translator?: ITranslator
   ): IChatModelRegistry => {
+    const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
+
     // Create ActiveCellManager if notebook tracker is available
     let activeCellManager: ActiveCellManager | undefined;
     if (notebookTracker) {
@@ -237,7 +246,8 @@ const chatModelRegistry: JupyterFrontEndPlugin<IChatModelRegistry> = {
       agentManagerFactory,
       docManager,
       providerRegistry,
-      toolRegistry
+      toolRegistry,
+      trans
     });
   }
 };
@@ -255,7 +265,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
     IChatModelRegistry,
     IAISettingsModel
   ],
-  optional: [IThemeManager, ILayoutRestorer, ILabShell],
+  optional: [IThemeManager, ILayoutRestorer, ILabShell, ITranslator],
   activate: (
     app: JupyterFrontEnd,
     rmRegistry: IRenderMimeRegistry,
@@ -264,8 +274,10 @@ const plugin: JupyterFrontEndPlugin<void> = {
     settingsModel: AISettingsModel,
     themeManager?: IThemeManager,
     restorer?: ILayoutRestorer,
-    labShell?: ILabShell
+    labShell?: ILabShell,
+    translator?: ITranslator
   ): void => {
+    const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
     // Create attachment opener registry to handle file attachments
     const attachmentOpenerRegistry = new AttachmentOpenerRegistry();
     attachmentOpenerRegistry.set('file', attachment => {
@@ -304,7 +316,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
 
     chatPanel.id = '@jupyterlite/ai:chat-panel';
     chatPanel.title.icon = chatIcon;
-    chatPanel.title.caption = 'Chat with AI assistant'; // TODO: i18n/
+    chatPanel.title.caption = trans.__('Chat with AI assistant');
 
     chatPanel.toolbar.addItem('spacer', Toolbar.createSpacerItem());
     chatPanel.toolbar.addItem(
@@ -314,7 +326,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
         onClick: () => {
           app.commands.execute('@jupyterlite/ai:open-settings');
         },
-        tooltip: 'Open AI Settings'
+        tooltip: trans.__('Open AI Settings')
       })
     );
 
@@ -335,7 +347,8 @@ const plugin: JupyterFrontEndPlugin<void> = {
       const tokenUsageWidget = new TokenUsageWidget({
         tokenUsageChanged: model.tokenUsageChanged,
         settingsModel,
-        initialTokenUsage: model.agentManager.tokenUsage
+        initialTokenUsage: model.agentManager.tokenUsage,
+        translator: trans
       });
       section.toolbar.insertBefore('markRead', 'token-usage', tokenUsageWidget);
       model.writersChanged?.connect((_, writers) => {
@@ -355,7 +368,8 @@ const plugin: JupyterFrontEndPlugin<void> = {
 
       // Associate an approval buttons object to the chat.
       const approvalButton = new ApprovalButtons({
-        chatPanel: widget
+        chatPanel: widget,
+        agentManager: model.agentManager
       });
 
       widget.disposed.connect(() => {
@@ -407,6 +421,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
       settingsModel,
       tracker,
       modelRegistry,
+      trans,
       themeManager,
       labShell
     );
@@ -422,6 +437,7 @@ function registerCommands(
   settingsModel: AISettingsModel,
   tracker: WidgetTracker<MainAreaChat | ChatWidget>,
   modelRegistry: IChatModelRegistry,
+  trans: TranslationBundle,
   themeManager?: IThemeManager,
   labShell?: ILabShell
 ) {
@@ -429,7 +445,7 @@ function registerCommands(
 
   if (labShell) {
     commands.addCommand(CommandIds.reposition, {
-      label: 'Reposition Widget',
+      label: trans.__('Reposition Widget'),
       execute: (args: any) => {
         const { widgetId, area, mode } = args;
         const widget = widgetId
@@ -456,17 +472,22 @@ function registerCommands(
           properties: {
             widgetId: {
               type: 'string',
-              description:
+              description: trans.__(
                 'The widget ID to reposition in the application shell'
+              )
             },
             area: {
               type: 'string',
-              description: 'The name of the area to reposition the widget to'
+              description: trans.__(
+                'The name of the area to reposition the widget to'
+              )
             },
             mode: {
               type: 'string',
               enum: ['split-left', 'split-right', 'split-top', 'split-bottom'],
-              description: 'The mode to use when repositioning the widget'
+              description: trans.__(
+                'The mode to use when repositioning the widget'
+              )
             }
           }
         }
@@ -481,7 +502,12 @@ function registerCommands(
         inputToolbarRegistry: inputToolbarFactory.create(),
         attachmentOpenerRegistry
       });
-      const widget = new MainAreaChat({ content, commands, settingsModel });
+      const widget = new MainAreaChat({
+        content,
+        commands,
+        settingsModel,
+        trans
+      });
       app.shell.add(widget, 'main');
 
       // Add the widget to the tracker.
@@ -501,7 +527,7 @@ function registerCommands(
     };
 
     commands.addCommand(CommandIds.openChat, {
-      label: 'Open a chat',
+      label: trans.__('Open a chat'),
       execute: async (args): Promise<boolean> => {
         const area = (args.area as string) === 'main' ? 'main' : 'side';
         const provider = (args.provider as string) ?? undefined;
@@ -532,15 +558,15 @@ function registerCommands(
             area: {
               type: 'string',
               enum: ['main', 'side'],
-              description: 'The name of the area to open the chat to'
+              description: trans.__('The name of the area to open the chat to')
             },
             name: {
               type: 'string',
-              description: 'The name of the chat'
+              description: trans.__('The name of the chat')
             },
             provider: {
               type: 'string',
-              description: 'The provider/model to use with this chat'
+              description: trans.__('The provider/model to use with this chat')
             }
           }
         }
@@ -548,7 +574,7 @@ function registerCommands(
     });
 
     commands.addCommand(CommandIds.moveChat, {
-      caption: 'Move chat between area',
+      caption: trans.__('Move chat between area'),
       execute: async (args): Promise<boolean> => {
         const area = args.area as string;
         if (!['side', 'main'].includes(area)) {
@@ -601,7 +627,7 @@ function registerCommands(
           trackerUpdated.promise,
           new Promise<boolean>(r =>
             setTimeout(() => {
-              return false;
+              r(false);
             }, 2000)
           )
         ]);
@@ -634,11 +660,11 @@ function registerCommands(
             area: {
               type: 'string',
               enum: ['main', 'side'],
-              description: 'The name of the area to move the chat to'
+              description: trans.__('The name of the area to move the chat to')
             },
             name: {
               type: 'string',
-              description: 'The name of the chat to move'
+              description: trans.__('The name of the chat to move')
             }
           },
           requires: ['area', 'name']
@@ -663,7 +689,8 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       ICompletionProviderManager,
       ILayoutRestorer,
       ISecretsManager,
-      IThemeManager
+      IThemeManager,
+      ITranslator
     ],
     activate: (
       app: JupyterFrontEnd,
@@ -673,8 +700,10 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       completionManager?: ICompletionProviderManager,
       restorer?: ILayoutRestorer,
       secretsManager?: ISecretsManager,
-      themeManager?: IThemeManager
+      themeManager?: IThemeManager,
+      translator?: ITranslator
     ): AgentManagerFactory => {
+      const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
       const agentManagerFactory = new AgentManagerFactory({
         settingsModel,
         secretsManager,
@@ -688,7 +717,8 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
         themeManager,
         providerRegistry,
         secretsManager,
-        token
+        token,
+        trans
       });
       settingsWidget.id = 'jupyterlite-ai-settings';
       settingsWidget.title.icon = settingsIcon;
@@ -715,8 +745,8 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       }
 
       app.commands.addCommand(CommandIds.openSettings, {
-        label: 'AI Settings',
-        caption: 'Configure AI providers and behavior',
+        label: trans.__('AI Settings'),
+        caption: trans.__('Configure AI providers and behavior'),
         icon: settingsIcon,
         iconClass: 'jp-ai-settings-icon',
         execute: () => {
@@ -744,7 +774,7 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       if (palette) {
         palette.addItem({
           command: CommandIds.openSettings,
-          category: 'AI Assistant'
+          category: trans.__('AI Assistant')
         });
       }
 
@@ -888,18 +918,22 @@ const inputToolbarFactory: JupyterFrontEndPlugin<IInputToolbarRegistryFactory> =
     autoStart: true,
     provides: IInputToolbarRegistryFactory,
     requires: [IAISettingsModel, IToolRegistry],
+    optional: [ITranslator],
     activate: (
       app: JupyterFrontEnd,
       settingsModel: AISettingsModel,
-      toolRegistry: IToolRegistry
+      toolRegistry: IToolRegistry,
+      translator?: ITranslator
     ): IInputToolbarRegistryFactory => {
-      const stopButton = stopItem();
-      const clearButton = clearItem();
+      const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
+      const stopButton = stopItem(trans);
+      const clearButton = clearItem(trans);
       const toolSelectButton = createToolSelectItem(
         toolRegistry,
-        settingsModel.config.toolsEnabled
+        settingsModel.config.toolsEnabled,
+        trans
       );
-      const modelSelectButton = createModelSelectItem(settingsModel);
+      const modelSelectButton = createModelSelectItem(settingsModel, trans);
 
       return {
         create() {
@@ -931,16 +965,21 @@ const completionStatus: JupyterFrontEndPlugin<void> = {
   description: 'The completion status displayed in the status bar',
   autoStart: true,
   requires: [IAISettingsModel],
-  optional: [IStatusBar],
+  optional: [IStatusBar, ITranslator],
   activate: (
     app: JupyterFrontEnd,
     settingsModel: AISettingsModel,
-    statusBar: IStatusBar | null
+    statusBar: IStatusBar | null,
+    translator?: ITranslator
   ) => {
     if (!statusBar) {
       return;
     }
-    const item = new CompletionStatusWidget({ settingsModel });
+    const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
+    const item = new CompletionStatusWidget({
+      settingsModel,
+      translator: trans
+    });
     statusBar?.registerStatusItem('completionState', {
       item,
       align: 'right',

package/src/models/settings-model.ts

@@ -5,7 +5,7 @@ const PLUGIN_ID = '@jupyterlite/ai:settings-model';
 
 export interface IProviderParameters {
   temperature?: number;
-  maxTokens?: number;
+  maxOutputTokens?: number;
   maxTurns?: number;
   supportsFillInMiddle?: boolean;
   useFilterText?: boolean;

package/src/providers/built-in-providers.ts

@@ -15,15 +15,17 @@ export const anthropicProvider: IProviderInfo = {
   name: 'Anthropic Claude',
   apiKeyRequirement: 'required',
   defaultModels: [
+    'claude-opus-4-5',
+    'claude-opus-4-5-20251101',
     'claude-sonnet-4-5',
     'claude-sonnet-4-5-20250929',
     'claude-haiku-4-5',
     'claude-haiku-4-5-20251001',
     'claude-opus-4-1',
-    'claude-opus-4-0',
-    'claude-sonnet-4-0',
     'claude-opus-4-1-20250805',
+    'claude-opus-4-0',
     'claude-opus-4-20250514',
+    'claude-sonnet-4-0',
     'claude-sonnet-4-20250514',
     'claude-3-7-sonnet-latest',
     'claude-3-7-sonnet-20250219',
@@ -58,8 +60,11 @@ export const googleProvider: IProviderInfo = {
   name: 'Google Generative AI',
   apiKeyRequirement: 'required',
   defaultModels: [
-    'gemini-2.5-flash',
+    'gemini-3-pro-preview',
+    'gemini-3-pro-image-preview',
+    'gemini-3-flash-preview',
     'gemini-2.5-pro',
+    'gemini-2.5-flash',
     'gemini-2.5-flash-image-preview',
     'gemini-2.5-flash-lite',
     'gemini-2.5-flash-lite-preview-09-2025',
@@ -84,6 +89,9 @@ export const googleProvider: IProviderInfo = {
     'gemini-1.5-pro-latest',
     'gemini-1.5-pro-001',
     'gemini-1.5-pro-002',
+    'gemini-pro-latest',
+    'gemini-flash-latest',
+    'gemini-flash-lite-latest',
     'gemini-exp-1206',
     'gemma-3-12b-it',
     'gemma-3-27b-it'
@@ -150,12 +158,32 @@ export const openaiProvider: IProviderInfo = {
   name: 'OpenAI',
   apiKeyRequirement: 'required',
   defaultModels: [
-    'o1',
-    'o1-2024-12-17',
-    'o3-mini',
-    'o3-mini-2025-01-31',
+    'gpt-5.2',
+    'gpt-5.2-chat-latest',
+    'gpt-5.2-pro',
+    'gpt-5.1',
+    'gpt-5.1-chat-latest',
+    'gpt-5.1-codex',
+    'gpt-5.1-codex-mini',
+    'gpt-5.1-codex-max',
+    'gpt-5',
+    'gpt-5-2025-08-07',
+    'gpt-5-chat-latest',
+    'gpt-5-codex',
+    'gpt-5-pro',
+    'gpt-5-pro-2025-10-06',
+    'gpt-5-mini',
+    'gpt-5-mini-2025-08-07',
+    'gpt-5-nano',
+    'gpt-5-nano-2025-08-07',
     'o3',
     'o3-2025-04-16',
+    'o3-mini',
+    'o3-mini-2025-01-31',
+    'o1',
+    'o1-2024-12-17',
+    'gpt-4.5-preview',
+    'gpt-4.5-preview-2025-02-27',
     'gpt-4.1',
     'gpt-4.1-2025-04-14',
     'gpt-4.1-mini',
@@ -168,23 +196,14 @@ export const openaiProvider: IProviderInfo = {
     'gpt-4o-2024-11-20',
     'gpt-4o-mini',
     'gpt-4o-mini-2024-07-18',
+    'chatgpt-4o-latest',
     'gpt-4-turbo',
     'gpt-4-turbo-2024-04-09',
     'gpt-4',
     'gpt-4-0613',
-    'gpt-4.5-preview',
-    'gpt-4.5-preview-2025-02-27',
-    'gpt-3.5-turbo-0125',
     'gpt-3.5-turbo',
-    'gpt-3.5-turbo-1106',
-    'chatgpt-4o-latest',
-    'gpt-5',
-    'gpt-5-2025-08-07',
-    'gpt-5-mini',
-    'gpt-5-mini-2025-08-07',
-    'gpt-5-nano',
-    'gpt-5-nano-2025-08-07',
-    'gpt-5-chat-latest'
+    'gpt-3.5-turbo-0125',
+    'gpt-3.5-turbo-1106'
   ],
   supportsBaseURL: true,
   supportsHeaders: true,

package/src/providers/models.ts

@@ -1,4 +1,4 @@
-import type { LanguageModelV2 } from '@ai-sdk/provider';
+import type { LanguageModel } from 'ai';
 import type { IProviderRegistry } from '../tokens';
 
 /**
@@ -38,7 +38,7 @@ export interface IModelOptions {
 export function createCompletionModel(
   options: IModelOptions,
   registry?: IProviderRegistry
-): LanguageModelV2 {
+): LanguageModel {
   if (!registry) {
     throw new Error('Provider registry not available');
   }
@@ -60,7 +60,7 @@ export function createCompletionModel(
 export function createModel(
   options: IModelOptions,
   registry?: IProviderRegistry
-) {
+): LanguageModel {
   if (!registry) {
     throw new Error('Provider registry not available');
   }

package/src/providers/provider-registry.ts

@@ -1,7 +1,5 @@
 import { ISignal, Signal } from '@lumino/signaling';
-import type { LanguageModelV2 } from '@ai-sdk/provider';
-import type { Model } from '@openai/agents';
-import { aisdk } from '@openai/agents-extensions';
+import type { LanguageModel } from 'ai';
 import type { IModelOptions } from './models';
 import { IProviderInfo, IProviderRegistry } from '../tokens';
 
@@ -50,15 +48,13 @@ export class ProviderRegistry implements IProviderRegistry {
    * @param options Model configuration options
    * @returns Chat model instance or null if creation fails
    */
-  createChatModel(id: string, options: IModelOptions): Model | null {
+  createChatModel(id: string, options: IModelOptions): LanguageModel | null {
     const provider = this._providers[id];
     if (!provider) {
       return null;
     }
 
-    const languageModel = provider.factory(options);
-    // wrap with aisdk for compatibility with the agent framework
-    return aisdk(languageModel);
+    return provider.factory(options);
   }
 
   /**
@@ -70,7 +66,7 @@ export class ProviderRegistry implements IProviderRegistry {
   createCompletionModel(
     id: string,
     options: IModelOptions
-  ): LanguageModelV2 | null {
+  ): LanguageModel | null {
     const provider = this._providers[id];
     if (!provider) {
       return null;

package/src/tokens.ts

@@ -1,7 +1,6 @@
 import { Token } from '@lumino/coreutils';
 import { ISignal } from '@lumino/signaling';
-import { FunctionTool, Model } from '@openai/agents';
-import { LanguageModelV2 } from '@ai-sdk/provider';
+import type { Tool, LanguageModel } from 'ai';
 import { AgentManager } from './agent';
 import type { AISettingsModel } from './models/settings-model';
 import type { IModelOptions } from './providers/models';
@@ -21,7 +20,7 @@ export namespace CommandIds {
 /**
  * Type definition for a tool
  */
-export type ITool = FunctionTool<any, any, any>;
+export type ITool = Tool;
 
 /**
  * Interface for token usage statistics from AI model interactions
@@ -109,7 +108,7 @@ export const IProviderRegistry = new Token<IProviderRegistry>(
  * Interface for a provider factory function that creates language models
  */
 export interface IProviderFactory {
-  (options: IModelOptions): LanguageModelV2;
+  (options: IModelOptions): LanguageModel;
 }
 
 /**
@@ -197,7 +196,7 @@ export interface IProviderRegistry {
   /**
    * Create a chat model instance for the given provider.
    */
-  createChatModel(id: string, options: IModelOptions): Model | null;
+  createChatModel(id: string, options: IModelOptions): LanguageModel | null;
 
   /**
    * Create a completion model instance for the given provider.
@@ -205,7 +204,7 @@ export interface IProviderRegistry {
   createCompletionModel(
     id: string,
     options: IModelOptions
-  ): LanguageModelV2 | null;
+  ): LanguageModel | null;
 
   /**
    * Get all available provider IDs.

package/src/tools/commands.ts

@@ -1,5 +1,5 @@
 import { CommandRegistry } from '@lumino/commands';
-import { tool } from '@openai/agents';
+import { tool } from 'ai';
 import { z } from 'zod';
 import { ITool } from '../tokens';
 import { AISettingsModel } from '../models/settings-model';
@@ -9,11 +9,10 @@ import { AISettingsModel } from '../models/settings-model';
  */
 export function createDiscoverCommandsTool(commands: CommandRegistry): ITool {
   return tool({
-    name: 'discover_commands',
+    title: 'Discover Commands',
     description:
       'Discover all available JupyterLab commands with their metadata, arguments, and descriptions',
-    parameters: z.object({
-      // currently unused, but could be used to filter commands by a search term
+    inputSchema: z.object({
       query: z
         .string()
         .optional()
@@ -75,31 +74,28 @@ export function createDiscoverCommandsTool(commands: CommandRegistry): ITool {
 }
 
 /**
- * Create a tool to execute a specific JupyterLab command
+ * Create a tool to execute a specific JupyterLab command.
+ * Commands in the settings' commandsRequiringApproval list will need approval.
  */
 export function createExecuteCommandTool(
   commands: CommandRegistry,
   settingsModel: AISettingsModel
 ): ITool {
   return tool({
-    name: 'execute_command',
+    title: 'Execute Command',
     description:
       'Execute a specific JupyterLab command with optional arguments',
-    parameters: z.object({
+    inputSchema: z.object({
       commandId: z.string().describe('The ID of the command to execute'),
       args: z
         .any()
         .optional()
         .describe('Optional arguments to pass to the command')
     }),
-    needsApproval: async (context, { commandId }) => {
-      // Use configurable list of commands requiring approval
+    needsApproval: (input: { commandId: string; args?: any }) => {
       const commandsRequiringApproval =
-        settingsModel.config.commandsRequiringApproval;
-
-      return commandsRequiringApproval.some(
-        cmd => commandId.includes(cmd) || cmd.includes(commandId)
-      );
+        settingsModel.config.commandsRequiringApproval || [];
+      return commandsRequiringApproval.includes(input.commandId);
     },
     execute: async (input: { commandId: string; args?: any }) => {
       const { commandId, args } = input;
@@ -112,45 +108,38 @@ export function createExecuteCommandTool(
         };
       }
 
-      try {
-        // Execute the command
-        const result = await commands.execute(commandId, args);
+      // Execute the command
+      const result = await commands.execute(commandId, args);
 
-        // Handle Widget objects specially (including subclasses like DocumentWidget)
-        let serializedResult;
-        if (
-          result &&
-          typeof result === 'object' &&
-          (result.constructor?.name?.includes('Widget') || result.id)
-        ) {
-          serializedResult = {
-            type: result.constructor?.name || 'Widget',
-            id: result.id,
-            title: result.title?.label || result.title,
-            className: result.className
-          };
-        } else {
-          // For other objects, try JSON serialization with fallback
-          try {
-            serializedResult = JSON.parse(JSON.stringify(result));
-          } catch {
-            serializedResult = result
-              ? '[Complex object - cannot serialize]'
-              : 'Command executed successfully';
-          }
-        }
-
-        return {
-          success: true,
-          commandId,
-          result: serializedResult
-        };
-      } catch (error) {
-        return {
-          success: false,
-          error: `Failed to execute command '${commandId}': ${error instanceof Error ? error.message : String(error)}`
+      // Handle Widget objects specially (including subclasses like DocumentWidget)
+      let serializedResult;
+      if (
+        result &&
+        typeof result === 'object' &&
+        (result.constructor?.name?.includes('Widget') || result.id)
+      ) {
+        serializedResult = {
+          type: result.constructor?.name || 'Widget',
+          id: result.id,
+          title: result.title?.label || result.title,
+          className: result.className
         };
+      } else {
+        // For other objects, try JSON serialization with fallback
+        try {
+          serializedResult = JSON.parse(JSON.stringify(result));
+        } catch {
+          serializedResult = result
+            ? '[Complex object - cannot serialize]'
+            : 'Command executed successfully';
+        }
       }
+
+      return {
+        success: true,
+        commandId,
+        result: serializedResult
+      };
     }
   });
 }