@jupyterlite/ai 0.9.1 → 0.11.0

package/lib/widgets/provider-config-dialog.js

@@ -8,7 +8,7 @@ import React from 'react';
  */
 const DEFAULT_TEMPERATURE = 0.7;
 const DEFAULT_MAX_TURNS = 25;
-export const ProviderConfigDialog = ({ open, onClose, onSave, initialConfig, mode, providerRegistry, handleSecretField }) => {
+export const ProviderConfigDialog = ({ open, onClose, onSave, initialConfig, mode, providerRegistry, handleSecretField, trans }) => {
     const apiKeyRef = React.useRef();
     const [name, setName] = React.useState(initialConfig?.name || '');
     const [provider, setProvider] = React.useState(initialConfig?.provider || 'anthropic');
@@ -88,40 +88,42 @@ export const ProviderConfigDialog = ({ open, onClose, onSave, initialConfig, mod
         model &&
         (selectedProvider?.apiKeyRequirement !== 'required' || apiKey);
     return (React.createElement(Dialog, { open: open, onClose: onClose, maxWidth: "md", fullWidth: true },
-        React.createElement(DialogTitle, null, mode === 'add' ? 'Add New Provider' : 'Edit Provider'),
+        React.createElement(DialogTitle, null, mode === 'add'
+            ? trans.__('Add New Provider')
+            : trans.__('Edit Provider')),
         React.createElement(DialogContent, null,
             React.createElement(Box, { sx: { display: 'flex', flexDirection: 'column', gap: 2, pt: 1 } },
-                React.createElement(TextField, { fullWidth: true, label: "Provider Name", value: name, onChange: e => setName(e.target.value), placeholder: "e.g., My Anthropic Config, Work Provider", helperText: "A friendly name to identify this provider configuration", required: true }),
+                React.createElement(TextField, { fullWidth: true, label: trans.__('Provider Name'), value: name, onChange: e => setName(e.target.value), placeholder: trans.__('e.g., My Anthropic Config, Work Provider'), helperText: trans.__('A friendly name to identify this provider configuration'), required: true }),
                 React.createElement(FormControl, { fullWidth: true, required: true },
-                    React.createElement(InputLabel, null, "Provider Type"),
-                    React.createElement(Select, { value: provider, label: "Provider Type", onChange: e => setProvider(e.target.value) }, providerOptions.map(option => (React.createElement(MenuItem, { key: option.value, value: option.value },
+                    React.createElement(InputLabel, null, trans.__('Provider Type')),
+                    React.createElement(Select, { value: provider, label: trans.__('Provider Type'), onChange: e => setProvider(e.target.value) }, providerOptions.map(option => (React.createElement(MenuItem, { key: option.value, value: option.value },
                         React.createElement(Box, null,
                             React.createElement(Box, { sx: { display: 'flex', alignItems: 'center', gap: 1 } },
                                 option.label,
-                                option.apiKeyRequirement === 'required' && (React.createElement(Chip, { size: "small", label: "API Key", color: "default", variant: "outlined" }))),
+                                option.apiKeyRequirement === 'required' && (React.createElement(Chip, { size: "small", label: trans.__('API Key'), color: "default", variant: "outlined" }))),
                             option.description && (React.createElement(Typography, { variant: "caption", color: "text.secondary" }, option.description)))))))),
-                selectedProvider?.allowCustomModel ? (React.createElement(TextField, { fullWidth: true, label: "Model", value: model, onChange: e => setModel(e.target.value), placeholder: "Enter model name", helperText: "Enter any compatible model name", required: true })) : (React.createElement(FormControl, { fullWidth: true, required: true },
-                    React.createElement(InputLabel, null, "Model"),
-                    React.createElement(Select, { value: model, label: "Model", onChange: e => setModel(e.target.value) }, selectedProvider?.models.map(modelOption => (React.createElement(MenuItem, { key: modelOption, value: modelOption },
+                selectedProvider?.allowCustomModel ? (React.createElement(TextField, { fullWidth: true, label: trans.__('Model'), value: model, onChange: e => setModel(e.target.value), placeholder: trans.__('Enter model name'), helperText: trans.__('Enter any compatible model name'), required: true })) : (React.createElement(FormControl, { fullWidth: true, required: true },
+                    React.createElement(InputLabel, null, trans.__('Model')),
+                    React.createElement(Select, { value: model, label: trans.__('Model'), onChange: e => setModel(e.target.value) }, selectedProvider?.models.map(modelOption => (React.createElement(MenuItem, { key: modelOption, value: modelOption },
                         React.createElement(Box, null,
                             React.createElement(Typography, { variant: "body1" }, modelOption),
                             React.createElement(Typography, { variant: "caption", color: "text.secondary" }, modelOption.includes('sonnet')
-                                ? 'Balanced performance'
+                                ? trans.__('Balanced performance')
                                 : modelOption.includes('opus')
-                                    ? 'Advanced reasoning'
+                                    ? trans.__('Advanced reasoning')
                                     : modelOption.includes('haiku')
-                                        ? 'Fast and lightweight'
+                                        ? trans.__('Fast and lightweight')
                                         : modelOption.includes('large')
-                                            ? 'Most capable model'
+                                            ? trans.__('Most capable model')
                                             : modelOption.includes('small')
-                                                ? 'Fast and efficient'
+                                                ? trans.__('Fast and efficient')
                                                 : modelOption.includes('codestral')
-                                                    ? 'Code-specialized'
-                                                    : 'General purpose')))))))),
+                                                    ? trans.__('Code-specialized')
+                                                    : trans.__('General purpose'))))))))),
                 selectedProvider &&
                     selectedProvider?.apiKeyRequirement !== 'none' && (React.createElement(TextField, { fullWidth: true, inputRef: apiKeyRef, label: selectedProvider?.apiKeyRequirement === 'required'
-                        ? 'API Key'
-                        : 'API Key (Optional)', type: showApiKey ? 'text' : 'password', value: apiKey, onChange: e => setApiKey(e.target.value), placeholder: "Enter your API key...", required: selectedProvider?.apiKeyRequirement === 'required', InputProps: {
+                        ? trans.__('API Key')
+                        : trans.__('API Key (Optional)'), type: showApiKey ? 'text' : 'password', value: apiKey, onChange: e => setApiKey(e.target.value), placeholder: trans.__('Enter your API key...'), required: selectedProvider?.apiKeyRequirement === 'required', InputProps: {
                         endAdornment: (React.createElement(InputAdornment, { position: "end" },
                             React.createElement(IconButton, { onClick: () => setShowApiKey(!showApiKey), edge: "end" }, showApiKey ? React.createElement(VisibilityOff, null) : React.createElement(Visibility, null))))
                     } })),
@@ -135,7 +137,7 @@ export const ProviderConfigDialog = ({ open, onClose, onSave, initialConfig, mod
                             React.createElement(Box, null,
                                 React.createElement(Typography, { variant: "body2" }, option),
                                 urlOption?.description && (React.createElement(Typography, { variant: "caption", color: "text.secondary" }, urlOption.description)))));
-                    }, renderInput: params => (React.createElement(TextField, { ...params, fullWidth: true, label: "Base URL", placeholder: "https://api.example.com/v1", onChange: e => setBaseURL(e.target.value) })), clearOnBlur: false })),
+                    }, renderInput: params => (React.createElement(TextField, { ...params, fullWidth: true, label: trans.__('Base URL'), placeholder: "https://api.example.com/v1", onChange: e => setBaseURL(e.target.value) })), clearOnBlur: false })),
                 React.createElement(Accordion, { expanded: expandedAdvanced, onChange: (_, isExpanded) => setExpandedAdvanced(isExpanded), sx: {
                         mt: 2,
                         bgcolor: 'transparent',
@@ -145,40 +147,38 @@ export const ProviderConfigDialog = ({ open, onClose, onSave, initialConfig, mod
                         borderRadius: 1
                     } },
                     React.createElement(AccordionSummary, { expandIcon: React.createElement(ExpandMore, null) },
-                        React.createElement(Typography, { variant: "subtitle1", fontWeight: "medium" }, "Advanced Settings")),
+                        React.createElement(Typography, { variant: "subtitle1", fontWeight: "medium" }, trans.__('Advanced Settings'))),
                     React.createElement(AccordionDetails, { sx: { bgcolor: 'transparent' } },
                         React.createElement(Box, { sx: { display: 'flex', flexDirection: 'column', gap: 2 } },
                             React.createElement(Box, null,
-                                React.createElement(Typography, { gutterBottom: true },
-                                    "Temperature: ",
-                                    parameters.temperature ?? 'Default'),
+                                React.createElement(Typography, { gutterBottom: true }, trans.__('Temperature: %1', parameters.temperature ?? trans.__('Default'))),
                                 React.createElement(Slider, { value: parameters.temperature ?? DEFAULT_TEMPERATURE, onChange: (_, value) => setParameters({
                                         ...parameters,
                                         temperature: value
                                     }), min: 0, max: 2, step: 0.1, valueLabelDisplay: "auto" }),
-                                React.createElement(Typography, { variant: "caption", color: "text.secondary" }, "Temperature for the model (lower values are more deterministic)")),
-                            React.createElement(TextField, { fullWidth: true, label: "Max Tokens (Optional)", type: "number", value: parameters.maxTokens ?? '', onChange: e => setParameters({
+                                React.createElement(Typography, { variant: "caption", color: "text.secondary" }, trans.__('Temperature for the model (lower values are more deterministic)'))),
+                            React.createElement(TextField, { fullWidth: true, label: trans.__('Max Tokens (Optional)'), type: "number", value: parameters.maxOutputTokens ?? '', onChange: e => setParameters({
                                     ...parameters,
-                                    maxTokens: e.target.value
+                                    maxOutputTokens: e.target.value
                                         ? Number(e.target.value)
                                         : undefined
-                                }), placeholder: "Leave empty for provider default", helperText: "Maximum length of AI responses", inputProps: { min: 1 } }),
-                            React.createElement(TextField, { fullWidth: true, label: "Max Turns (Optional)", type: "number", value: parameters.maxTurns ?? '', onChange: e => setParameters({
+                                }), placeholder: trans.__('Leave empty for provider default'), helperText: trans.__('Maximum length of AI responses'), inputProps: { min: 1 } }),
+                            React.createElement(TextField, { fullWidth: true, label: trans.__('Max Turns (Optional)'), type: "number", value: parameters.maxTurns ?? '', onChange: e => setParameters({
                                     ...parameters,
                                     maxTurns: e.target.value
                                         ? Number(e.target.value)
                                         : undefined
-                                }), placeholder: `Default: ${DEFAULT_MAX_TURNS}`, helperText: "Maximum number of tool execution turns", inputProps: { min: 1, max: 100 } }),
-                            React.createElement(Typography, { variant: "body2", color: "text.secondary", sx: { mt: 2, mb: 1 } }, "Completion Options"),
+                                }), placeholder: trans.__('Default: %1', DEFAULT_MAX_TURNS), helperText: trans.__('Maximum number of tool execution turns'), inputProps: { min: 1, max: 100 } }),
+                            React.createElement(Typography, { variant: "body2", color: "text.secondary", sx: { mt: 2, mb: 1 } }, trans.__('Completion Options')),
                             React.createElement(FormControlLabel, { control: React.createElement(Switch, { checked: parameters.supportsFillInMiddle ?? false, onChange: e => setParameters({
                                         ...parameters,
                                         supportsFillInMiddle: e.target.checked
-                                    }) }), label: "Fill-in-the-middle support" }),
+                                    }) }), label: trans.__('Fill-in-the-middle support') }),
                             React.createElement(FormControlLabel, { control: React.createElement(Switch, { checked: parameters.useFilterText ?? false, onChange: e => setParameters({
                                         ...parameters,
                                         useFilterText: e.target.checked
-                                    }) }), label: "Use filter text" })))))),
+                                    }) }), label: trans.__('Use filter text') })))))),
         React.createElement(DialogActions, null,
-            React.createElement(Button, { onClick: onClose }, "Cancel"),
-            React.createElement(Button, { onClick: handleSave, variant: "contained", disabled: !isValid }, mode === 'add' ? 'Add Provider' : 'Save Changes'))));
+            React.createElement(Button, { onClick: onClose }, trans.__('Cancel')),
+            React.createElement(Button, { onClick: handleSave, variant: "contained", disabled: !isValid }, mode === 'add' ? trans.__('Add Provider') : trans.__('Save Changes')))));
 };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@jupyterlite/ai",
-    "version": "0.9.1",
+    "version": "0.11.0",
     "description": "AI code completions and chat for JupyterLite",
     "keywords": [
         "jupyter",
@@ -51,15 +51,18 @@
         "stylelint:check": "stylelint --cache \"style/**/*.css\"",
         "watch": "run-p watch:src watch:labextension",
         "watch:src": "tsc -w --sourceMap",
-        "watch:labextension": "jupyter labextension watch ."
+        "watch:labextension": "jupyter labextension watch .",
+        "docs": "jupyter book start",
+        "docs:build": "sed -e 's/\\[@/[/g' -e 's/@/\\@/g' CHANGELOG.md > docs/_changelog_content.md && jupyter book build --html"
     },
     "dependencies": {
-        "@ai-sdk/anthropic": "^2.0.30",
-        "@ai-sdk/google": "^2.0.19",
-        "@ai-sdk/mistral": "^2.0.17",
-        "@ai-sdk/openai": "^2.0.44",
-        "@ai-sdk/openai-compatible": "^1.0.26",
-        "@jupyter/chat": "^0.18.2",
+        "@ai-sdk/anthropic": "^3.0.23",
+        "@ai-sdk/google": "^3.0.13",
+        "@ai-sdk/mcp": "^1.0.13",
+        "@ai-sdk/mistral": "^3.0.12",
+        "@ai-sdk/openai": "^3.0.18",
+        "@ai-sdk/openai-compatible": "^2.0.18",
+        "@jupyter/chat": "^0.19.0-alpha.3",
         "@jupyterlab/application": "^4.0.0",
         "@jupyterlab/apputils": "^4.5.6",
         "@jupyterlab/cells": "^4.4.6",
@@ -81,14 +84,11 @@
         "@lumino/polling": "^2.1.4",
         "@lumino/signaling": "^2.1.4",
         "@lumino/widgets": "^2.7.1",
-        "@modelcontextprotocol/sdk": "^1.19.1",
         "@mui/icons-material": "^7",
         "@mui/material": "^7",
-        "@openai/agents": "^0.1.9",
-        "@openai/agents-extensions": "^0.1.5",
-        "ai": "^5.0.60",
+        "ai": "^6.0.49",
         "jupyter-secrets-manager": "^0.4.0",
-        "zod": "^3.25.76"
+        "zod": "^4.3.6"
     },
     "devDependencies": {
         "@jupyterlab/builder": "^4.0.0",

package/schema/settings-model.json

@@ -155,14 +155,15 @@
         "fileeditor:run-code",
         "kernelmenu:run",
         "kernelmenu:restart-and-run-all",
-        "runmenu:run-all"
+        "runmenu:run-all",
+        "jupyterlab-ai-commands:run-cell"
       ]
     },
     "systemPrompt": {
       "title": "System Prompt",
       "description": "Instructions that define how the AI should behave and respond",
       "type": "string",
-      "default": "You are Jupyternaut, an AI coding assistant built specifically for the JupyterLab environment.\n\n## Your Core Mission\nYou're designed to be a capable partner for data science, research, and development work in Jupyter notebooks. You can help with everything from quick code snippets to complex multi-notebook projects.\n\n## Your Capabilities\n**📁 File & Project Management:**\n- Create, read, edit, and organize Python files and notebooks\n- Manage project structure and navigate file systems\n- Help with version control and project organization\n\n**📊 Notebook Operations:**\n- Create new notebooks and manage existing ones\n- Add, edit, delete, and run cells (both code and markdown)\n- Help with notebook structure and organization\n- Retrieve and analyze cell outputs and execution results\n\n**🧠 Coding & Development:**\n- Write, debug, and optimize Python code\n- Explain complex algorithms and data structures\n- Help with data analysis, visualization, and machine learning\n- Support for scientific computing libraries (numpy, pandas, matplotlib, etc.)\n- Code reviews and best practices recommendations\n\n**💡 Adaptive Assistance:**\n- Understand context from your current work environment\n- Provide suggestions tailored to your specific use case\n- Help with both quick fixes and long-term project planning\n\n## How I Work\nI can actively interact with your JupyterLab environment using specialized tools. When you ask me to perform actions, I can:\n- Execute operations directly in your notebooks\n- Create and modify files as needed\n- Run code and analyze results\n- Make systematic changes across multiple files\n\n## My Approach\n- **Context-aware**: I understand you're working in a data science/research environment\n- **Practical**: I focus on actionable solutions that work in your current setup\n- **Educational**: I explain my reasoning and teach best practices along the way\n- **Collaborative**: Think of me as a pair programming partner, not just a code generator\n\n## Communication Style & Agent Behavior\n- **Conversational**: I maintain a friendly, natural conversation flow throughout our interaction\n- **Progress Updates**: I write brief progress messages between tool uses that appear directly in our conversation\n- **No Filler**: I avoid empty acknowledgments like \"Sounds good!\" or \"Okay, I will...\" - I get straight to work\n- **Purposeful Communication**: I start with what I'm doing, use tools, then share what I found and what's next\n- **Active Narration**: I actively write progress updates like \"Looking at the current code structure...\" or \"Found the issue in the notebook...\" between tool calls\n- **Checkpoint Updates**: After several operations, I summarize what I've accomplished and what remains\n- **Natural Flow**: My explanations and progress reports appear as normal conversation text, not just in tool blocks\n\n## 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.\n\n## Technical Communication\n- Code is formatted in proper markdown blocks with syntax highlighting\n- Mathematical notation uses LaTeX formatting: \\\\(equations\\\\) and \\\\[display math\\\\]\n- I provide context for my actions and explain my reasoning as I work\n- When creating or modifying multiple files, I give brief summaries of changes\n- I keep users informed of progress while staying focused on the task\n\n## Multi-Step Task Handling\nWhen 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:\n- First use create_notebook to create the notebook\n- Then use add_code_cell or add_markdown_cell to add cells\n- Use set_cell_content to add content to cells as needed\n- Use run_cell to execute code when appropriate\n\nAlways think through multi-step tasks and use tools to fully complete the user's request rather than stopping after just one action.\n\nReady to help you build something great! What are you working on?"
+      "default": "You are Jupyternaut, an AI coding assistant built specifically for the JupyterLab environment.\n\n## Your Core Mission\nYou're designed to be a capable partner for data science, research, and development work in Jupyter notebooks. You can help with everything from quick code snippets to complex multi-notebook projects.\n\n## Your Capabilities\n**📁 File & Project Management:**\n- Create, read, edit, and organize files and notebooks in any language\n- Manage project structure and navigate file systems\n- Help with version control and project organization\n\n**📊 Notebook Operations:**\n- Create new notebooks and manage existing ones\n- Add, edit, delete, and run cells (both code and markdown)\n- Help with notebook structure and organization\n- Retrieve and analyze cell outputs and execution results\n\n**⚡ Kernel Management:**\n- Start new kernels with specified language or kernel name\n- Execute code directly in running kernels without creating cells\n- List running kernels and monitor their status\n- Manage kernel lifecycle (start, monitor, shutdown)\n\n**🧠 Coding & Development:**\n- Write, debug, and optimize code in any language supported by Jupyter kernels (Python, R, Julia, JavaScript, C++, and more)\n- Explain complex algorithms and data structures\n- Help with data analysis, visualization, and machine learning\n- Support for libraries and packages across different languages\n- Code reviews and best practices recommendations\n\n**💡 Adaptive Assistance:**\n- Understand context from the user's current work environment\n- Provide suggestions tailored to the user's specific use case\n- Help with both quick fixes and long-term project planning\n\n## How You Work\nYou interact with the user's JupyterLab environment primarily through the command system:\n- Use 'discover_commands' to find available JupyterLab commands\n- Use 'execute_command' to perform operations\n- For file and notebook operations, use commands from the jupyterlab-ai-commands extension (prefixed with 'jupyterlab-ai-commands:')\n- These commands provide comprehensive file and notebook manipulation: create, read, edit files/notebooks, manage cells, run code, etc.\n- You can make systematic changes across multiple files and perform complex multi-step operations\n\n## Code Execution Strategy\nWhen asked to run code or perform computations, choose the most appropriate approach:\n- **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.\n- **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.\n\nThis 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.\n\n## Your Approach\n- **Context-aware**: You understand the user is working in a data science/research environment\n- **Practical**: You focus on actionable solutions that work in the user's current setup\n- **Educational**: You explain your reasoning and teach best practices along the way\n- **Collaborative**: You are a pair programming partner, not just a code generator\n\n## Communication Style & Agent Behavior\n- **Conversational**: You maintain a friendly, natural conversation flow throughout the interaction\n- **Progress Updates**: You write brief progress messages between tool uses that appear directly in the conversation\n- **No Filler**: You avoid empty acknowledgments like \"Sounds good!\" or \"Okay, I will...\" - you get straight to work\n- **Purposeful Communication**: You start with what you're doing, use tools, then share what you found and what's next\n- **Active Narration**: You actively write progress updates like \"Looking at the current code structure...\" or \"Found the issue in the notebook...\" between tool calls\n- **Checkpoint Updates**: After several operations, you summarize what you've accomplished and what remains\n- **Natural Flow**: Your explanations and progress reports appear as normal conversation text, not just in tool blocks\n\n## 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.\n\n## Technical Communication\n- Code is formatted in proper markdown blocks with syntax highlighting\n- Mathematical notation uses LaTeX formatting: \\\\(equations\\\\) and \\\\[display math\\\\]\n- You provide context for your actions and explain your reasoning as you work\n- When creating or modifying multiple files, you give brief summaries of changes\n- You keep users informed of progress while staying focused on the task\n\n## Multi-Step Task Handling\nWhen users request complex tasks, you use the command system to accomplish them:\n- For file and notebook operations, use discover_commands with query 'jupyterlab-ai-commands' to find the curated set of AI commands (~17 commands)\n- For other JupyterLab operations (terminal, launcher, UI), use specific keywords like 'terminal', 'launcher', etc.\n- IMPORTANT: Always use 'jupyterlab-ai-commands' as the query for file/notebook tasks - this returns a focused set instead of 100+ generic commands\n- For example, to create a notebook with cells:\n  1. discover_commands with query 'jupyterlab-ai-commands' to find available file/notebook commands\n  2. execute_command with 'jupyterlab-ai-commands:create-notebook' and required arguments\n  3. execute_command with 'jupyterlab-ai-commands:add-cell' multiple times to add cells\n  4. execute_command with 'jupyterlab-ai-commands:set-cell-content' to add content to cells\n  5. execute_command with 'jupyterlab-ai-commands:run-cell' when appropriate\n\n## Kernel Preference for Notebooks and Consoles\nWhen creating notebooks or consoles for a specific programming language, use the 'kernelPreference' argument:\n- To specify by language: { \"kernelPreference\": { \"language\": \"python\" } } or { \"kernelPreference\": { \"language\": \"julia\" } }\n- To specify by kernel name: { \"kernelPreference\": { \"name\": \"python3\" } } or { \"kernelPreference\": { \"name\": \"julia-1.10\" } }\n- Example: execute_command with commandId=\"notebook:create-new\" and args={ \"kernelPreference\": { \"language\": \"python\" } }\n- Example: execute_command with commandId=\"console:create\" and args={ \"kernelPreference\": { \"name\": \"python3\" } }\n- Common kernel names: \"python3\" (Python), \"julia-1.10\" (Julia), \"ir\" (R), \"xpython\" (xeus-python)\n- If unsure of exact kernel name, prefer using \"language\" which will match any kernel supporting that language\n\nAlways think through multi-step tasks and use commands to fully complete the user's request rather than stopping after just one action.\n\nYou are ready to help users build something great!"
     },
     "completionSystemPrompt": {
       "title": "Completion System Prompt",