@jupyterlite/ai 0.9.1 → 0.10.0

package/lib/providers/built-in-providers.js

@@ -11,15 +11,17 @@ export const anthropicProvider = {
     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',
@@ -53,8 +55,11 @@ export const googleProvider = {
     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',
@@ -79,6 +84,9 @@ export const googleProvider = {
         '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'
@@ -143,12 +151,32 @@ export const openaiProvider = {
     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',
@@ -161,23 +189,14 @@ export const openaiProvider = {
         '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/lib/providers/models.d.ts

@@ -1,4 +1,4 @@
-import type { LanguageModelV2 } from '@ai-sdk/provider';
+import type { LanguageModel } from 'ai';
 import type { IProviderRegistry } from '../tokens';
 /**
  * Configuration options for creating language models.
@@ -29,9 +29,9 @@ export interface IModelOptions {
  * Create a completion model using the provider registry.
  * Built-in providers are automatically registered during extension initialization.
  */
-export declare function createCompletionModel(options: IModelOptions, registry?: IProviderRegistry): LanguageModelV2;
+export declare function createCompletionModel(options: IModelOptions, registry?: IProviderRegistry): LanguageModel;
 /**
  * Create a chat model using the provider registry.
  * Built-in providers are automatically registered during extension initialization.
  */
-export declare function createModel(options: IModelOptions, registry?: IProviderRegistry): import("@openai/agents-core").Model;
+export declare function createModel(options: IModelOptions, registry?: IProviderRegistry): LanguageModel;

package/lib/providers/provider-registry.d.ts

@@ -1,6 +1,5 @@
 import { ISignal } from '@lumino/signaling';
-import type { LanguageModelV2 } from '@ai-sdk/provider';
-import type { Model } from '@openai/agents';
+import type { LanguageModel } from 'ai';
 import type { IModelOptions } from './models';
 import { IProviderInfo, IProviderRegistry } from '../tokens';
 /**
@@ -32,14 +31,14 @@ export declare 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;
     /**
      * Create a completion model instance using the specified provider
      * @param id Provider ID
      * @param options Model configuration options
      * @returns Language model instance or null if creation fails
      */
-    createCompletionModel(id: string, options: IModelOptions): LanguageModelV2 | null;
+    createCompletionModel(id: string, options: IModelOptions): LanguageModel | null;
     /**
      * Get list of all available provider IDs
      * @returns Array of provider IDs

package/lib/providers/provider-registry.js

@@ -1,5 +1,4 @@
 import { Signal } from '@lumino/signaling';
-import { aisdk } from '@openai/agents-extensions';
 /**
  * Implementation of the provider registry
  */
@@ -46,9 +45,7 @@ export class ProviderRegistry {
         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);
     }
     /**
      * Create a completion model instance using the specified provider

package/lib/tokens.d.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';
@@ -19,7 +18,7 @@ export declare 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
  */
@@ -89,7 +88,7 @@ export declare const IProviderRegistry: Token<IProviderRegistry>;
  * Interface for a provider factory function that creates language models
  */
 export interface IProviderFactory {
-    (options: IModelOptions): LanguageModelV2;
+    (options: IModelOptions): LanguageModel;
 }
 /**
  * Provider information
@@ -165,11 +164,11 @@ 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.
      */
-    createCompletionModel(id: string, options: IModelOptions): LanguageModelV2 | null;
+    createCompletionModel(id: string, options: IModelOptions): LanguageModel | null;
     /**
      * Get all available provider IDs.
      */

package/lib/tools/commands.d.ts

@@ -6,6 +6,7 @@ import { AISettingsModel } from '../models/settings-model';
  */
 export declare 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 declare function createExecuteCommandTool(commands: CommandRegistry, settingsModel: AISettingsModel): ITool;

package/lib/tools/commands.js

@@ -1,14 +1,13 @@
-import { tool } from '@openai/agents';
+import { tool } from 'ai';
 import { z } from 'zod';
 /**
  * Create a tool to discover all available commands and their metadata
  */
 export function createDiscoverCommandsTool(commands) {
     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()
@@ -57,23 +56,23 @@ export function createDiscoverCommandsTool(commands) {
     });
 }
 /**
- * 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, settingsModel) {
     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
-            const commandsRequiringApproval = settingsModel.config.commandsRequiringApproval;
-            return commandsRequiringApproval.some(cmd => commandId.includes(cmd) || cmd.includes(commandId));
+        needsApproval: (input) => {
+            const commandsRequiringApproval = settingsModel.config.commandsRequiringApproval || [];
+            return commandsRequiringApproval.includes(input.commandId);
         },
         execute: async (input) => {
             const { commandId, args } = input;
@@ -84,44 +83,36 @@ export function createExecuteCommandTool(commands, settingsModel) {
                     error: `Command '${commandId}' does not exist. Use 'discover_commands' to see available commands.`
                 };
             }
-            try {
-                // 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
+            // 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
                 };
             }
-            catch (error) {
-                return {
-                    success: false,
-                    error: `Failed to execute command '${commandId}': ${error instanceof Error ? error.message : String(error)}`
-                };
+            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
+            };
         }
     });
 }

package/lib/tools/file.js

@@ -1,14 +1,14 @@
 import { PathExt } from '@jupyterlab/coreutils';
-import { tool } from '@openai/agents';
+import { tool } from 'ai';
 import { z } from 'zod';
 /**
  * Create a tool for creating new files of various types
  */
 export function createNewFileTool(docManager) {
     return tool({
-        name: 'create_file',
+        title: 'New File',
         description: 'Create a new file of specified type (text, python, markdown, json, etc.)',
-        parameters: z.object({
+        inputSchema: z.object({
             fileName: z.string().describe('Name of the file to create'),
             fileType: z
                 .string()
@@ -25,12 +25,6 @@ export function createNewFileTool(docManager) {
                 .nullable()
                 .describe('Directory where to create the file (optional)')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to create file: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { fileName, content = '', cwd, fileType = 'text' } = input;
             const registeredFileType = docManager.registry.getFileType(fileType);
@@ -77,22 +71,19 @@ export function createNewFileTool(docManager) {
  */
 export function createOpenFileTool(docManager) {
     return tool({
-        name: 'open_file',
+        title: 'Open File',
         description: 'Open a file in the editor',
-        parameters: z.object({
+        inputSchema: z.object({
             filePath: z.string().describe('Path to the file to open')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to open file: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { filePath } = input;
             const widget = docManager.openOrReveal(filePath);
             if (!widget) {
-                throw new Error(`Could not open file: ${filePath}`);
+                return {
+                    success: false,
+                    error: `Could not open file: ${filePath}`
+                };
             }
             return {
                 success: true,
@@ -108,17 +99,11 @@ export function createOpenFileTool(docManager) {
  */
 export function createDeleteFileTool(docManager) {
     return tool({
-        name: 'delete_file',
+        title: 'Delete File',
         description: 'Delete a file from the file system',
-        parameters: z.object({
+        inputSchema: z.object({
             filePath: z.string().describe('Path to the file to delete')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to delete file: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { filePath } = input;
             await docManager.services.contents.delete(filePath);
@@ -135,18 +120,12 @@ export function createDeleteFileTool(docManager) {
  */
 export function createRenameFileTool(docManager) {
     return tool({
-        name: 'rename_file',
+        title: 'Rename File',
         description: 'Rename a file or move it to a different location',
-        parameters: z.object({
+        inputSchema: z.object({
             oldPath: z.string().describe('Current path of the file'),
             newPath: z.string().describe('New path/name for the file')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to rename file: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { oldPath, newPath } = input;
             await docManager.services.contents.rename(oldPath, newPath);
@@ -164,20 +143,14 @@ export function createRenameFileTool(docManager) {
  */
 export function createCopyFileTool(docManager) {
     return tool({
-        name: 'copy_file',
+        title: 'Copy File',
         description: 'Copy a file to a new location',
-        parameters: z.object({
+        inputSchema: z.object({
             sourcePath: z.string().describe('Path of the file to copy'),
             destinationPath: z
                 .string()
                 .describe('Destination path for the copied file')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to copy file: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { sourcePath, destinationPath } = input;
             await docManager.services.contents.copy(sourcePath, destinationPath);
@@ -195,17 +168,11 @@ export function createCopyFileTool(docManager) {
  */
 export function createNavigateToDirectoryTool(commands) {
     return tool({
-        name: 'navigate_to_directory',
+        title: 'Navigate to Directory',
         description: 'Navigate to a specific directory in the file browser',
-        parameters: z.object({
+        inputSchema: z.object({
             directoryPath: z.string().describe('Path to the directory to navigate to')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to navigate to directory: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { directoryPath } = input;
             await commands.execute('filebrowser:go-to-path', {
@@ -224,21 +191,15 @@ export function createNavigateToDirectoryTool(commands) {
  */
 export function createGetFileInfoTool(docManager, editorTracker) {
     return tool({
-        name: 'get_file_info',
+        title: 'Get File Info',
         description: 'Get information about a file including its path, name, extension, and content. Works with text-based files like Python files, markdown, JSON, etc. For Jupyter notebooks, use dedicated notebook tools instead. If no file path is provided, returns information about the currently active file in the editor.',
-        parameters: z.object({
+        inputSchema: z.object({
             filePath: z
                 .string()
                 .optional()
                 .nullable()
                 .describe('Path to the file to read (e.g., "script.py", "README.md", "config.json"). If not provided, uses the currently active file in the editor.')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to get file info: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { filePath } = input;
             let widget = null;
@@ -248,22 +209,34 @@ export function createGetFileInfoTool(docManager, editorTracker) {
                         docManager.openOrReveal(filePath) ??
                         null;
                 if (!widget) {
-                    throw new Error(`Failed to open file at path: ${filePath}`);
+                    return JSON.stringify({
+                        success: false,
+                        error: `Failed to open file at path: ${filePath}`
+                    });
                 }
             }
             else {
                 widget = editorTracker?.currentWidget ?? null;
                 if (!widget) {
-                    throw new Error('No active file in the editor and no file path provided');
+                    return JSON.stringify({
+                        success: false,
+                        error: 'No active file in the editor and no file path provided'
+                    });
                 }
             }
             if (!widget.context) {
-                throw new Error('Widget is not a document');
+                return JSON.stringify({
+                    success: false,
+                    error: 'Widget is not a document'
+                });
             }
             await widget.context.ready;
             const model = widget.context.model;
             if (!model) {
-                throw new Error('File model not available');
+                return JSON.stringify({
+                    success: false,
+                    error: 'File model not available'
+                });
             }
             const sharedModel = model.sharedModel;
             const content = sharedModel.getSource();
@@ -288,9 +261,9 @@ export function createGetFileInfoTool(docManager, editorTracker) {
  */
 export function createSetFileContentTool(docManager, diffManager) {
     return tool({
-        name: 'set_file_content',
+        title: 'Set File Content',
         description: 'Set or update the content of an existing file. This will replace the entire content of the file. For Jupyter notebooks, use dedicated notebook tools instead.',
-        parameters: z.object({
+        inputSchema: z.object({
             filePath: z
                 .string()
                 .describe('Path to the file to update (e.g., "script.py", "README.md", "config.json")'),
@@ -301,12 +274,6 @@ export function createSetFileContentTool(docManager, diffManager) {
                 .default(true)
                 .describe('Whether to save the file after updating (default: true)')
         }),
-        errorFunction: (context, error) => {
-            return JSON.stringify({
-                success: false,
-                error: `Failed to set file content: ${error instanceof Error ? error.message : String(error)}`
-            });
-        },
         execute: async (input) => {
             const { filePath, content, save = true } = input;
             let widget = docManager.findWidget(filePath);
@@ -314,15 +281,24 @@ export function createSetFileContentTool(docManager, diffManager) {
                 widget = docManager.openOrReveal(filePath);
             }
             if (!widget) {
-                throw new Error(`Failed to open file at path: ${filePath}`);
+                return JSON.stringify({
+                    success: false,
+                    error: `Failed to open file at path: ${filePath}`
+                });
             }
             await widget.context.ready;
             const model = widget.context.model;
             if (!model) {
-                throw new Error('File model not available');
+                return JSON.stringify({
+                    success: false,
+                    error: 'File model not available'
+                });
             }
             if (model.readOnly) {
-                throw new Error('File is read-only and cannot be modified');
+                return JSON.stringify({
+                    success: false,
+                    error: 'File is read-only and cannot be modified'
+                });
             }
             const sharedModel = model.sharedModel;
             const originalContent = sharedModel.getSource();