@notebook-intelligence/notebook-intelligence 1.3.5 → 2.1.0

package/src/index.ts

@@ -104,6 +104,23 @@ namespace CommandIDs {
     'notebook-intelligence:open-github-copilot-login-dialog';
   export const openConfigurationDialog =
     'notebook-intelligence:open-configuration-dialog';
+  export const addMarkdownCellToActiveNotebook =
+    'notebook-intelligence:add-markdown-cell-to-active-notebook';
+  export const addCodeCellToActiveNotebook =
+    'notebook-intelligence:add-code-cell-to-active-notebook';
+  export const deleteCellAtIndex = 'notebook-intelligence:delete-cell-at-index';
+  export const insertCellAtIndex = 'notebook-intelligence:insert-cell-at-index';
+  export const setCellTypeAndSource =
+    'notebook-intelligence:set-cell-type-and-source';
+  export const getNumberOfCells = 'notebook-intelligence:get-number-of-cells';
+  export const getCellType = 'notebook-intelligence:get-cell-type';
+  export const getCellSource = 'notebook-intelligence:get-cell-source';
+  export const getCellOutput = 'notebook-intelligence:get-cell-output';
+  export const runCellAtIndex = 'notebook-intelligence:run-cell-at-index';
+  export const getCurrentFileContent =
+    'notebook-intelligence:get-current-file-content';
+  export const setCurrentFileContent =
+    'notebook-intelligence:set-current-file-content';
 }
 
 const DOCUMENT_WATCH_INTERVAL = 1000;
@@ -856,6 +873,240 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
       }
     });
 
+    const ensureANotebookIsActive = (): boolean => {
+      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 = (): boolean => {
+      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 as NotebookPanel;
+        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 as string
+        });
+
+        return true;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.addCodeCellToActiveNotebook, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        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 as string
+        });
+
+        return true;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.setCellTypeAndSource, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        const model = np.model.sharedModel;
+
+        const cellIndex = args.cellIndex as number;
+        const cellType = args.cellType as 'code' | 'markdown';
+        const cell = model.getCell(cellIndex);
+
+        model.deleteCell(cellIndex);
+        model.insertCell(cellIndex, {
+          cell_type: cellType,
+          metadata: cell.metadata,
+          source: args.source as string
+        });
+
+        return true;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.getNumberOfCells, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        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 as NotebookPanel;
+        const model = np.model.sharedModel;
+
+        return model.cells[args.cellIndex as number].cell_type;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.getCellSource, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        const model = np.model.sharedModel;
+
+        return model.cells[args.cellIndex as number].source;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.getCellOutput, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        const cellIndex = args.cellIndex as number;
+
+        const cell = np.content.widgets[cellIndex];
+
+        if (!(cell instanceof CodeCell)) {
+          return '';
+        }
+
+        const content = cellOutputAsText(cell as CodeCell);
+
+        return content;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.insertCellAtIndex, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        const model = np.model.sharedModel;
+        const cellIndex = args.cellIndex as number;
+        const cellType = args.cellType as 'code' | 'markdown';
+
+        model.insertCell(cellIndex, {
+          cell_type: cellType,
+          metadata: { trusted: true },
+          source: args.source as string
+        });
+
+        return true;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.deleteCellAtIndex, {
+      execute: args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const np = app.shell.currentWidget as NotebookPanel;
+        const model = np.model.sharedModel;
+        const cellIndex = args.cellIndex as number;
+
+        model.deleteCell(cellIndex);
+
+        return true;
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.runCellAtIndex, {
+      execute: async args => {
+        if (!ensureANotebookIsActive()) {
+          return false;
+        }
+
+        const currentWidget = app.shell.currentWidget as NotebookPanel;
+        currentWidget.content.activeCellIndex = args.cellIndex as number;
+
+        await app.commands.execute('notebook:run-cell');
+      }
+    });
+
+    app.commands.addCommand(CommandIDs.getCurrentFileContent, {
+      execute: async args => {
+        if (!ensureAFileEditorIsActive()) {
+          return false;
+        }
+
+        const currentWidget = app.shell.currentWidget as FileEditorWidget;
+        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 as FileEditorWidget;
+        const editor = currentWidget.content.editor;
+        editor.model.sharedModel.setSource(args.content as string);
+        return editor.model.sharedModel.getSource();
+      }
+    });
+
     app.commands.addCommand(CommandIDs.openGitHubCopilotLoginDialog, {
       execute: args => {
         let dialog: Dialog<unknown> | null = null;
@@ -1358,7 +1609,7 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
 
     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/src/tokens.ts

@@ -74,6 +74,21 @@ export interface IChatParticipant {
   commands: string[];
 }
 
+export interface IToolSelections {
+  builtinToolsets?: string[];
+  mcpServers?: {
+    [key: string]: string[];
+  };
+  extensions?: {
+    [key: string]: string[];
+  };
+}
+
+export enum BuiltinToolsetType {
+  NotebookEdit = 'nbi-notebook-edit',
+  NotebookExecute = 'nbi-notebook-execute'
+}
+
 export const GITHUB_COPILOT_PROVIDER_ID = 'github-copilot';
 
 export enum TelemetryEventType {

package/style/base.css

@@ -108,6 +108,57 @@
   display: flex;
   flex-direction: row;
   align-items: center;
+  gap: 10px;
+}
+
+.chat-mode-widgets-container {
+  display: flex;
+  flex-direction: row;
+  align-items: center;
+  gap: 5px;
+  margin-right: 5px;
+}
+
+.chat-mode-select {
+  height: 24px;
+  background-color: initial;
+  color: var(--jp-ui-font-color1);
+  padding: 0 5px;
+  border-color: var(--jp-border-color0);
+  outline: none;
+}
+
+.user-input-footer-button {
+  width: 24px;
+  height: 24px;
+  box-sizing: border-box;
+  padding: 2px;
+  cursor: pointer;
+  color: var(--jp-ui-font-color2);
+}
+
+.user-input-footer-button.tools-button {
+  width: auto;
+  border: 1px solid var(--jp-border-color0);
+  display: flex;
+  align-items: center;
+  gap: 2px;
+  padding-right: 3px;
+}
+
+.user-input-footer-button.tools-button-warning {
+  color: var(--jp-warn-color0);
+  border-color: var(--jp-warn-color0);
+}
+
+.user-input-footer-button.tools-button-active {
+  color: var(--jp-brand-color0);
+  border-color: var(--jp-brand-color0);
+}
+
+.user-input-footer-button svg {
+  width: 18px;
+  height: 18px;
 }
 
 .chat-message {
@@ -292,17 +343,99 @@ pre:has(.code-block-header) {
   margin-top: 2px;
 }
 
-.user-input-autocomplete {
+.user-input-autocomplete,
+.mode-tools-popover {
   display: flex;
   background-color: var(--jp-layout-color2);
   border: 1px solid var(--jp-border-color1);
   flex-direction: column;
   position: absolute;
-  bottom: 92px;
+  bottom: 98px;
   left: 4px;
   gap: 2px;
-  max-height: 300px;
+  /* stylelint-disable */
+  max-height: min(calc(100% - 120px), 400px);
+  /* stylelint-enable */
   overflow-y: auto;
+  user-select: none;
+}
+
+.mode-tools-popover {
+  width: calc(100% - 10px);
+  /* stylelint-disable */
+  max-height: min(calc(100% - 200px), 400px);
+  /* stylelint-enable */
+  box-shadow: var(--jp-brand-color0) 0 0 1px 1px;
+  display: flex;
+  flex-direction: column;
+}
+
+.mode-tools-popover-title {
+  font-weight: bold;
+}
+
+.mode-tools-popover-header {
+  display: flex;
+  align-items: center;
+  gap: 5px;
+  height: 24px;
+  background-color: var(--jp-layout-color1);
+  padding: 2px;
+}
+
+.mode-tools-popover-header-icon svg {
+  width: 18px;
+  height: 18px;
+}
+
+.mode-tools-popover-clear-tools-button {
+  flex-grow: 1;
+  color: var(--jp-ui-font-color2);
+  display: flex;
+  padding-top: 3px;
+}
+
+.mode-tools-popover-clear-tools-button svg {
+  width: 16px;
+  height: 16px;
+}
+
+.mode-tools-popover-close-button {
+  display: flex;
+  align-items: center;
+  cursor: pointer;
+  width: auto;
+  height: 24px;
+  color: white;
+  background-color: var(--jp-brand-color0);
+  padding-right: 6px;
+  border-radius: 2px;
+}
+
+.mode-tools-popover-close-button svg {
+  width: 16px;
+  height: 16px;
+  padding: 4px;
+  margin-top: 2px;
+}
+
+.mode-tools-popover-tool-list {
+  flex-grow: 1;
+  overflow-y: auto;
+}
+
+.mode-tools-group {
+  display: flex;
+  flex-direction: column;
+  padding-left: 5px;
+  margin-bottom: 5px;
+}
+
+.mode-tools-group-header {
+  padding: 5px 0;
+  border-bottom: 1px solid var(--jp-border-color1);
+  margin: 0 5px;
+  font-weight: bold;
 }
 
 .user-input-autocomplete-item {
@@ -626,3 +759,63 @@ svg.access-token-warning {
   width: 18px;
   height: 18px;
 }
+
+.checkbox-item {
+  display: flex;
+  flex-direction: column;
+  cursor: pointer;
+}
+
+.checkbox-item:hover {
+  background-color: var(--jp-layout-color1);
+}
+
+.checkbox-item-indent-0 {
+  padding-left: 0;
+}
+
+.checkbox-item-indent-1 {
+  padding-left: 12px;
+}
+
+.checkbox-item-indent-2 {
+  padding-left: 24px;
+}
+
+.checkbox-item-indent-3 {
+  padding-left: 36px;
+}
+
+.checkbox-icon {
+  width: 16px;
+  height: 16px;
+  color: var(--jp-brand-color1);
+  padding-top: 1px;
+}
+
+.checkbox-item-toggle {
+  display: flex;
+  align-items: center;
+  gap: 5px;
+  margin: 0 2px;
+  padding: 3px 2px;
+  color: var(--jp-ui-font-color1);
+}
+
+.checkbox-item-header .checkbox-item-toggle {
+  color: var(--jp-ui-font-color0);
+  font-weight: bold;
+}
+
+.mode-tools-group-built-in .checkbox-item-toggle {
+  font-weight: normal;
+}
+
+.checkbox-item-description {
+  padding-left: 25px;
+  color: var(--jp-ui-font-color2);
+  padding-bottom: 4px;
+  overflow: hidden;
+  text-overflow: ellipsis;
+  white-space: nowrap;
+}