@notebook-intelligence/notebook-intelligence 2.4.2 → 2.6.0

package/lib/tokens.d.ts

@@ -26,7 +26,8 @@ export declare enum BackendMessageType {
     StreamMessage = "stream-message",
     StreamEnd = "stream-end",
     RunUICommand = "run-ui-command",
-    GitHubCopilotLoginStatusChange = "github-copilot-login-status-change"
+    GitHubCopilotLoginStatusChange = "github-copilot-login-status-change",
+    MCPServerStatusChange = "mcp-server-status-change"
 }
 export declare enum ResponseStreamDataType {
     LLMRaw = "llm-raw",
@@ -43,6 +44,17 @@ export declare enum ContextType {
     Custom = "custom",
     CurrentFile = "current-file"
 }
+export declare enum MCPServerStatus {
+    NotConnected = "not-connected",
+    Connecting = "connecting",
+    Disconnecting = "disconnecting",
+    FailedToConnect = "failed-to-connect",
+    Connected = "connected",
+    UpdatingToolList = "updating-tool-list",
+    UpdatedToolList = "updated-tool-list",
+    UpdatingPromptList = "updating-prompt-list",
+    UpdatedPromptList = "updated-prompt-list"
+}
 export interface IContextItem {
     type: ContextType;
     content: string;

package/lib/tokens.js

@@ -17,6 +17,7 @@ export var BackendMessageType;
     BackendMessageType["StreamEnd"] = "stream-end";
     BackendMessageType["RunUICommand"] = "run-ui-command";
     BackendMessageType["GitHubCopilotLoginStatusChange"] = "github-copilot-login-status-change";
+    BackendMessageType["MCPServerStatusChange"] = "mcp-server-status-change";
 })(BackendMessageType || (BackendMessageType = {}));
 export var ResponseStreamDataType;
 (function (ResponseStreamDataType) {
@@ -35,6 +36,18 @@ export var ContextType;
     ContextType["Custom"] = "custom";
     ContextType["CurrentFile"] = "current-file";
 })(ContextType || (ContextType = {}));
+export var MCPServerStatus;
+(function (MCPServerStatus) {
+    MCPServerStatus["NotConnected"] = "not-connected";
+    MCPServerStatus["Connecting"] = "connecting";
+    MCPServerStatus["Disconnecting"] = "disconnecting";
+    MCPServerStatus["FailedToConnect"] = "failed-to-connect";
+    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) {
     BuiltinToolsetType["NotebookEdit"] = "nbi-notebook-edit";

package/package.json

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

package/src/api.ts

@@ -76,6 +76,28 @@ 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;
+  }
+
   capabilities: any = {};
   chatParticipants: IChatParticipant[] = [];
 
@@ -102,7 +124,11 @@ export class NBIAPI {
 
     this._messageReceived.connect((_, msg) => {
       msg = JSON.parse(msg);
-      if (msg.type === BackendMessageType.GitHubCopilotLoginStatusChange) {
+      if (msg.type === BackendMessageType.MCPServerStatusChange) {
+        this.fetchCapabilities();
+      } else if (
+        msg.type === BackendMessageType.GitHubCopilotLoginStatusChange
+      ) {
         this.updateGitHubLoginStatus().then(() => {
           this.githubLoginStatusChanged.emit();
         });
@@ -144,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) => {
@@ -201,11 +251,21 @@ export class NBIAPI {
     return new Promise<void>((resolve, reject) => {
       requestAPI<any>('capabilities', { method: 'GET' })
         .then(data => {
+          const oldConfig = {
+            capabilities: structuredClone(this.config.capabilities),
+            chatParticipants: structuredClone(this.config.chatParticipants)
+          };
           this.config.capabilities = structuredClone(data);
           this.config.chatParticipants = structuredClone(
             data.chat_participants
           );
-          this.configChanged.emit();
+          const newConfig = {
+            capabilities: structuredClone(this.config.capabilities),
+            chatParticipants: structuredClone(this.config.chatParticipants)
+          };
+          if (JSON.stringify(newConfig) !== JSON.stringify(oldConfig)) {
+            this.configChanged.emit();
+          }
           resolve();
         })
         .catch(reason => {
@@ -245,20 +305,6 @@ export class NBIAPI {
     });
   }
 
-  static async reloadMCPServerList(): Promise<any> {
-    return new Promise<any>((resolve, reject) => {
-      requestAPI<any>('reload-mcp-servers', { method: 'POST' })
-        .then(async data => {
-          await NBIAPI.fetchCapabilities();
-          resolve(data);
-        })
-        .catch(reason => {
-          console.error(`Failed to reload MCP server list.\n${reason}`);
-          reject(reason);
-        });
-    });
-  }
-
   static async getMCPConfigFile(): Promise<any> {
     return new Promise<any>((resolve, reject) => {
       requestAPI<any>('mcp-config-file', { method: 'GET' })
@@ -322,6 +368,20 @@ export class NBIAPI {
     );
   }
 
+  static async reloadMCPServers(): Promise<any> {
+    return new Promise<any>((resolve, reject) => {
+      requestAPI<any>('reload-mcp-servers', { method: 'POST' })
+        .then(async data => {
+          await NBIAPI.fetchCapabilities();
+          resolve(data);
+        })
+        .catch(reason => {
+          console.error(`Failed to reload MCP servers.\n${reason}`);
+          reject(reason);
+        });
+    });
+  }
+
   static async generateCode(
     chatId: string,
     prompt: string,