@jupyterlite/ai 0.9.1 → 0.11.0

package/src/index.ts

@@ -26,18 +26,20 @@ 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, KernelSpec } from '@jupyterlab/services';
-
 import { ISettingRegistry } from '@jupyterlab/settingregistry';
 
 import { IStatusBar } from '@jupyterlab/statusbar';
 
+import {
+  ITranslator,
+  nullTranslator,
+  TranslationBundle
+} from '@jupyterlab/translation';
+
 import {
   settingsIcon,
   Toolbar,
@@ -94,29 +96,6 @@ 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
@@ -212,7 +191,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 +200,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 +219,8 @@ const chatModelRegistry: JupyterFrontEndPlugin<IChatModelRegistry> = {
       agentManagerFactory,
       docManager,
       providerRegistry,
-      toolRegistry
+      toolRegistry,
+      trans
     });
   }
 };
@@ -255,7 +238,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
     IChatModelRegistry,
     IAISettingsModel
   ],
-  optional: [IThemeManager, ILayoutRestorer, ILabShell],
+  optional: [IThemeManager, ILayoutRestorer, ILabShell, ITranslator],
   activate: (
     app: JupyterFrontEnd,
     rmRegistry: IRenderMimeRegistry,
@@ -264,8 +247,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 +289,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 +299,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
         onClick: () => {
           app.commands.execute('@jupyterlite/ai:open-settings');
         },
-        tooltip: 'Open AI Settings'
+        tooltip: trans.__('Open AI Settings')
       })
     );
 
@@ -335,7 +320,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 +341,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 +394,7 @@ const plugin: JupyterFrontEndPlugin<void> = {
       settingsModel,
       tracker,
       modelRegistry,
+      trans,
       themeManager,
       labShell
     );
@@ -422,6 +410,7 @@ function registerCommands(
   settingsModel: AISettingsModel,
   tracker: WidgetTracker<MainAreaChat | ChatWidget>,
   modelRegistry: IChatModelRegistry,
+  trans: TranslationBundle,
   themeManager?: IThemeManager,
   labShell?: ILabShell
 ) {
@@ -429,7 +418,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 +445,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 +475,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 +500,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 +531,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 +547,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 +600,7 @@ function registerCommands(
           trackerUpdated.promise,
           new Promise<boolean>(r =>
             setTimeout(() => {
-              return false;
+              r(false);
             }, 2000)
           )
         ]);
@@ -634,11 +633,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 +662,8 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       ICompletionProviderManager,
       ILayoutRestorer,
       ISecretsManager,
-      IThemeManager
+      IThemeManager,
+      ITranslator
     ],
     activate: (
       app: JupyterFrontEnd,
@@ -673,8 +673,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 +690,8 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
         themeManager,
         providerRegistry,
         secretsManager,
-        token
+        token,
+        trans
       });
       settingsWidget.id = 'jupyterlite-ai-settings';
       settingsWidget.title.icon = settingsIcon;
@@ -715,8 +718,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 +747,7 @@ const agentManagerFactory: JupyterFrontEndPlugin<AgentManagerFactory> =
       if (palette) {
         palette.addItem({
           command: CommandIds.openSettings,
-          category: 'AI Assistant'
+          category: trans.__('AI Assistant')
         });
       }
 
@@ -790,80 +793,11 @@ const toolRegistry: JupyterFrontEndPlugin<IToolRegistry> = {
   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: JupyterFrontEnd,
-    settingsModel: AISettingsModel,
-    docManager: IDocumentManager,
-    kernelSpecManager: KernelSpec.IManager,
-    notebookTracker?: INotebookTracker,
-    diffManager?: IDiffManager,
-    editorTracker?: IEditorTracker
-  ) => {
+  activate: (app: JupyterFrontEnd, settingsModel: AISettingsModel) => {
     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(
@@ -888,18 +822,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 +869,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',
@@ -968,3 +911,4 @@ export default [
 
 // Export extension points for other extensions to use
 export * from './tokens';
+export * from './icons';

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;
@@ -91,7 +91,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.
 
@@ -100,7 +101,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
 
@@ -110,51 +111,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
@@ -162,7 +176,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.