@notebook-intelligence/notebook-intelligence 4.1.2 → 4.2.1

package/lib/api.js

@@ -5,7 +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, BackendMessageType } from './tokens';
+import { GITHUB_COPILOT_PROVIDER_ID, RequestDataType, BackendMessageType, AssistantMode } from './tokens';
 export var GitHubCopilotLoginStatus;
 (function (GitHubCopilotLoginStatus) {
     GitHubCopilotLoginStatus["NotLoggedIn"] = "NOT_LOGGED_IN";
@@ -360,6 +360,13 @@ class NBIAPI {
         }));
     }
     static async emitTelemetryEvent(event) {
+        const assistantMode = this.config.isInClaudeCodeMode
+            ? AssistantMode.Claude
+            : AssistantMode.Default;
+        event.data = {
+            ...(event.data || {}),
+            assistantMode
+        };
         return new Promise((resolve, reject) => {
             requestAPI('emit-telemetry-event', {
                 method: 'POST',

package/lib/chat-sidebar.js

@@ -327,6 +327,11 @@ function ChatResponse(props) {
                                     runCommand('notebook-intelligence:chat-user-input', item.content.confirmArgs);
                                 } },
                                 React.createElement("div", { className: "jp-Dialog-buttonLabel" }, item.content.confirmLabel)),
+                            item.content.confirmSessionArgs ? (React.createElement("button", { className: "jp-Dialog-button jp-mod-accept jp-mod-styled", onClick: () => {
+                                    markFormConfirmed(item.id);
+                                    runCommand('notebook-intelligence:chat-user-input', item.content.confirmSessionArgs);
+                                } },
+                                React.createElement("div", { className: "jp-Dialog-buttonLabel" }, item.content.confirmSessionLabel))) : null,
                             React.createElement("button", { className: "jp-Dialog-button jp-mod-reject jp-mod-styled", onClick: () => {
                                     markFormCanceled(item.id);
                                     runCommand('notebook-intelligence:chat-user-input', item.content.cancelArgs);
@@ -1319,7 +1324,7 @@ function SidebarComponent(props) {
             "Chat is disabled as you don't have a model configured.",
             React.createElement("button", { className: "jp-Dialog-button jp-mod-accept jp-mod-styled", onClick: handleConfigurationClick },
                 React.createElement("div", { className: "jp-Dialog-buttonLabel" }, "Configure models")))),
-        ghLoginRequired && (React.createElement("div", { className: "sidebar-login-info" },
+        !NBIAPI.config.isInClaudeCodeMode && ghLoginRequired && (React.createElement("div", { className: "sidebar-login-info" },
             React.createElement("div", null, "You are not logged in to GitHub Copilot. Please login now to activate chat."),
             React.createElement("div", { className: "sidebar-login-buttons" },
                 React.createElement("button", { className: "jp-Dialog-button jp-mod-accept jp-mod-styled", onClick: handleLoginClick },

package/lib/components/settings-panel.js

@@ -431,7 +431,7 @@ function SettingsPanelComponentClaude(props) {
         settingSources,
         tools
     ]);
-    return (React.createElement("div", { className: "config-dialog" },
+    return (React.createElement("div", { className: "config-dialog claude-mode-config-dialog" },
         React.createElement("div", { className: "config-dialog-body" },
             React.createElement("div", { className: "model-config-section" },
                 React.createElement("div", { className: "model-config-section-header" }, "Enable Claude mode"),

package/lib/index.js

@@ -760,11 +760,12 @@ const plugin = {
                     const messageCheckTimeout = 5000;
                     const messageCheckInterval = 1000;
                     let output = '';
-                    session.messageReceived.connect((sender, message) => {
+                    const messageReceivedHandler = (sender, message) => {
                         const content = stripAnsi(message.content.join(''));
                         output += content;
                         lastMessageReceivedTime = Date.now();
-                    });
+                    };
+                    session.messageReceived.connect(messageReceivedHandler);
                     session.send({
                         type: 'stdin',
                         content: [command + '\n'] // Add newline to execute the command
@@ -774,6 +775,7 @@ const plugin = {
                     lastMessageCheckInterval = setInterval(() => {
                         if (Date.now() - lastMessageReceivedTime > messageCheckTimeout) {
                             clearInterval(lastMessageCheckInterval);
+                            session.messageReceived.disconnect(messageReceivedHandler);
                             resolve(`Command executed in Jupyter terminal, output: ${output}`);
                         }
                     }, messageCheckInterval);

package/lib/tokens.d.ts

@@ -96,6 +96,10 @@ export declare enum BuiltinToolsetType {
 }
 export declare const GITHUB_COPILOT_PROVIDER_ID = "github-copilot";
 export declare const CLAUDE_CODE_CHAT_PARTICIPANT_ID = "claude-code";
+export declare enum AssistantMode {
+    Default = "default",
+    Claude = "claude"
+}
 export declare enum TelemetryEventType {
     InlineCompletionRequest = "inline-completion-request",
     ExplainThisRequest = "explain-this-request",

package/lib/tokens.js

@@ -61,6 +61,11 @@ export var BuiltinToolsetType;
 })(BuiltinToolsetType || (BuiltinToolsetType = {}));
 export const GITHUB_COPILOT_PROVIDER_ID = 'github-copilot';
 export const CLAUDE_CODE_CHAT_PARTICIPANT_ID = 'claude-code';
+export var AssistantMode;
+(function (AssistantMode) {
+    AssistantMode["Default"] = "default";
+    AssistantMode["Claude"] = "claude";
+})(AssistantMode || (AssistantMode = {}));
 export var TelemetryEventType;
 (function (TelemetryEventType) {
     TelemetryEventType["InlineCompletionRequest"] = "inline-completion-request";

package/package.json

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

package/src/api.ts

@@ -13,7 +13,8 @@ import {
   ITelemetryEvent,
   IToolSelections,
   RequestDataType,
-  BackendMessageType
+  BackendMessageType,
+  AssistantMode
 } from './tokens';
 
 export enum GitHubCopilotLoginStatus {
@@ -497,6 +498,15 @@ export class NBIAPI {
   }
 
   static async emitTelemetryEvent(event: ITelemetryEvent): Promise<void> {
+    const assistantMode = this.config.isInClaudeCodeMode
+      ? AssistantMode.Claude
+      : AssistantMode.Default;
+
+    event.data = {
+      ...(event.data || {}),
+      assistantMode
+    };
+
     return new Promise<void>((resolve, reject) => {
       requestAPI<any>('emit-telemetry-event', {
         method: 'POST',

package/src/chat-sidebar.tsx

@@ -613,6 +613,22 @@ function ChatResponse(props: any) {
                       {item.content.confirmLabel}
                     </div>
                   </button>
+                  {item.content.confirmSessionArgs ? (
+                    <button
+                      className="jp-Dialog-button jp-mod-accept jp-mod-styled"
+                      onClick={() => {
+                        markFormConfirmed(item.id);
+                        runCommand(
+                          'notebook-intelligence:chat-user-input',
+                          item.content.confirmSessionArgs
+                        );
+                      }}
+                    >
+                      <div className="jp-Dialog-buttonLabel">
+                        {item.content.confirmSessionLabel}
+                      </div>
+                    </button>
+                  ) : null}
                   <button
                     className="jp-Dialog-button jp-mod-reject jp-mod-styled"
                     onClick={() => {
@@ -1937,7 +1953,7 @@ function SidebarComponent(props: any) {
           </button>
         </div>
       )}
-      {ghLoginRequired && (
+      {!NBIAPI.config.isInClaudeCodeMode && ghLoginRequired && (
         <div className="sidebar-login-info">
           <div>
             You are not logged in to GitHub Copilot. Please login now to

package/src/components/settings-panel.tsx

@@ -877,7 +877,7 @@ function SettingsPanelComponentClaude(props: any) {
   ]);
 
   return (
-    <div className="config-dialog">
+    <div className="config-dialog claude-mode-config-dialog">
       <div className="config-dialog-body">
         <div className="model-config-section">
           <div className="model-config-section-header">Enable Claude mode</div>

package/src/index.ts

@@ -1006,11 +1006,12 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
           const messageCheckTimeout = 5000;
           const messageCheckInterval = 1000;
           let output = '';
-          session.messageReceived.connect((sender, message) => {
+          const messageReceivedHandler = (sender: any, message: any) => {
             const content = stripAnsi(message.content.join(''));
             output += content;
             lastMessageReceivedTime = Date.now();
-          });
+          };
+          session.messageReceived.connect(messageReceivedHandler);
 
           session.send({
             type: 'stdin',
@@ -1022,6 +1023,7 @@ const plugin: JupyterFrontEndPlugin<INotebookIntelligence> = {
           lastMessageCheckInterval = setInterval(() => {
             if (Date.now() - lastMessageReceivedTime > messageCheckTimeout) {
               clearInterval(lastMessageCheckInterval);
+              session.messageReceived.disconnect(messageReceivedHandler);
               resolve(
                 `Command executed in Jupyter terminal, output: ${output}`
               );

package/src/tokens.ts

@@ -112,6 +112,11 @@ export enum BuiltinToolsetType {
 export const GITHUB_COPILOT_PROVIDER_ID = 'github-copilot';
 export const CLAUDE_CODE_CHAT_PARTICIPANT_ID = 'claude-code';
 
+export enum AssistantMode {
+  Default = 'default',
+  Claude = 'claude'
+}
+
 export enum TelemetryEventType {
   InlineCompletionRequest = 'inline-completion-request',
   ExplainThisRequest = 'explain-this-request',

package/style/base.css

@@ -1044,3 +1044,7 @@ svg.access-token-warning {
 .ask-user-question-footer {
   padding: 0 0 5px 10px;
 }
+
+.claude-mode-config-dialog .form-field-description {
+  color: var(--jp-ui-font-color1);
+}