@jupyterlite/ai 0.18.0 → 0.19.0

package/package.json

@@ -1,261 +1,114 @@
 {
-    "name": "@jupyterlite/ai",
-    "version": "0.18.0",
-    "description": "AI code completions and chat for JupyterLite",
-    "keywords": [
-        "jupyter",
-        "jupyterlab",
-        "jupyterlab-extension"
-    ],
-    "homepage": "https://github.com/jupyterlite/ai",
-    "bugs": {
-        "url": "https://github.com/jupyterlite/ai/issues"
-    },
-    "license": "BSD-3-Clause",
-    "author": "JupyterLite Contributors",
-    "files": [
-        "lib/**/*.{d.ts,eot,gif,html,jpg,js,js.map,json,png,svg,woff2,ttf,md}",
-        "style/**/*.{css,js,eot,gif,html,jpg,json,png,svg,woff2,ttf}",
-        "src/**/*.{ts,tsx}",
-        "schema/*.json"
-    ],
-    "main": "lib/index.js",
-    "types": "lib/index.d.ts",
-    "style": "style/index.css",
-    "repository": {
-        "type": "git",
-        "url": "https://github.com/jupyterlite/ai.git"
-    },
-    "scripts": {
-        "build": "jlpm build:lib && jlpm build:labextension:dev",
-        "build:dev": "jlpm build:lib && jlpm build:labextension:dev",
-        "build:prod": "jlpm clean && jlpm build:lib:prod && jlpm build:labextension",
-        "build:labextension": "jupyter labextension build .",
-        "build:labextension:dev": "jupyter labextension build --development True .",
-        "build:lib": "tsc --sourceMap",
-        "build:lib:prod": "tsc",
-        "clean": "jlpm clean:lib",
-        "clean:lib": "rimraf lib tsconfig.tsbuildinfo",
-        "clean:lintcache": "rimraf .eslintcache .stylelintcache",
-        "clean:labextension": "rimraf jupyterlite_ai/labextension jupyterlite_ai/_version.py",
-        "clean:all": "jlpm clean:lib && jlpm clean:labextension && jlpm clean:lintcache",
-        "eslint": "jlpm eslint:check --fix",
-        "eslint:check": "eslint . --cache --ext .ts,.tsx",
-        "install:extension": "jlpm build",
-        "lint": "jlpm stylelint && jlpm prettier && jlpm eslint",
-        "lint:check": "jlpm stylelint:check && jlpm prettier:check && jlpm eslint:check",
-        "prettier": "jlpm prettier:base --write --list-different",
-        "prettier:base": "prettier \"**/*{.ts,.tsx,.js,.jsx,.css,.json,.md}\"",
-        "prettier:check": "jlpm prettier:base --check",
-        "stylelint": "jlpm stylelint:check --fix",
-        "stylelint:check": "stylelint --cache \"style/**/*.css\"",
-        "watch": "run-p watch:src watch:labextension",
-        "watch:src": "tsc -w --sourceMap",
-        "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",
-        "sync:model-info": "node scripts/sync-model-info.mjs && prettier --write src/providers/generated-model-info.ts && eslint --fix src/providers/generated-model-info.ts"
-    },
-    "dependencies": {
-        "@ai-sdk/anthropic": "^3.0.58",
-        "@ai-sdk/google": "^3.0.43",
-        "@ai-sdk/mcp": "^1.0.25",
-        "@ai-sdk/mistral": "^3.0.24",
-        "@ai-sdk/openai": "^3.0.41",
-        "@ai-sdk/openai-compatible": "^2.0.35",
-        "@jupyter/chat": "^0.21.1",
-        "@jupyterlab/application": "^4.0.0",
-        "@jupyterlab/apputils": "^4.5.6",
-        "@jupyterlab/cells": "^4.4.6",
-        "@jupyterlab/completer": "^4.0.0",
-        "@jupyterlab/coreutils": "^6.4.6",
-        "@jupyterlab/docmanager": "^4.4.6",
-        "@jupyterlab/docregistry": "^4.4.6",
-        "@jupyterlab/filebrowser": "^4.5.6",
-        "@jupyterlab/fileeditor": "^4.4.6",
-        "@jupyterlab/notebook": "^4.4.6",
-        "@jupyterlab/rendermime": "^4.4.6",
-        "@jupyterlab/services": "^7.4.6",
-        "@jupyterlab/settingregistry": "^4.0.0",
-        "@jupyterlab/statusbar": "^4.4.6",
-        "@jupyterlab/ui-components": "^4.4.6",
-        "@lumino/commands": "^2.3.2",
-        "@lumino/coreutils": "^2.2.1",
-        "@lumino/disposable": "^2.1.4",
-        "@lumino/messaging": "^2.0.3",
-        "@lumino/polling": "^2.1.4",
-        "@lumino/signaling": "^2.1.4",
-        "@lumino/widgets": "^2.7.1",
-        "@mui/icons-material": "^7",
-        "@mui/material": "^7",
-        "ai": "^6.0.116",
-        "jupyter-chat-components": "^0.5.0",
-        "jupyter-secrets-manager": "^0.5.0",
-        "yaml": "^2.8.1",
-        "zod": "^4.3.6"
-    },
-    "devDependencies": {
-        "@jupyterlab/builder": "^4.0.0",
-        "@jupyterlab/testutils": "^4.0.0",
-        "@types/json-schema": "^7.0.11",
-        "@types/node": "^24.3.0",
-        "@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",
-        "css-loader": "^6.7.1",
-        "eslint": "^8.36.0",
-        "eslint-config-prettier": "^8.8.0",
-        "eslint-plugin-prettier": "^5.0.0",
-        "npm-run-all2": "^7.0.1",
-        "prettier": "^3.0.0",
-        "rimraf": "^5.0.1",
-        "source-map-loader": "^1.0.2",
-        "style-loader": "^3.3.1",
-        "stylelint": "^15.10.1",
-        "stylelint-config-recommended": "^13.0.0",
-        "stylelint-config-standard": "^34.0.0",
-        "stylelint-csstree-validator": "^3.0.0",
-        "stylelint-prettier": "^4.0.0",
-        "typescript": "~5.8.0",
-        "yjs": "^13.5.0"
-    },
-    "sideEffects": [
-        "style/*.css",
-        "style/index.js"
-    ],
-    "styleModule": "style/index.js",
-    "publishConfig": {
-        "access": "public"
-    },
-    "jupyterlab": {
-        "extension": true,
-        "outputDir": "jupyterlite_ai/labextension",
-        "schemaDir": "schema",
-        "sharedPackages": {
-            "@jupyter/chat": {
-                "bundled": true,
-                "singleton": true
-            },
-            "jupyter-chat-components": {
-                "bundled": false,
-                "singleton": true
-            }
-        }
-    },
-    "eslintIgnore": [
-        "node_modules",
-        "dist",
-        "coverage",
-        "**/*.d.ts",
-        "tests",
-        "**/__tests__",
-        "ui-tests",
-        ".venv",
-        "webpack.config.js"
-    ],
-    "eslintConfig": {
-        "extends": [
-            "eslint:recommended",
-            "plugin:@typescript-eslint/eslint-recommended",
-            "plugin:@typescript-eslint/recommended",
-            "plugin:prettier/recommended"
-        ],
-        "parser": "@typescript-eslint/parser",
-        "parserOptions": {
-            "project": "tsconfig.json",
-            "sourceType": "module"
-        },
-        "plugins": [
-            "@typescript-eslint"
-        ],
-        "rules": {
-            "@typescript-eslint/naming-convention": [
-                "error",
-                {
-                    "selector": "interface",
-                    "format": [
-                        "PascalCase"
-                    ],
-                    "custom": {
-                        "regex": "^I[A-Z]",
-                        "match": true
-                    }
-                }
-            ],
-            "@typescript-eslint/no-unused-vars": [
-                "warn",
-                {
-                    "args": "none"
-                }
-            ],
-            "@typescript-eslint/no-explicit-any": "off",
-            "@typescript-eslint/no-namespace": "off",
-            "@typescript-eslint/no-use-before-define": "off",
-            "@typescript-eslint/quotes": [
-                "error",
-                "single",
-                {
-                    "avoidEscape": true,
-                    "allowTemplateLiterals": false
-                }
-            ],
-            "curly": [
-                "error",
-                "all"
-            ],
-            "eqeqeq": "error",
-            "prefer-arrow-callback": "error",
-            "no-restricted-imports": [
-                "error",
-                {
-                    "paths": [
-                        {
-                            "name": "@mui/icons-material",
-                            "message": "Please import icons using path imports, e.g. `import AddIcon from '@mui/icons-material/Add'`"
-                        }
-                    ],
-                    "patterns": [
-                        {
-                            "group": [
-                                "@mui/*/*/*"
-                            ],
-                            "message": "3rd level imports in mui are considered private"
-                        }
-                    ]
-                }
-            ]
-        }
-    },
-    "prettier": {
-        "singleQuote": true,
-        "trailingComma": "none",
-        "arrowParens": "avoid",
-        "endOfLine": "auto",
-        "overrides": [
-            {
-                "files": "package.json",
-                "options": {
-                    "tabWidth": 4
-                }
-            }
-        ]
-    },
-    "stylelint": {
-        "extends": [
-            "stylelint-config-recommended",
-            "stylelint-config-standard",
-            "stylelint-prettier/recommended"
-        ],
-        "plugins": [
-            "stylelint-csstree-validator"
-        ],
-        "rules": {
-            "csstree/validator": true,
-            "property-no-vendor-prefix": null,
-            "selector-class-pattern": "^([a-z][A-z\\d]*)(-[A-z\\d]+)*$|^Mui[A-Z]",
-            "selector-no-vendor-prefix": null,
-            "value-no-vendor-prefix": null
-        }
+  "name": "@jupyterlite/ai",
+  "version": "0.19.0",
+  "description": "AI code completions and chat for JupyterLite",
+  "keywords": [
+    "jupyter",
+    "jupyterlab",
+    "jupyterlab-extension"
+  ],
+  "homepage": "https://github.com/jupyterlite/ai",
+  "bugs": {
+    "url": "https://github.com/jupyterlite/ai/issues"
+  },
+  "license": "BSD-3-Clause",
+  "author": "JupyterLite Contributors",
+  "files": [
+    "lib/**/*.{d.ts,eot,gif,html,jpg,js,js.map,json,png,svg,woff2,ttf,md}",
+    "style/**/*.{css,js,eot,gif,html,jpg,json,png,svg,woff2,ttf}",
+    "src/**/*.{ts,tsx}",
+    "schema/*.json"
+  ],
+  "main": "lib/index.js",
+  "types": "lib/index.d.ts",
+  "style": "style/index.css",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/jupyterlite/ai.git",
+    "directory": "packages/ai"
+  },
+  "scripts": {
+    "build": "echo 'Building @jupyterlite/ai' && jlpm build:lib && jlpm build:labextension:dev",
+    "build:dev": "echo 'Building @jupyterlite/ai' && jlpm build:lib && jlpm build:labextension:dev",
+    "build:prod": "echo 'Building @jupyterlite/ai' && jlpm clean && jlpm build:lib:prod && jlpm build:labextension",
+    "build:labextension": "jupyter labextension build .",
+    "build:labextension:dev": "jupyter labextension build --development True .",
+    "build:lib": "tsc --sourceMap",
+    "build:lib:prod": "tsc",
+    "clean": "jlpm clean:lib",
+    "clean:lib": "rimraf lib tsconfig.tsbuildinfo",
+    "install:extension": "jlpm build",
+    "watch": "run-p watch:src watch:labextension",
+    "watch:src": "tsc -w --sourceMap",
+    "watch:labextension": "jupyter labextension watch ."
+  },
+  "dependencies": {
+    "@jupyter/chat": "^0.21.1",
+    "@jupyter/ydoc": "^3.1.0",
+    "@jupyterlab/application": "^4.5.8",
+    "@jupyterlab/apputils": "^4.6.8",
+    "@jupyterlab/completer": "^4.5.8",
+    "@jupyterlab/coreutils": "^6.5.8",
+    "@jupyterlab/docmanager": "^4.5.8",
+    "@jupyterlab/docregistry": "^4.5.8",
+    "@jupyterlab/filebrowser": "^4.5.8",
+    "@jupyterlab/nbformat": "^4.5.8",
+    "@jupyterlab/notebook": "^4.5.8",
+    "@jupyterlab/rendermime": "^4.5.8",
+    "@jupyterlab/services": "^7.5.8",
+    "@jupyterlab/settingregistry": "^4.5.8",
+    "@jupyterlab/statusbar": "^4.5.8",
+    "@jupyterlab/translation": "^4.5.8",
+    "@jupyterlab/ui-components": "^4.5.8",
+    "@jupyternaut/agent": "^0.19.0",
+    "@lumino/commands": "^2.3.2",
+    "@lumino/coreutils": "^2.2.1",
+    "@lumino/disposable": "^2.1.4",
+    "@lumino/polling": "^2.1.4",
+    "@lumino/signaling": "^2.1.4",
+    "@lumino/widgets": "^2.7.1",
+    "@mui/icons-material": "^7",
+    "@mui/material": "^7",
+    "ai": "^6.0.116",
+    "jupyter-chat-components": "^0.6.0",
+    "jupyter-secrets-manager": "^0.5.0",
+    "react": "^18.3.1",
+    "yaml": "^2.8.1"
+  },
+  "devDependencies": {
+    "@jupyterlab/builder": "^4.0.0",
+    "@jupyterlab/testutils": "^4.0.0",
+    "@types/json-schema": "^7.0.11",
+    "@types/node": "^24.3.0",
+    "@types/react": "^18.0.26",
+    "@types/react-addons-linked-state-mixin": "^0.14.22",
+    "css-loader": "^6.7.1",
+    "npm-run-all2": "^7.0.1",
+    "rimraf": "^5.0.1",
+    "source-map-loader": "^1.0.2",
+    "typescript": "~5.8.0"
+  },
+  "sideEffects": [
+    "style/*.css",
+    "style/index.js"
+  ],
+  "styleModule": "style/index.js",
+  "publishConfig": {
+    "access": "public"
+  },
+  "jupyterlab": {
+    "extension": true,
+    "outputDir": "../../jupyterlite_ai/labextension",
+    "schemaDir": "schema",
+    "sharedPackages": {
+      "@jupyter/chat": {
+        "bundled": true,
+        "singleton": true
+      },
+      "jupyter-chat-components": {
+        "bundled": false,
+        "singleton": true
+      }
     }
+  }
 }

package/src/chat-commands/skills.ts

@@ -1,9 +1,9 @@
+import type { ISkillRegistry } from '@jupyternaut/agent';
 import { ChatCommand, IChatCommandProvider, IInputModel } from '@jupyter/chat';
-
 import { CommandRegistry } from '@lumino/commands';
 
 import { AIChatModel } from '../chat-model';
-import { CommandIds, ISkillRegistry } from '../tokens';
+import { CommandIds } from '../tokens';
 
 export class SkillsCommandProvider implements IChatCommandProvider {
   constructor(options: SkillsCommandProvider.IOptions) {

package/src/chat-model-handler.ts

@@ -1,3 +1,8 @@
+import type {
+  IAgentManagerFactory,
+  IProviderRegistry,
+  IToolRegistry
+} from '@jupyternaut/agent';
 import { ActiveCellManager } from '@jupyter/chat';
 import { IDocumentManager } from '@jupyterlab/docmanager';
 import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
@@ -5,13 +10,10 @@ import { Contents } from '@jupyterlab/services';
 
 import { AIChatModel } from './chat-model';
 import type {
-  IAgentManagerFactory,
   IAIChatModel,
   IAISettingsModel,
   IChatModelHandler,
-  ICreateChatOptions,
-  IProviderRegistry,
-  IToolRegistry
+  ICreateChatOptions
 } from './tokens';
 
 /**

package/src/chat-model.ts

@@ -26,6 +26,14 @@ import { IRenderMime } from '@jupyterlab/rendermime';
 
 import { Contents } from '@jupyterlab/services';
 
+import { AI_AVATAR } from '@jupyternaut/agent';
+
+import type {
+  IAgentManager,
+  IProviderRegistry,
+  ITokenUsage
+} from '@jupyternaut/agent';
+
 import { UUID } from '@lumino/coreutils';
 
 import { Debouncer } from '@lumino/polling';
@@ -34,21 +42,13 @@ import { ISignal, Signal } from '@lumino/signaling';
 
 import type { UserContent, ImagePart, FilePart, ModelMessage } from 'ai';
 
-import { AI_AVATAR } from './icons';
-
-import type {
-  IAgentManager,
-  IAIChatModel,
-  IAISettingsModel,
-  IProviderRegistry,
-  ITokenUsage
-} from './tokens';
+import type { IAIChatModel, IAISettingsModel } from './tokens';
 
 import {
   modelSupportsAudio,
   modelSupportsImages,
   modelSupportsPdf
-} from './providers/model-info';
+} from '@jupyternaut/agent';
 
 /**
  * Tool call status types.
@@ -273,7 +273,11 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
       stopStreaming: () => this.stopStreaming(),
       clearMessages: () => this.clearMessages(),
       agentManager: this._agentManager,
-      addSystemMessage: (body: string) => this._addSystemMessage(body)
+      addSystemMessage: (body: string) => this._addSystemMessage(body),
+      removeQueuedMessage: (id: string) => this.removeQueuedMessage(id),
+      reorderQueuedMessages: (ids: string[]) => this.reorderQueuedMessages(ids),
+      editQueuedMessage: (id: string, body: string) =>
+        this.editQueuedMessage(id, body)
     };
   }
 
@@ -438,7 +442,9 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
 
       if (
         this._settingsModel.config.autoTitle &&
-        (this.messages.length <= 5 || this.title === null)
+        (this.messages.filter(msg => msg.sender.username !== 'ai-assistant')
+          .length <= 5 ||
+          this.title === null)
       ) {
         try {
           this.title = await this.requestTitle();
@@ -528,8 +534,32 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
    * @param messageId The ID of the queued message to remove
    */
   removeQueuedMessage(messageId: string): void {
-    this._messageQueue = this._messageQueue.filter(msg => msg.id !== messageId);
-    this._updateQueueUI();
+    this.messageQueue = this._messageQueue.filter(msg => msg.id !== messageId);
+  }
+
+  /**
+   * Reorders queued messages by their IDs.
+   * @param messageIds Array of message IDs in the desired order
+   */
+  reorderQueuedMessages(messageIds: string[]): void {
+    const byId = new Map(this._messageQueue.map(m => [m.id, m]));
+    this.messageQueue = messageIds
+      .map(id => byId.get(id))
+      .filter((m): m is Private.IQueuedItem => m !== undefined);
+  }
+
+  /**
+   * Edits a queued message by its ID.
+   * @param messageId The ID of the queued message to edit
+   * @param newBody The new body of the message
+   */
+  editQueuedMessage(messageId: string, newBody: string): void {
+    const queue = [...this._messageQueue];
+    const index = queue.findIndex(m => m.id === messageId);
+    if (index !== -1) {
+      queue[index] = { ...queue[index], body: newBody };
+      this.messageQueue = queue;
+    }
   }
 
   /**
@@ -600,7 +630,7 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
         );
       }
     } else if (!silent) {
-      console.log(`Provider not providing when restoring ${filepath}.`);
+      console.log(`Provider not provided when restoring ${filepath}.`);
     }
 
     const messages: IMessageContent[] = content.messages.map(message => {
@@ -855,6 +885,7 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
     this.messageAdded(aiMessage);
     this._currentStreamingMessage =
       this.messages.find(message => message.id === aiMessage.id) ?? null;
+    this._updateQueueUI();
   }
 
   /**
@@ -1023,6 +1054,7 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
     };
 
     this.messageAdded(toolCallMessage);
+    this._updateQueueUI();
   }
 
   /**
@@ -1099,6 +1131,7 @@ export class AIChatModel extends AbstractChatModel implements IAIChatModel {
       type: 'msg',
       raw_time: false
     });
+    this._updateQueueUI();
   }
 
   /**
@@ -1241,11 +1274,13 @@ namespace Private {
     | nbformat.IDisplayUpdate
     | nbformat.IExecuteResult;
 
-  const isPlainObject = (value: unknown): value is Record<string, unknown> => {
+  export const isPlainObject = (
+    value: unknown
+  ): value is Record<string, unknown> => {
     return typeof value === 'object' && value !== null && !Array.isArray(value);
   };
 
-  const isDisplayOutput = (value: unknown): value is IDisplayOutput => {
+  export const isDisplayOutput = (value: unknown): value is IDisplayOutput => {
     if (!isPlainObject(value)) {
       return false;
     }
@@ -1258,7 +1293,7 @@ namespace Private {
     );
   };
 
-  const toMimeBundle = (
+  export const toMimeBundle = (
     value: IDisplayOutput,
     trustedMimeTypes: ReadonlySet<string>
   ): IMimeModelBody | null => {
@@ -1286,7 +1321,7 @@ namespace Private {
    * Tool outputs are not guaranteed to be raw Jupyter IOPub messages; they are
    * often wrapped objects (for example `{ success, result: { outputs: [...] } }`).
    */
-  const toDisplayOutputs = (value: unknown): IDisplayOutput[] => {
+  export const toDisplayOutputs = (value: unknown): IDisplayOutput[] => {
     if (isDisplayOutput(value)) {
       return [value];
     }
@@ -1833,6 +1868,18 @@ export namespace AIChatModel {
      * The agent manager of the chat.
      */
     agentManager: IAgentManager;
+    /**
+     * Removes a queued message by its ID.
+     */
+    removeQueuedMessage: (id: string) => void;
+    /**
+     * Reorders queued messages by their IDs.
+     */
+    reorderQueuedMessages: (ids: string[]) => void;
+    /**
+     * Edits a queued message by its ID.
+     */
+    editQueuedMessage: (id: string, body: string) => void;
   }
 
   /**

package/src/completion/completion-provider.ts

@@ -1,3 +1,8 @@
+import {
+  createCompletionModel,
+  IProviderRegistry,
+  SECRETS_NAMESPACE
+} from '@jupyternaut/agent';
 import {
   CompletionHandler,
   IInlineCompletionContext,
@@ -8,12 +13,7 @@ import { NotebookPanel } from '@jupyterlab/notebook';
 import { generateText, type LanguageModel } from 'ai';
 import { ISecretsManager } from 'jupyter-secrets-manager';
 
-import { createCompletionModel } from '../providers/models';
-import {
-  type IAISettingsModel,
-  SECRETS_NAMESPACE,
-  type IProviderRegistry
-} from '../tokens';
+import type { IAISettingsModel } from '../tokens';
 
 /**
  * Configuration interface for provider-specific completion behavior

package/src/components/clear-button.tsx

@@ -4,8 +4,6 @@ import type { TranslationBundle } from '@jupyterlab/translation';
 
 import ClearIcon from '@mui/icons-material/Clear';
 
-import React from 'react';
-
 import { AIChatModel } from '../chat-model';
 
 /**

package/src/components/completion-status.tsx

@@ -1,7 +1,7 @@
-import React, { useEffect, useState } from 'react';
+import { useEffect, useState } from 'react';
 import { ReactWidget } from '@jupyterlab/ui-components';
 import type { TranslationBundle } from '@jupyterlab/translation';
-import { jupyternautIcon } from '../icons';
+import { jupyternautIcon } from '@jupyternaut/agent';
 import type { IAISettingsModel } from '../tokens';
 
 const COMPLETION_STATUS_CLASS = 'jp-ai-completion-status';

package/src/components/model-select.tsx

@@ -2,7 +2,7 @@ import { InputToolbarRegistry, TooltippedButton } from '@jupyter/chat';
 import type { TranslationBundle } from '@jupyterlab/translation';
 import CheckIcon from '@mui/icons-material/Check';
 import { Menu, MenuItem, Typography } from '@mui/material';
-import React, { useCallback, useEffect, useState } from 'react';
+import { useCallback, useEffect, useState } from 'react';
 import { AIChatModel } from '../chat-model';
 import type { IAISettingsModel } from '../tokens';
 

package/src/components/stop-button.tsx

@@ -4,8 +4,6 @@ import type { TranslationBundle } from '@jupyterlab/translation';
 
 import StopIcon from '@mui/icons-material/Stop';
 
-import React from 'react';
-
 import { AIChatModel } from '../chat-model';
 
 /**