@jupyterlite/ai 0.9.1 → 0.11.0

package/lib/index.js

@@ -3,12 +3,11 @@ import { ActiveCellManager, AttachmentOpenerRegistry, chatIcon, ChatWidget, IInp
 import { ICommandPalette, IThemeManager, WidgetTracker } from '@jupyterlab/apputils';
 import { ICompletionProviderManager } from '@jupyterlab/completer';
 import { IDocumentManager } from '@jupyterlab/docmanager';
-import { IEditorTracker } from '@jupyterlab/fileeditor';
 import { INotebookTracker } from '@jupyterlab/notebook';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
-import { IKernelSpecManager } from '@jupyterlab/services';
 import { ISettingRegistry } from '@jupyterlab/settingregistry';
 import { IStatusBar } from '@jupyterlab/statusbar';
+import { ITranslator, nullTranslator } from '@jupyterlab/translation';
 import { settingsIcon, Toolbar, ToolbarButton } from '@jupyterlab/ui-components';
 import { ISecretsManager, SecretsManager } from 'jupyter-secrets-manager';
 import { PromiseDelegate, UUID } from '@lumino/coreutils';
@@ -23,8 +22,6 @@ import { clearItem, createModelSelectItem, createToolSelectItem, stopItem, Compl
 import { AISettingsModel } from './models/settings-model';
 import { DiffManager } from './diff-manager';
 import { ToolRegistry } from './tools/tool-registry';
-import { createAddCellTool, createDeleteCellTool, createExecuteActiveCellTool, createGetCellInfoTool, createGetNotebookInfoTool, createNotebookCreationTool, createRunCellTool, createSaveNotebookTool, createSetCellContentTool } from './tools/notebook';
-import { createCopyFileTool, createDeleteFileTool, createGetFileInfoTool, createNavigateToDirectoryTool, createNewFileTool, createOpenFileTool, createRenameFileTool, createSetFileContentTool } from './tools/file';
 import { createDiscoverCommandsTool, createExecuteCommandTool } from './tools/commands';
 import { AISettingsWidget } from './widgets/ai-settings';
 import { MainAreaChat } from './widgets/main-area-chat';
@@ -108,9 +105,10 @@ const chatModelRegistry = {
     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, settingsModel, agentManagerFactory, docManager, providerRegistry, notebookTracker, toolRegistry) => {
+    activate: (app, settingsModel, agentManagerFactory, docManager, providerRegistry, notebookTracker, toolRegistry, translator) => {
+        const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
         // Create ActiveCellManager if notebook tracker is available
         let activeCellManager;
         if (notebookTracker) {
@@ -125,7 +123,8 @@ const chatModelRegistry = {
             agentManagerFactory,
             docManager,
             providerRegistry,
-            toolRegistry
+            toolRegistry,
+            trans
         });
     }
 };
@@ -142,8 +141,9 @@ const plugin = {
         IChatModelRegistry,
         IAISettingsModel
     ],
-    optional: [IThemeManager, ILayoutRestorer, ILabShell],
-    activate: (app, rmRegistry, inputToolbarFactory, modelRegistry, settingsModel, themeManager, restorer, labShell) => {
+    optional: [IThemeManager, ILayoutRestorer, ILabShell, ITranslator],
+    activate: (app, rmRegistry, inputToolbarFactory, modelRegistry, settingsModel, themeManager, restorer, labShell, translator) => {
+        const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
         // Create attachment opener registry to handle file attachments
         const attachmentOpenerRegistry = new AttachmentOpenerRegistry();
         attachmentOpenerRegistry.set('file', attachment => {
@@ -178,14 +178,14 @@ const plugin = {
         });
         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('settings', new ToolbarButton({
             icon: settingsIcon,
             onClick: () => {
                 app.commands.execute('@jupyterlite/ai:open-settings');
             },
-            tooltip: 'Open AI Settings'
+            tooltip: trans.__('Open AI Settings')
         }));
         chatPanel.sectionAdded.connect((_, section) => {
             const { widget } = section;
@@ -199,7 +199,8 @@ const plugin = {
             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) => {
@@ -216,7 +217,8 @@ const plugin = {
             });
             // Associate an approval buttons object to the chat.
             const approvalButton = new ApprovalButtons({
-                chatPanel: widget
+                chatPanel: widget,
+                agentManager: model.agentManager
             });
             widget.disposed.connect(() => {
                 // Dispose of the approval buttons widget when the chat is disposed.
@@ -251,14 +253,14 @@ const plugin = {
                 app.commands.execute(CommandIds.openChat);
             }
         });
-        registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry, inputToolbarFactory, settingsModel, tracker, modelRegistry, themeManager, labShell);
+        registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry, inputToolbarFactory, settingsModel, tracker, modelRegistry, trans, themeManager, labShell);
     }
 };
-function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry, inputToolbarFactory, settingsModel, tracker, modelRegistry, themeManager, labShell) {
+function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry, inputToolbarFactory, settingsModel, tracker, modelRegistry, trans, themeManager, labShell) {
     const { commands } = app;
     if (labShell) {
         commands.addCommand(CommandIds.reposition, {
-            label: 'Reposition Widget',
+            label: trans.__('Reposition Widget'),
             execute: (args) => {
                 const { widgetId, area, mode } = args;
                 const widget = widgetId
@@ -284,16 +286,16 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
                     properties: {
                         widgetId: {
                             type: 'string',
-                            description: 'The widget ID to reposition in the application shell'
+                            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')
                         }
                     }
                 }
@@ -307,7 +309,12 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
                 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.
             tracker.add(widget);
@@ -321,7 +328,7 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
             });
         };
         commands.addCommand(CommandIds.openChat, {
-            label: 'Open a chat',
+            label: trans.__('Open a chat'),
             execute: async (args) => {
                 const area = args.area === 'main' ? 'main' : 'side';
                 const provider = args.provider ?? undefined;
@@ -348,22 +355,22 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
                         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')
                         }
                     }
                 }
             }
         });
         commands.addCommand(CommandIds.moveChat, {
-            caption: 'Move chat between area',
+            caption: trans.__('Move chat between area'),
             execute: async (args) => {
                 const area = args.area;
                 if (!['side', 'main'].includes(area)) {
@@ -399,7 +406,7 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
                 const status = await Promise.any([
                     trackerUpdated.promise,
                     new Promise(r => setTimeout(() => {
-                        return false;
+                        r(false);
                     }, 2000))
                 ]);
                 tracker.widgetUpdated.disconnect(widgetUpdated);
@@ -427,11 +434,11 @@ function registerCommands(app, rmRegistry, chatPanel, attachmentOpenerRegistry,
                         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']
@@ -454,9 +461,11 @@ const agentManagerFactory = SecretsManager.sign(SECRETS_NAMESPACE, token => ({
         ICompletionProviderManager,
         ILayoutRestorer,
         ISecretsManager,
-        IThemeManager
+        IThemeManager,
+        ITranslator
     ],
-    activate: (app, settingsModel, providerRegistry, palette, completionManager, restorer, secretsManager, themeManager) => {
+    activate: (app, settingsModel, providerRegistry, palette, completionManager, restorer, secretsManager, themeManager, translator) => {
+        const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
         const agentManagerFactory = new AgentManagerFactory({
             settingsModel,
             secretsManager,
@@ -469,7 +478,8 @@ const agentManagerFactory = SecretsManager.sign(SECRETS_NAMESPACE, token => ({
             themeManager,
             providerRegistry,
             secretsManager,
-            token
+            token,
+            trans
         });
         settingsWidget.id = 'jupyterlite-ai-settings';
         settingsWidget.title.icon = settingsIcon;
@@ -491,8 +501,8 @@ const agentManagerFactory = SecretsManager.sign(SECRETS_NAMESPACE, token => ({
             restorer.add(settingsWidget, settingsWidget.id);
         }
         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: () => {
@@ -515,7 +525,7 @@ const agentManagerFactory = SecretsManager.sign(SECRETS_NAMESPACE, token => ({
         if (palette) {
             palette.addItem({
                 command: CommandIds.openSettings,
-                category: 'AI Assistant'
+                category: trans.__('AI Assistant')
             });
         }
         return agentManagerFactory;
@@ -554,47 +564,10 @@ const toolRegistry = {
     id: '@jupyterlite/ai:tool-registry',
     description: 'Provide the AI tool registry',
     autoStart: true,
-    requires: [IAISettingsModel, IDocumentManager, IKernelSpecManager],
-    optional: [INotebookTracker, IDiffManager, IEditorTracker],
+    requires: [IAISettingsModel],
     provides: IToolRegistry,
-    activate: (app, settingsModel, docManager, kernelSpecManager, notebookTracker, diffManager, editorTracker) => {
+    activate: (app, settingsModel) => {
         const toolRegistry = new ToolRegistry();
-        const notebookCreationTool = createNotebookCreationTool(docManager, kernelSpecManager);
-        toolRegistry.add('create_notebook', notebookCreationTool);
-        // Add high-level notebook operation tools
-        const addCellTool = createAddCellTool(docManager, notebookTracker);
-        const getNotebookInfoTool = createGetNotebookInfoTool(docManager, notebookTracker);
-        const getCellInfoTool = createGetCellInfoTool(docManager, notebookTracker);
-        const setCellContentTool = createSetCellContentTool(docManager, notebookTracker, diffManager);
-        const runCellTool = createRunCellTool(docManager, notebookTracker);
-        const deleteCellTool = createDeleteCellTool(docManager, notebookTracker);
-        const saveNotebookTool = createSaveNotebookTool(docManager, notebookTracker);
-        const executeActiveCellTool = createExecuteActiveCellTool(docManager, notebookTracker);
-        toolRegistry.add('add_cell', addCellTool);
-        toolRegistry.add('get_notebook_info', getNotebookInfoTool);
-        toolRegistry.add('get_cell_info', getCellInfoTool);
-        toolRegistry.add('set_cell_content', setCellContentTool);
-        toolRegistry.add('run_cell', runCellTool);
-        toolRegistry.add('delete_cell', deleteCellTool);
-        toolRegistry.add('save_notebook', saveNotebookTool);
-        toolRegistry.add('execute_active_cell', executeActiveCellTool);
-        // Add file operation tools
-        const newFileTool = createNewFileTool(docManager);
-        const openFileTool = createOpenFileTool(docManager);
-        const deleteFileTool = createDeleteFileTool(docManager);
-        const renameFileTool = createRenameFileTool(docManager);
-        const copyFileTool = createCopyFileTool(docManager);
-        const navigateToDirectoryTool = createNavigateToDirectoryTool(app.commands);
-        const getFileInfoTool = createGetFileInfoTool(docManager, editorTracker);
-        const setFileContentTool = createSetFileContentTool(docManager, diffManager);
-        toolRegistry.add('create_file', newFileTool);
-        toolRegistry.add('open_file', openFileTool);
-        toolRegistry.add('delete_file', deleteFileTool);
-        toolRegistry.add('rename_file', renameFileTool);
-        toolRegistry.add('copy_file', copyFileTool);
-        toolRegistry.add('navigate_to_directory', navigateToDirectoryTool);
-        toolRegistry.add('get_file_info', getFileInfoTool);
-        toolRegistry.add('set_file_content', setFileContentTool);
         // Add command operation tools
         const discoverCommandsTool = createDiscoverCommandsTool(app.commands);
         const executeCommandTool = createExecuteCommandTool(app.commands, settingsModel);
@@ -612,11 +585,13 @@ const inputToolbarFactory = {
     autoStart: true,
     provides: IInputToolbarRegistryFactory,
     requires: [IAISettingsModel, IToolRegistry],
-    activate: (app, settingsModel, toolRegistry) => {
-        const stopButton = stopItem();
-        const clearButton = clearItem();
-        const toolSelectButton = createToolSelectItem(toolRegistry, settingsModel.config.toolsEnabled);
-        const modelSelectButton = createModelSelectItem(settingsModel);
+    optional: [ITranslator],
+    activate: (app, settingsModel, toolRegistry, translator) => {
+        const trans = (translator ?? nullTranslator).load('jupyterlite_ai');
+        const stopButton = stopItem(trans);
+        const clearButton = clearItem(trans);
+        const toolSelectButton = createToolSelectItem(toolRegistry, settingsModel.config.toolsEnabled, trans);
+        const modelSelectButton = createModelSelectItem(settingsModel, trans);
         return {
             create() {
                 const inputToolbarRegistry = InputToolbarRegistry.defaultToolbarRegistry();
@@ -644,12 +619,16 @@ const completionStatus = {
     description: 'The completion status displayed in the status bar',
     autoStart: true,
     requires: [IAISettingsModel],
-    optional: [IStatusBar],
-    activate: (app, settingsModel, statusBar) => {
+    optional: [IStatusBar, ITranslator],
+    activate: (app, settingsModel, statusBar, translator) => {
         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',
@@ -675,3 +654,4 @@ export default [
 ];
 // Export extension points for other extensions to use
 export * from './tokens';
+export * from './icons';

package/lib/models/settings-model.d.ts

@@ -2,7 +2,7 @@ import { VDomModel } from '@jupyterlab/ui-components';
 import { ISettingRegistry } from '@jupyterlab/settingregistry';
 export interface IProviderParameters {
     temperature?: number;
-    maxTokens?: number;
+    maxOutputTokens?: number;
     maxTurns?: number;
     supportsFillInMiddle?: boolean;
     useFilterText?: boolean;

package/lib/models/settings-model.js

@@ -29,7 +29,8 @@ export class AISettingsModel extends VDomModel {
             'fileeditor:run-code',
             'kernelmenu:run',
             'kernelmenu:restart-and-run-all',
-            'runmenu:run-all'
+            'runmenu:run-all',
+            'jupyterlab-ai-commands:run-cell'
         ],
         systemPrompt: `You are Jupyternaut, an AI coding assistant built specifically for the JupyterLab environment.
 
@@ -38,7 +39,7 @@ You're designed to be a capable partner for data science, research, and developm
 
 ## Your Capabilities
 **📁 File & Project Management:**
-- Create, read, edit, and organize Python files and notebooks
+- Create, read, edit, and organize files and notebooks in any language
 - Manage project structure and navigate file systems
 - Help with version control and project organization
 
@@ -48,51 +49,64 @@ You're designed to be a capable partner for data science, research, and developm
 - Help with notebook structure and organization
 - Retrieve and analyze cell outputs and execution results
 
+**⚡ Kernel Management:**
+- Start new kernels with specified language or kernel name
+- Execute code directly in running kernels without creating cells
+- List running kernels and monitor their status
+- Manage kernel lifecycle (start, monitor, shutdown)
+
 **🧠 Coding & Development:**
-- Write, debug, and optimize Python code
+- Write, debug, and optimize code in any language supported by Jupyter kernels (Python, R, Julia, JavaScript, C++, and more)
 - Explain complex algorithms and data structures
 - Help with data analysis, visualization, and machine learning
-- Support for scientific computing libraries (numpy, pandas, matplotlib, etc.)
+- Support for libraries and packages across different languages
 - Code reviews and best practices recommendations
 
 **💡 Adaptive Assistance:**
-- Understand context from your current work environment
-- Provide suggestions tailored to your specific use case
+- Understand context from the user's current work environment
+- Provide suggestions tailored to the user's specific use case
 - Help with both quick fixes and long-term project planning
 
-## How I Work
-I can actively interact with your JupyterLab environment using specialized tools. When you ask me to perform actions, I can:
-- Execute operations directly in your notebooks
+## How You Work
+You can actively interact with the user's JupyterLab environment using specialized tools. When asked to perform actions, you can:
+- Execute operations directly in notebooks
 - Create and modify files as needed
 - Run code and analyze results
 - Make systematic changes across multiple files
 
-## My Approach
-- **Context-aware**: I understand you're working in a data science/research environment
-- **Practical**: I focus on actionable solutions that work in your current setup
-- **Educational**: I explain my reasoning and teach best practices along the way
-- **Collaborative**: Think of me as a pair programming partner, not just a code generator
+## Code Execution Strategy
+When asked to run code or perform computations, choose the most appropriate approach:
+- **For quick computations or one-off code execution**: Use kernel commands to start a kernel and execute code directly, without creating notebook files. This is ideal for calculations, data lookups, or testing code snippets.
+- **For work that should be saved**: Create or use notebooks when the user needs a persistent record of their work, wants to iterate on code, or is building something they'll return to later.
+
+This means if the user asks you to "calculate the factorial of 100" or "check what library version is installed", run that directly in a kernel rather than creating a new notebook file.
+
+## Your Approach
+- **Context-aware**: You understand the user is working in a data science/research environment
+- **Practical**: You focus on actionable solutions that work in the user's current setup
+- **Educational**: You explain your reasoning and teach best practices along the way
+- **Collaborative**: You are a pair programming partner, not just a code generator
 
 ## Communication Style & Agent Behavior
-- **Conversational**: I maintain a friendly, natural conversation flow throughout our interaction
-- **Progress Updates**: I write brief progress messages between tool uses that appear directly in our conversation
-- **No Filler**: I avoid empty acknowledgments like "Sounds good!" or "Okay, I will..." - I get straight to work
-- **Purposeful Communication**: I start with what I'm doing, use tools, then share what I found and what's next
-- **Active Narration**: I actively write progress updates like "Looking at the current code structure..." or "Found the issue in the notebook..." between tool calls
-- **Checkpoint Updates**: After several operations, I summarize what I've accomplished and what remains
-- **Natural Flow**: My explanations and progress reports appear as normal conversation text, not just in tool blocks
+- **Conversational**: You maintain a friendly, natural conversation flow throughout the interaction
+- **Progress Updates**: You write brief progress messages between tool uses that appear directly in the conversation
+- **No Filler**: You avoid empty acknowledgments like "Sounds good!" or "Okay, I will..." - you get straight to work
+- **Purposeful Communication**: You start with what you're doing, use tools, then share what you found and what's next
+- **Active Narration**: You actively write progress updates like "Looking at the current code structure..." or "Found the issue in the notebook..." between tool calls
+- **Checkpoint Updates**: After several operations, you summarize what you've accomplished and what remains
+- **Natural Flow**: Your explanations and progress reports appear as normal conversation text, not just in tool blocks
 
 ## IMPORTANT: Always write progress messages between tools that explain what you're doing and what you found. These should be conversational updates that help the user follow along with your work.
 
 ## Technical Communication
 - Code is formatted in proper markdown blocks with syntax highlighting
 - Mathematical notation uses LaTeX formatting: \\(equations\\) and \\[display math\\]
-- I provide context for my actions and explain my reasoning as I work
-- When creating or modifying multiple files, I give brief summaries of changes
-- I keep users informed of progress while staying focused on the task
+- You provide context for your actions and explain your reasoning as you work
+- When creating or modifying multiple files, you give brief summaries of changes
+- You keep users informed of progress while staying focused on the task
 
 ## Multi-Step Task Handling
-When users request complex tasks that require multiple steps (like "create a notebook with example cells"), I use tools in sequence to accomplish the complete task. For example:
+When users request complex tasks that require multiple steps (like "create a notebook with example cells"), you use tools in sequence to accomplish the complete task. For example:
 - First use create_notebook to create the notebook
 - Then use add_code_cell or add_markdown_cell to add cells
 - Use set_cell_content to add content to cells as needed
@@ -100,7 +114,7 @@ When users request complex tasks that require multiple steps (like "create a not
 
 Always think through multi-step tasks and use tools to fully complete the user's request rather than stopping after just one action.
 
-Ready to help you build something great! What are you working on?`,
+You are ready to help users build something great!`,
         // Completion system prompt - also defined in schema/settings-model.json
         // This serves as a fallback if settings fail to load or are not available
         completionSystemPrompt: `You are an AI code completion assistant. Complete the given code fragment with appropriate code.

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