@jupyterlite/ai 0.17.0 → 0.19.0

package/src/widgets/ai-settings.tsx

@@ -1,3 +1,15 @@
+import {
+  getEffectiveContextWindow,
+  SECRETS_REPLACEMENT
+} from '@jupyternaut/agent';
+import type {
+  IAgentManagerFactory,
+  IAIConfig,
+  IAISecretsAccess,
+  IMCPServerConfig,
+  IProviderConfig,
+  IProviderRegistry
+} from '@jupyternaut/agent';
 import { IThemeManager } from '@jupyterlab/apputils';
 import { ReactWidget } from '@jupyterlab/ui-components';
 import type { TranslationBundle } from '@jupyterlab/translation';
@@ -45,17 +57,8 @@ import {
   createTheme
 } from '@mui/material';
 import React, { useEffect, useMemo, useState } from 'react';
-import { getEffectiveContextWindow } from '../providers/model-info';
-import {
-  type IAgentManagerFactory,
-  type IAIConfig,
-  type IAISecretsAccess,
-  type IAISettingsModel,
-  type IMCPServerConfig,
-  type IProviderConfig,
-  type IProviderRegistry,
-  SECRETS_REPLACEMENT
-} from '../tokens';
+
+import type { IAISettingsModel } from '../tokens';
 import { ProviderConfigDialog } from './provider-config-dialog';
 
 /**
@@ -925,6 +928,32 @@ const AISettingsComponent: React.FC<IAISettingsComponentProps> = ({
                   }
                 />
 
+                <FormControlLabel
+                  control={
+                    <Switch
+                      checked={config.autoTitle}
+                      onChange={e =>
+                        handleConfigUpdate({
+                          autoTitle: e.target.checked
+                        })
+                      }
+                      color="primary"
+                    />
+                  }
+                  label={
+                    <Box>
+                      <Typography variant="body1">
+                        {trans.__('Auto Title')}
+                      </Typography>
+                      <Typography variant="caption" color="text.secondary">
+                        {trans.__(
+                          'Automatically generate a chat title from the model for every message until there are 5 messages'
+                        )}
+                      </Typography>
+                    </Box>
+                  }
+                />
+
                 <FormControlLabel
                   control={
                     <Switch

package/src/widgets/main-area-chat.ts

@@ -4,11 +4,10 @@ import { launchIcon } from '@jupyterlab/ui-components';
 import type { TranslationBundle } from '@jupyterlab/translation';
 import { CommandRegistry } from '@lumino/commands';
 
-import { AIChatModel } from '../chat-model';
 import { SaveComponentWidget } from '../components/save-button';
 import { UsageWidget } from '../components/usage-display';
 import { RenderedMessageOutputAreaCompat } from '../rendered-message-outputarea';
-import { CommandIds, type IAISettingsModel } from '../tokens';
+import { CommandIds, IAIChatModel, type IAISettingsModel } from '../tokens';
 
 export namespace MainAreaChat {
   export interface IOptions extends MainAreaWidget.IOptions<ChatWidget> {
@@ -69,7 +68,7 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
       chatPanel: this.content
     });
 
-    this.model.writersChanged.connect(this._writersChanged);
+    this.model.writersChanged?.connect(this._writersChanged);
 
     this.model.titleChanged.connect(this._titleChanged);
   }
@@ -78,15 +77,15 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
     super.dispose();
     // Dispose of the approval buttons widget when the chat is disposed.
     this._outputAreaCompat.dispose();
-    this.model.writersChanged.disconnect(this._writersChanged);
+    this.model.writersChanged?.disconnect(this._writersChanged);
     this.model.titleChanged.disconnect(this._titleChanged);
   }
 
   /**
    * Get the model of the chat.
    */
-  get model(): AIChatModel {
-    return this.content.model as AIChatModel;
+  get model(): IAIChatModel {
+    return this.content.model as IAIChatModel;
   }
 
   /**
@@ -103,11 +102,9 @@ export class MainAreaChat extends MainAreaWidget<ChatWidget> {
     );
 
     if (aiWriting) {
-      this.content.inputToolbarRegistry?.hide('send');
       this.content.inputToolbarRegistry?.show('stop');
     } else {
       this.content.inputToolbarRegistry?.hide('stop');
-      this.content.inputToolbarRegistry?.show('send');
     }
   };
 

package/src/widgets/provider-config-dialog.tsx

@@ -1,3 +1,11 @@
+import { getProviderModelInfo } from '@jupyternaut/agent';
+import type {
+  IProviderConfig,
+  IProviderParameters,
+  IProviderRegistry,
+  IProviderToolCapabilities
+} from '@jupyternaut/agent';
+import type { TranslationBundle } from '@jupyterlab/translation';
 import ExpandMore from '@mui/icons-material/ExpandMore';
 import Delete from '@mui/icons-material/Delete';
 import Visibility from '@mui/icons-material/Visibility';
@@ -29,15 +37,7 @@ import {
   TextField,
   Typography
 } from '@mui/material';
-import type { TranslationBundle } from '@jupyterlab/translation';
 import React from 'react';
-import type {
-  IProviderConfig,
-  IProviderParameters,
-  IProviderRegistry,
-  IProviderToolCapabilities
-} from '../tokens';
-import { getProviderModelInfo } from '../providers/model-info';
 
 /**
  * Default parameter values for provider configuration

package/LICENSE

@@ -1,30 +0,0 @@
-BSD 3-Clause License
-
-Copyright (c) 2024, JupyterLite Contributors
-Copyright (c) 2024, Jeremy Tuloup
-All rights reserved.
-
-Redistribution and use in source and binary forms, with or without
-modification, are permitted provided that the following conditions are met:
-
-1. Redistributions of source code must retain the above copyright notice, this
-   list of conditions and the following disclaimer.
-
-2. Redistributions in binary form must reproduce the above copyright notice,
-   this list of conditions and the following disclaimer in the documentation
-   and/or other materials provided with the distribution.
-
-3. Neither the name of the copyright holder nor the names of its
-   contributors may be used to endorse or promote products derived from
-   this software without specific prior written permission.
-
-THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
-AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
-IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
-DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
-FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
-DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
-SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
-CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
-OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
-OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

package/README.md

@@ -1,49 +0,0 @@
-# jupyterlite-ai
-
-[![Github Actions Status](https://github.com/jupyterlite/ai/workflows/Build/badge.svg)](https://github.com/jupyterlite/ai/actions/workflows/build.yml)
-[![Documentation Status](https://readthedocs.org/projects/jupyterlite-ai/badge/?version=latest)](https://jupyterlite-ai.readthedocs.io/en/latest/?badge=latest)
-[![lite-badge](https://jupyterlite.rtfd.io/en/latest/_static/badge.svg)](https://jupyterlite.github.io/ai/lab/index.html)
-
-AI code completions and chat for JupyterLab, Notebook 7 and JupyterLite.
-
-[a screencast showing the Jupyterlite AI extension in JupyterLite](https://github.com/user-attachments/assets/e33d7d84-53ca-4835-a034-b6757476c98b)
-
-## Requirements
-
-- JupyterLab >= 4.4.0 or Notebook >= 7.4.0
-
-## Try it in your browser
-
-You can try the extension in your browser using JupyterLite:
-
-[![lite-badge](https://jupyterlite.rtfd.io/en/latest/_static/badge.svg)](https://jupyterlite.github.io/ai/lab/index.html)
-
-## Install
-
-To install the extension, execute:
-
-```bash
-pip install jupyterlite-ai
-```
-
-To install requirements (JupyterLab, JupyterLite and Notebook):
-
-```bash
-pip install jupyterlite-ai[jupyter]
-```
-
-## Documentation
-
-For detailed usage instructions, including how to configure AI providers, see the [documentation](https://jupyterlite-ai.readthedocs.io/).
-
-## Uninstall
-
-To remove the extension, execute:
-
-```bash
-pip uninstall jupyterlite-ai
-```
-
-## Contributing
-
-See [CONTRIBUTING](CONTRIBUTING.md)

package/lib/agent.d.ts

@@ -1,277 +0,0 @@
-import type { IMessageContent } from '@jupyter/chat';
-import { ISignal } from '@lumino/signaling';
-import { type ModelMessage, type UserContent } from 'ai';
-import { ISecretsManager } from 'jupyter-secrets-manager';
-import { type IAgentManager, type IAgentManagerFactory, type IAISettingsModel, type ISkillRegistry, type ITokenUsage, type ToolMap } from './tokens';
-/**
- * The agent manager factory namespace.
- */
-export declare namespace AgentManagerFactory {
-    interface IOptions {
-        /**
-         * The settings model.
-         */
-        settingsModel: IAISettingsModel;
-        /**
-         * The skill registry for discovering skills.
-         */
-        skillRegistry?: ISkillRegistry;
-        /**
-         * The secrets manager.
-         */
-        secretsManager?: ISecretsManager;
-        /**
-         * The token used to request the secrets manager.
-         */
-        token: symbol | null;
-    }
-}
-/**
- * The agent manager factory.
- */
-export declare class AgentManagerFactory implements IAgentManagerFactory {
-    constructor(options: AgentManagerFactory.IOptions);
-    /**
-     * Create a new agent.
-     */
-    createAgent(options: IAgentManager.IOptions): IAgentManager;
-    /**
-     * Signal emitted when MCP connection status changes
-     */
-    get mcpConnectionChanged(): ISignal<this, boolean>;
-    /**
-     * Checks whether a specific MCP server is connected.
-     * @param serverName The name of the MCP server to check
-     * @returns True if the server is connected, false otherwise
-     */
-    isMCPServerConnected(serverName: string): boolean;
-    /**
-     * Gets the MCP tools from connected servers
-     */
-    getMCPTools(): Promise<ToolMap>;
-    /**
-     * Handles settings changes and reinitializes the agent.
-     */
-    private _onSettingsChanged;
-    /**
-     * Initializes MCP (Model Context Protocol) clients based on current settings.
-     * Closes existing clients and connects to enabled servers from configuration.
-     */
-    private _initializeMCPClients;
-    /**
-     * Initializes the AI agent with current settings and tools.
-     * Sets up the agent with model configuration, tools, and MCP servers.
-     */
-    private _initializeAgents;
-    /**
-     * Refresh skill snapshots across all agents.
-     */
-    refreshSkillSnapshots(): void;
-    private _agentManagers;
-    private _settingsModel;
-    private _skillRegistry?;
-    private _secretsManager?;
-    private _mcpClients;
-    private _mcpConnectionChanged;
-    private _initQueue;
-}
-/**
- * Manages the AI agent lifecycle and execution loop.
- * Provides agent initialization, tool management, MCP server integration,
- * and handles the complete agent execution cycle.
- * Emits events for UI updates instead of directly manipulating the chat interface.
- */
-export declare class AgentManager implements IAgentManager {
-    /**
-     * Creates a new AgentManager instance.
-     * @param options Configuration options for the agent manager
-     */
-    constructor(options: IAgentManager.IOptions);
-    /**
-     * Signal emitted when agent events occur
-     */
-    get agentEvent(): ISignal<this, IAgentManager.IAgentEvent>;
-    /**
-     * Signal emitted when the active provider has changed.
-     */
-    get activeProviderChanged(): ISignal<this, string | undefined>;
-    /**
-     * Gets the current token usage statistics.
-     */
-    get tokenUsage(): ITokenUsage;
-    /**
-     * Signal emitted when token usage statistics change.
-     */
-    get tokenUsageChanged(): ISignal<this, ITokenUsage>;
-    /**
-     * Refresh the skills snapshot and rebuild the agent if resources are ready.
-     */
-    refreshSkills(): void;
-    /**
-     * The active provider for this agent.
-     */
-    get activeProvider(): string;
-    set activeProvider(value: string);
-    /**
-     * Sets the selected tools by name and reinitializes the agent.
-     * @param toolNames Array of tool names to select
-     */
-    setSelectedTools(toolNames: string[]): void;
-    /**
-     * Gets the currently selected tools as a record.
-     * @returns Record of selected tools
-     */
-    get selectedAgentTools(): ToolMap;
-    /**
-     * Checks if the current configuration is valid for agent operations.
-     * Uses the provider registry to determine if an API key is required.
-     * @returns True if the configuration is valid, false otherwise
-     */
-    hasValidConfig(): boolean;
-    /**
-     * Clears conversation history and resets agent state.
-     */
-    clearHistory(): Promise<void>;
-    /**
-     * Sets the history with a list of messages from the chat.
-     * @param messages The chat messages to set as history
-     */
-    setHistory(messages: IMessageContent[]): void;
-    /**
-     * Stops the current streaming response by aborting the request.
-     * Resolve any pending approval.
-     */
-    stopStreaming(reason?: string): void;
-    /**
-     * Approves a pending tool call.
-     * @param approvalId The approval ID to approve
-     * @param reason Optional reason for approval
-     */
-    approveToolCall(approvalId: string, reason?: string): void;
-    /**
-     * Rejects a pending tool call.
-     * @param approvalId The approval ID to reject
-     * @param reason Optional reason for rejection
-     */
-    rejectToolCall(approvalId: string, reason?: string): void;
-    /**
-     * Generates AI response to user message using the agent.
-     * Handles the complete execution cycle including tool calls.
-     * @param message The user message to respond to (may include processed attachment content)
-     */
-    generateResponse(message: UserContent): Promise<void>;
-    /**
-     * Create a transient language model to request a text response which won't be added to history.
-     * @param messages - the messages sequence to send to the model.
-     */
-    textResponse(messages: ModelMessage[]): Promise<string>;
-    /**
-     * Updates cumulative token usage statistics from a completed model step.
-     */
-    private _updateTokenUsage;
-    /**
-     * Gets the configured context window for the active provider.
-     */
-    private _getActiveContextWindow;
-    /**
-     * Initializes the AI agent with current settings and tools.
-     * Sets up the agent with model configuration, tools, and MCP tools.
-     */
-    initializeAgent: (mcpTools?: ToolMap) => Promise<void>;
-    /**
-     * Refresh the in-memory skills snapshot from the skill registry.
-     */
-    private _refreshSkills;
-    /**
-     * Prepare model, tools, and settings needed to (re)build the agent.
-     */
-    private _prepareAgentConfig;
-    /**
-     * Build the runtime tool map used by the agent.
-     */
-    private _buildRuntimeTools;
-    /**
-     * Rebuild the agent using cached resources and the current skills snapshot.
-     */
-    private _rebuildAgent;
-    /**
-     * Processes the stream result from agent execution.
-     * Handles message streaming, tool calls, and emits appropriate events.
-     * @param result The stream result from agent execution
-     * @returns Processing result including approval info if applicable
-     */
-    private _processStreamResult;
-    /**
-     * Emits a message_complete event.
-     */
-    private _emitMessageComplete;
-    /**
-     * Handles tool-result stream parts.
-     */
-    private _handleToolResult;
-    /**
-     * Handles tool-error stream parts.
-     */
-    private _handleToolError;
-    /**
-     * Handles tool-output-denied stream parts.
-     */
-    private _handleToolOutputDenied;
-    /**
-     * Handles tool-approval-request stream parts.
-     */
-    private _handleApprovalRequest;
-    /**
-     * Waits for user approval of a tool call.
-     * @param approvalId The approval ID to wait for
-     * @returns Promise that resolves to true if approved, false if rejected
-     */
-    private _waitForApproval;
-    /**
-     * Formats tool input for display by pretty-printing JSON strings.
-     * @param input The tool input string to format
-     * @returns Pretty-printed JSON string
-     */
-    private _formatToolInput;
-    /**
-     * Checks if the current provider supports tool calling.
-     * @returns True if the provider supports tool calling, false otherwise
-     */
-    private _supportsToolCalling;
-    /**
-     * Creates a model instance based on current settings.
-     * @returns The configured model instance for the agent
-     */
-    private _createModel;
-    /**
-     * Enhances the base system prompt with dynamic context like skills.
-     * @param baseSystemPrompt The base system prompt from settings
-     * @returns The enhanced system prompt with dynamic additions
-     */
-    private _getEnhancedSystemPrompt;
-    /**
-     * Build an instruction line describing MIME types supported by this session.
-     */
-    private _getSupportedMimeTypesInstruction;
-    private _settingsModel;
-    private _toolRegistry?;
-    private _providerRegistry?;
-    private _skillRegistry?;
-    private _secretsManager?;
-    private _selectedToolNames;
-    private _agent;
-    private _history;
-    private _mcpTools;
-    private _controller;
-    private _agentEvent;
-    private _tokenUsage;
-    private _tokenUsageChanged;
-    private _activeProvider;
-    private _activeProviderChanged;
-    private _skills;
-    private _renderMimeRegistry?;
-    private _initQueue;
-    private _agentConfig;
-    private _pendingApprovals;
-    private _streaming;
-}