@jupyterlite/ai 0.6.0 → 0.6.2

package/lib/base-completer.d.ts

@@ -5,7 +5,7 @@ export interface IBaseCompleter {
     /**
      * The LLM completer.
      */
-    provider: BaseLanguageModel;
+    completer: BaseLanguageModel;
     /**
      * The completion prompt.
      */

package/lib/default-providers/Anthropic/completer.d.ts

@@ -3,7 +3,7 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { BaseCompleter, IBaseCompleter } from '../../base-completer';
 export declare class AnthropicCompleter implements IBaseCompleter {
     constructor(options: BaseCompleter.IOptions);
-    get provider(): BaseChatModel;
+    get completer(): BaseChatModel;
     /**
      * Getter and setter for the initial prompt.
      */
@@ -14,6 +14,6 @@ export declare class AnthropicCompleter implements IBaseCompleter {
             insertText: string;
         }[];
     }>;
-    private _anthropicProvider;
+    private _completer;
     private _prompt;
 }

package/lib/default-providers/Anthropic/completer.js

@@ -4,10 +4,10 @@ import { COMPLETION_SYSTEM_PROMPT } from '../../provider';
 export class AnthropicCompleter {
     constructor(options) {
         this._prompt = COMPLETION_SYSTEM_PROMPT;
-        this._anthropicProvider = new ChatAnthropic({ ...options.settings });
+        this._completer = new ChatAnthropic({ ...options.settings });
     }
-    get provider() {
-        return this._anthropicProvider;
+    get completer() {
+        return this._completer;
     }
     /**
      * Getter and setter for the initial prompt.
@@ -28,7 +28,7 @@ export class AnthropicCompleter {
             new AIMessage(trimmedPrompt)
         ];
         try {
-            const response = await this._anthropicProvider.invoke(messages);
+            const response = await this._completer.invoke(messages);
             const items = [];
             // Anthropic can return string or complex content, a list of string/images/other.
             if (typeof response.content === 'string') {

package/lib/default-providers/ChromeAI/completer.d.ts

@@ -8,12 +8,12 @@ export declare class ChromeCompleter implements IBaseCompleter {
      */
     get prompt(): string;
     set prompt(value: string);
-    get provider(): LLM;
+    get completer(): LLM;
     fetch(request: CompletionHandler.IRequest, context: IInlineCompletionContext): Promise<{
         items: {
             insertText: string;
         }[];
     }>;
-    private _chromeProvider;
+    private _completer;
     private _prompt;
 }

package/lib/default-providers/ChromeAI/completer.js

@@ -23,7 +23,7 @@ const CODE_BLOCK_END_REGEX = /```$/;
 export class ChromeCompleter {
     constructor(options) {
         this._prompt = COMPLETION_SYSTEM_PROMPT;
-        this._chromeProvider = new ChromeAI({ ...options.settings });
+        this._completer = new ChromeAI({ ...options.settings });
     }
     /**
      * Getter and setter for the initial prompt.
@@ -34,8 +34,8 @@ export class ChromeCompleter {
     set prompt(value) {
         this._prompt = value;
     }
-    get provider() {
-        return this._chromeProvider;
+    get completer() {
+        return this._completer;
     }
     async fetch(request, context) {
         const { text, offset: cursorOffset } = request;
@@ -46,7 +46,7 @@ export class ChromeCompleter {
             new HumanMessage(trimmedPrompt)
         ];
         try {
-            let response = await this._chromeProvider.invoke(messages);
+            let response = await this._completer.invoke(messages);
             // ChromeAI sometimes returns a string starting with '```',
             // so process the response to remove the code block delimiters
             if (CODE_BLOCK_START_REGEX.test(response)) {

package/lib/default-providers/MistralAI/completer.d.ts

@@ -3,7 +3,7 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { BaseCompleter, IBaseCompleter } from '../../base-completer';
 export declare class CodestralCompleter implements IBaseCompleter {
     constructor(options: BaseCompleter.IOptions);
-    get provider(): BaseChatModel;
+    get completer(): BaseChatModel;
     /**
      * Getter and setter for the initial prompt.
      */
@@ -11,6 +11,6 @@ export declare class CodestralCompleter implements IBaseCompleter {
     set prompt(value: string);
     fetch(request: CompletionHandler.IRequest, context: IInlineCompletionContext): Promise<any>;
     private _throttler;
-    private _mistralProvider;
+    private _completer;
     private _prompt;
 }

package/lib/default-providers/MistralAI/completer.js

@@ -9,9 +9,9 @@ const INTERVAL = 1000;
 export class CodestralCompleter {
     constructor(options) {
         this._prompt = COMPLETION_SYSTEM_PROMPT;
-        this._mistralProvider = new ChatMistralAI({ ...options.settings });
+        this._completer = new ChatMistralAI({ ...options.settings });
         this._throttler = new Throttler(async (messages) => {
-            const response = await this._mistralProvider.invoke(messages);
+            const response = await this._completer.invoke(messages);
             // Extract results of completion request.
             const items = [];
             if (typeof response.content === 'string') {
@@ -32,8 +32,8 @@ export class CodestralCompleter {
             return { items };
         }, { limit: INTERVAL });
     }
-    get provider() {
-        return this._mistralProvider;
+    get completer() {
+        return this._completer;
     }
     /**
      * Getter and setter for the initial prompt.

package/lib/default-providers/OpenAI/completer.d.ts

@@ -3,7 +3,7 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import { BaseCompleter, IBaseCompleter } from '../../base-completer';
 export declare class OpenAICompleter implements IBaseCompleter {
     constructor(options: BaseCompleter.IOptions);
-    get provider(): BaseChatModel;
+    get completer(): BaseChatModel;
     /**
      * Getter and setter for the initial prompt.
      */
@@ -14,6 +14,6 @@ export declare class OpenAICompleter implements IBaseCompleter {
             insertText: string;
         }[];
     }>;
-    private _openAIProvider;
+    private _completer;
     private _prompt;
 }

package/lib/default-providers/OpenAI/completer.js

@@ -4,10 +4,10 @@ import { COMPLETION_SYSTEM_PROMPT } from '../../provider';
 export class OpenAICompleter {
     constructor(options) {
         this._prompt = COMPLETION_SYSTEM_PROMPT;
-        this._openAIProvider = new ChatOpenAI({ ...options.settings });
+        this._completer = new ChatOpenAI({ ...options.settings });
     }
-    get provider() {
-        return this._openAIProvider;
+    get completer() {
+        return this._completer;
     }
     /**
      * Getter and setter for the initial prompt.
@@ -23,7 +23,7 @@ export class OpenAICompleter {
         const prompt = text.slice(0, cursorOffset);
         const messages = [new SystemMessage(this._prompt), new AIMessage(prompt)];
         try {
-            const response = await this._openAIProvider.invoke(messages);
+            const response = await this._completer.invoke(messages);
             const items = [];
             if (typeof response.content === 'string') {
                 items.push({

package/lib/index.js

@@ -5,16 +5,16 @@ import { INotebookTracker } from '@jupyterlab/notebook';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { ISettingConnector, ISettingRegistry } from '@jupyterlab/settingregistry';
 import { IFormRendererRegistry } from '@jupyterlab/ui-components';
-import { ISecretsManager } from 'jupyter-secrets-manager';
+import { ISecretsManager, SecretsManager } from 'jupyter-secrets-manager';
 import { ChatHandler } from './chat-handler';
 import { CompletionProvider } from './completion-provider';
 import { defaultProviderPlugins } from './default-providers';
 import { AIProviderRegistry } from './provider';
 import { aiSettingsRenderer, SettingConnector } from './settings';
-import { IAIProviderRegistry } from './tokens';
+import { IAIProviderRegistry, PLUGIN_IDS } from './tokens';
 import { stopItem } from './components/stop-button';
 const chatCommandRegistryPlugin = {
-    id: '@jupyterlite/ai:autocompletion-registry',
+    id: PLUGIN_IDS.chatCommandRegistry,
     description: 'Autocompletion registry',
     autoStart: true,
     provides: IChatCommandRegistry,
@@ -25,7 +25,7 @@ const chatCommandRegistryPlugin = {
     }
 };
 const chatPlugin = {
-    id: '@jupyterlite/ai:chat',
+    id: PLUGIN_IDS.chat,
     description: 'LLM chat extension',
     autoStart: true,
     requires: [IAIProviderRegistry, IRenderMimeRegistry, IChatCommandRegistry],
@@ -98,7 +98,7 @@ const chatPlugin = {
     }
 };
 const completerPlugin = {
-    id: '@jupyterlite/ai:completer',
+    id: PLUGIN_IDS.completer,
     autoStart: true,
     requires: [IAIProviderRegistry, ICompletionProviderManager],
     activate: (app, providerRegistry, manager) => {
@@ -109,16 +109,20 @@ const completerPlugin = {
         manager.registerInlineProvider(completer);
     }
 };
-const providerRegistryPlugin = {
-    id: '@jupyterlite/ai:provider-registry',
+const providerRegistryPlugin = SecretsManager.sign(PLUGIN_IDS.providerRegistry, token => ({
+    id: PLUGIN_IDS.providerRegistry,
     autoStart: true,
     requires: [IFormRendererRegistry, ISettingRegistry],
     optional: [IRenderMimeRegistry, ISecretsManager, ISettingConnector],
     provides: IAIProviderRegistry,
     activate: (app, editorRegistry, settingRegistry, rmRegistry, secretsManager, settingConnector) => {
-        const providerRegistry = new AIProviderRegistry({ secretsManager });
-        editorRegistry.addRenderer('@jupyterlite/ai:provider-registry.AIprovider', aiSettingsRenderer({
+        const providerRegistry = new AIProviderRegistry({
+            token,
+            secretsManager
+        });
+        editorRegistry.addRenderer(`${PLUGIN_IDS.providerRegistry}.AIprovider`, aiSettingsRenderer({
             providerRegistry,
+            secretsToken: token,
             rmRegistry,
             secretsManager,
             settingConnector
@@ -145,14 +149,14 @@ const providerRegistryPlugin = {
         });
         return providerRegistry;
     }
-};
+}));
 /**
  * Provides the settings connector as a separate plugin to allow for alternative
  * implementations that may want to fetch settings from a different source or
  * endpoint.
  */
 const settingsConnector = {
-    id: '@jupyterlite/ai:settings-connector',
+    id: PLUGIN_IDS.settingsConnector,
     description: 'Provides a settings connector which does not save passwords.',
     autoStart: true,
     provides: ISettingConnector,

package/lib/provider.d.ts

@@ -84,6 +84,10 @@ export declare namespace AIProviderRegistry {
          * The secrets manager used in the application.
          */
         secretsManager?: ISecretsManager;
+        /**
+         * The token used to request the secrets manager.
+         */
+        token: symbol;
     }
     /**
      * The options for the Chat system prompt.

package/lib/provider.js

@@ -1,5 +1,7 @@
 import { Signal } from '@lumino/signaling';
-import { getSecretId, SECRETS_NAMESPACE, SECRETS_REPLACEMENT } from './settings';
+import { getSecretId, SECRETS_REPLACEMENT } from './settings';
+import { PLUGIN_IDS } from './tokens';
+const SECRETS_NAMESPACE = PLUGIN_IDS.providerRegistry;
 export const chatSystemPrompt = (options) => `
 You are Jupyternaut, a conversational assistant living in JupyterLab to help users.
 You are not a language model, but rather an application built on a foundation model from ${options.provider_name}.
@@ -41,6 +43,7 @@ export class AIProviderRegistry {
         this._providers = new Map();
         this._deferredProvider = null;
         this._secretsManager = options.secretsManager || null;
+        Private.setToken(options.token);
     }
     /**
      * Get the list of provider names.
@@ -149,8 +152,10 @@ export class AIProviderRegistry {
         for (const key of Object.keys(settings)) {
             if (settings[key] === SECRETS_REPLACEMENT) {
                 const id = getSecretId(name, key);
-                const secrets = await ((_b = this._secretsManager) === null || _b === void 0 ? void 0 : _b.get(SECRETS_NAMESPACE, id));
-                fullSettings[key] = (secrets === null || secrets === void 0 ? void 0 : secrets.value) || settings[key];
+                const secrets = await ((_b = this._secretsManager) === null || _b === void 0 ? void 0 : _b.get(Private.getToken(), SECRETS_NAMESPACE, id));
+                if (secrets !== undefined) {
+                    fullSettings[key] = secrets.value;
+                }
                 continue;
             }
             fullSettings[key] = settings[key];
@@ -158,7 +163,7 @@ export class AIProviderRegistry {
         if (((_c = this._currentProvider) === null || _c === void 0 ? void 0 : _c.completer) !== undefined) {
             try {
                 this._completer = new this._currentProvider.completer({
-                    ...fullSettings
+                    settings: fullSettings
                 });
                 this._completerError = '';
             }
@@ -231,3 +236,24 @@ export class AIProviderRegistry {
     }
     AIProviderRegistry.updateConfig = updateConfig;
 })(AIProviderRegistry || (AIProviderRegistry = {}));
+var Private;
+(function (Private) {
+    /**
+     * The token to use with the secrets manager.
+     */
+    let secretsToken;
+    /**
+     * Set of the token.
+     */
+    function setToken(value) {
+        secretsToken = value;
+    }
+    Private.setToken = setToken;
+    /**
+     * get the token.
+     */
+    function getToken() {
+        return secretsToken;
+    }
+    Private.getToken = getToken;
+})(Private || (Private = {}));

package/lib/settings/panel.d.ts

@@ -8,6 +8,7 @@ import React from 'react';
 import { IAIProviderRegistry, IDict } from '../tokens';
 export declare const aiSettingsRenderer: (options: {
     providerRegistry: IAIProviderRegistry;
+    secretsToken?: symbol;
     rmRegistry?: IRenderMimeRegistry;
     secretsManager?: ISecretsManager;
     settingConnector?: ISettingConnector;
@@ -19,6 +20,7 @@ export interface ISettingsFormStates {
 export declare class AiSettings extends React.Component<FieldProps, ISettingsFormStates> {
     constructor(props: FieldProps);
     componentDidUpdate(): Promise<void>;
+    componentWillUnmount(): void;
     /**
      * Get the current provider from the local storage.
      */
@@ -35,12 +37,12 @@ export declare class AiSettings extends React.Component<FieldProps, ISettingsFor
      * Save settings in local storage for a given provider.
      */
     saveSettings(value: IDict<any>): void;
-    private updateUseSecretsManager;
     /**
-     * Update the UI schema of the form.
-     * Currently use to hide API keys.
+     * Update the settings whether the secrets manager is used or not.
+     *
+     * @param value - whether to use the secrets manager or not.
      */
-    private _updateUiSchema;
+    private _updateUseSecretsManager;
     /**
      * Build the schema for a given provider.
      */
@@ -75,6 +77,7 @@ export declare class AiSettings extends React.Component<FieldProps, ISettingsFor
     private _provider;
     private _providerSchema;
     private _useSecretsManager;
+    private _hideSecretFields;
     private _rmRegistry;
     private _secretsManager;
     private _settingConnector;
@@ -82,6 +85,5 @@ export declare class AiSettings extends React.Component<FieldProps, ISettingsFor
     private _uiSchema;
     private _settings;
     private _formRef;
-    private _unsavedFields;
-    private _formInputs;
+    private _secretFields;
 }

package/lib/settings/panel.js

@@ -1,14 +1,20 @@
 import { FormComponent } from '@jupyterlab/ui-components';
-import { ArrayExt } from '@lumino/algorithm';
 import { JSONExt } from '@lumino/coreutils';
 import validator from '@rjsf/validator-ajv8';
 import React from 'react';
-import { getSecretId, SECRETS_NAMESPACE, SettingConnector } from '.';
+import { getSecretId, SettingConnector } from '.';
 import baseSettings from './base.json';
+import { PLUGIN_IDS } from '../tokens';
 const MD_MIME_TYPE = 'text/markdown';
 const STORAGE_NAME = '@jupyterlite/ai:settings';
 const INSTRUCTION_CLASS = 'jp-AISettingsInstructions';
+const SECRETS_NAMESPACE = PLUGIN_IDS.providerRegistry;
 export const aiSettingsRenderer = (options) => {
+    const { secretsToken } = options;
+    delete options.secretsToken;
+    if (secretsToken) {
+        Private.setToken(secretsToken);
+    }
     return {
         fieldRenderer: (props) => {
             props.formContext = { ...props.formContext, ...options };
@@ -21,25 +27,27 @@ const WrappedFormComponent = (props) => {
 };
 export class AiSettings extends React.Component {
     constructor(props) {
-        var _a, _b, _c, _d, _e;
+        var _a, _b, _c, _d, _e, _f;
         super(props);
-        this.updateUseSecretsManager = (value) => {
+        /**
+         * Update the settings whether the secrets manager is used or not.
+         *
+         * @param value - whether to use the secrets manager or not.
+         */
+        this._updateUseSecretsManager = (value) => {
             var _a;
             this._useSecretsManager = value;
             if (!value) {
                 // Detach all the password inputs attached to the secrets manager, and save the
                 // current settings to the local storage to save the password.
-                (_a = this._secretsManager) === null || _a === void 0 ? void 0 : _a.detachAll(SECRETS_NAMESPACE);
-                this._formInputs = [];
-                this._unsavedFields = [];
+                (_a = this._secretsManager) === null || _a === void 0 ? void 0 : _a.detachAll(Private.getToken(), SECRETS_NAMESPACE);
                 if (this._settingConnector instanceof SettingConnector) {
                     this._settingConnector.doNotSave = [];
                 }
                 this.saveSettings(this._currentSettings);
             }
             else {
-                // Remove all the keys stored locally and attach the password inputs to the
-                // secrets manager.
+                // Remove all the keys stored locally.
                 const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) || '{}');
                 Object.keys(settings).forEach(provider => {
                     Object.keys(settings[provider])
@@ -49,6 +57,11 @@ export class AiSettings extends React.Component {
                     });
                 });
                 localStorage.setItem(STORAGE_NAME, JSON.stringify(settings));
+                // Update the fields not to save in settings.
+                if (this._settingConnector instanceof SettingConnector) {
+                    this._settingConnector.doNotSave = this._secretFields;
+                }
+                // Attach the password inputs to the secrets manager.
                 this.componentDidUpdate();
             }
             this._settings
@@ -98,8 +111,7 @@ export class AiSettings extends React.Component {
         this._currentSettings = { provider: 'None' };
         this._uiSchema = {};
         this._formRef = React.createRef();
-        this._unsavedFields = [];
-        this._formInputs = [];
+        this._secretFields = [];
         if (!props.formContext.providerRegistry) {
             throw new Error('The provider registry is needed to enable the jupyterlite-ai settings panel');
         }
@@ -110,6 +122,8 @@ export class AiSettings extends React.Component {
         this._settings = props.formContext.settings;
         this._useSecretsManager =
             (_d = this._settings.get('UseSecretsManager').composite) !== null && _d !== void 0 ? _d : true;
+        this._hideSecretFields =
+            (_e = this._settings.get('HideSecretFields').composite) !== null && _e !== void 0 ? _e : true;
         // Initialize the providers schema.
         const providerSchema = JSONExt.deepCopy(baseSettings);
         providerSchema.properties.provider = {
@@ -127,7 +141,7 @@ export class AiSettings extends React.Component {
             const labSettings = this._settings.get('AIprovider').composite;
             if (labSettings && Object.keys(labSettings).includes('provider')) {
                 // Get the provider name.
-                const provider = (_e = Object.entries(labSettings).find(v => v[0] === 'provider')) === null || _e === void 0 ? void 0 : _e[1];
+                const provider = (_f = Object.entries(labSettings).find(v => v[0] === 'provider')) === null || _f === void 0 ? void 0 : _f[1];
                 // Save the settings.
                 const settings = {
                     _current: provider
@@ -148,10 +162,15 @@ export class AiSettings extends React.Component {
             .set('AIprovider', this._currentSettings)
             .catch(console.error);
         this._settings.changed.connect(() => {
-            var _a;
+            var _a, _b;
             const useSecretsManager = (_a = this._settings.get('UseSecretsManager').composite) !== null && _a !== void 0 ? _a : true;
             if (useSecretsManager !== this._useSecretsManager) {
-                this.updateUseSecretsManager(useSecretsManager);
+                this._updateUseSecretsManager(useSecretsManager);
+            }
+            const hideSecretFields = (_b = this._settings.get('HideSecretFields').composite) !== null && _b !== void 0 ? _b : true;
+            if (hideSecretFields !== this._hideSecretFields) {
+                this._hideSecretFields = hideSecretFields;
+                this._updateSchema();
             }
         });
     }
@@ -160,27 +179,24 @@ export class AiSettings extends React.Component {
         if (!this._secretsManager || !this._useSecretsManager) {
             return;
         }
-        // Attach the password inputs to the secrets manager only if they have changed.
+        // Attach the password inputs to the secrets manager.
+        await this._secretsManager.detachAll(Private.getToken(), SECRETS_NAMESPACE);
         const inputs = ((_a = this._formRef.current) === null || _a === void 0 ? void 0 : _a.getElementsByTagName('input')) || [];
-        if (ArrayExt.shallowEqual(inputs, this._formInputs)) {
-            return;
-        }
-        await this._secretsManager.detachAll(SECRETS_NAMESPACE);
-        this._formInputs = [...inputs];
-        this._unsavedFields = [];
         for (let i = 0; i < inputs.length; i++) {
             if (inputs[i].type.toLowerCase() === 'password') {
                 const label = inputs[i].getAttribute('label');
                 if (label) {
                     const id = getSecretId(this._provider, label);
-                    this._secretsManager.attach(SECRETS_NAMESPACE, id, inputs[i], (value) => this._onPasswordUpdated(label, value));
-                    this._unsavedFields.push(label);
+                    this._secretsManager.attach(Private.getToken(), SECRETS_NAMESPACE, id, inputs[i], (value) => this._onPasswordUpdated(label, value));
                 }
             }
         }
-        if (this._settingConnector instanceof SettingConnector) {
-            this._settingConnector.doNotSave = this._unsavedFields;
+    }
+    componentWillUnmount() {
+        if (!this._secretsManager || !this._useSecretsManager) {
+            return;
         }
+        this._secretsManager.detachAll(Private.getToken(), SECRETS_NAMESPACE);
     }
     /**
      * Get the current provider from the local storage.
@@ -213,19 +229,13 @@ export class AiSettings extends React.Component {
         var _a;
         const currentSettings = { ...value };
         const settings = JSON.parse((_a = localStorage.getItem(STORAGE_NAME)) !== null && _a !== void 0 ? _a : '{}');
-        this._unsavedFields.forEach(field => delete currentSettings[field]);
+        // Do not save secrets in local storage if using the secrets manager.
+        if (this._secretsManager && this._useSecretsManager) {
+            this._secretFields.forEach(field => delete currentSettings[field]);
+        }
         settings[this._provider] = currentSettings;
         localStorage.setItem(STORAGE_NAME, JSON.stringify(settings));
     }
-    /**
-     * Update the UI schema of the form.
-     * Currently use to hide API keys.
-     */
-    _updateUiSchema(key) {
-        if (key.toLowerCase().includes('key')) {
-            this._uiSchema[key] = { 'ui:widget': 'password' };
-        }
-    }
     /**
      * Build the schema for a given provider.
      */
@@ -233,12 +243,25 @@ export class AiSettings extends React.Component {
         const schema = JSONExt.deepCopy(baseSettings);
         this._uiSchema = {};
         const settingsSchema = this._providerRegistry.getSettingsSchema(this._provider);
+        this._secretFields = [];
         if (settingsSchema) {
             Object.entries(settingsSchema).forEach(([key, value]) => {
+                if (key.toLowerCase().includes('key')) {
+                    this._secretFields.push(key);
+                    if (this._hideSecretFields) {
+                        return;
+                    }
+                    this._uiSchema[key] = { 'ui:widget': 'password' };
+                }
                 schema.properties[key] = value;
-                this._updateUiSchema(key);
             });
         }
+        // Do not save secrets in settings if using the secrets manager.
+        if (this._secretsManager &&
+            this._useSecretsManager &&
+            this._settingConnector instanceof SettingConnector) {
+            this._settingConnector.doNotSave = this._secretFields;
+        }
         return schema;
     }
     /**
@@ -275,3 +298,24 @@ export class AiSettings extends React.Component {
             React.createElement(WrappedFormComponent, { formData: this._currentSettings, schema: this.state.schema, onChange: this._onFormChange, uiSchema: this._uiSchema })));
     }
 }
+var Private;
+(function (Private) {
+    /**
+     * The token to use with the secrets manager.
+     */
+    let secretsToken;
+    /**
+     * Set of the token.
+     */
+    function setToken(value) {
+        secretsToken = value;
+    }
+    Private.setToken = setToken;
+    /**
+     * get the token.
+     */
+    function getToken() {
+        return secretsToken;
+    }
+    Private.getToken = getToken;
+})(Private || (Private = {}));

package/lib/settings/settings-connector.js

@@ -49,10 +49,10 @@ export class SettingConnector extends DataConnector {
     }
     async save(id, raw) {
         const settings = json5.parse(raw);
+        // Replace secrets fields with the replacement string.
+        // Create the field if it does not exist in settings.
         this._doNotSave.forEach(field => {
-            if (settings['AIprovider'] !== undefined &&
-                settings['AIprovider'][field] !== undefined &&
-                settings['AIprovider'][field] !== '') {
+            if (settings['AIprovider'] !== undefined) {
                 settings['AIprovider'][field] = SECRETS_REPLACEMENT;
             }
         });

package/lib/settings/utils.d.ts

@@ -1,3 +1,2 @@
-export declare const SECRETS_NAMESPACE = "@jupyterlite/ai";
 export declare const SECRETS_REPLACEMENT = "***";
 export declare function getSecretId(provider: string, label: string): string;

package/lib/settings/utils.js

@@ -1,4 +1,3 @@
-export const SECRETS_NAMESPACE = '@jupyterlite/ai';
 export const SECRETS_REPLACEMENT = '***';
 export function getSecretId(provider, label) {
     return `${provider}-${label}`;

package/lib/tokens.d.ts

@@ -3,6 +3,13 @@ import { ReadonlyPartialJSONObject, Token } from '@lumino/coreutils';
 import { ISignal } from '@lumino/signaling';
 import { JSONSchema7 } from 'json-schema';
 import { IBaseCompleter } from './base-completer';
+export declare const PLUGIN_IDS: {
+    chat: string;
+    chatCommandRegistry: string;
+    completer: string;
+    providerRegistry: string;
+    settingsConnector: string;
+};
 export interface IDict<T = any> {
     [key: string]: T;
 }

package/lib/tokens.js

@@ -1,4 +1,11 @@
 import { Token } from '@lumino/coreutils';
+export const PLUGIN_IDS = {
+    chat: '@jupyterlite/ai:chat',
+    chatCommandRegistry: '@jupyterlite/ai:autocompletion-registry',
+    completer: '@jupyterlite/ai:completer',
+    providerRegistry: '@jupyterlite/ai:provider-registry',
+    settingsConnector: '@jupyterlite/ai:settings-connector'
+};
 /**
  * The provider registry token.
  */