nebula-notebook-mcp 0.1.0

package/dist/tools/index.js

@@ -0,0 +1,217 @@
+/**
+ * Tools Index
+ *
+ * Single source of truth for all tool definitions.
+ * MCP server and other clients use these tools directly.
+ */
+// Import all tool categories
+import { notebookTools } from './notebook.js';
+import { kernelTools } from './kernel.js';
+import { executionTools } from './execution.js';
+import { fileTools } from './files.js';
+// Re-export tool categories for direct access
+export { notebookTools } from './notebook.js';
+export { kernelTools } from './kernel.js';
+export { executionTools } from './execution.js';
+export { fileTools } from './files.js';
+// Re-export individual tools for TypeScript consumers who need type safety
+export * from './notebook.js';
+export * from './kernel.js';
+export * from './execution.js';
+export * from './files.js';
+/**
+ * All tools combined in a single array
+ */
+export const allTools = [
+    ...notebookTools,
+    ...kernelTools,
+    ...executionTools,
+    ...fileTools,
+];
+/**
+ * Map of tool name to tool object for quick lookup
+ */
+export const toolsByName = new Map(allTools.map(tool => [tool.definition.name, tool]));
+/**
+ * Get all tool definitions (MCP-compatible format)
+ */
+export function getToolDefinitions() {
+    return allTools.map(tool => tool.definition);
+}
+/**
+ * Execute a tool by name with given parameters
+ *
+ * @param name - Tool name
+ * @param params - Tool parameters
+ * @param client - NebulaClient instance
+ * @returns Tool result
+ */
+export async function executeToolByName(name, params, client) {
+    const tool = toolsByName.get(name);
+    if (!tool) {
+        return { success: false, error: `Unknown tool: ${name}` };
+    }
+    try {
+        return await tool.execute(params, client);
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        return { success: false, error: `Tool execution failed: ${message}` };
+    }
+}
+// Tools that operate on notebooks and should track user changes
+const NOTEBOOK_TOOLS = new Set([
+    'read_notebook', 'read_cell', 'read_output',
+    'insert_cell', 'update_cell', 'delete_cell', 'delete_cells',
+    'move_cell', 'duplicate_cell', 'insert_cells',
+    'create_notebook', 'search_cells', 'update_metadata',
+    'clear_notebook', 'clear_outputs', 'execute_cell',
+    'start_agent_session', 'end_agent_session',
+]);
+/**
+ * Execute a tool and format result for MCP display
+ *
+ * Automatically includes user changes since the last tool call for notebook operations.
+ *
+ * @param name - Tool name
+ * @param params - Tool parameters
+ * @param client - NebulaClient instance
+ * @returns MCP-formatted content
+ */
+export async function executeToolForMCP(name, params, client) {
+    const tool = toolsByName.get(name);
+    if (!tool) {
+        return {
+            content: [{ type: 'text', text: `Unknown tool: ${name}` }],
+            isError: true,
+        };
+    }
+    // Extract notebook path for user change tracking
+    const notebookPath = params.path || undefined;
+    const isNotebookTool = NOTEBOOK_TOOLS.has(name) && Boolean(notebookPath);
+    const hasActiveSession = isNotebookTool ? client.hasActiveAgentSession(notebookPath) : false;
+    // Query updates BEFORE executing the tool (to capture what happened since last call)
+    let updatesSince = [];
+    const shouldCheckUpdates = isNotebookTool && (name === 'start_agent_session' || !hasActiveSession);
+    if (shouldCheckUpdates) {
+        try {
+            const updatesResult = await client.getUpdatesSinceOp(notebookPath);
+            if (updatesResult.success && updatesResult.data?.updates.length) {
+                updatesSince = updatesResult.data.updates;
+            }
+        }
+        catch (e) {
+            // Silently ignore errors in update tracking
+            console.error('[executeToolForMCP] Error getting updates:', e);
+        }
+    }
+    // Validate required parameters before execution
+    const schema = tool.definition.inputSchema;
+    if (schema.required) {
+        const missing = schema.required.filter(key => params[key] === undefined || params[key] === null);
+        if (missing.length > 0) {
+            return {
+                content: [{ type: 'text', text: `Error: Missing required parameter(s): ${missing.join(', ')}` }],
+                isError: true,
+            };
+        }
+    }
+    // Warn about unexpected parameters (likely typos like "source" instead of "content")
+    const knownParams = new Set(Object.keys(schema.properties || {}));
+    const unknownParams = Object.keys(params).filter(key => !knownParams.has(key));
+    if (unknownParams.length > 0) {
+        return {
+            content: [{ type: 'text', text: `Error: Unknown parameter(s): ${unknownParams.join(', ')}. Valid parameters: ${[...knownParams].join(', ')}` }],
+            isError: true,
+        };
+    }
+    try {
+        // Clear any stale backend info before executing the tool
+        client.consumeLastBackend();
+        client.consumeAutoStartWarning();
+        const result = await tool.execute(params, client);
+        const backend = client.consumeLastBackend();
+        const autoStartWarning = client.consumeAutoStartWarning();
+        // Record the tool call timestamp for future change tracking
+        if (isNotebookTool) {
+            client.recordToolCallTimestamp(notebookPath);
+        }
+        // Build content array
+        let content;
+        // Use custom formatter if available, otherwise default JSON
+        if (tool.formatForMCP) {
+            content = await tool.formatForMCP(result);
+        }
+        else if (!result.success) {
+            content = [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        else {
+            content = [{ type: 'text', text: JSON.stringify(result.data, null, 2) }];
+        }
+        // Add backend info
+        if (autoStartWarning) {
+            content.push({ type: 'text', text: `Warning: ${autoStartWarning}` });
+        }
+        if (backend) {
+            content.push({ type: 'text', text: `Backend: ${backend}` });
+        }
+        // Add updates if any (silent if none)
+        if (updatesSince.length > 0) {
+            const changesSummary = updatesSince.map(c => `  • ${c.description} (${new Date(c.timestamp).toLocaleTimeString()})`).join('\n');
+            content.push({
+                type: 'text',
+                text: `\n📝 Updates since last tool call:\n${changesSummary}`,
+            });
+        }
+        return {
+            content,
+            isError: !result.success,
+        };
+    }
+    catch (error) {
+        client.consumeLastBackend();
+        client.consumeAutoStartWarning();
+        // Still record timestamp even on error
+        if (isNotebookTool) {
+            client.recordToolCallTimestamp(notebookPath);
+        }
+        const message = error instanceof Error ? error.message : String(error);
+        return {
+            content: [{ type: 'text', text: `Tool execution failed: ${message}` }],
+            isError: true,
+        };
+    }
+}
+/**
+ * Get tool by name
+ */
+export function getTool(name) {
+    return toolsByName.get(name);
+}
+/**
+ * Check if a tool exists
+ */
+export function hasTool(name) {
+    return toolsByName.has(name);
+}
+/**
+ * Tool categories for organized access
+ */
+export const toolCategories = {
+    notebook: notebookTools,
+    kernel: kernelTools,
+    execution: executionTools,
+    files: fileTools,
+};
+/**
+ * Get tool names by category
+ */
+export function getToolNamesByCategory() {
+    return {
+        notebook: notebookTools.map(t => t.definition.name),
+        kernel: kernelTools.map(t => t.definition.name),
+        execution: executionTools.map(t => t.definition.name),
+        files: fileTools.map(t => t.definition.name),
+    };
+}
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAQH,6BAA6B;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,8CAA8C;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEvC,2EAA2E;AAC3E,cAAc,eAAe,CAAC;AAC9B,cAAc,aAAa,CAAC;AAC5B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,YAAY,CAAC;AAE3B;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAqB;IACxC,GAAG,aAAa;IAChB,GAAG,WAAW;IACd,GAAG,cAAc;IACjB,GAAG,SAAS;CACb,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,WAAW,GAAgC,IAAI,GAAG,CAC7D,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CACnD,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;AAC/C,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAY,EACZ,MAA+B,EAC/B,MAAoB;IAEpB,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;IAC5D,CAAC;IAED,IAAI,CAAC;QACH,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,0BAA0B,OAAO,EAAE,EAAE,CAAC;IACxE,CAAC;AACH,CAAC;AAED,gEAAgE;AAChE,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;IAC7B,eAAe,EAAE,WAAW,EAAE,aAAa;IAC3C,aAAa,EAAE,aAAa,EAAE,aAAa,EAAE,cAAc;IAC3D,WAAW,EAAE,gBAAgB,EAAE,cAAc;IAC7C,iBAAiB,EAAE,cAAc,EAAE,iBAAiB;IACpD,gBAAgB,EAAE,eAAe,EAAE,cAAc;IACjD,qBAAqB,EAAE,mBAAmB;CAC3C,CAAC,CAAC;AAEH;;;;;;;;;GASG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,IAAY,EACZ,MAA+B,EAC/B,MAAoB;IAEpB,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC;YAC1D,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,iDAAiD;IACjD,MAAM,YAAY,GAAI,MAAM,CAAC,IAAe,IAAI,SAAS,CAAC;IAC1D,MAAM,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,YAAY,CAAC,CAAC;IACzE,MAAM,gBAAgB,GAAG,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,qBAAqB,CAAC,YAAsB,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEvG,qFAAqF;IACrF,IAAI,YAAY,GAAyG,EAAE,CAAC;IAC5H,MAAM,kBAAkB,GAAG,cAAc,IAAI,CAC3C,IAAI,KAAK,qBAAqB,IAAI,CAAC,gBAAgB,CACpD,CAAC;IACF,IAAI,kBAAkB,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,YAAsB,CAAC,CAAC;YAC7E,IAAI,aAAa,CAAC,OAAO,IAAI,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;gBAChE,YAAY,GAAG,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC;YAC5C,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,4CAA4C;YAC5C,OAAO,CAAC,KAAK,CAAC,4CAA4C,EAAE,CAAC,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,gDAAgD;IAChD,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;IAC3C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC;QACjG,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,yCAAyC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;gBAChG,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED,qFAAqF;IACrF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/E,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gCAAgC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;YAC/I,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,yDAAyD;QACzD,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAC5B,MAAM,CAAC,uBAAuB,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAC5C,MAAM,gBAAgB,GAAG,MAAM,CAAC,uBAAuB,EAAE,CAAC;QAE1D,4DAA4D;QAC5D,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,CAAC,uBAAuB,CAAC,YAAsB,CAAC,CAAC;QACzD,CAAC;QAED,sBAAsB;QACtB,IAAI,OAAmB,CAAC;QAExB,4DAA4D;QAC5D,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,OAAO,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC;aAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAC3B,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/D,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3E,CAAC;QAED,mBAAmB;QACnB,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,gBAAgB,EAAE,EAAE,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,sCAAsC;QACtC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,cAAc,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAC1C,OAAO,CAAC,CAAC,WAAW,KAAK,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,kBAAkB,EAAE,GAAG,CACvE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,uCAAuC,cAAc,EAAE;aAC9D,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,OAAO;YACP,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO;SACzB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,kBAAkB,EAAE,CAAC;QAC5B,MAAM,CAAC,uBAAuB,EAAE,CAAC;QACjC,uCAAuC;QACvC,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,CAAC,uBAAuB,CAAC,YAAsB,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0BAA0B,OAAO,EAAE,EAAE,CAAC;YACtE,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,OAAO,CAAC,IAAY;IAClC,OAAO,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,OAAO,CAAC,IAAY;IAClC,OAAO,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,QAAQ,EAAE,aAAa;IACvB,MAAM,EAAE,WAAW;IACnB,SAAS,EAAE,cAAc;IACzB,KAAK,EAAE,SAAS;CACR,CAAC;AAEX;;GAEG;AACH,MAAM,UAAU,sBAAsB;IACpC,OAAO;QACL,QAAQ,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;QACnD,MAAM,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;QAC/C,SAAS,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;QACrD,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC;KAC7C,CAAC;AACJ,CAAC"}

package/dist/tools/kernel.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Kernel Tools
+ *
+ * Tools for managing Jupyter kernel sessions.
+ */
+import type { Tool } from './types.js';
+export interface ListKernelsParams {
+}
+export interface KernelInfo {
+    name: string;
+    displayName: string;
+    language: string;
+}
+export interface ListKernelsResult {
+    kernels: KernelInfo[];
+}
+export declare const listKernelsTool: Tool<ListKernelsParams, ListKernelsResult>;
+export interface KernelStartParams {
+    kernel_name?: string;
+}
+export interface KernelStartResult {
+    sessionId: string;
+    kernelName: string;
+}
+export declare const kernelStartTool: Tool<KernelStartParams, KernelStartResult>;
+export interface KernelStopParams {
+}
+export declare const kernelStopTool: Tool<KernelStopParams, void>;
+export interface KernelRestartParams {
+}
+export declare const kernelRestartTool: Tool<KernelRestartParams, void>;
+export interface KernelInterruptParams {
+}
+export declare const kernelInterruptTool: Tool<KernelInterruptParams, void>;
+export declare const kernelTools: (Tool<ListKernelsParams, ListKernelsResult> | Tool<KernelStartParams, KernelStartResult> | Tool<KernelStopParams, void>)[];
+//# sourceMappingURL=kernel.d.ts.map

package/dist/tools/kernel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernel.d.ts","sourceRoot":"","sources":["../../src/tools/kernel.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,IAAI,EAA0B,MAAM,YAAY,CAAC;AAc/D,MAAM,WAAW,iBAAiB;CAEjC;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,UAAU,EAAE,CAAC;CACvB;AAED,eAAO,MAAM,eAAe,EAAE,IAAI,CAAC,iBAAiB,EAAE,iBAAiB,CA+BtE,CAAC;AAMF,MAAM,WAAW,iBAAiB;IAChC,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,eAAO,MAAM,eAAe,EAAE,IAAI,CAAC,iBAAiB,EAAE,iBAAiB,CAsCtE,CAAC;AAMF,MAAM,WAAW,gBAAgB;CAChC;AAED,eAAO,MAAM,cAAc,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,CA6BvD,CAAC;AAMF,MAAM,WAAW,mBAAmB;CACnC;AAED,eAAO,MAAM,iBAAiB,EAAE,IAAI,CAAC,mBAAmB,EAAE,IAAI,CAuC7D,CAAC;AAMF,MAAM,WAAW,qBAAqB;CACrC;AAED,eAAO,MAAM,mBAAmB,EAAE,IAAI,CAAC,qBAAqB,EAAE,IAAI,CA6BjE,CAAC;AAMF,eAAO,MAAM,WAAW,4HAMvB,CAAC"}

package/dist/tools/kernel.js

@@ -0,0 +1,182 @@
+/**
+ * Kernel Tools
+ *
+ * Tools for managing Jupyter kernel sessions.
+ */
+async function resolveKernelSessionId(client, notebookPath) {
+    return client.resolveKernelSessionIdForNotebook(notebookPath, { createIfMissing: false });
+}
+export const listKernelsTool = {
+    definition: {
+        name: 'list_kernels',
+        description: 'List available kernel specifications with display names (includes version info)',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            required: [],
+        },
+        annotations: { readOnlyHint: true },
+    },
+    async execute(_params, client) {
+        const result = await client.listKernels();
+        if (!result.success) {
+            return { success: false, error: result.error };
+        }
+        return { success: true, data: { kernels: result.data || [] } };
+    },
+    formatForMCP(result) {
+        if (!result.success) {
+            return [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        const { kernels } = result.data;
+        if (kernels.length === 0) {
+            return [{ type: 'text', text: 'No kernels available' }];
+        }
+        const lines = kernels.map(k => `  - ${k.name}: ${k.displayName} [${k.language}]`);
+        return [{ type: 'text', text: `Available kernels:\n${lines.join('\n')}` }];
+    },
+};
+export const kernelStartTool = {
+    definition: {
+        name: 'kernel_start',
+        description: 'Start a new kernel session for the active notebook (requires start_agent_session)',
+        inputSchema: {
+            type: 'object',
+            properties: {
+                kernel_name: { type: 'string', description: 'Kernel name (e.g., python3). Default: python3.' },
+            },
+            required: [],
+        },
+        annotations: { destructiveHint: false },
+    },
+    async execute(params, client) {
+        const notebookPath = client.getActiveNotebookPath();
+        if (!notebookPath) {
+            return { success: false, error: 'No active agent session. Call start_agent_session first.' };
+        }
+        const kernelName = params.kernel_name || 'python3';
+        const result = await client.getOrCreateKernelForFile(notebookPath, kernelName);
+        if (!result.success)
+            return { success: false, error: result.error };
+        return {
+            success: true,
+            data: {
+                sessionId: result.data.sessionId,
+                kernelName: result.data.kernelName || kernelName,
+            },
+        };
+    },
+    formatForMCP(result) {
+        if (!result.success) {
+            return [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        const { sessionId, kernelName } = result.data;
+        return [{ type: 'text', text: `Kernel started: ${sessionId} (${kernelName})` }];
+    },
+};
+export const kernelStopTool = {
+    definition: {
+        name: 'kernel_stop',
+        description: 'Stop and shutdown the active notebook kernel (requires start_agent_session)',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            required: [],
+        },
+        annotations: { destructiveHint: true },
+    },
+    async execute(params, client) {
+        const notebookPath = client.getActiveNotebookPath();
+        if (!notebookPath) {
+            return { success: false, error: 'No active agent session. Call start_agent_session first.' };
+        }
+        const session = await resolveKernelSessionId(client, notebookPath);
+        if (!session.success)
+            return { success: false, error: session.error };
+        const result = await client.shutdownKernel(session.data.sessionId);
+        return result.success ? { success: true } : { success: false, error: result.error };
+    },
+    formatForMCP(result) {
+        if (!result.success) {
+            return [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        return [{ type: 'text', text: 'Kernel stopped' }];
+    },
+};
+export const kernelRestartTool = {
+    definition: {
+        name: 'kernel_restart',
+        description: 'Restart the active notebook kernel (clears all variables, requires start_agent_session)',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            required: [],
+        },
+        annotations: { destructiveHint: true },
+    },
+    async execute(params, client) {
+        const notebookPath = client.getActiveNotebookPath();
+        if (!notebookPath) {
+            return { success: false, error: 'No active agent session. Call start_agent_session first.' };
+        }
+        // Prefer a robust restart over the direct restart endpoint, which can be flaky
+        // depending on backend implementation. Shutdown (best effort), then start a new
+        // session for this file.
+        const existing = await resolveKernelSessionId(client, notebookPath);
+        if (existing.success) {
+            const stopped = await client.shutdownKernel(existing.data.sessionId);
+            // Ignore not-found errors: the session may have already been cleaned up.
+            if (!stopped.success && !stopped.error?.includes('not found')) {
+                return { success: false, error: stopped.error };
+            }
+        }
+        const started = await client.getOrCreateKernelForFile(notebookPath, 'python3');
+        return started.success ? { success: true } : { success: false, error: started.error };
+    },
+    formatForMCP(result) {
+        if (!result.success) {
+            return [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        return [{ type: 'text', text: 'Kernel restarted' }];
+    },
+};
+export const kernelInterruptTool = {
+    definition: {
+        name: 'kernel_interrupt',
+        description: 'Interrupt execution in the active notebook kernel (requires start_agent_session)',
+        inputSchema: {
+            type: 'object',
+            properties: {},
+            required: [],
+        },
+        annotations: { destructiveHint: false },
+    },
+    async execute(params, client) {
+        const notebookPath = client.getActiveNotebookPath();
+        if (!notebookPath) {
+            return { success: false, error: 'No active agent session. Call start_agent_session first.' };
+        }
+        const session = await resolveKernelSessionId(client, notebookPath);
+        if (!session.success)
+            return { success: false, error: session.error };
+        const result = await client.interruptKernel(session.data.sessionId);
+        return result.success ? { success: true } : { success: false, error: result.error };
+    },
+    formatForMCP(result) {
+        if (!result.success) {
+            return [{ type: 'text', text: `Error: ${result.error}` }];
+        }
+        return [{ type: 'text', text: 'Kernel interrupted' }];
+    },
+};
+// =============================================================================
+// Export all kernel tools
+// =============================================================================
+export const kernelTools = [
+    listKernelsTool,
+    kernelStartTool,
+    kernelStopTool,
+    kernelRestartTool,
+    kernelInterruptTool,
+];
+//# sourceMappingURL=kernel.js.map

package/dist/tools/kernel.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"kernel.js","sourceRoot":"","sources":["../../src/tools/kernel.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,KAAK,UAAU,sBAAsB,CACnC,MAAoB,EACpB,YAAoB;IAEpB,OAAO,MAAM,CAAC,iCAAiC,CAAC,YAAY,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;AAC5F,CAAC;AAoBD,MAAM,CAAC,MAAM,eAAe,GAA+C;IACzE,UAAU,EAAE;QACV,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,iFAAiF;QAC9F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE;SACb;QACD,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE;KACpC;IAED,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM;QAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACjD,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,IAAI,IAAI,EAAE,EAAE,EAAE,CAAC;IACjE,CAAC;IAED,YAAY,CAAC,MAAM;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,IAAK,CAAC;QACjC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAC;QAC1D,CAAC;QACD,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,KAAK,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC;QAClF,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;IAC7E,CAAC;CACF,CAAC;AAeF,MAAM,CAAC,MAAM,eAAe,GAA+C;IACzE,UAAU,EAAE;QACV,IAAI,EAAE,cAAc;QACpB,WAAW,EAAE,mFAAmF;QAChG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gDAAgD,EAAE;aAC/F;YACD,QAAQ,EAAE,EAAE;SACb;QACD,WAAW,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE;KACxC;IAED,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM;QAC1B,MAAM,YAAY,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,0DAA0D,EAAE,CAAC;QAC/F,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,IAAI,SAAS,CAAC;QACnD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;QAC/E,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACpE,OAAO;YACL,OAAO,EAAE,IAAI;YACb,IAAI,EAAE;gBACJ,SAAS,EAAE,MAAM,CAAC,IAAK,CAAC,SAAS;gBACjC,UAAU,EAAE,MAAM,CAAC,IAAK,CAAC,UAAU,IAAI,UAAU;aAClD;SACF,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,MAAM;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC,IAAK,CAAC;QAC/C,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,SAAS,KAAK,UAAU,GAAG,EAAE,CAAC,CAAC;IAClF,CAAC;CACF,CAAC;AASF,MAAM,CAAC,MAAM,cAAc,GAAiC;IAC1D,UAAU,EAAE;QACV,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,6EAA6E;QAC1F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE;SACb;QACD,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE;KACvC;IAED,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM;QAC1B,MAAM,YAAY,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,0DAA0D,EAAE,CAAC;QAC/F,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QACtE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,IAAK,CAAC,SAAS,CAAC,CAAC;QACpE,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;IACtF,CAAC;IAED,YAAY,CAAC,MAAM;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC;IACpD,CAAC;CACF,CAAC;AASF,MAAM,CAAC,MAAM,iBAAiB,GAAoC;IAChE,UAAU,EAAE;QACV,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,yFAAyF;QACtG,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE;SACb;QACD,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE;KACvC;IAED,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM;QAC1B,MAAM,YAAY,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,0DAA0D,EAAE,CAAC;QAC/F,CAAC;QACD,+EAA+E;QAC/E,gFAAgF;QAChF,yBAAyB;QACzB,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACpE,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrB,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAK,CAAC,SAAS,CAAC,CAAC;YACtE,yEAAyE;YACzE,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC9D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;YAClD,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;QAC/E,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;IACxF,CAAC;IAED,YAAY,CAAC,MAAM;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;IACtD,CAAC;CACF,CAAC;AASF,MAAM,CAAC,MAAM,mBAAmB,GAAsC;IACpE,UAAU,EAAE;QACV,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,kFAAkF;QAC/F,WAAW,EAAE;YACX,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;YACd,QAAQ,EAAE,EAAE;SACb;QACD,WAAW,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE;KACxC;IAED,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM;QAC1B,MAAM,YAAY,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;QACpD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,0DAA0D,EAAE,CAAC;QAC/F,CAAC;QACD,MAAM,OAAO,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QACtE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,IAAK,CAAC,SAAS,CAAC,CAAC;QACrE,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;IACtF,CAAC;IAED,YAAY,CAAC,MAAM;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,MAAM,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5D,CAAC;QACD,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC,CAAC;IACxD,CAAC;CACF,CAAC;AAEF,gFAAgF;AAChF,0BAA0B;AAC1B,gFAAgF;AAEhF,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,eAAe;IACf,eAAe;IACf,cAAc;IACd,iBAAiB;IACjB,mBAAmB;CACpB,CAAC"}