@jupyterlite/ai 0.6.2 → 0.7.0

package/lib/settings/panel.js

@@ -1,13 +1,14 @@
-import { FormComponent } from '@jupyterlab/ui-components';
+import { Button, FormComponent } from '@jupyterlab/ui-components';
 import { JSONExt } from '@lumino/coreutils';
 import validator from '@rjsf/validator-ajv8';
 import React from 'react';
-import { getSecretId, SettingConnector } from '.';
+import { getSecretId, SECRETS_REPLACEMENT } 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 ERROR_CLASS = 'jp-AISettingsError';
 const SECRETS_NAMESPACE = PLUGIN_IDS.providerRegistry;
 export const aiSettingsRenderer = (options) => {
     const { secretsToken } = options;
@@ -27,103 +28,17 @@ const WrappedFormComponent = (props) => {
 };
 export class AiSettings extends React.Component {
     constructor(props) {
-        var _a, _b, _c, _d, _e, _f;
         super(props);
-        /**
-         * 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(Private.getToken(), SECRETS_NAMESPACE);
-                if (this._settingConnector instanceof SettingConnector) {
-                    this._settingConnector.doNotSave = [];
-                }
-                this.saveSettings(this._currentSettings);
-            }
-            else {
-                // Remove all the keys stored locally.
-                const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) || '{}');
-                Object.keys(settings).forEach(provider => {
-                    Object.keys(settings[provider])
-                        .filter(key => key.toLowerCase().includes('key'))
-                        .forEach(key => {
-                        delete settings[provider][key];
-                    });
-                });
-                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
-                .set('AIprovider', { provider: this._provider, ...this._currentSettings })
-                .catch(console.error);
-        };
-        /**
-         * Triggered when the provider hes changed, to update the schema and values.
-         * Update the Jupyterlab settings accordingly.
-         */
-        this._onProviderChanged = (e) => {
-            const provider = e.formData.provider;
-            if (provider === this._currentSettings.provider) {
-                return;
-            }
-            this._provider = provider;
-            this.saveCurrentProvider();
-            this._currentSettings = this.getSettings();
-            this._updateSchema();
-            this._renderInstruction();
-            this._settings
-                .set('AIprovider', { provider: this._provider, ...this._currentSettings })
-                .catch(console.error);
-        };
-        /**
-         * Callback function called when the password input has been programmatically updated
-         * with the secret manager.
-         */
-        this._onPasswordUpdated = (fieldName, value) => {
-            this._currentSettings[fieldName] = value;
-            this._settings
-                .set('AIprovider', { provider: this._provider, ...this._currentSettings })
-                .catch(console.error);
-        };
-        /**
-         * Triggered when the form value has changed, to update the current settings and save
-         * it in local storage.
-         * Update the Jupyterlab settings accordingly.
-         */
-        this._onFormChange = (e) => {
-            this._currentSettings = JSONExt.deepCopy(e.formData);
-            this.saveSettings(this._currentSettings);
-            this._settings
-                .set('AIprovider', { provider: this._provider, ...this._currentSettings })
-                .catch(console.error);
-        };
-        this._currentSettings = { provider: 'None' };
-        this._uiSchema = {};
-        this._formRef = React.createRef();
-        this._secretFields = [];
         if (!props.formContext.providerRegistry) {
             throw new Error('The provider registry is needed to enable the jupyterlite-ai settings panel');
         }
         this._providerRegistry = props.formContext.providerRegistry;
-        this._rmRegistry = (_a = props.formContext.rmRegistry) !== null && _a !== void 0 ? _a : null;
-        this._secretsManager = (_b = props.formContext.secretsManager) !== null && _b !== void 0 ? _b : null;
-        this._settingConnector = (_c = props.formContext.settingConnector) !== null && _c !== void 0 ? _c : null;
+        this._rmRegistry = props.formContext.rmRegistry ?? null;
+        this._secretsManager = props.formContext.secretsManager ?? null;
         this._settings = props.formContext.settings;
+        const useSecretsManagerSetting = this._settings.get('UseSecretsManager').composite ?? true;
         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;
+            useSecretsManagerSetting && this._secretsManager !== null;
         // Initialize the providers schema.
         const providerSchema = JSONExt.deepCopy(baseSettings);
         providerSchema.properties.provider = {
@@ -135,13 +50,13 @@ export class AiSettings extends React.Component {
         };
         this._providerSchema = providerSchema;
         // Check if there is saved values in local storage, otherwise use the settings from
-        // the setting registry (led to default if there are no user settings).
+        // the setting registry (leads to default if there are no user settings).
         const storageSettings = localStorage.getItem(STORAGE_NAME);
         if (storageSettings === null) {
             const labSettings = this._settings.get('AIprovider').composite;
             if (labSettings && Object.keys(labSettings).includes('provider')) {
                 // Get the provider name.
-                const provider = (_f = Object.entries(labSettings).find(v => v[0] === 'provider')) === null || _f === void 0 ? void 0 : _f[1];
+                const provider = Object.entries(labSettings).find(v => v[0] === 'provider')?.[1];
                 // Save the settings.
                 const settings = {
                     _current: provider
@@ -152,36 +67,30 @@ export class AiSettings extends React.Component {
         }
         // Initialize the settings from the saved ones.
         this._provider = this.getCurrentProvider();
-        this._currentSettings = this.getSettings();
         // Initialize the schema.
         const schema = this._buildSchema();
-        this.state = { schema, instruction: null };
+        // Initialize the current settings.
+        const isModified = this._updatedFormData(this.getSettingsFromLocalStorage());
+        this.state = {
+            schema,
+            instruction: null,
+            compatibilityError: null,
+            isModified: isModified
+        };
         this._renderInstruction();
+        this._checkProviderCompatibility();
         // Update the setting registry.
-        this._settings
-            .set('AIprovider', this._currentSettings)
-            .catch(console.error);
-        this._settings.changed.connect(() => {
-            var _a, _b;
-            const useSecretsManager = (_a = this._settings.get('UseSecretsManager').composite) !== null && _a !== void 0 ? _a : true;
-            if (useSecretsManager !== this._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();
-            }
-        });
+        this.saveSettingsToRegistry();
+        this._secretsManager?.fieldVisibilityChanged.connect(this._fieldVisibilityChanged);
+        this._settings.changed.connect(this._settingsChanged);
     }
     async componentDidUpdate() {
-        var _a;
         if (!this._secretsManager || !this._useSecretsManager) {
             return;
         }
         // 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')) || [];
+        const inputs = this._formRef.current?.getElementsByTagName('input') || [];
         for (let i = 0; i < inputs.length; i++) {
             if (inputs[i].type.toLowerCase() === 'password') {
                 const label = inputs[i].getAttribute('label');
@@ -193,6 +102,8 @@ export class AiSettings extends React.Component {
         }
     }
     componentWillUnmount() {
+        this._settings.changed.disconnect(this._settingsChanged);
+        this._secretsManager?.fieldVisibilityChanged.disconnect(this._fieldVisibilityChanged);
         if (!this._secretsManager || !this._useSecretsManager) {
             return;
         }
@@ -202,9 +113,8 @@ export class AiSettings extends React.Component {
      * Get the current provider from the local storage.
      */
     getCurrentProvider() {
-        var _a;
         const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) || '{}');
-        return (_a = settings['_current']) !== null && _a !== void 0 ? _a : 'None';
+        return settings['_current'] ?? 'None';
     }
     /**
      * Save the current provider to the local storage.
@@ -217,25 +127,84 @@ export class AiSettings extends React.Component {
     /**
      * Get settings from local storage for a given provider.
      */
-    getSettings() {
-        var _a;
+    getSettingsFromLocalStorage() {
         const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) || '{}');
-        return (_a = settings[this._provider]) !== null && _a !== void 0 ? _a : { provider: this._provider };
+        return settings[this._provider] ?? { provider: this._provider };
     }
     /**
      * Save settings in local storage for a given provider.
      */
-    saveSettings(value) {
-        var _a;
-        const currentSettings = { ...value };
-        const settings = JSON.parse((_a = localStorage.getItem(STORAGE_NAME)) !== null && _a !== void 0 ? _a : '{}');
+    saveSettingsToLocalStorage() {
+        const currentSettings = { ...this._currentSettings };
+        const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) ?? '{}');
         // Do not save secrets in local storage if using the secrets manager.
-        if (this._secretsManager && this._useSecretsManager) {
+        if (this._useSecretsManager) {
             this._secretFields.forEach(field => delete currentSettings[field]);
         }
         settings[this._provider] = currentSettings;
         localStorage.setItem(STORAGE_NAME, JSON.stringify(settings));
     }
+    /**
+     * Save the settings to the setting registry.
+     */
+    saveSettingsToRegistry() {
+        const sanitizedSettings = { ...this._currentSettings };
+        if (this._useSecretsManager) {
+            this._secretFields.forEach(field => {
+                sanitizedSettings[field] = SECRETS_REPLACEMENT;
+            });
+        }
+        this._settings
+            .set('AIprovider', { provider: this._provider, ...sanitizedSettings })
+            .catch(console.error);
+    }
+    /**
+     * Triggered when the settings has changed.
+     */
+    _settingsChanged = (settings) => {
+        this._updateUseSecretsManager(this._settings.get('UseSecretsManager').composite ?? true);
+    };
+    /**
+     * Triggered when the secret fields visibility has changed.
+     */
+    _fieldVisibilityChanged = (_, value) => {
+        if (this._useSecretsManager) {
+            this._updateSchema();
+        }
+    };
+    /**
+     * Update the settings whether the secrets manager is used or not.
+     *
+     * @param value - whether to use the secrets manager or not.
+     */
+    _updateUseSecretsManager = (value) => {
+        // No-op if the value did not change or the secrets manager has not been provided.
+        if (value === this._useSecretsManager || this._secretsManager === null) {
+            return;
+        }
+        // Update the secrets manager.
+        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.
+            this._secretsManager.detachAll(Private.getToken(), SECRETS_NAMESPACE);
+        }
+        else {
+            // Remove all the keys stored locally.
+            const settings = JSON.parse(localStorage.getItem(STORAGE_NAME) || '{}');
+            Object.keys(settings).forEach(provider => {
+                Object.keys(settings[provider])
+                    .filter(key => key.toLowerCase().includes('key'))
+                    .forEach(key => {
+                    delete settings[provider][key];
+                });
+            });
+            localStorage.setItem(STORAGE_NAME, JSON.stringify(settings));
+        }
+        this._updateSchema();
+        this.saveSettingsToLocalStorage();
+        this.saveSettingsToRegistry();
+    };
     /**
      * Build the schema for a given provider.
      */
@@ -244,24 +213,26 @@ export class AiSettings extends React.Component {
         this._uiSchema = {};
         const settingsSchema = this._providerRegistry.getSettingsSchema(this._provider);
         this._secretFields = [];
+        this._defaultFormData = {};
         if (settingsSchema) {
             Object.entries(settingsSchema).forEach(([key, value]) => {
                 if (key.toLowerCase().includes('key')) {
                     this._secretFields.push(key);
-                    if (this._hideSecretFields) {
+                    // If the secrets manager is not used, do not show the secrets fields.
+                    // If the secrets manager is used, check if the fields should be visible.
+                    const showSecretFields = !this._useSecretsManager ||
+                        (this._secretsManager?.secretFieldsVisibility ?? true);
+                    if (!showSecretFields) {
                         return;
                     }
                     this._uiSchema[key] = { 'ui:widget': 'password' };
                 }
                 schema.properties[key] = value;
+                if (value.default !== undefined) {
+                    this._defaultFormData[key] = value.default;
+                }
             });
         }
-        // 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;
     }
     /**
@@ -289,14 +260,137 @@ export class AiSettings extends React.Component {
         await renderer.renderModel(model);
         this.setState({ instruction: renderer.node });
     }
+    /**
+     * Check for compatibility of the provider with the current environment.
+     * If the provider is not compatible, display an error message.
+     */
+    async _checkProviderCompatibility() {
+        const compatibilityCheck = this._providerRegistry.getCompatibilityCheck(this._provider);
+        if (!compatibilityCheck) {
+            this.setState({ compatibilityError: null });
+            return;
+        }
+        const error = await compatibilityCheck();
+        if (!error) {
+            this.setState({ compatibilityError: null });
+            return;
+        }
+        const errorDiv = document.createElement('div');
+        errorDiv.className = ERROR_CLASS;
+        errorDiv.innerHTML = error;
+        this.setState({ compatibilityError: error });
+    }
+    /**
+     * Triggered when the provider has changed, to update the schema and values.
+     * Update the Jupyterlab settings accordingly.
+     */
+    _onProviderChanged = (e) => {
+        const provider = e.formData.provider;
+        if (provider === this._currentSettings.provider) {
+            return;
+        }
+        this._provider = provider;
+        this.saveCurrentProvider();
+        this._updateSchema();
+        this._renderInstruction();
+        this._checkProviderCompatibility();
+        // Initialize the current settings.
+        const isModified = this._updatedFormData(this.getSettingsFromLocalStorage());
+        if (isModified !== this.state.isModified) {
+            this.setState({ isModified });
+        }
+        this.saveSettingsToRegistry();
+    };
+    /**
+     * Callback function called when the password input has been programmatically updated
+     * with the secret manager.
+     */
+    _onPasswordUpdated = (fieldName, value) => {
+        this._currentSettings[fieldName] = value;
+        this.saveSettingsToRegistry();
+    };
+    /**
+     * Update the current settings with the new values from the form.
+     *
+     * @param data - The form data to update.
+     * @returns - Boolean whether the form is not the default one.
+     */
+    _updatedFormData(data) {
+        let isModified = false;
+        Object.entries(data).forEach(([key, value]) => {
+            if (this._defaultFormData[key] !== undefined) {
+                if (value === undefined) {
+                    const schemaProperty = this.state.schema.properties?.[key];
+                    if (schemaProperty.type === 'string') {
+                        data[key] = '';
+                    }
+                }
+                if (value !== this._defaultFormData[key]) {
+                    isModified = true;
+                }
+            }
+        });
+        this._currentSettings = JSONExt.deepCopy(data);
+        return isModified;
+    }
+    /**
+     * Triggered when the form value has changed, to update the current settings and save
+     * it in local storage.
+     * Update the Jupyterlab settings accordingly.
+     */
+    _onFormChanged = (e) => {
+        const { formData } = e;
+        const isModified = this._updatedFormData(formData);
+        this.saveSettingsToLocalStorage();
+        this.saveSettingsToRegistry();
+        if (isModified !== this.state.isModified) {
+            this.setState({ isModified });
+        }
+    };
+    /**
+     * Handler for the "Restore to defaults" button - clears all
+     * modified settings then calls `setFormData` to restore the
+     * values.
+     */
+    _reset = async (event) => {
+        event.stopPropagation();
+        this._currentSettings = {
+            ...this._currentSettings,
+            ...this._defaultFormData
+        };
+        this.saveSettingsToLocalStorage();
+        this.saveSettingsToRegistry();
+        this.setState({ isModified: false });
+    };
     render() {
         return (React.createElement("div", { ref: this._formRef },
             React.createElement(WrappedFormComponent, { formData: { provider: this._provider }, schema: this._providerSchema, onChange: this._onProviderChanged }),
+            this.state.compatibilityError !== null && (React.createElement("div", { className: ERROR_CLASS },
+                React.createElement("i", { className: 'fas fa-exclamation-triangle' }),
+                React.createElement("span", null, this.state.compatibilityError))),
             this.state.instruction !== null && (React.createElement("details", null,
                 React.createElement("summary", { className: INSTRUCTION_CLASS }, "Instructions"),
                 React.createElement("span", { ref: node => node && node.replaceChildren(this.state.instruction) }))),
-            React.createElement(WrappedFormComponent, { formData: this._currentSettings, schema: this.state.schema, onChange: this._onFormChange, uiSchema: this._uiSchema })));
+            React.createElement("div", { className: "jp-SettingsHeader" },
+                React.createElement("h3", { title: this._provider }, this._provider),
+                React.createElement("div", { className: "jp-SettingsHeader-buttonbar" }, this.state.isModified && (React.createElement(Button, { className: "jp-RestoreButton", onClick: this._reset }, "Restore to Defaults")))),
+            React.createElement(WrappedFormComponent, { formData: this._currentSettings, schema: this.state.schema, onChange: this._onFormChanged, uiSchema: this._uiSchema, idPrefix: `jp-SettingsEditor-${PLUGIN_IDS.providerRegistry}`, formContext: {
+                    ...this.props.formContext,
+                    defaultFormData: this._defaultFormData
+                } })));
     }
+    _providerRegistry;
+    _provider;
+    _providerSchema;
+    _useSecretsManager;
+    _rmRegistry;
+    _secretsManager;
+    _currentSettings = { provider: 'None' };
+    _uiSchema = {};
+    _settings;
+    _formRef = React.createRef();
+    _secretFields = [];
+    _defaultFormData = {};
 }
 var Private;
 (function (Private) {

package/lib/tokens.d.ts

@@ -3,6 +3,7 @@ import { ReadonlyPartialJSONObject, Token } from '@lumino/coreutils';
 import { ISignal } from '@lumino/signaling';
 import { JSONSchema7 } from 'json-schema';
 import { IBaseCompleter } from './base-completer';
+import { AIChatModel, AICompleter } from './types/ai-model';
 export declare const PLUGIN_IDS: {
     chat: string;
     chatCommandRegistry: string;
@@ -46,6 +47,20 @@ export interface IAIProvider {
      * Default to `(error) => error.message`.
      */
     errorMessage?: (error: any) => string;
+    /**
+     * Compatibility check function, to determine if the provider is compatible with the
+     * current environment.
+     */
+    compatibilityCheck?: () => Promise<string | null>;
+    /**
+     * Whether to expose or not the chat model.
+     *
+     * ### CAUTION
+     * This flag will expose the whole chat model API, which may contain private keys.
+     * Be sure to use it with a model that does not expose sensitive information in the
+     * API.
+     */
+    exposeChatModel?: boolean;
 }
 /**
  * The provider registry interface.
@@ -66,11 +81,11 @@ export interface IAIProviderRegistry {
     /**
      * Get the current completer of the completion provider.
      */
-    currentCompleter: IBaseCompleter | null;
+    currentCompleter: AICompleter | null;
     /**
      * Get the current llm chat model.
      */
-    currentChatModel: BaseChatModel | null;
+    currentChatModel: AIChatModel | null;
     /**
      * Get the settings schema of a given provider.
      */
@@ -79,6 +94,10 @@ export interface IAIProviderRegistry {
      * Get the instructions of a given provider.
      */
     getInstructions(provider: string): string | undefined;
+    /**
+     * Get the compatibility check function of a given provider.
+     */
+    getCompatibilityCheck(provider: string): (() => Promise<string | null>) | undefined;
     /**
      * Format an error message from the current provider.
      */

package/lib/types/ai-model.d.ts

@@ -0,0 +1,24 @@
+import { CompletionHandler, IInlineCompletionContext } from '@jupyterlab/completer';
+import { IterableReadableStream } from '@langchain/core/utils/stream';
+/**
+ * The reduced AI chat model interface.
+ */
+export type AIChatModel = {
+    /**
+     * The stream function of the chat model.
+     */
+    stream: (input: any, options?: any) => Promise<IterableReadableStream<any>>;
+};
+/**
+ * The reduced AI completer interface.
+ */
+export type AICompleter = {
+    /**
+     * The fetch function of the completer.
+     */
+    fetch: (request: CompletionHandler.IRequest, context: IInlineCompletionContext) => Promise<any>;
+    /**
+     * The optional request completion function of the completer.
+     */
+    requestCompletion?: () => void;
+};

package/lib/types/ai-model.js

@@ -0,0 +1,5 @@
+/*
+ * Copyright (c) Jupyter Development Team.
+ * Distributed under the terms of the Modified BSD License.
+ */
+export {};

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@jupyterlite/ai",
-    "version": "0.6.2",
+    "version": "0.7.0",
     "description": "AI code completions and chat for JupyterLite",
     "keywords": [
         "jupyter",
@@ -56,7 +56,7 @@
         "watch:labextension": "jupyter labextension watch ."
     },
     "dependencies": {
-        "@jupyter/chat": "^0.9.0",
+        "@jupyter/chat": "^0.12.0",
         "@jupyterlab/application": "^4.4.0",
         "@jupyterlab/apputils": "^4.5.0",
         "@jupyterlab/completer": "^4.4.0",
@@ -66,31 +66,37 @@
         "@jupyterlab/settingregistry": "^4.4.0",
         "@jupyterlab/ui-components": "^4.4.0",
         "@langchain/anthropic": "^0.3.9",
-        "@langchain/community": "^0.3.31",
-        "@langchain/core": "^0.3.40",
+        "@langchain/community": "^0.3.44",
+        "@langchain/core": "^0.3.57",
         "@langchain/mistralai": "^0.1.1",
+        "@langchain/ollama": "^0.2.0",
         "@langchain/openai": "^0.4.4",
         "@lumino/coreutils": "^2.1.2",
         "@lumino/polling": "^2.1.2",
         "@lumino/signaling": "^2.1.2",
+        "@mlc-ai/web-llm": "^0.2.79",
+        "@mlc-ai/web-runtime": "^0.18.0-dev2",
+        "@mlc-ai/web-tokenizers": "^0.1.6",
         "@mui/icons-material": "^5.11.0",
         "@mui/material": "^5.11.0",
         "@rjsf/core": "^5.18.4",
         "@rjsf/utils": "^5.18.4",
         "@rjsf/validator-ajv8": "^5.18.4",
         "json5": "^2.2.3",
-        "jupyter-secrets-manager": "^0.3.0",
+        "jupyter-secrets-manager": "^0.4.0",
         "react": "^18.2.0",
         "react-dom": "^18.2.0"
     },
     "devDependencies": {
         "@jupyterlab/builder": "^4.4.0",
         "@stylistic/eslint-plugin": "^3.0.1",
+        "@types/chrome": "^0.0.304",
         "@types/json-schema": "^7.0.11",
         "@types/react": "^18.0.26",
         "@types/react-addons-linked-state-mixin": "^0.14.22",
         "@typescript-eslint/eslint-plugin": "^6.1.0",
         "@typescript-eslint/parser": "^6.1.0",
+        "@webgpu/types": "^0.1.54",
         "css-loader": "^6.7.1",
         "eslint": "^8.36.0",
         "eslint-config-prettier": "^8.8.0",
@@ -105,8 +111,8 @@
         "stylelint-config-standard": "^34.0.0",
         "stylelint-csstree-validator": "^3.0.0",
         "stylelint-prettier": "^4.0.0",
-        "ts-json-schema-generator": "^2.3.0",
-        "typescript": "~5.1.6",
+        "ts-json-schema-generator": "^2.4.0",
+        "typescript": "~5.8.3",
         "yjs": "^13.5.0"
     },
     "sideEffects": [
@@ -120,9 +126,6 @@
     "jupyterlab": {
         "extension": true,
         "outputDir": "jupyterlite_ai/labextension",
-        "schemaDir": "schema",
-        "disabledExtensions": [
-            "@jupyterlab/apputils-extension:settings-connector"
-        ]
+        "schemaDir": "schema"
     }
 }

package/schema/provider-registry.json

@@ -11,12 +11,6 @@
       "description": "Whether to use or not the secrets manager. If not, secrets will be stored in the browser (local storage)",
       "default": true
     },
-    "HideSecretFields": {
-      "type": "boolean",
-      "title": "Hide secret fields",
-      "description": "Whether to hide the secret fields in the UI or not",
-      "default": true
-    },
     "AIprovider": {
       "type": "object",
       "title": "AI provider",

package/src/base-completer.ts

@@ -2,15 +2,9 @@ import {
   CompletionHandler,
   IInlineCompletionContext
 } from '@jupyterlab/completer';
-import { BaseLanguageModel } from '@langchain/core/language_models/base';
 import { ReadonlyPartialJSONObject } from '@lumino/coreutils';
 
 export interface IBaseCompleter {
-  /**
-   * The LLM completer.
-   */
-  completer: BaseLanguageModel;
-
   /**
    * The completion prompt.
    */