@notebook-intelligence/notebook-intelligence 1.3.4 → 2.0.0

package/lib/index.js

@@ -37,6 +37,18 @@ var CommandIDs;
     CommandIDs.editorTroubleshootThisOutput = 'notebook-intelligence:editor-troubleshoot-this-output';
     CommandIDs.openGitHubCopilotLoginDialog = 'notebook-intelligence:open-github-copilot-login-dialog';
     CommandIDs.openConfigurationDialog = 'notebook-intelligence:open-configuration-dialog';
+    CommandIDs.addMarkdownCellToActiveNotebook = 'notebook-intelligence:add-markdown-cell-to-active-notebook';
+    CommandIDs.addCodeCellToActiveNotebook = 'notebook-intelligence:add-code-cell-to-active-notebook';
+    CommandIDs.deleteCellAtIndex = 'notebook-intelligence:delete-cell-at-index';
+    CommandIDs.insertCellAtIndex = 'notebook-intelligence:insert-cell-at-index';
+    CommandIDs.setCellTypeAndSource = 'notebook-intelligence:set-cell-type-and-source';
+    CommandIDs.getNumberOfCells = 'notebook-intelligence:get-number-of-cells';
+    CommandIDs.getCellType = 'notebook-intelligence:get-cell-type';
+    CommandIDs.getCellSource = 'notebook-intelligence:get-cell-source';
+    CommandIDs.getCellOutput = 'notebook-intelligence:get-cell-output';
+    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 || (CommandIDs = {}));
 const DOCUMENT_WATCH_INTERVAL = 1000;
 const MAX_TOKENS = 4096;
@@ -625,6 +637,192 @@ const plugin = {
                 return addCellToNotebook(args.path, 'markdown', args.markdown);
             }
         });
+        const ensureANotebookIsActive = () => {
+            const currentWidget = app.shell.currentWidget;
+            const notebookOpen = currentWidget instanceof NotebookPanel && currentWidget.model;
+            if (!notebookOpen) {
+                app.commands.execute('apputils:notify', {
+                    message: 'Failed to find active notebook',
+                    type: 'error',
+                    options: { autoClose: true }
+                });
+                return false;
+            }
+            return true;
+        };
+        const ensureAFileEditorIsActive = () => {
+            const currentWidget = app.shell.currentWidget;
+            const textFileOpen = currentWidget instanceof FileEditorWidget;
+            if (!textFileOpen) {
+                app.commands.execute('apputils:notify', {
+                    message: 'Failed to find active file',
+                    type: 'error',
+                    options: { autoClose: true }
+                });
+                return false;
+            }
+            return true;
+        };
+        app.commands.addCommand(CommandIDs.addMarkdownCellToActiveNotebook, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                const newCellIndex = isNewEmptyNotebook(model)
+                    ? 0
+                    : model.cells.length - 1;
+                model.insertCell(newCellIndex, {
+                    cell_type: 'markdown',
+                    metadata: { trusted: true },
+                    source: args.source
+                });
+                return true;
+            }
+        });
+        app.commands.addCommand(CommandIDs.addCodeCellToActiveNotebook, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                const newCellIndex = isNewEmptyNotebook(model)
+                    ? 0
+                    : model.cells.length - 1;
+                model.insertCell(newCellIndex, {
+                    cell_type: 'code',
+                    metadata: { trusted: true },
+                    source: args.source
+                });
+                return true;
+            }
+        });
+        app.commands.addCommand(CommandIDs.setCellTypeAndSource, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                const cellIndex = args.cellIndex;
+                const cellType = args.cellType;
+                const cell = model.getCell(cellIndex);
+                model.deleteCell(cellIndex);
+                model.insertCell(cellIndex, {
+                    cell_type: cellType,
+                    metadata: cell.metadata,
+                    source: args.source
+                });
+                return true;
+            }
+        });
+        app.commands.addCommand(CommandIDs.getNumberOfCells, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                return model.cells.length;
+            }
+        });
+        app.commands.addCommand(CommandIDs.getCellType, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                return model.cells[args.cellIndex].cell_type;
+            }
+        });
+        app.commands.addCommand(CommandIDs.getCellSource, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                return model.cells[args.cellIndex].source;
+            }
+        });
+        app.commands.addCommand(CommandIDs.getCellOutput, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const cellIndex = args.cellIndex;
+                const cell = np.content.widgets[cellIndex];
+                if (!(cell instanceof CodeCell)) {
+                    return '';
+                }
+                const content = cellOutputAsText(cell);
+                return content;
+            }
+        });
+        app.commands.addCommand(CommandIDs.insertCellAtIndex, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                const cellIndex = args.cellIndex;
+                const cellType = args.cellType;
+                model.insertCell(cellIndex, {
+                    cell_type: cellType,
+                    metadata: { trusted: true },
+                    source: args.source
+                });
+                return true;
+            }
+        });
+        app.commands.addCommand(CommandIDs.deleteCellAtIndex, {
+            execute: args => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const np = app.shell.currentWidget;
+                const model = np.model.sharedModel;
+                const cellIndex = args.cellIndex;
+                model.deleteCell(cellIndex);
+                return true;
+            }
+        });
+        app.commands.addCommand(CommandIDs.runCellAtIndex, {
+            execute: async (args) => {
+                if (!ensureANotebookIsActive()) {
+                    return false;
+                }
+                const currentWidget = app.shell.currentWidget;
+                currentWidget.content.activeCellIndex = args.cellIndex;
+                await app.commands.execute('notebook:run-cell');
+            }
+        });
+        app.commands.addCommand(CommandIDs.getCurrentFileContent, {
+            execute: async (args) => {
+                if (!ensureAFileEditorIsActive()) {
+                    return false;
+                }
+                const currentWidget = app.shell.currentWidget;
+                const editor = currentWidget.content.editor;
+                return editor.model.sharedModel.getSource();
+            }
+        });
+        app.commands.addCommand(CommandIDs.setCurrentFileContent, {
+            execute: async (args) => {
+                if (!ensureAFileEditorIsActive()) {
+                    return false;
+                }
+                const currentWidget = app.shell.currentWidget;
+                const editor = currentWidget.content.editor;
+                editor.model.sharedModel.setSource(args.content);
+                return editor.model.sharedModel.getSource();
+            }
+        });
         app.commands.addCommand(CommandIDs.openGitHubCopilotLoginDialog, {
             execute: args => {
                 let dialog = null;
@@ -1044,7 +1242,7 @@ const plugin = {
         });
         const copilotContextMenu = new Menu({ commands: copilotMenuCommands });
         copilotContextMenu.id = 'notebook-intelligence:editor-context-menu';
-        copilotContextMenu.title.label = 'Copilot';
+        copilotContextMenu.title.label = 'Notebook Intelligence';
         copilotContextMenu.title.icon = sidebarIcon;
         copilotContextMenu.addItem({ command: CommandIDs.editorGenerateCode });
         copilotContextMenu.addItem({ command: CommandIDs.editorExplainThisCode });

package/lib/tokens.d.ts

@@ -62,6 +62,19 @@ export interface IChatParticipant {
     iconPath: string;
     commands: string[];
 }
+export interface IToolSelections {
+    builtinToolsets?: string[];
+    mcpServers?: {
+        [key: string]: string[];
+    };
+    extensions?: {
+        [key: string]: string[];
+    };
+}
+export declare enum BuiltinToolsetType {
+    NotebookEdit = "nbi-notebook-edit",
+    NotebookExecute = "nbi-notebook-execute"
+}
 export declare const GITHUB_COPILOT_PROVIDER_ID = "github-copilot";
 export declare enum TelemetryEventType {
     InlineCompletionRequest = "inline-completion-request",

package/lib/tokens.js

@@ -34,6 +34,11 @@ export var ContextType;
     ContextType["Custom"] = "custom";
     ContextType["CurrentFile"] = "current-file";
 })(ContextType || (ContextType = {}));
+export var BuiltinToolsetType;
+(function (BuiltinToolsetType) {
+    BuiltinToolsetType["NotebookEdit"] = "nbi-notebook-edit";
+    BuiltinToolsetType["NotebookExecute"] = "nbi-notebook-execute";
+})(BuiltinToolsetType || (BuiltinToolsetType = {}));
 export const GITHUB_COPILOT_PROVIDER_ID = 'github-copilot';
 export var TelemetryEventType;
 (function (TelemetryEventType) {

package/package.json

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

package/src/api.ts

@@ -11,6 +11,7 @@ import {
   IChatParticipant,
   IContextItem,
   ITelemetryEvent,
+  IToolSelections,
   RequestDataType
 } from './tokens';
 
@@ -60,6 +61,10 @@ export class NBIConfig {
     return this.capabilities.store_github_access_token === true;
   }
 
+  get toolConfig(): any {
+    return this.capabilities.tool_config;
+  }
+
   capabilities: any = {};
   chatParticipants: IChatParticipant[] = [];
 
@@ -230,6 +235,8 @@ export class NBIAPI {
     language: string,
     filename: string,
     additionalContext: IContextItem[],
+    chatMode: string,
+    toolSelections: IToolSelections,
     responseEmitter: IChatCompletionResponseEmitter
   ) {
     this._messageReceived.connect((_, msg) => {
@@ -242,7 +249,15 @@ export class NBIAPI {
       JSON.stringify({
         id: messageId,
         type: RequestDataType.ChatRequest,
-        data: { chatId, prompt, language, filename, additionalContext }
+        data: {
+          chatId,
+          prompt,
+          language,
+          filename,
+          additionalContext,
+          chatMode,
+          toolSelections
+        }
       })
     );
   }