@notebook-intelligence/notebook-intelligence 2.5.0 → 2.6.0

package/README.md

@@ -175,6 +175,119 @@ If you have multiple servers configured but you would like to disable some for a
 }
 ```
 
+### Ruleset System
+
+NBI includes a powerful ruleset system that allows you to define custom guidelines and best practices that are automatically injected into AI prompts. This helps ensure consistent coding standards, project-specific conventions, and domain knowledge across all AI interactions.
+
+#### How It Works
+
+Rules are markdown files with optional YAML frontmatter stored in `~/.jupyter/nbi/rules/`. They are automatically discovered and applied based on context (file type, notebook kernel, chat mode).
+
+#### Creating Rules
+
+**Global Rules** - Apply to all contexts:
+
+Create a file like `~/.jupyter/nbi/rules/01-coding-standards.md`:
+
+```markdown
+---
+priority: 10
+---
+
+# Coding Standards
+
+- Always use type hints in Python functions
+- Prefer list comprehensions over loops when appropriate
+- Add docstrings to all public functions
+```
+
+**Mode-Specific Rules** - Apply only to specific chat modes:
+
+NBI supports mode-specific rules for three modes:
+
+- **ask** - Question/answer mode
+- **agent** - Autonomous agent mode with tool access
+- **inline-chat** - Inline code generation and editing
+
+Create a file like `~/.jupyter/nbi/rules/modes/agent/01-testing.md`:
+
+```markdown
+---
+priority: 20
+scope:
+  kernels: ['python3']
+---
+
+# Testing Guidelines
+
+When writing code in agent mode:
+
+- Always include error handling
+- Add logging for debugging
+- Test edge cases
+```
+
+#### Rule Frontmatter Options
+
+```yaml
+---
+apply: always # 'always', 'auto', or 'manual'
+active: true # Enable/disable the rule
+priority: 10 # Lower numbers = higher priority
+scope:
+  file_patterns: # Apply to specific file patterns
+    - '*.py'
+    - 'test_*.ipynb'
+  kernels: # Apply to specific notebook kernels
+    - 'python3'
+    - 'ir'
+  directories: # Apply to specific directories
+    - '/projects/ml'
+---
+```
+
+#### Configuration
+
+**Enable/Disable Rules System:**
+
+Edit `~/.jupyter/nbi/config.json`:
+
+```json
+{
+  "rules_enabled": true
+}
+```
+
+**Auto-Reload Configuration:**
+
+Rules are automatically reloaded when changed (enabled by default). This behavior is controlled by the `NBI_RULES_AUTO_RELOAD` environment variable.
+
+To disable auto-reload:
+
+```bash
+export NBI_RULES_AUTO_RELOAD=false
+jupyter lab
+```
+
+Or to enable (default):
+
+```bash
+export NBI_RULES_AUTO_RELOAD=true
+jupyter lab
+```
+
+#### Managing Rules
+
+Rules are automatically discovered from:
+
+- **Global rules**: `~/.jupyter/nbi/rules/*.md`
+- **Mode-specific rules**: `~/.jupyter/nbi/rules/modes/{mode}/*.md` where `{mode}` can be:
+  - `ask` - For question/answer interactions
+  - `agent` - For autonomous agent operations
+  - `inline-chat` - For inline code generation
+
+Rules are applied in priority order (lower numbers first) and can be toggled on/off without deleting the files.
+
 ### Developer documentation
 
 For building locally and contributing see the [developer documentatation](CONTRIBUTING.md).

package/lib/api.d.ts

@@ -22,6 +22,9 @@ export declare class NBIConfig {
     get usingGitHubCopilotModel(): boolean;
     get storeGitHubAccessToken(): boolean;
     get toolConfig(): any;
+    get mcpServers(): any;
+    getMCPServer(serverId: string): any;
+    getMCPServerPrompt(serverId: string, promptName: string): any;
     get mcpServerSettings(): any;
     capabilities: any;
     chatParticipants: IChatParticipant[];
@@ -39,6 +42,9 @@ export declare class NBIAPI {
     static initializeWebsocket(): Promise<void>;
     static getLoginStatus(): GitHubCopilotLoginStatus;
     static getDeviceVerificationInfo(): IDeviceVerificationInfo;
+    static getGHLoginRequired(): boolean;
+    static getChatEnabled(): any;
+    static getInlineCompletionEnabled(): any;
     static loginToGitHub(): Promise<unknown>;
     static logoutFromGitHub(): Promise<unknown>;
     static updateGitHubLoginStatus(): Promise<void>;

package/lib/api.js

@@ -53,6 +53,19 @@ export class NBIConfig {
     get toolConfig() {
         return this.capabilities.tool_config;
     }
+    get mcpServers() {
+        return this.toolConfig.mcpServers;
+    }
+    getMCPServer(serverId) {
+        return this.toolConfig.mcpServers.find((server) => server.id === serverId);
+    }
+    getMCPServerPrompt(serverId, promptName) {
+        const server = this.getMCPServer(serverId);
+        if (server) {
+            return server.prompts.find((prompt) => prompt.name === promptName);
+        }
+        return null;
+    }
     get mcpServerSettings() {
         return this.capabilities.mcp_server_settings;
     }
@@ -98,6 +111,21 @@ class NBIAPI {
     static getDeviceVerificationInfo() {
         return this._deviceVerificationInfo;
     }
+    static getGHLoginRequired() {
+        return (this.config.usingGitHubCopilotModel &&
+            NBIAPI.getLoginStatus() === GitHubCopilotLoginStatus.NotLoggedIn);
+    }
+    static getChatEnabled() {
+        return this.config.chatModel.provider === GITHUB_COPILOT_PROVIDER_ID
+            ? !this.getGHLoginRequired()
+            : this.config.llmProviders.find(provider => provider.id === this.config.chatModel.provider);
+    }
+    static getInlineCompletionEnabled() {
+        return this.config.inlineCompletionModel.provider ===
+            GITHUB_COPILOT_PROVIDER_ID
+            ? !this.getGHLoginRequired()
+            : this.config.llmProviders.find(provider => provider.id === this.config.inlineCompletionModel.provider);
+    }
     static async loginToGitHub() {
         this._loginStatus = GitHubCopilotLoginStatus.ActivatingDevice;
         return new Promise((resolve, reject) => {

package/lib/chat-sidebar.d.ts

@@ -42,6 +42,8 @@ export interface IInlinePromptWidgetOptions {
     existingCode: string;
     prefix: string;
     suffix: string;
+    language?: string;
+    filename?: string;
     onRequestSubmitted: (prompt: string) => void;
     onRequestCancelled: () => void;
     onContentStream: (content: string) => void;
@@ -73,3 +75,12 @@ export declare class GitHubCopilotLoginDialogBody extends ReactWidget {
     render(): JSX.Element;
     private _onLoggedIn;
 }
+export declare class FormInputDialogBody extends ReactWidget {
+    constructor(options: {
+        fields: any;
+        onDone: (formData: any) => void;
+    });
+    render(): JSX.Element;
+    private _fields;
+    private _onDone;
+}

package/lib/chat-sidebar.js

@@ -4,7 +4,7 @@ import { ReactWidget } from '@jupyterlab/apputils';
 import { UUID } from '@lumino/coreutils';
 import * as monaco from 'monaco-editor/esm/vs/editor/editor.api.js';
 import { NBIAPI, GitHubCopilotLoginStatus } from './api';
-import { BackendMessageType, BuiltinToolsetType, ContextType, GITHUB_COPILOT_PROVIDER_ID, RequestDataType, ResponseStreamDataType, TelemetryEventType } from './tokens';
+import { BackendMessageType, BuiltinToolsetType, ContextType, RequestDataType, ResponseStreamDataType, TelemetryEventType } from './tokens';
 import { MarkdownRenderer as OriginalMarkdownRenderer } from './markdown-renderer';
 const MarkdownRenderer = memo(OriginalMarkdownRenderer);
 import copySvgstr from '../style/icons/copy.svg';
@@ -613,29 +613,36 @@ function SidebarComponent(props) {
         setToolSelections(newConfig);
     };
     useEffect(() => {
+        var _a;
         const prefixes = [];
-        if (chatMode !== 'ask') {
-            prefixes.push('/clear');
-            setOriginalPrefixes(prefixes);
-            setPrefixSuggestions(prefixes);
-            return;
-        }
-        const chatParticipants = NBIAPI.config.chatParticipants;
-        for (const participant of chatParticipants) {
-            const id = participant.id;
-            const commands = participant.commands;
-            const participantPrefix = id === 'default' ? '' : `@${id}`;
-            if (participantPrefix !== '') {
-                prefixes.push(participantPrefix);
+        prefixes.push('/clear');
+        if (chatMode === 'ask') {
+            const chatParticipants = NBIAPI.config.chatParticipants;
+            for (const participant of chatParticipants) {
+                const id = participant.id;
+                const commands = participant.commands;
+                const participantPrefix = id === 'default' ? '' : `@${id}`;
+                if (participantPrefix !== '') {
+                    prefixes.push(participantPrefix);
+                }
+                const commandPrefix = participantPrefix === '' ? '' : `${participantPrefix} `;
+                for (const command of commands) {
+                    prefixes.push(`${commandPrefix}/${command}`);
+                }
             }
-            const commandPrefix = participantPrefix === '' ? '' : `${participantPrefix} `;
-            for (const command of commands) {
-                prefixes.push(`${commandPrefix}/${command}`);
+        }
+        const mcpServers = NBIAPI.config.toolConfig.mcpServers;
+        const mcpServerSettings = NBIAPI.config.mcpServerSettings;
+        for (const mcpServer of mcpServers) {
+            if (((_a = mcpServerSettings[mcpServer.id]) === null || _a === void 0 ? void 0 : _a.disabled) !== true) {
+                for (const prompt of mcpServer.prompts) {
+                    prefixes.push(`/mcp:${mcpServer.id}:${prompt.name}`);
+                }
             }
         }
         setOriginalPrefixes(prefixes);
         setPrefixSuggestions(prefixes);
-    }, [chatMode]);
+    }, [chatMode, renderCount]);
     useEffect(() => {
         const fetchData = () => {
             setGHLoginStatus(NBIAPI.getLoginStatus());
@@ -664,13 +671,37 @@ function SidebarComponent(props) {
             setShowPopover(false);
         }
     };
-    const applyPrefixSuggestion = (prefix) => {
+    const applyPrefixSuggestion = async (prefix) => {
         var _a;
+        let mcpArguments = '';
+        if (prefix.startsWith('/mcp:')) {
+            mcpArguments = ':';
+            const serverId = prefix.split(':')[1];
+            const promptName = prefix.split(':')[2];
+            const promptConfig = NBIAPI.config.getMCPServerPrompt(serverId, promptName);
+            if (promptConfig &&
+                promptConfig.arguments &&
+                promptConfig.arguments.length > 0) {
+                const result = await props
+                    .getApp()
+                    .commands.execute('notebook-intelligence:show-form-input-dialog', {
+                    title: 'Input Parameters',
+                    fields: promptConfig.arguments
+                });
+                const argumentValues = [];
+                for (const argument of promptConfig.arguments) {
+                    if (result[argument.name] !== undefined) {
+                        argumentValues.push(`${argument.name}=${result[argument.name]}`);
+                    }
+                }
+                mcpArguments = `(${argumentValues.join(', ')}):`;
+            }
+        }
         if (prefix.includes(prompt)) {
-            setPrompt(`${prefix} `);
+            setPrompt(`${prefix}${mcpArguments} `);
         }
         else {
-            setPrompt(`${prefix} ${prompt} `);
+            setPrompt(`${prefix} ${prompt}${mcpArguments} `);
         }
         setShowPopover(false);
         (_a = promptInputRef.current) === null || _a === void 0 ? void 0 : _a.focus();
@@ -721,7 +752,6 @@ function SidebarComponent(props) {
                 }
             }
         }
-        const promptPrefix = promptPrefixParts.length > 0 ? promptPrefixParts.join(' ') + ' ' : '';
         lastMessageId.current = UUID.uuid4();
         lastRequestTime.current = new Date();
         const newList = [
@@ -780,7 +810,7 @@ function SidebarComponent(props) {
             type: RunChatCompletionType.Chat,
             content: extractedPrompt,
             language: activeDocInfo.language,
-            filename: activeDocInfo.filename,
+            filename: activeDocInfo.filePath,
             additionalContext,
             chatMode,
             toolSelections: toolSelections
@@ -863,7 +893,7 @@ function SidebarComponent(props) {
                 ]);
             }
         });
-        const newPrompt = prompt.startsWith('/settings') ? '' : promptPrefix;
+        const newPrompt = '';
         setPrompt(newPrompt);
         filterPrefixSuggestions(newPrompt);
         telemetryEmitter.emitTelemetryEvent({
@@ -1130,27 +1160,17 @@ function SidebarComponent(props) {
         }
         return `${activeDocumentInfo.filename}${cellAndLineIndicator}`;
     };
-    const nbiConfig = NBIAPI.config;
-    const getGHLoginRequired = () => {
-        return (nbiConfig.usingGitHubCopilotModel &&
-            NBIAPI.getLoginStatus() === GitHubCopilotLoginStatus.NotLoggedIn);
-    };
-    const getChatEnabled = () => {
-        return nbiConfig.chatModel.provider === GITHUB_COPILOT_PROVIDER_ID
-            ? !getGHLoginRequired()
-            : nbiConfig.llmProviders.find(provider => provider.id === nbiConfig.chatModel.provider);
-    };
-    const [ghLoginRequired, setGHLoginRequired] = useState(getGHLoginRequired());
-    const [chatEnabled, setChatEnabled] = useState(getChatEnabled());
+    const [ghLoginRequired, setGHLoginRequired] = useState(NBIAPI.getGHLoginRequired());
+    const [chatEnabled, setChatEnabled] = useState(NBIAPI.getChatEnabled());
     useEffect(() => {
         NBIAPI.configChanged.connect(() => {
-            setGHLoginRequired(getGHLoginRequired());
-            setChatEnabled(getChatEnabled());
+            setGHLoginRequired(NBIAPI.getGHLoginRequired());
+            setChatEnabled(NBIAPI.getChatEnabled());
         });
     }, []);
     useEffect(() => {
-        setGHLoginRequired(getGHLoginRequired());
-        setChatEnabled(getChatEnabled());
+        setGHLoginRequired(NBIAPI.getGHLoginRequired());
+        setChatEnabled(NBIAPI.getChatEnabled());
     }, [ghLoginStatus]);
     return (React.createElement("div", { className: "sidebar" },
         React.createElement("div", { className: "sidebar-header" },
@@ -1242,7 +1262,7 @@ function SidebarComponent(props) {
                         } })))),
                     renderCount > 0 &&
                         mcpServerEnabledState.size > 0 &&
-                        toolConfigRef.current.mcpServers.length > 0 && (React.createElement("div", { className: "mode-tools-group-header" }, "MCP Servers")),
+                        toolConfigRef.current.mcpServers.length > 0 && (React.createElement("div", { className: "mode-tools-group-header" }, "MCP Server Tools")),
                     renderCount > 0 &&
                         toolConfigRef.current.mcpServers
                             .filter(mcpServer => mcpServerEnabledState.has(mcpServer.id))
@@ -1350,8 +1370,8 @@ function InlinePromptComponent(props) {
             chatId: UUID.uuid4(),
             type: RunChatCompletionType.GenerateCode,
             content: prompt,
-            language: undefined,
-            filename: undefined,
+            language: props.language || 'python',
+            filename: props.filename || 'Untitled.ipynb',
             prefix: props.prefix,
             suffix: props.suffix,
             existingCode: props.existingCode,
@@ -1508,3 +1528,31 @@ function GitHubCopilotLoginDialogBodyComponent(props) {
             React.createElement("button", { className: "jp-Dialog-button jp-mod-reject jp-mod-styled", onClick: handleLogoutClick },
                 React.createElement("div", { className: "jp-Dialog-buttonLabel" }, "Cancel activation"))))));
 }
+export class FormInputDialogBody extends ReactWidget {
+    constructor(options) {
+        super();
+        this._fields = options.fields || [];
+        this._onDone = options.onDone || (() => { });
+    }
+    render() {
+        return (React.createElement(FormInputDialogBodyComponent, { fields: this._fields, onDone: this._onDone }));
+    }
+}
+function FormInputDialogBodyComponent(props) {
+    const [formData, setFormData] = useState({});
+    const handleInputChange = (event) => {
+        setFormData({ ...formData, [event.target.name]: event.target.value });
+    };
+    return (React.createElement("div", { className: "form-input-dialog-body" },
+        React.createElement("div", { className: "form-input-dialog-body-content" },
+            React.createElement("div", { className: "form-input-dialog-body-content-title" }, props.title),
+            React.createElement("div", { className: "form-input-dialog-body-content-fields" }, props.fields.map((field) => (React.createElement("div", { className: "form-input-dialog-body-content-field", key: field.name },
+                React.createElement("label", { className: "form-input-dialog-body-content-field-label jp-mod-styled", htmlFor: field.name },
+                    field.name,
+                    field.required ? ' (required)' : ''),
+                React.createElement("input", { className: "form-input-dialog-body-content-field-input jp-mod-styled", type: field.type, id: field.name, name: field.name, onChange: handleInputChange, value: formData[field.name] || '' }))))),
+            React.createElement("div", null,
+                React.createElement("div", { style: { marginTop: '10px' } },
+                    React.createElement("button", { className: "jp-Dialog-button jp-mod-accept jp-mod-styled", onClick: () => props.onDone(formData) },
+                        React.createElement("div", { className: "jp-Dialog-buttonLabel" }, "Done")))))));
+}

package/lib/index.js

@@ -14,11 +14,12 @@ import { LabIcon } from '@jupyterlab/ui-components';
 import { Menu, Panel, Widget } from '@lumino/widgets';
 import { CommandRegistry } from '@lumino/commands';
 import { IStatusBar } from '@jupyterlab/statusbar';
-import { ChatSidebar, GitHubCopilotLoginDialogBody, GitHubCopilotStatusBarItem, InlinePromptWidget, RunChatCompletionType } from './chat-sidebar';
+import { ChatSidebar, FormInputDialogBody, GitHubCopilotLoginDialogBody, GitHubCopilotStatusBarItem, InlinePromptWidget, RunChatCompletionType } from './chat-sidebar';
 import { NBIAPI, GitHubCopilotLoginStatus } from './api';
 import { BackendMessageType, GITHUB_COPILOT_PROVIDER_ID, INotebookIntelligence, RequestDataType, TelemetryEventType } from './tokens';
 import sparklesSvgstr from '../style/icons/sparkles.svg';
 import copilotSvgstr from '../style/icons/copilot.svg';
+import sparklesWarningSvgstr from '../style/icons/sparkles-warning.svg';
 import { applyCodeToSelectionInEditor, cellOutputAsText, compareSelections, extractLLMGeneratedCode, getSelectionInEditor, getTokenCount, getWholeNotebookContent, isSelectionEmpty, markdownToComment, waitForDuration } from './utils';
 import { UUID } from '@lumino/coreutils';
 import * as path from 'path';
@@ -52,6 +53,7 @@ var CommandIDs;
     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.showFormInputDialog = 'notebook-intelligence:show-form-input-dialog';
 })(CommandIDs || (CommandIDs = {}));
 const DOCUMENT_WATCH_INTERVAL = 1000;
 const MAX_TOKENS = 4096;
@@ -63,6 +65,10 @@ const sparkleIcon = new LabIcon({
     name: 'notebook-intelligence:sparkles-icon',
     svgstr: sparklesSvgstr
 });
+const sparkleWarningIcon = new LabIcon({
+    name: 'notebook-intelligence:sparkles-warning-icon',
+    svgstr: sparklesWarningSvgstr
+});
 const emptyNotebookContent = {
     cells: [],
     metadata: {},
@@ -499,7 +505,7 @@ const plugin = {
         panel.id = 'notebook-intelligence-tab';
         panel.title.caption = 'Notebook Intelligence';
         const sidebarIcon = new LabIcon({
-            name: 'ui-components:palette',
+            name: 'notebook-intelligence:sidebar-icon',
             svgstr: sparklesSvgstr
         });
         panel.title.icon = sidebarIcon;
@@ -526,6 +532,23 @@ const plugin = {
         panel.addWidget(sidebar);
         app.shell.add(panel, 'left', { rank: 1000 });
         app.shell.activateById(panel.id);
+        const updateSidebarIcon = () => {
+            if (NBIAPI.getChatEnabled()) {
+                panel.title.icon = sidebarIcon;
+            }
+            else {
+                panel.title.icon = sparkleWarningIcon;
+            }
+        };
+        NBIAPI.githubLoginStatusChanged.connect((_, args) => {
+            updateSidebarIcon();
+        });
+        NBIAPI.configChanged.connect((_, args) => {
+            updateSidebarIcon();
+        });
+        setTimeout(() => {
+            updateSidebarIcon();
+        }, 2000);
         app.commands.addCommand(CommandIDs.chatuserInput, {
             execute: args => {
                 NBIAPI.sendChatUserInput(args.id, args.data);
@@ -595,6 +618,36 @@ const plugin = {
                 return newPyFile;
             }
         });
+        app.commands.addCommand(CommandIDs.showFormInputDialog, {
+            execute: async (args) => {
+                const title = args.title;
+                const fields = args.fields;
+                return new Promise((resolve, reject) => {
+                    let dialog = null;
+                    const dialogBody = new FormInputDialogBody({
+                        fields: fields,
+                        onDone: (formData) => {
+                            dialog.dispose();
+                            resolve(formData);
+                        }
+                    });
+                    dialog = new Dialog({
+                        title: title,
+                        hasClose: true,
+                        body: dialogBody,
+                        buttons: []
+                    });
+                    dialog
+                        .launch()
+                        .then((result) => {
+                        reject();
+                    })
+                        .catch(() => {
+                        reject(new Error('Failed to show form input dialog'));
+                    });
+                });
+            }
+        });
         app.commands.addCommand(CommandIDs.createNewNotebookFromPython, {
             execute: async (args) => {
                 var _a;
@@ -1143,6 +1196,8 @@ const plugin = {
                 existingCode,
                 prefix: prefix,
                 suffix: suffix,
+                language: ActiveDocumentWatcher.activeDocumentInfo.language,
+                filename: ActiveDocumentWatcher.activeDocumentInfo.filePath,
                 onRequestSubmitted: (prompt) => {
                     userPrompt = prompt;
                     generatedContent = '';

package/lib/tokens.d.ts

@@ -51,7 +51,9 @@ export declare enum MCPServerStatus {
     FailedToConnect = "failed-to-connect",
     Connected = "connected",
     UpdatingToolList = "updating-tool-list",
-    UpdatedToolList = "updated-tool-list"
+    UpdatedToolList = "updated-tool-list",
+    UpdatingPromptList = "updating-prompt-list",
+    UpdatedPromptList = "updated-prompt-list"
 }
 export interface IContextItem {
     type: ContextType;

package/lib/tokens.js

@@ -45,6 +45,8 @@ export var MCPServerStatus;
     MCPServerStatus["Connected"] = "connected";
     MCPServerStatus["UpdatingToolList"] = "updating-tool-list";
     MCPServerStatus["UpdatedToolList"] = "updated-tool-list";
+    MCPServerStatus["UpdatingPromptList"] = "updating-prompt-list";
+    MCPServerStatus["UpdatedPromptList"] = "updated-prompt-list";
 })(MCPServerStatus || (MCPServerStatus = {}));
 export var BuiltinToolsetType;
 (function (BuiltinToolsetType) {

package/package.json

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

package/src/api.ts

@@ -76,6 +76,24 @@ export class NBIConfig {
     return this.capabilities.tool_config;
   }
 
+  get mcpServers(): any {
+    return this.toolConfig.mcpServers;
+  }
+
+  getMCPServer(serverId: string): any {
+    return this.toolConfig.mcpServers.find(
+      (server: any) => server.id === serverId
+    );
+  }
+
+  getMCPServerPrompt(serverId: string, promptName: string): any {
+    const server = this.getMCPServer(serverId);
+    if (server) {
+      return server.prompts.find((prompt: any) => prompt.name === promptName);
+    }
+    return null;
+  }
+
   get mcpServerSettings(): any {
     return this.capabilities.mcp_server_settings;
   }
@@ -152,6 +170,30 @@ export class NBIAPI {
     return this._deviceVerificationInfo;
   }
 
+  static getGHLoginRequired() {
+    return (
+      this.config.usingGitHubCopilotModel &&
+      NBIAPI.getLoginStatus() === GitHubCopilotLoginStatus.NotLoggedIn
+    );
+  }
+
+  static getChatEnabled() {
+    return this.config.chatModel.provider === GITHUB_COPILOT_PROVIDER_ID
+      ? !this.getGHLoginRequired()
+      : this.config.llmProviders.find(
+          provider => provider.id === this.config.chatModel.provider
+        );
+  }
+
+  static getInlineCompletionEnabled() {
+    return this.config.inlineCompletionModel.provider ===
+      GITHUB_COPILOT_PROVIDER_ID
+      ? !this.getGHLoginRequired()
+      : this.config.llmProviders.find(
+          provider => provider.id === this.config.inlineCompletionModel.provider
+        );
+  }
+
   static async loginToGitHub() {
     this._loginStatus = GitHubCopilotLoginStatus.ActivatingDevice;
     return new Promise((resolve, reject) => {

package/src/chat-sidebar.tsx

@@ -20,7 +20,6 @@ import {
   BackendMessageType,
   BuiltinToolsetType,
   ContextType,
-  GITHUB_COPILOT_PROVIDER_ID,
   IActiveDocumentInfo,
   ICellContents,
   IChatCompletionResponseEmitter,
@@ -118,6 +117,8 @@ export interface IInlinePromptWidgetOptions {
   existingCode: string;
   prefix: string;
   suffix: string;
+  language?: string;
+  filename?: string;
   onRequestSubmitted: (prompt: string) => void;
   onRequestCancelled: () => void;
   onContentStream: (content: string) => void;
@@ -1078,30 +1079,38 @@ function SidebarComponent(props: any) {
 
   useEffect(() => {
     const prefixes: string[] = [];
-    if (chatMode !== 'ask') {
-      prefixes.push('/clear');
-      setOriginalPrefixes(prefixes);
-      setPrefixSuggestions(prefixes);
-      return;
+    prefixes.push('/clear');
+
+    if (chatMode === 'ask') {
+      const chatParticipants = NBIAPI.config.chatParticipants;
+      for (const participant of chatParticipants) {
+        const id = participant.id;
+        const commands = participant.commands;
+        const participantPrefix = id === 'default' ? '' : `@${id}`;
+        if (participantPrefix !== '') {
+          prefixes.push(participantPrefix);
+        }
+        const commandPrefix =
+          participantPrefix === '' ? '' : `${participantPrefix} `;
+        for (const command of commands) {
+          prefixes.push(`${commandPrefix}/${command}`);
+        }
+      }
     }
 
-    const chatParticipants = NBIAPI.config.chatParticipants;
-    for (const participant of chatParticipants) {
-      const id = participant.id;
-      const commands = participant.commands;
-      const participantPrefix = id === 'default' ? '' : `@${id}`;
-      if (participantPrefix !== '') {
-        prefixes.push(participantPrefix);
-      }
-      const commandPrefix =
-        participantPrefix === '' ? '' : `${participantPrefix} `;
-      for (const command of commands) {
-        prefixes.push(`${commandPrefix}/${command}`);
+    const mcpServers = NBIAPI.config.toolConfig.mcpServers;
+    const mcpServerSettings = NBIAPI.config.mcpServerSettings;
+    for (const mcpServer of mcpServers) {
+      if (mcpServerSettings[mcpServer.id]?.disabled !== true) {
+        for (const prompt of mcpServer.prompts) {
+          prefixes.push(`/mcp:${mcpServer.id}:${prompt.name}`);
+        }
       }
     }
+
     setOriginalPrefixes(prefixes);
     setPrefixSuggestions(prefixes);
-  }, [chatMode]);
+  }, [chatMode, renderCount]);
 
   useEffect(() => {
     const fetchData = () => {
@@ -1137,11 +1146,41 @@ function SidebarComponent(props: any) {
     }
   };
 
-  const applyPrefixSuggestion = (prefix: string) => {
+  const applyPrefixSuggestion = async (prefix: string) => {
+    let mcpArguments = '';
+    if (prefix.startsWith('/mcp:')) {
+      mcpArguments = ':';
+      const serverId = prefix.split(':')[1];
+      const promptName = prefix.split(':')[2];
+      const promptConfig = NBIAPI.config.getMCPServerPrompt(
+        serverId,
+        promptName
+      );
+      if (
+        promptConfig &&
+        promptConfig.arguments &&
+        promptConfig.arguments.length > 0
+      ) {
+        const result = await props
+          .getApp()
+          .commands.execute('notebook-intelligence:show-form-input-dialog', {
+            title: 'Input Parameters',
+            fields: promptConfig.arguments
+          });
+        const argumentValues: string[] = [];
+        for (const argument of promptConfig.arguments) {
+          if (result[argument.name] !== undefined) {
+            argumentValues.push(`${argument.name}=${result[argument.name]}`);
+          }
+        }
+        mcpArguments = `(${argumentValues.join(', ')}):`;
+      }
+    }
+
     if (prefix.includes(prompt)) {
-      setPrompt(`${prefix} `);
+      setPrompt(`${prefix}${mcpArguments} `);
     } else {
-      setPrompt(`${prefix} ${prompt} `);
+      setPrompt(`${prefix} ${prompt}${mcpArguments} `);
     }
     setShowPopover(false);
     promptInputRef.current?.focus();
@@ -1201,9 +1240,6 @@ function SidebarComponent(props: any) {
       }
     }
 
-    const promptPrefix =
-      promptPrefixParts.length > 0 ? promptPrefixParts.join(' ') + ' ' : '';
-
     lastMessageId.current = UUID.uuid4();
     lastRequestTime.current = new Date();
 
@@ -1275,7 +1311,7 @@ function SidebarComponent(props: any) {
         type: RunChatCompletionType.Chat,
         content: extractedPrompt,
         language: activeDocInfo.language,
-        filename: activeDocInfo.filename,
+        filename: activeDocInfo.filePath,
         additionalContext,
         chatMode,
         toolSelections: toolSelections
@@ -1368,7 +1404,7 @@ function SidebarComponent(props: any) {
       }
     );
 
-    const newPrompt = prompt.startsWith('/settings') ? '' : promptPrefix;
+    const newPrompt = '';
 
     setPrompt(newPrompt);
     filterPrefixSuggestions(newPrompt);
@@ -1697,34 +1733,21 @@ function SidebarComponent(props: any) {
     return `${activeDocumentInfo.filename}${cellAndLineIndicator}`;
   };
 
-  const nbiConfig = NBIAPI.config;
-  const getGHLoginRequired = () => {
-    return (
-      nbiConfig.usingGitHubCopilotModel &&
-      NBIAPI.getLoginStatus() === GitHubCopilotLoginStatus.NotLoggedIn
-    );
-  };
-  const getChatEnabled = () => {
-    return nbiConfig.chatModel.provider === GITHUB_COPILOT_PROVIDER_ID
-      ? !getGHLoginRequired()
-      : nbiConfig.llmProviders.find(
-          provider => provider.id === nbiConfig.chatModel.provider
-        );
-  };
-
-  const [ghLoginRequired, setGHLoginRequired] = useState(getGHLoginRequired());
-  const [chatEnabled, setChatEnabled] = useState(getChatEnabled());
+  const [ghLoginRequired, setGHLoginRequired] = useState(
+    NBIAPI.getGHLoginRequired()
+  );
+  const [chatEnabled, setChatEnabled] = useState(NBIAPI.getChatEnabled());
 
   useEffect(() => {
     NBIAPI.configChanged.connect(() => {
-      setGHLoginRequired(getGHLoginRequired());
-      setChatEnabled(getChatEnabled());
+      setGHLoginRequired(NBIAPI.getGHLoginRequired());
+      setChatEnabled(NBIAPI.getChatEnabled());
     });
   }, []);
 
   useEffect(() => {
-    setGHLoginRequired(getGHLoginRequired());
-    setChatEnabled(getChatEnabled());
+    setGHLoginRequired(NBIAPI.getGHLoginRequired());
+    setChatEnabled(NBIAPI.getChatEnabled());
   }, [ghLoginStatus]);
 
   return (
@@ -1986,7 +2009,9 @@ function SidebarComponent(props: any) {
                 {renderCount > 0 &&
                   mcpServerEnabledState.size > 0 &&
                   toolConfigRef.current.mcpServers.length > 0 && (
-                    <div className="mode-tools-group-header">MCP Servers</div>
+                    <div className="mode-tools-group-header">
+                      MCP Server Tools
+                    </div>
                   )}
                 {renderCount > 0 &&
                   toolConfigRef.current.mcpServers
@@ -2236,8 +2261,8 @@ function InlinePromptComponent(props: any) {
         chatId: UUID.uuid4(),
         type: RunChatCompletionType.GenerateCode,
         content: prompt,
-        language: undefined,
-        filename: undefined,
+        language: props.language || 'python',
+        filename: props.filename || 'Untitled.ipynb',
         prefix: props.prefix,
         suffix: props.suffix,
         existingCode: props.existingCode,
@@ -2509,3 +2534,76 @@ function GitHubCopilotLoginDialogBodyComponent(props: any) {
     </div>
   );
 }
+
+export class FormInputDialogBody extends ReactWidget {
+  constructor(options: { fields: any; onDone: (formData: any) => void }) {
+    super();
+
+    this._fields = options.fields || [];
+    this._onDone = options.onDone || (() => {});
+  }
+
+  render(): JSX.Element {
+    return (
+      <FormInputDialogBodyComponent
+        fields={this._fields}
+        onDone={this._onDone}
+      />
+    );
+  }
+
+  private _fields: any;
+  private _onDone: (formData: any) => void;
+}
+
+function FormInputDialogBodyComponent(props: any) {
+  const [formData, setFormData] = useState<any>({});
+
+  const handleInputChange = (event: any) => {
+    setFormData({ ...formData, [event.target.name]: event.target.value });
+  };
+
+  return (
+    <div className="form-input-dialog-body">
+      <div className="form-input-dialog-body-content">
+        <div className="form-input-dialog-body-content-title">
+          {props.title}
+        </div>
+        <div className="form-input-dialog-body-content-fields">
+          {props.fields.map((field: any) => (
+            <div
+              className="form-input-dialog-body-content-field"
+              key={field.name}
+            >
+              <label
+                className="form-input-dialog-body-content-field-label jp-mod-styled"
+                htmlFor={field.name}
+              >
+                {field.name}
+                {field.required ? ' (required)' : ''}
+              </label>
+              <input
+                className="form-input-dialog-body-content-field-input jp-mod-styled"
+                type={field.type}
+                id={field.name}
+                name={field.name}
+                onChange={handleInputChange}
+                value={formData[field.name] || ''}
+              />
+            </div>
+          ))}
+        </div>
+        <div>
+          <div style={{ marginTop: '10px' }}>
+            <button
+              className="jp-Dialog-button jp-mod-accept jp-mod-styled"
+              onClick={() => props.onDone(formData)}
+            >
+              <div className="jp-Dialog-buttonLabel">Done</div>
+            </button>
+          </div>
+        </div>
+      </div>
+    </div>
+  );
+}

package/src/index.ts

@@ -44,6 +44,7 @@ import { IStatusBar } from '@jupyterlab/statusbar';
 
 import {
   ChatSidebar,
+  FormInputDialogBody,
   GitHubCopilotLoginDialogBody,
   GitHubCopilotStatusBarItem,
   InlinePromptWidget,
@@ -64,6 +65,7 @@ import {
 } from './tokens';
 import sparklesSvgstr from '../style/icons/sparkles.svg';
 import copilotSvgstr from '../style/icons/copilot.svg';
+import sparklesWarningSvgstr from '../style/icons/sparkles-warning.svg';
 
 import {
   applyCodeToSelectionInEditor,
@@ -126,6 +128,8 @@ namespace CommandIDs {
     'notebook-intelligence:set-current-file-content';
   export const openMCPConfigEditor =
     'notebook-intelligence:open-mcp-config-editor';
+  export const showFormInputDialog =
+    'notebook-intelligence:show-form-input-dialog';
 }
 
 const DOCUMENT_WATCH_INTERVAL = 1000;
@@ -140,6 +144,10 @@ const sparkleIcon = new LabIcon({
   svgstr: sparklesSvgstr
 });
 
+const sparkleWarningIcon = new LabIcon({
+  name: 'notebook-intelligence:sparkles-warning-icon',
+  svgstr: sparklesWarningSvgstr
+});
 const emptyNotebookContent: any = {
   cells: [],
   metadata: {},
@@ -711,7 +719,7 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
     panel.id = 'notebook-intelligence-tab';
     panel.title.caption = 'Notebook Intelligence';
     const sidebarIcon = new LabIcon({
-      name: 'ui-components:palette',
+      name: 'notebook-intelligence:sidebar-icon',
       svgstr: sparklesSvgstr
     });
     panel.title.icon = sidebarIcon;
@@ -739,6 +747,26 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
     app.shell.add(panel, 'left', { rank: 1000 });
     app.shell.activateById(panel.id);
 
+    const updateSidebarIcon = () => {
+      if (NBIAPI.getChatEnabled()) {
+        panel.title.icon = sidebarIcon;
+      } else {
+        panel.title.icon = sparkleWarningIcon;
+      }
+    };
+
+    NBIAPI.githubLoginStatusChanged.connect((_, args) => {
+      updateSidebarIcon();
+    });
+
+    NBIAPI.configChanged.connect((_, args) => {
+      updateSidebarIcon();
+    });
+
+    setTimeout(() => {
+      updateSidebarIcon();
+    }, 2000);
+
     app.commands.addCommand(CommandIDs.chatuserInput, {
       execute: args => {
         NBIAPI.sendChatUserInput(args.id as string, args.data);
@@ -820,6 +848,39 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
       }
     });
 
+    app.commands.addCommand(CommandIDs.showFormInputDialog, {
+      execute: async args => {
+        const title = args.title as string;
+        const fields = args.fields;
+
+        return new Promise<any>((resolve, reject) => {
+          let dialog: Dialog<unknown> | null = null;
+          const dialogBody = new FormInputDialogBody({
+            fields: fields,
+            onDone: (formData: any) => {
+              dialog.dispose();
+              resolve(formData);
+            }
+          });
+          dialog = new Dialog({
+            title: title,
+            hasClose: true,
+            body: dialogBody,
+            buttons: []
+          });
+
+          dialog
+            .launch()
+            .then((result: any) => {
+              reject();
+            })
+            .catch(() => {
+              reject(new Error('Failed to show form input dialog'));
+            });
+        });
+      }
+    });
+
     app.commands.addCommand(CommandIDs.createNewNotebookFromPython, {
       execute: async args => {
         let pythonKernelSpec = null;
@@ -1501,6 +1562,8 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
         existingCode,
         prefix: prefix,
         suffix: suffix,
+        language: ActiveDocumentWatcher.activeDocumentInfo.language,
+        filename: ActiveDocumentWatcher.activeDocumentInfo.filePath,
         onRequestSubmitted: (prompt: string) => {
           userPrompt = prompt;
           generatedContent = '';

package/src/tokens.ts

@@ -60,7 +60,9 @@ export enum MCPServerStatus {
   FailedToConnect = 'failed-to-connect',
   Connected = 'connected',
   UpdatingToolList = 'updating-tool-list',
-  UpdatedToolList = 'updated-tool-list'
+  UpdatedToolList = 'updated-tool-list',
+  UpdatingPromptList = 'updating-prompt-list',
+  UpdatedPromptList = 'updated-prompt-list'
 }
 
 export interface IContextItem {

package/style/base.css

@@ -890,13 +890,15 @@ svg.access-token-warning {
 }
 
 .server-status-indicator.connected,
-.server-status-indicator.updated-tool-list {
+.server-status-indicator.updated-tool-list,
+.server-status-indicator.updated-prompt-list {
   background-color: var(--jp-success-color1);
 }
 
 .server-status-indicator.connecting,
 .server-status-indicator.disconnecting,
-.server-status-indicator.updating-tool-list {
+.server-status-indicator.updating-tool-list,
+.server-status-indicator.updating-prompt-list {
   background-color: var(--jp-warn-color1);
 }
 
@@ -904,3 +906,27 @@ svg.access-token-warning {
 .server-status-indicator.failed-to-connect {
   background-color: var(--jp-error-color1);
 }
+
+.form-input-dialog-body {
+  width: 500px;
+}
+
+.form-input-dialog-body-content-fields {
+  display: flex;
+  flex-direction: column;
+  gap: 10px;
+}
+
+.form-input-dialog-body-content-field {
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+}
+
+.form-input-dialog-body-content-field-label {
+  width: 50%;
+}
+
+.form-input-dialog-body-content-field-input {
+  width: 50%;
+}

package/style/icons/sparkles-warning.svg

@@ -0,0 +1,5 @@
+<svg width="16" height="16" viewBox="0 0 16 16" xmlns="http://www.w3.org/2000/svg" fill="var(--jp-inverse-layout-color3)"><path d="M 5.398 10.807 C 5.574 10.931 5.785 10.998 6 10.997 C 6.216 10.998 6.427 10.93 6.602 10.804 C 6.78 10.674 6.915 10.494 6.989 10.286 L 7.436 8.913 C 7.551 8.569 7.744 8.256 8 7.999 C 8.257 7.743 8.569 7.549 8.913 7.434 L 10.304 6.983 C 10.456 6.929 10.594 6.84 10.706 6.724 C 10.817 6.608 10.901 6.467 10.949 6.313 C 10.998 6.159 11.01 5.996 10.985 5.837 C 10.96 5.677 10.898 5.526 10.804 5.394 C 10.67 5.208 10.479 5.071 10.26 5.003 L 8.885 4.556 C 8.541 4.442 8.228 4.249 7.971 3.993 C 7.714 3.736 7.52 3.424 7.405 3.079 L 6.953 1.691 C 6.881 1.489 6.748 1.314 6.571 1.191 C 6.439 1.098 6.286 1.036 6.125 1.012 C 5.965 0.987 5.801 1.001 5.646 1.051 C 5.492 1.101 5.351 1.187 5.236 1.301 C 5.12 1.415 5.033 1.555 4.98 1.708 L 4.523 3.108 C 4.409 3.443 4.22 3.748 3.97 3.999 C 3.721 4.25 3.418 4.441 3.083 4.557 L 1.692 5.005 C 1.541 5.06 1.404 5.149 1.292 5.265 C 1.18 5.381 1.097 5.521 1.048 5.675 C 1 5.829 0.988 5.992 1.013 6.151 C 1.038 6.31 1.099 6.462 1.192 6.593 C 1.32 6.773 1.501 6.908 1.709 6.979 L 3.083 7.424 C 3.524 7.571 3.91 7.845 4.193 8.212 C 4.356 8.423 4.481 8.66 4.564 8.912 L 5.016 10.303 C 5.088 10.507 5.222 10.683 5.398 10.807 Z M 11.535 14.849 C 11.671 14.946 11.834 14.997 12 14.997 C 12.165 14.997 12.326 14.946 12.461 14.851 C 12.601 14.753 12.706 14.613 12.761 14.451 L 13.009 13.689 C 13.063 13.531 13.152 13.387 13.269 13.268 C 13.387 13.15 13.531 13.061 13.689 13.009 L 14.461 12.757 C 14.619 12.703 14.756 12.6 14.852 12.464 C 14.926 12.361 14.974 12.242 14.992 12.117 C 15.011 11.992 14.999 11.865 14.959 11.745 C 14.918 11.625 14.85 11.516 14.76 11.428 C 14.669 11.34 14.559 11.274 14.438 11.236 L 13.674 10.987 C 13.516 10.935 13.372 10.846 13.254 10.729 C 13.136 10.611 13.047 10.467 12.994 10.309 L 12.742 9.536 C 12.689 9.379 12.586 9.242 12.449 9.146 C 12.347 9.073 12.23 9.025 12.106 9.006 C 11.982 8.987 11.855 8.998 11.736 9.037 C 11.616 9.076 11.508 9.142 11.419 9.231 C 11.33 9.319 11.264 9.427 11.224 9.546 L 10.977 10.308 C 10.925 10.466 10.838 10.61 10.721 10.728 C 10.607 10.845 10.467 10.934 10.312 10.987 L 9.539 11.239 C 9.38 11.293 9.242 11.396 9.145 11.533 C 9.047 11.669 8.995 11.833 8.996 12.001 C 8.997 12.169 9.051 12.333 9.15 12.468 C 9.249 12.604 9.388 12.705 9.547 12.757 L 10.31 13.004 C 10.469 13.058 10.614 13.147 10.732 13.265 C 10.851 13.384 10.939 13.528 10.99 13.687 L 11.243 14.461 C 11.298 14.618 11.4 14.753 11.535 14.849 Z"/>
+<g class="layer">
+<circle cx="12" cy="4" fill="#FF0000" r="4" stroke="#FFFFFF" stroke-width="0"/>
+</g>
+</svg>