@notebook-intelligence/notebook-intelligence 2.3.1 → 2.4.0

package/README.md

@@ -62,7 +62,7 @@ for the frontend extension.
 Notebook Intelligence can remember your GitHub Copilot login so that you don't need to re-login after a JupyterLab or system restart. Please be aware of the security implications of using this feature.
 
 > [!CAUTION]
-> If you configure NBI to remember your GitHub Copilot login, it will encrypt the token and store into a data file at `~/.jupyter/nbi-data.json`. You should never share this file with others as they can access your tokens.
+> If you configure NBI to remember your GitHub Copilot login, it will encrypt the token and store into a data file at `~/.jupyter/nbi/user-data.json`. You should never share this file with others as they can access your tokens.
 > Even though the token is encrypted, it is done so by using a default password and that's why it can be decrypted by others. In order to prevent that you can specify a custom password using the environment variable `NBI_GH_ACCESS_TOKEN_PASSWORD`.
 
 ```bash

package/lib/api.d.ts

@@ -12,10 +12,11 @@ export interface IDeviceVerificationInfo {
 }
 export declare class NBIConfig {
     get userHomeDir(): string;
-    get configFilePath(): string;
+    get userConfigDir(): string;
     get llmProviders(): [any];
     get chatModels(): [any];
     get inlineCompletionModels(): [any];
+    get defaultChatMode(): string;
     get chatModel(): any;
     get inlineCompletionModel(): any;
     get usingGitHubCopilotModel(): boolean;
@@ -32,6 +33,7 @@ export declare class NBIAPI {
     static _messageReceived: Signal<unknown, any>;
     static config: NBIConfig;
     static configChanged: Signal<NBIConfig, void>;
+    static githubLoginStatusChanged: Signal<unknown, void>;
     static initialize(): Promise<void>;
     static initializeWebsocket(): Promise<void>;
     static getLoginStatus(): GitHubCopilotLoginStatus;
@@ -43,6 +45,8 @@ export declare class NBIAPI {
     static setConfig(config: any): Promise<void>;
     static updateOllamaModelList(): Promise<void>;
     static reloadMCPServerList(): Promise<any>;
+    static getMCPConfigFile(): Promise<any>;
+    static setMCPConfigFile(config: any): Promise<any>;
     static chatRequest(messageId: string, chatId: string, prompt: string, language: string, filename: string, additionalContext: IContextItem[], chatMode: string, toolSelections: IToolSelections, responseEmitter: IChatCompletionResponseEmitter): Promise<void>;
     static generateCode(chatId: string, prompt: string, prefix: string, suffix: string, existingCode: string, language: string, filename: string, responseEmitter: IChatCompletionResponseEmitter): Promise<void>;
     static sendChatUserInput(messageId: string, data: any): Promise<void>;

package/lib/api.js

@@ -5,8 +5,7 @@ import { requestAPI } from './handler';
 import { URLExt } from '@jupyterlab/coreutils';
 import { UUID } from '@lumino/coreutils';
 import { Signal } from '@lumino/signaling';
-import { GITHUB_COPILOT_PROVIDER_ID, RequestDataType } from './tokens';
-const LOGIN_STATUS_UPDATE_INTERVAL = 2000;
+import { GITHUB_COPILOT_PROVIDER_ID, RequestDataType, BackendMessageType } from './tokens';
 export var GitHubCopilotLoginStatus;
 (function (GitHubCopilotLoginStatus) {
     GitHubCopilotLoginStatus["NotLoggedIn"] = "NOT_LOGGED_IN";
@@ -23,8 +22,8 @@ export class NBIConfig {
     get userHomeDir() {
         return this.capabilities.user_home_dir;
     }
-    get configFilePath() {
-        return this.capabilities.config_file_path;
+    get userConfigDir() {
+        return this.capabilities.nbi_user_config_dir;
     }
     get llmProviders() {
         return this.capabilities.llm_providers;
@@ -35,6 +34,9 @@ export class NBIConfig {
     get inlineCompletionModels() {
         return this.capabilities.inline_completion_models;
     }
+    get defaultChatMode() {
+        return this.capabilities.default_chat_mode;
+    }
     get chatModel() {
         return this.capabilities.chat_model;
     }
@@ -56,10 +58,15 @@ class NBIAPI {
     static async initialize() {
         await this.fetchCapabilities();
         this.updateGitHubLoginStatus();
-        setInterval(() => {
-            this.updateGitHubLoginStatus();
-        }, LOGIN_STATUS_UPDATE_INTERVAL);
         NBIAPI.initializeWebsocket();
+        this._messageReceived.connect((_, msg) => {
+            msg = JSON.parse(msg);
+            if (msg.type === BackendMessageType.GitHubCopilotLoginStatusChange) {
+                this.updateGitHubLoginStatus().then(() => {
+                    this.githubLoginStatusChanged.emit();
+                });
+            }
+        });
     }
     static async initializeWebsocket() {
         const serverSettings = ServerConnection.makeSettings();
@@ -189,6 +196,33 @@ class NBIAPI {
             });
         });
     }
+    static async getMCPConfigFile() {
+        return new Promise((resolve, reject) => {
+            requestAPI('mcp-config-file', { method: 'GET' })
+                .then(async (data) => {
+                resolve(data);
+            })
+                .catch(reason => {
+                console.error(`Failed to get MCP config file.\n${reason}`);
+                reject(reason);
+            });
+        });
+    }
+    static async setMCPConfigFile(config) {
+        return new Promise((resolve, reject) => {
+            requestAPI('mcp-config-file', {
+                method: 'POST',
+                body: JSON.stringify(config)
+            })
+                .then(async (data) => {
+                resolve(data);
+            })
+                .catch(reason => {
+                console.error(`Failed to set MCP config file.\n${reason}`);
+                reject(reason);
+            });
+        });
+    }
     static async chatRequest(messageId, chatId, prompt, language, filename, additionalContext, chatMode, toolSelections, responseEmitter) {
         this._messageReceived.connect((_, msg) => {
             msg = JSON.parse(msg);
@@ -286,4 +320,5 @@ NBIAPI._deviceVerificationInfo = {
 NBIAPI._messageReceived = new Signal(_a);
 NBIAPI.config = new NBIConfig();
 NBIAPI.configChanged = _a.config.changed;
+NBIAPI.githubLoginStatusChanged = new Signal(_a);
 export { NBIAPI };

package/lib/chat-sidebar.d.ts

@@ -76,7 +76,9 @@ export declare class GitHubCopilotLoginDialogBody extends ReactWidget {
 export declare class ConfigurationDialogBody extends ReactWidget {
     constructor(options: {
         onSave: () => void;
+        onEditMCPConfigClicked: () => void;
     });
     render(): JSX.Element;
+    private _onEditMCPConfigClicked;
     private _onSave;
 }

package/lib/chat-sidebar.js

@@ -13,6 +13,7 @@ import copilotWarningSvgstr from '../style/icons/copilot-warning.svg';
 import { VscSend, VscStopCircle, VscEye, VscEyeClosed, VscTriangleRight, VscTriangleDown, VscWarning, VscSettingsGear, VscPassFilled, VscTools, VscTrash } from 'react-icons/vsc';
 import { MdOutlineCheckBoxOutlineBlank, MdCheckBox } from 'react-icons/md';
 import { extractLLMGeneratedCode, isDarkTheme } from './utils';
+import * as path from 'path';
 const OPENAI_COMPATIBLE_CHAT_MODEL_ID = 'openai-compatible-chat-model';
 const LITELLM_COMPATIBLE_CHAT_MODEL_ID = 'litellm-compatible-chat-model';
 const OPENAI_COMPATIBLE_INLINE_COMPLETION_MODEL_ID = 'openai-compatible-inline-completion-model';
@@ -130,10 +131,11 @@ export class GitHubCopilotLoginDialogBody extends ReactWidget {
 export class ConfigurationDialogBody extends ReactWidget {
     constructor(options) {
         super();
+        this._onEditMCPConfigClicked = options.onEditMCPConfigClicked;
         this._onSave = options.onSave;
     }
     render() {
-        return React.createElement(ConfigurationDialogBodyComponent, { onSave: this._onSave });
+        return (React.createElement(ConfigurationDialogBodyComponent, { onEditMCPConfigClicked: this._onEditMCPConfigClicked, onSave: this._onSave }));
     }
 }
 const answeredForms = new Map();
@@ -358,7 +360,7 @@ function SidebarComponent(props) {
     const [activeDocumentInfo, setActiveDocumentInfo] = useState(null);
     const [currentFileContextTitle, setCurrentFileContextTitle] = useState('');
     const telemetryEmitter = props.getTelemetryEmitter();
-    const [chatMode, setChatMode] = useState('ask');
+    const [chatMode, setChatMode] = useState(NBIAPI.config.defaultChatMode);
     const [toolSelectionTitle, setToolSelectionTitle] = useState('Tool selection');
     const [selectedToolCount, setSelectedToolCount] = useState(0);
     const [notebookExecuteToolSelected, setNotebookExecuteToolSelected] = useState(false);
@@ -1508,6 +1510,7 @@ function ConfigurationDialogBodyComponent(props) {
     const [mcpServerNames, setMcpServerNames] = useState(((_a = nbiConfig.toolConfig.mcpServers) === null || _a === void 0 ? void 0 : _a.map((server) => server.id)) || []);
     const handleSaveClick = async () => {
         const config = {
+            default_chat_mode: defaultChatMode,
             chat_model: {
                 provider: chatModelProvider,
                 model: chatModel,
@@ -1532,6 +1535,7 @@ function ConfigurationDialogBodyComponent(props) {
     };
     const [chatModelProvider, setChatModelProvider] = useState(nbiConfig.chatModel.provider || 'none');
     const [inlineCompletionModelProvider, setInlineCompletionModelProvider] = useState(nbiConfig.inlineCompletionModel.provider || 'none');
+    const [defaultChatMode, setDefaultChatMode] = useState(nbiConfig.defaultChatMode);
     const [chatModel, setChatModel] = useState(nbiConfig.chatModel.model);
     const [chatModelProperties, setChatModelProperties] = useState([]);
     const [inlineCompletionModelProperties, setInlineCompletionModelProperties] = useState([]);
@@ -1608,6 +1612,16 @@ function ConfigurationDialogBodyComponent(props) {
     }, []);
     return (React.createElement("div", { className: "config-dialog" },
         React.createElement("div", { className: "config-dialog-body" },
+            React.createElement("div", { className: "model-config-section" },
+                React.createElement("div", { className: "model-config-section-header" }, "Default chat mode"),
+                React.createElement("div", { className: "model-config-section-body" },
+                    React.createElement("div", { className: "model-config-section-row" },
+                        React.createElement("div", { className: "model-config-section-column" },
+                            React.createElement("div", null,
+                                React.createElement("select", { className: "jp-mod-styled", value: defaultChatMode, onChange: event => setDefaultChatMode(event.target.value) },
+                                    React.createElement("option", { value: "ask" }, "Ask"),
+                                    React.createElement("option", { value: "agent" }, "Agent")))),
+                        React.createElement("div", { className: "model-config-section-column" }, " ")))),
             React.createElement("div", { className: "model-config-section" },
                 React.createElement("div", { className: "model-config-section-header" }, "Chat model"),
                 React.createElement("div", { className: "model-config-section-body" },
@@ -1689,8 +1703,10 @@ function ConfigurationDialogBodyComponent(props) {
                                 "Remember my GitHub Copilot access token")))))),
             React.createElement("div", { className: "model-config-section" },
                 React.createElement("div", { className: "model-config-section-header" },
-                    "MCP Servers [",
+                    "MCP Servers (",
                     mcpServerNames.length,
+                    ") [",
+                    React.createElement("a", { href: "javascript:void(0)", onClick: props.onEditMCPConfigClicked }, "edit"),
                     "]"),
                 React.createElement("div", { className: "model-config-section-body" },
                     React.createElement("div", { className: "model-config-section-row" },
@@ -1706,10 +1722,21 @@ function ConfigurationDialogBodyComponent(props) {
                     React.createElement("div", { className: "model-config-section-row" },
                         React.createElement("div", { className: "model-config-section-column" },
                             React.createElement("span", { className: "user-code-span", onClick: () => {
-                                    navigator.clipboard.writeText(NBIAPI.config.configFilePath);
+                                    navigator.clipboard.writeText(path.join(NBIAPI.config.userConfigDir, 'config.json'));
+                                    return true;
+                                } },
+                                path.join(NBIAPI.config.userConfigDir, 'config.json'),
+                                ' ',
+                                React.createElement("span", { className: "copy-icon", dangerouslySetInnerHTML: { __html: copySvgstr } }))))),
+                React.createElement("div", { className: "model-config-section-header" }, "MCP config file path"),
+                React.createElement("div", { className: "model-config-section-body" },
+                    React.createElement("div", { className: "model-config-section-row" },
+                        React.createElement("div", { className: "model-config-section-column" },
+                            React.createElement("span", { className: "user-code-span", onClick: () => {
+                                    navigator.clipboard.writeText(path.join(NBIAPI.config.userConfigDir, 'mcp.json'));
                                     return true;
                                 } },
-                                NBIAPI.config.configFilePath,
+                                path.join(NBIAPI.config.userConfigDir, 'mcp.json'),
                                 ' ',
                                 React.createElement("span", { className: "copy-icon", dangerouslySetInnerHTML: { __html: copySvgstr } }))))))),
         React.createElement("div", { className: "config-dialog-footer" },

package/lib/index.js

@@ -21,6 +21,7 @@ import sparklesSvgstr from '../style/icons/sparkles.svg';
 import copilotSvgstr from '../style/icons/copilot.svg';
 import { applyCodeToSelectionInEditor, cellOutputAsText, compareSelections, extractLLMGeneratedCode, getSelectionInEditor, getTokenCount, getWholeNotebookContent, isSelectionEmpty, markdownToComment, waitForDuration } from './utils';
 import { UUID } from '@lumino/coreutils';
+import * as path from 'path';
 var CommandIDs;
 (function (CommandIDs) {
     CommandIDs.chatuserInput = 'notebook-intelligence:chat-user-input';
@@ -28,6 +29,7 @@ var CommandIDs;
     CommandIDs.addCodeAsNewCell = 'notebook-intelligence:add-code-as-new-cell';
     CommandIDs.createNewFile = 'notebook-intelligence:create-new-file';
     CommandIDs.createNewNotebookFromPython = 'notebook-intelligence:create-new-notebook-from-py';
+    CommandIDs.renameNotebook = 'notebook-intelligence:rename-notebook';
     CommandIDs.addCodeCellToNotebook = 'notebook-intelligence:add-code-cell-to-notebook';
     CommandIDs.addMarkdownCellToNotebook = 'notebook-intelligence:add-markdown-cell-to-notebook';
     CommandIDs.editorGenerateCode = 'notebook-intelligence:editor-generate-code';
@@ -48,6 +50,7 @@ var CommandIDs;
     CommandIDs.runCellAtIndex = 'notebook-intelligence:run-cell-at-index';
     CommandIDs.getCurrentFileContent = 'notebook-intelligence:get-current-file-content';
     CommandIDs.setCurrentFileContent = 'notebook-intelligence:set-current-file-content';
+    CommandIDs.openMCPConfigEditor = 'notebook-intelligence:open-mcp-config-editor';
 })(CommandIDs || (CommandIDs = {}));
 const DOCUMENT_WATCH_INTERVAL = 1000;
 const MAX_TOKENS = 4096;
@@ -354,6 +357,67 @@ class TelemetryEmitter {
         });
     }
 }
+class MCPConfigEditor {
+    constructor(docManager) {
+        this._docWidget = null;
+        this._tmpMCPConfigFilename = 'nbi.mcp.temp.json';
+        this._isOpen = false;
+        this._docManager = docManager;
+    }
+    async open() {
+        const contents = new ContentsManager();
+        const newJSONFile = await contents.newUntitled({
+            ext: '.json'
+        });
+        const mcpConfig = await NBIAPI.getMCPConfigFile();
+        try {
+            await contents.delete(this._tmpMCPConfigFilename);
+        }
+        catch (error) {
+            // ignore
+        }
+        await contents.save(newJSONFile.path, {
+            content: JSON.stringify(mcpConfig, null, 2),
+            format: 'text',
+            type: 'file'
+        });
+        await contents.rename(newJSONFile.path, this._tmpMCPConfigFilename);
+        this._docWidget = this._docManager.openOrReveal(this._tmpMCPConfigFilename, 'Editor');
+        this._addListeners();
+        // tab closed
+        this._docWidget.disposed.connect((_, args) => {
+            this._removeListeners();
+            contents.delete(this._tmpMCPConfigFilename);
+        });
+        this._isOpen = true;
+    }
+    close() {
+        if (!this._isOpen) {
+            return;
+        }
+        this._isOpen = false;
+        this._docWidget.dispose();
+        this._docWidget = null;
+    }
+    get isOpen() {
+        return this._isOpen;
+    }
+    _addListeners() {
+        this._docWidget.context.model.stateChanged.connect(this._onStateChanged, this);
+    }
+    _removeListeners() {
+        this._docWidget.context.model.stateChanged.disconnect(this._onStateChanged, this);
+    }
+    _onStateChanged(model, args) {
+        if (args.name === 'dirty' && args.newValue === false) {
+            this._onSave();
+        }
+    }
+    async _onSave() {
+        const mcpConfig = this._docWidget.context.model.toJSON();
+        await NBIAPI.setMCPConfigFile(mcpConfig);
+    }
+}
 /**
  * Initialization data for the @notebook-intelligence/notebook-intelligence extension.
  */
@@ -390,6 +454,7 @@ const plugin = {
         };
         await NBIAPI.initialize();
         let openPopover = null;
+        let mcpConfigEditor = null;
         completionManager.registerInlineProvider(new NBIInlineCompletionProvider(telemetryEmitter));
         if (settingRegistry) {
             settingRegistry
@@ -577,6 +642,32 @@ const plugin = {
                 return newNBFile;
             }
         });
+        app.commands.addCommand(CommandIDs.renameNotebook, {
+            execute: async (args) => {
+                const activeWidget = app.shell.currentWidget;
+                if (activeWidget instanceof NotebookPanel) {
+                    const oldPath = activeWidget.context.path;
+                    const oldParentPath = path.dirname(oldPath);
+                    let newPath = path.join(oldParentPath, args.newName);
+                    if (path.extname(newPath) !== '.ipynb') {
+                        newPath += '.ipynb';
+                    }
+                    if (path.dirname(newPath) !== oldParentPath) {
+                        return 'Failed to rename notebook. New path is outside the old parent directory';
+                    }
+                    try {
+                        await app.serviceManager.contents.rename(oldPath, newPath);
+                        return 'Successfully renamed notebook';
+                    }
+                    catch (error) {
+                        return `Failed to rename notebook: ${error}`;
+                    }
+                }
+                else {
+                    return 'Cannot rename non notebook files';
+                }
+            }
+        });
         const isNewEmptyNotebook = (model) => {
             return (model.cells.length === 1 &&
                 model.cells[0].cell_type === 'code' &&
@@ -838,6 +929,10 @@ const plugin = {
                     onSave: () => {
                         dialog === null || dialog === void 0 ? void 0 : dialog.dispose();
                         NBIAPI.fetchCapabilities();
+                    },
+                    onEditMCPConfigClicked: () => {
+                        dialog === null || dialog === void 0 ? void 0 : dialog.dispose();
+                        app.commands.execute('notebook-intelligence:open-mcp-config-editor');
                     }
                 });
                 dialog = new Dialog({
@@ -850,6 +945,16 @@ const plugin = {
                 dialog.launch();
             }
         });
+        app.commands.addCommand(CommandIDs.openMCPConfigEditor, {
+            label: 'Open MCP Config Editor',
+            execute: args => {
+                if (mcpConfigEditor && mcpConfigEditor.isOpen) {
+                    mcpConfigEditor.close();
+                }
+                mcpConfigEditor = new MCPConfigEditor(docManager);
+                mcpConfigEditor.open();
+            }
+        });
         palette.addItem({
             command: CommandIDs.openConfigurationDialog,
             category: 'Notebook Intelligence'

package/lib/tokens.d.ts

@@ -25,7 +25,8 @@ export declare enum RequestDataType {
 export declare enum BackendMessageType {
     StreamMessage = "stream-message",
     StreamEnd = "stream-end",
-    RunUICommand = "run-ui-command"
+    RunUICommand = "run-ui-command",
+    GitHubCopilotLoginStatusChange = "github-copilot-login-status-change"
 }
 export declare enum ResponseStreamDataType {
     LLMRaw = "llm-raw",

package/lib/tokens.js

@@ -16,6 +16,7 @@ export var BackendMessageType;
     BackendMessageType["StreamMessage"] = "stream-message";
     BackendMessageType["StreamEnd"] = "stream-end";
     BackendMessageType["RunUICommand"] = "run-ui-command";
+    BackendMessageType["GitHubCopilotLoginStatusChange"] = "github-copilot-login-status-change";
 })(BackendMessageType || (BackendMessageType = {}));
 export var ResponseStreamDataType;
 (function (ResponseStreamDataType) {

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@notebook-intelligence/notebook-intelligence",
-    "version": "2.3.1",
+    "version": "2.4.0",
     "description": "AI coding assistant for JupyterLab",
     "keywords": [
         "AI",

package/src/api.ts

@@ -12,11 +12,10 @@ import {
   IContextItem,
   ITelemetryEvent,
   IToolSelections,
-  RequestDataType
+  RequestDataType,
+  BackendMessageType
 } from './tokens';
 
-const LOGIN_STATUS_UPDATE_INTERVAL = 2000;
-
 export enum GitHubCopilotLoginStatus {
   NotLoggedIn = 'NOT_LOGGED_IN',
   ActivatingDevice = 'ACTIVATING_DEVICE',
@@ -34,8 +33,8 @@ export class NBIConfig {
     return this.capabilities.user_home_dir;
   }
 
-  get configFilePath(): string {
-    return this.capabilities.config_file_path;
+  get userConfigDir(): string {
+    return this.capabilities.nbi_user_config_dir;
   }
 
   get llmProviders(): [any] {
@@ -50,6 +49,10 @@ export class NBIConfig {
     return this.capabilities.inline_completion_models;
   }
 
+  get defaultChatMode(): string {
+    return this.capabilities.default_chat_mode;
+  }
+
   get chatModel(): any {
     return this.capabilities.chat_model;
   }
@@ -89,16 +92,22 @@ export class NBIAPI {
   static _messageReceived = new Signal<unknown, any>(this);
   static config = new NBIConfig();
   static configChanged = this.config.changed;
+  static githubLoginStatusChanged = new Signal<unknown, void>(this);
 
   static async initialize() {
     await this.fetchCapabilities();
     this.updateGitHubLoginStatus();
 
-    setInterval(() => {
-      this.updateGitHubLoginStatus();
-    }, LOGIN_STATUS_UPDATE_INTERVAL);
-
     NBIAPI.initializeWebsocket();
+
+    this._messageReceived.connect((_, msg) => {
+      msg = JSON.parse(msg);
+      if (msg.type === BackendMessageType.GitHubCopilotLoginStatusChange) {
+        this.updateGitHubLoginStatus().then(() => {
+          this.githubLoginStatusChanged.emit();
+        });
+      }
+    });
   }
 
   static async initializeWebsocket() {
@@ -250,6 +259,35 @@ export class NBIAPI {
     });
   }
 
+  static async getMCPConfigFile(): Promise<any> {
+    return new Promise<any>((resolve, reject) => {
+      requestAPI<any>('mcp-config-file', { method: 'GET' })
+        .then(async data => {
+          resolve(data);
+        })
+        .catch(reason => {
+          console.error(`Failed to get MCP config file.\n${reason}`);
+          reject(reason);
+        });
+    });
+  }
+
+  static async setMCPConfigFile(config: any): Promise<any> {
+    return new Promise<any>((resolve, reject) => {
+      requestAPI<any>('mcp-config-file', {
+        method: 'POST',
+        body: JSON.stringify(config)
+      })
+        .then(async data => {
+          resolve(data);
+        })
+        .catch(reason => {
+          console.error(`Failed to set MCP config file.\n${reason}`);
+          reject(reason);
+        });
+    });
+  }
+
   static async chatRequest(
     messageId: string,
     chatId: string,

package/src/chat-sidebar.tsx

@@ -56,6 +56,7 @@ import {
 import { MdOutlineCheckBoxOutlineBlank, MdCheckBox } from 'react-icons/md';
 
 import { extractLLMGeneratedCode, isDarkTheme } from './utils';
+import * as path from 'path';
 
 const OPENAI_COMPATIBLE_CHAT_MODEL_ID = 'openai-compatible-chat-model';
 const LITELLM_COMPATIBLE_CHAT_MODEL_ID = 'litellm-compatible-chat-model';
@@ -264,16 +265,26 @@ export class GitHubCopilotLoginDialogBody extends ReactWidget {
 }
 
 export class ConfigurationDialogBody extends ReactWidget {
-  constructor(options: { onSave: () => void }) {
+  constructor(options: {
+    onSave: () => void;
+    onEditMCPConfigClicked: () => void;
+  }) {
     super();
 
+    this._onEditMCPConfigClicked = options.onEditMCPConfigClicked;
     this._onSave = options.onSave;
   }
 
   render(): JSX.Element {
-    return <ConfigurationDialogBodyComponent onSave={this._onSave} />;
+    return (
+      <ConfigurationDialogBodyComponent
+        onEditMCPConfigClicked={this._onEditMCPConfigClicked}
+        onSave={this._onSave}
+      />
+    );
   }
 
+  private _onEditMCPConfigClicked: () => void;
   private _onSave: () => void;
 }
 
@@ -725,7 +736,8 @@ function SidebarComponent(props: any) {
     useState<IActiveDocumentInfo | null>(null);
   const [currentFileContextTitle, setCurrentFileContextTitle] = useState('');
   const telemetryEmitter: ITelemetryEmitter = props.getTelemetryEmitter();
-  const [chatMode, setChatMode] = useState('ask');
+  const [chatMode, setChatMode] = useState(NBIAPI.config.defaultChatMode);
+
   const [toolSelectionTitle, setToolSelectionTitle] =
     useState('Tool selection');
   const [selectedToolCount, setSelectedToolCount] = useState(0);
@@ -2497,6 +2509,7 @@ function ConfigurationDialogBodyComponent(props: any) {
 
   const handleSaveClick = async () => {
     const config: any = {
+      default_chat_mode: defaultChatMode,
       chat_model: {
         provider: chatModelProvider,
         model: chatModel,
@@ -2531,6 +2544,9 @@ function ConfigurationDialogBodyComponent(props: any) {
   );
   const [inlineCompletionModelProvider, setInlineCompletionModelProvider] =
     useState(nbiConfig.inlineCompletionModel.provider || 'none');
+  const [defaultChatMode, setDefaultChatMode] = useState<string>(
+    nbiConfig.defaultChatMode
+  );
   const [chatModel, setChatModel] = useState<string>(nbiConfig.chatModel.model);
   const [chatModelProperties, setChatModelProperties] = useState<any[]>([]);
   const [inlineCompletionModelProperties, setInlineCompletionModelProperties] =
@@ -2629,6 +2645,27 @@ function ConfigurationDialogBodyComponent(props: any) {
   return (
     <div className="config-dialog">
       <div className="config-dialog-body">
+        <div className="model-config-section">
+          <div className="model-config-section-header">Default chat mode</div>
+          <div className="model-config-section-body">
+            <div className="model-config-section-row">
+              <div className="model-config-section-column">
+                <div>
+                  <select
+                    className="jp-mod-styled"
+                    value={defaultChatMode}
+                    onChange={event => setDefaultChatMode(event.target.value)}
+                  >
+                    <option value="ask">Ask</option>
+                    <option value="agent">Agent</option>
+                  </select>
+                </div>
+              </div>
+              <div className="model-config-section-column"> </div>
+            </div>
+          </div>
+        </div>
+
         <div className="model-config-section">
           <div className="model-config-section-header">Chat model</div>
           <div className="model-config-section-body">
@@ -2889,7 +2926,11 @@ function ConfigurationDialogBodyComponent(props: any) {
 
         <div className="model-config-section">
           <div className="model-config-section-header">
-            MCP Servers [{mcpServerNames.length}]
+            MCP Servers ({mcpServerNames.length}) [
+            <a href="javascript:void(0)" onClick={props.onEditMCPConfigClicked}>
+              edit
+            </a>
+            ]
           </div>
           <div className="model-config-section-body">
             <div className="model-config-section-row">
@@ -2927,11 +2968,37 @@ function ConfigurationDialogBodyComponent(props: any) {
                 <span
                   className="user-code-span"
                   onClick={() => {
-                    navigator.clipboard.writeText(NBIAPI.config.configFilePath);
+                    navigator.clipboard.writeText(
+                      path.join(NBIAPI.config.userConfigDir, 'config.json')
+                    );
+                    return true;
+                  }}
+                >
+                  {path.join(NBIAPI.config.userConfigDir, 'config.json')}{' '}
+                  <span
+                    className="copy-icon"
+                    dangerouslySetInnerHTML={{ __html: copySvgstr }}
+                  ></span>
+                </span>
+              </div>
+            </div>
+          </div>
+          <div className="model-config-section-header">
+            MCP config file path
+          </div>
+          <div className="model-config-section-body">
+            <div className="model-config-section-row">
+              <div className="model-config-section-column">
+                <span
+                  className="user-code-span"
+                  onClick={() => {
+                    navigator.clipboard.writeText(
+                      path.join(NBIAPI.config.userConfigDir, 'mcp.json')
+                    );
                     return true;
                   }}
                 >
-                  {NBIAPI.config.configFilePath}{' '}
+                  {path.join(NBIAPI.config.userConfigDir, 'mcp.json')}{' '}
                   <span
                     className="copy-icon"
                     dangerouslySetInnerHTML={{ __html: copySvgstr }}

package/src/index.ts

@@ -7,7 +7,7 @@ import {
 } from '@jupyterlab/application';
 
 import { IDocumentManager } from '@jupyterlab/docmanager';
-import { DocumentWidget } from '@jupyterlab/docregistry';
+import { DocumentWidget, IDocumentWidget } from '@jupyterlab/docregistry';
 
 import { Dialog, ICommandPalette } from '@jupyterlab/apputils';
 import { IMainMenu } from '@jupyterlab/mainmenu';
@@ -80,6 +80,8 @@ import {
 } from './utils';
 import { UUID } from '@lumino/coreutils';
 
+import * as path from 'path';
+
 namespace CommandIDs {
   export const chatuserInput = 'notebook-intelligence:chat-user-input';
   export const insertAtCursor = 'notebook-intelligence:insert-at-cursor';
@@ -87,6 +89,7 @@ namespace CommandIDs {
   export const createNewFile = 'notebook-intelligence:create-new-file';
   export const createNewNotebookFromPython =
     'notebook-intelligence:create-new-notebook-from-py';
+  export const renameNotebook = 'notebook-intelligence:rename-notebook';
   export const addCodeCellToNotebook =
     'notebook-intelligence:add-code-cell-to-notebook';
   export const addMarkdownCellToNotebook =
@@ -121,6 +124,8 @@ namespace CommandIDs {
     'notebook-intelligence:get-current-file-content';
   export const setCurrentFileContent =
     'notebook-intelligence:set-current-file-content';
+  export const openMCPConfigEditor =
+    'notebook-intelligence:open-mcp-config-editor';
 }
 
 const DOCUMENT_WATCH_INTERVAL = 1000;
@@ -511,6 +516,87 @@ class TelemetryEmitter implements ITelemetryEmitter {
   private _listeners: Set<ITelemetryListener> = new Set<ITelemetryListener>();
 }
 
+class MCPConfigEditor {
+  constructor(docManager: IDocumentManager) {
+    this._docManager = docManager;
+  }
+
+  async open() {
+    const contents = new ContentsManager();
+    const newJSONFile = await contents.newUntitled({
+      ext: '.json'
+    });
+    const mcpConfig = await NBIAPI.getMCPConfigFile();
+
+    try {
+      await contents.delete(this._tmpMCPConfigFilename);
+    } catch (error) {
+      // ignore
+    }
+
+    await contents.save(newJSONFile.path, {
+      content: JSON.stringify(mcpConfig, null, 2),
+      format: 'text',
+      type: 'file'
+    });
+    await contents.rename(newJSONFile.path, this._tmpMCPConfigFilename);
+    this._docWidget = this._docManager.openOrReveal(
+      this._tmpMCPConfigFilename,
+      'Editor'
+    );
+    this._addListeners();
+    // tab closed
+    this._docWidget.disposed.connect((_, args) => {
+      this._removeListeners();
+      contents.delete(this._tmpMCPConfigFilename);
+    });
+    this._isOpen = true;
+  }
+
+  close() {
+    if (!this._isOpen) {
+      return;
+    }
+    this._isOpen = false;
+    this._docWidget.dispose();
+    this._docWidget = null;
+  }
+
+  get isOpen(): boolean {
+    return this._isOpen;
+  }
+
+  private _addListeners() {
+    this._docWidget.context.model.stateChanged.connect(
+      this._onStateChanged,
+      this
+    );
+  }
+
+  private _removeListeners() {
+    this._docWidget.context.model.stateChanged.disconnect(
+      this._onStateChanged,
+      this
+    );
+  }
+
+  private _onStateChanged(model: any, args: any) {
+    if (args.name === 'dirty' && args.newValue === false) {
+      this._onSave();
+    }
+  }
+
+  private async _onSave() {
+    const mcpConfig = this._docWidget.context.model.toJSON();
+    await NBIAPI.setMCPConfigFile(mcpConfig);
+  }
+
+  private _docManager: IDocumentManager;
+  private _docWidget: IDocumentWidget = null;
+  private _tmpMCPConfigFilename = 'nbi.mcp.temp.json';
+  private _isOpen = false;
+}
+
 /**
  * Initialization data for the @notebook-intelligence/notebook-intelligence extension.
  */
@@ -564,6 +650,7 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
     await NBIAPI.initialize();
 
     let openPopover: InlinePromptWidget | null = null;
+    let mcpConfigEditor: MCPConfigEditor | null = null;
 
     completionManager.registerInlineProvider(
       new NBIInlineCompletionProvider(telemetryEmitter)
@@ -786,6 +873,33 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
       }
     });
 
+    app.commands.addCommand(CommandIDs.renameNotebook, {
+      execute: async args => {
+        const activeWidget = app.shell.currentWidget;
+        if (activeWidget instanceof NotebookPanel) {
+          const oldPath = activeWidget.context.path;
+          const oldParentPath = path.dirname(oldPath);
+          let newPath = path.join(oldParentPath, args.newName as string);
+          if (path.extname(newPath) !== '.ipynb') {
+            newPath += '.ipynb';
+          }
+
+          if (path.dirname(newPath) !== oldParentPath) {
+            return 'Failed to rename notebook. New path is outside the old parent directory';
+          }
+
+          try {
+            await app.serviceManager.contents.rename(oldPath, newPath);
+            return 'Successfully renamed notebook';
+          } catch (error) {
+            return `Failed to rename notebook: ${error}`;
+          }
+        } else {
+          return 'Cannot rename non notebook files';
+        }
+      }
+    });
+
     const isNewEmptyNotebook = (model: ISharedNotebook) => {
       return (
         model.cells.length === 1 &&
@@ -1122,6 +1236,12 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
           onSave: () => {
             dialog?.dispose();
             NBIAPI.fetchCapabilities();
+          },
+          onEditMCPConfigClicked: () => {
+            dialog?.dispose();
+            app.commands.execute(
+              'notebook-intelligence:open-mcp-config-editor'
+            );
           }
         });
         dialog = new Dialog({
@@ -1136,6 +1256,17 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
       }
     });
 
+    app.commands.addCommand(CommandIDs.openMCPConfigEditor, {
+      label: 'Open MCP Config Editor',
+      execute: args => {
+        if (mcpConfigEditor && mcpConfigEditor.isOpen) {
+          mcpConfigEditor.close();
+        }
+        mcpConfigEditor = new MCPConfigEditor(docManager);
+        mcpConfigEditor.open();
+      }
+    });
+
     palette.addItem({
       command: CommandIDs.openConfigurationDialog,
       category: 'Notebook Intelligence'

package/src/tokens.ts

@@ -31,7 +31,8 @@ export enum RequestDataType {
 export enum BackendMessageType {
   StreamMessage = 'stream-message',
   StreamEnd = 'stream-end',
-  RunUICommand = 'run-ui-command'
+  RunUICommand = 'run-ui-command',
+  GitHubCopilotLoginStatusChange = 'github-copilot-login-status-change'
 }
 
 export enum ResponseStreamDataType {

package/style/base.css

@@ -30,7 +30,7 @@
 }
 
 .sidebar-user-input {
-  height: 80px;
+  height: auto;
   padding: 5px;
   display: flex;
   flex-direction: column;
@@ -67,6 +67,11 @@
   border: none;
   resize: none;
   background: none;
+  /* stylelint-disable */
+  field-sizing: content;
+  /* stylelint-enable */
+  min-height: 36px;
+  max-height: 200px;
 }
 
 .sidebar-user-input .user-input-context-row {