@jupyterlite/ai 0.9.1 → 0.11.0

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,16 +9,17 @@ 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()
         .nullable()
-        .describe('Optional search query to filter commands')
+        .describe(
+          "Optional search query to filter commands. It doesn't need to be provided to list all commands"
+        )
     }),
     execute: async (input: { query?: string | null }) => {
       const { query } = input;
@@ -75,31 +76,30 @@ 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()
+        .record(z.string(), z.unknown())
         .optional()
-        .describe('Optional arguments to pass to the command')
+        .describe(
+          'Optional arguments object to pass to the command (must be an object, not a string)'
+        )
     }),
-    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 +112,32 @@ 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 by extracting id and title
+      let serializedResult;
+      if (result && typeof result === 'object' && result.id) {
+        serializedResult = {
+          id: result.id,
+          title: result.title?.label || result.title
         };
+      } 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
+      };
     }
   });
 }