@datalayer/jupyter-react 1.1.7 → 1.1.9

package/lib/tools/core/zodUtils.js

@@ -0,0 +1,179 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+/**
+ * Zod utility functions for tool parameter validation
+ *
+ * @module tools/core/zodUtils
+ */
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+/**
+ * Converts a Zod schema to ToolDefinition parameters (JSON Schema format).
+ *
+ * This enables a single source of truth: define the schema once with Zod,
+ * and automatically generate the JSON Schema for LLM tool calling.
+ *
+ * @param schema - Zod schema defining the tool's input parameters
+ * @returns JSON Schema object compatible with ToolDefinition.parameters
+ *
+ * @example
+ * ```typescript
+ * const schema = z.object({
+ *   index: z.number().int().describe('Cell index'),
+ *   source: z.string().describe('Cell content')
+ * });
+ *
+ * const parameters = zodToToolParameters(schema);
+ * // Returns: { type: 'object', properties: {...}, required: [...] }
+ * ```
+ */
+export function zodToToolParameters(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+schema) {
+    // Manual conversion for Zod v3/v4 compatibility
+    // zod-to-json-schema@3.x doesn't fully support Zod v4
+    // Handle both Zod v3 (_def.type === 'object') and Zod v4 (_def.typeName === 'ZodObject')
+    const isZodObject = schema?._def?.typeName === 'ZodObject' || schema?._def?.type === 'object';
+    if (isZodObject && schema._def?.shape) {
+        // Get shape object - handle both Zod v3 (shape is object) and Zod v4 (shape is function)
+        const shape = typeof schema._def.shape === 'function'
+            ? schema._def.shape()
+            : schema._def.shape;
+        const properties = {};
+        const required = [];
+        for (const [key, value] of Object.entries(shape)) {
+            let zodField = value;
+            // Build property schema
+            const prop = {};
+            // Unwrap optional/default modifiers to get to the base type
+            // Support both Zod v3 (def/def) and Zod v4 (_def)
+            let isOptional = false;
+            while (zodField._def?.typeName === 'ZodOptional' ||
+                zodField._def?.typeName === 'ZodDefault' ||
+                zodField._def?.type === 'optional' ||
+                zodField._def?.type === 'default' ||
+                zodField.def?.type === 'optional' ||
+                zodField.def?.type === 'default') {
+                if (zodField._def?.typeName === 'ZodOptional' ||
+                    zodField._def?.type === 'optional' ||
+                    zodField.def?.type === 'optional') {
+                    isOptional = true;
+                }
+                // Unwrap to get the inner type - try all possible paths
+                zodField =
+                    zodField._def?.innerType || zodField.def?.innerType || zodField;
+            }
+            // Extract description (check at each level)
+            // Support both Zod v3 (def.description) and Zod v4 (_def.description)
+            const description = value._def?.description ||
+                value.def?.description ||
+                value.description;
+            if (description) {
+                prop.description = description;
+            }
+            // Handle type based on unwrapped field
+            // Support both Zod v3 (def.type) and Zod v4 (_def.typeName or _def.type)
+            const typeName = zodField._def?.typeName || zodField._def?.type || zodField.def?.type;
+            if (typeName === 'ZodString' || typeName === 'string') {
+                prop.type = 'string';
+            }
+            else if (typeName === 'ZodNumber' || typeName === 'number') {
+                prop.type = 'number';
+            }
+            else if (typeName === 'ZodBoolean' || typeName === 'boolean') {
+                prop.type = 'boolean';
+            }
+            else if (typeName === 'ZodEnum' || typeName === 'enum') {
+                prop.type = 'string';
+                // Enum values can be in different places depending on Zod version:
+                // - Zod v4: _def.values (array)
+                // - Zod v3: def.entries (object), options (array), or enum (object)
+                const enumValues = zodField._def?.values ||
+                    zodField.def?.values ||
+                    zodField.options ||
+                    (zodField.def?.entries ? Object.keys(zodField.def.entries) : null) ||
+                    (zodField.enum ? Object.keys(zodField.enum) : null) ||
+                    [];
+                prop.enum = Array.isArray(enumValues)
+                    ? enumValues
+                    : Object.keys(enumValues || {});
+            }
+            properties[key] = prop;
+            // Mark as required if not optional
+            if (!isOptional) {
+                required.push(key);
+            }
+        }
+        return {
+            type: 'object',
+            properties,
+            required,
+        };
+    }
+    // Fallback to zod-to-json-schema for other schema types
+    // Note: This may not work correctly with Zod v4
+    const jsonSchema = zodToJsonSchema(schema, {
+        target: 'openApi3',
+        $refStrategy: 'none',
+    });
+    const schemaProperties = jsonSchema.properties || {};
+    const requiredFields = jsonSchema.required || [];
+    return {
+        type: 'object',
+        properties: schemaProperties,
+        required: requiredFields,
+    };
+}
+/**
+ * Validates parameters using a Zod schema with user-friendly error messages.
+ *
+ * This replaces manual type guard functions with automatic runtime validation.
+ * On validation failure, throws a descriptive error that helps both developers
+ * and LLMs understand what went wrong.
+ *
+ * @param schema - Zod schema to validate against
+ * @param params - Unknown parameters to validate
+ * @param operationName - Name of the operation (for error messages)
+ * @returns Validated parameters with proper TypeScript type
+ * @throws Error with detailed validation failure information
+ *
+ * @example
+ * ```typescript
+ * const schema = z.object({ index: z.number().int() });
+ *
+ * // Valid params
+ * const validated = validateWithZod(schema, { index: 5 }, 'readCell');
+ * // Returns: { index: 5 } with type { index: number }
+ *
+ * // Invalid params
+ * validateWithZod(schema, { index: 'invalid' }, 'readCell');
+ * // Throws: Error: Invalid parameters for readCell:
+ * //   - index: Expected number, received string
+ * ```
+ */
+export function validateWithZod(schema, params, operationName) {
+    try {
+        // Parse and validate parameters
+        return schema.parse(params);
+    }
+    catch (error) {
+        if (error instanceof z.ZodError) {
+            // Convert Zod validation errors to user-friendly format
+            const issues = error.issues
+                .map(issue => {
+                // Build path string (e.g., "cells[0].type" for nested errors)
+                const path = issue.path.length > 0 ? issue.path.join('.') : 'root';
+                return `  - ${path}: ${issue.message}`;
+            })
+                .join('\n');
+            throw new Error(`Invalid parameters for ${operationName}:\n${issues}\n\n` +
+                `Received: ${JSON.stringify(params)}`);
+        }
+        // Re-throw non-Zod errors
+        throw error;
+    }
+}
+//# sourceMappingURL=zodUtils.js.map

package/lib/tools/core/zodUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zodUtils.js","sourceRoot":"","sources":["../../../src/tools/core/zodUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAGrD;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAM,UAAU,mBAAmB;AACjC,8DAA8D;AAC9D,MAAW;IAEX,gDAAgD;IAChD,sDAAsD;IACtD,yFAAyF;IACzF,MAAM,WAAW,GACf,MAAM,EAAE,IAAI,EAAE,QAAQ,KAAK,WAAW,IAAI,MAAM,EAAE,IAAI,EAAE,IAAI,KAAK,QAAQ,CAAC;IAE5E,IAAI,WAAW,IAAI,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;QACtC,yFAAyF;QACzF,MAAM,KAAK,GACT,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,KAAK,UAAU;YACrC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE;YACrB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,UAAU,GAAwB,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAa,EAAE,CAAC;QAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YACjD,IAAI,QAAQ,GAAG,KAAY,CAAC;YAE5B,wBAAwB;YACxB,MAAM,IAAI,GAAwB,EAAE,CAAC;YAErC,4DAA4D;YAC5D,kDAAkD;YAClD,IAAI,UAAU,GAAG,KAAK,CAAC;YACvB,OACE,QAAQ,CAAC,IAAI,EAAE,QAAQ,KAAK,aAAa;gBACzC,QAAQ,CAAC,IAAI,EAAE,QAAQ,KAAK,YAAY;gBACxC,QAAQ,CAAC,IAAI,EAAE,IAAI,KAAK,UAAU;gBAClC,QAAQ,CAAC,IAAI,EAAE,IAAI,KAAK,SAAS;gBACjC,QAAQ,CAAC,GAAG,EAAE,IAAI,KAAK,UAAU;gBACjC,QAAQ,CAAC,GAAG,EAAE,IAAI,KAAK,SAAS,EAChC,CAAC;gBACD,IACE,QAAQ,CAAC,IAAI,EAAE,QAAQ,KAAK,aAAa;oBACzC,QAAQ,CAAC,IAAI,EAAE,IAAI,KAAK,UAAU;oBAClC,QAAQ,CAAC,GAAG,EAAE,IAAI,KAAK,UAAU,EACjC,CAAC;oBACD,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;gBACD,wDAAwD;gBACxD,QAAQ;oBACN,QAAQ,CAAC,IAAI,EAAE,SAAS,IAAI,QAAQ,CAAC,GAAG,EAAE,SAAS,IAAI,QAAQ,CAAC;YACpE,CAAC;YAED,4CAA4C;YAC5C,sEAAsE;YACtE,MAAM,WAAW,GACd,KAAa,CAAC,IAAI,EAAE,WAAW;gBAC/B,KAAa,CAAC,GAAG,EAAE,WAAW;gBAC9B,KAAa,CAAC,WAAW,CAAC;YAC7B,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;YACjC,CAAC;YAED,uCAAuC;YACvC,yEAAyE;YACzE,MAAM,QAAQ,GACZ,QAAQ,CAAC,IAAI,EAAE,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,IAAI,IAAI,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC;YACvE,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBACtD,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC;YACvB,CAAC;iBAAM,IAAI,QAAQ,KAAK,WAAW,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC7D,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC;YACvB,CAAC;iBAAM,IAAI,QAAQ,KAAK,YAAY,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC/D,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;YACxB,CAAC;iBAAM,IAAI,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;gBACzD,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAC;gBACrB,mEAAmE;gBACnE,gCAAgC;gBAChC,oEAAoE;gBACpE,MAAM,UAAU,GACd,QAAQ,CAAC,IAAI,EAAE,MAAM;oBACrB,QAAQ,CAAC,GAAG,EAAE,MAAM;oBACpB,QAAQ,CAAC,OAAO;oBAChB,CAAC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;oBAClE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;oBACnD,EAAE,CAAC;gBACL,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC;oBACnC,CAAC,CAAC,UAAU;oBACZ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;YACpC,CAAC;YAED,UAAU,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;YAEvB,mCAAmC;YACnC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE,QAAiB;YACvB,UAAU;YACV,QAAQ;SACT,CAAC;IACJ,CAAC;IAED,wDAAwD;IACxD,gDAAgD;IAChD,MAAM,UAAU,GAAG,eAAe,CAAC,MAAM,EAAE;QACzC,MAAM,EAAE,UAAU;QAClB,YAAY,EAAE,MAAM;KACrB,CAGA,CAAC;IAEF,MAAM,gBAAgB,GAAG,UAAU,CAAC,UAAU,IAAI,EAAE,CAAC;IACrD,MAAM,cAAc,GAAG,UAAU,CAAC,QAAQ,IAAI,EAAE,CAAC;IAEjD,OAAO;QACL,IAAI,EAAE,QAAiB;QACvB,UAAU,EAAE,gBAAgB;QAC5B,QAAQ,EAAE,cAAc;KACzB,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,MAAM,UAAU,eAAe,CAC7B,MAAoB,EACpB,MAAe,EACf,aAAqB;IAErB,IAAI,CAAC;QACH,gCAAgC;QAChC,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC;YAChC,wDAAwD;YACxD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM;iBACxB,GAAG,CAAC,KAAK,CAAC,EAAE;gBACX,8DAA8D;gBAC9D,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;gBAEnE,OAAO,OAAO,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;YACzC,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,MAAM,IAAI,KAAK,CACb,0BAA0B,aAAa,MAAM,MAAM,MAAM;gBACvD,aAAa,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CACxC,CAAC;QACJ,CAAC;QAED,0BAA0B;QAC1B,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/lib/tools/definitions/deleteCell.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Delete cell tool definition.
+ *
+ * @module tools/definitions/deleteCell
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const deleteCellTool: ToolDefinition;

package/lib/tools/definitions/deleteCell.js

@@ -0,0 +1,25 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { deleteCellParamsSchema } from '../schemas/deleteCell';
+export const deleteCellTool = {
+    name: 'datalayer_deleteCell',
+    displayName: 'Delete Notebook Cell(s)',
+    toolReferenceName: 'deleteCell',
+    description: 'Deletes one or more cells from a Jupyter notebook at the specified indices. Cells are deleted in reverse order to prevent index shifting.',
+    // Generate parameters from Zod schema (single source of truth)
+    parameters: zodToToolParameters(deleteCellParamsSchema),
+    operation: 'deleteCell',
+    config: {
+        confirmationMessage: (params) => `Delete ${params.indices.length} cell${params.indices.length !== 1 ? 's' : ''} at ${params.indices.length === 1 ? `index ${params.indices[0]}` : `indices ${params.indices.join(', ')}`}?`,
+        invocationMessage: (params) => `Deleting ${params.indices.length} cell${params.indices.length !== 1 ? 's' : ''}`,
+        requiresConfirmation: true,
+        canBeReferencedInPrompt: true,
+        priority: 'medium',
+    },
+    tags: ['cell', 'notebook', 'manipulation', 'delete'],
+};
+//# sourceMappingURL=deleteCell.js.map

package/lib/tools/definitions/deleteCell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"deleteCell.js","sourceRoot":"","sources":["../../../src/tools/definitions/deleteCell.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,MAAM,CAAC,MAAM,cAAc,GAAmB;IAC5C,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE,yBAAyB;IACtC,iBAAiB,EAAE,YAAY;IAC/B,WAAW,EACT,2IAA2I;IAE7I,+DAA+D;IAC/D,UAAU,EAAE,mBAAmB,CAAC,sBAAsB,CAAC;IAEvD,SAAS,EAAE,YAAY;IAEvB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAA6B,EAAE,EAAE,CACrD,UAAU,MAAM,CAAC,OAAO,CAAC,MAAM,QAAQ,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,GAAG;QAC5L,iBAAiB,EAAE,CAAC,MAA6B,EAAE,EAAE,CACnD,YAAY,MAAM,CAAC,OAAO,CAAC,MAAM,QAAQ,MAAM,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACnF,oBAAoB,EAAE,IAAI;QAC1B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,QAAQ;KACnB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,QAAQ,CAAC;CACrD,CAAC"}

package/lib/tools/definitions/executeCode.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Execute code tool definition.
+ *
+ * @module tools/definitions/executeCode
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const executeCodeTool: ToolDefinition;

package/lib/tools/definitions/executeCode.js

@@ -0,0 +1,33 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { executeCodeParamsSchema } from '../schemas/executeCode';
+export const executeCodeTool = {
+    name: 'datalayer_executeCode',
+    displayName: 'Execute Code in Kernel',
+    toolReferenceName: 'executeCode',
+    description: 'Execute code directly in the kernel (not saved to notebook) on the current activated notebook.\n\n' +
+        'Recommended to use in following cases:\n' +
+        '1. Execute Jupyter magic commands (e.g., %timeit, %pip install xxx)\n' +
+        '2. Performance profiling and debugging\n' +
+        '3. View intermediate variable values (e.g., print(xxx), df.head())\n' +
+        "4. Temporary calculations and quick tests (e.g., np.mean(df['xxx']))\n" +
+        '5. Execute Shell commands in Jupyter server (e.g., !git xxx)\n\n' +
+        'Under no circumstances should you use this tool to:\n' +
+        '1. Import new modules or perform variable assignments that affect subsequent Notebook execution\n' +
+        "2. Execute dangerous code that may harm the Jupyter server or the user's data without permission",
+    parameters: zodToToolParameters(executeCodeParamsSchema),
+    operation: 'executeCode',
+    config: {
+        confirmationMessage: (params) => `Execute code: ${params.code.substring(0, 50)}${params.code.length > 50 ? '...' : ''}?`,
+        invocationMessage: () => 'Executing code in kernel',
+        requiresConfirmation: false,
+        canBeReferencedInPrompt: true,
+        priority: 'high',
+    },
+    tags: ['kernel', 'notebook', 'execute', 'code', 'inspection', 'magic'],
+};
+//# sourceMappingURL=executeCode.js.map

package/lib/tools/definitions/executeCode.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"executeCode.js","sourceRoot":"","sources":["../../../src/tools/definitions/executeCode.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,uBAAuB,EAAE,MAAM,wBAAwB,CAAC;AAEjE,MAAM,CAAC,MAAM,eAAe,GAAmB;IAC7C,IAAI,EAAE,uBAAuB;IAC7B,WAAW,EAAE,wBAAwB;IACrC,iBAAiB,EAAE,aAAa;IAChC,WAAW,EACT,oGAAoG;QACpG,0CAA0C;QAC1C,uEAAuE;QACvE,0CAA0C;QAC1C,sEAAsE;QACtE,wEAAwE;QACxE,kEAAkE;QAClE,uDAAuD;QACvD,mGAAmG;QACnG,kGAAkG;IAEpG,UAAU,EAAE,mBAAmB,CAAC,uBAAuB,CAAC;IAExD,SAAS,EAAE,aAAa;IAExB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAAwB,EAAE,EAAE,CAChD,iBAAiB,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG;QACzF,iBAAiB,EAAE,GAAG,EAAE,CAAC,0BAA0B;QACnD,oBAAoB,EAAE,KAAK;QAC3B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,MAAM;KACjB;IAED,IAAI,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,CAAC;CACvE,CAAC"}

package/lib/tools/definitions/index.d.ts

@@ -0,0 +1,12 @@
+/**
+ * Tool definitions (schemas) for notebook operations.
+ *
+ * @module tools/definitions
+ */
+export * from './insertCell';
+export * from './deleteCell';
+export * from './updateCell';
+export * from './readCell';
+export * from './readAllCells';
+export * from './runCell';
+export * from './executeCode';

package/lib/tools/definitions/index.js

@@ -0,0 +1,18 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+/**
+ * Tool definitions (schemas) for notebook operations.
+ *
+ * @module tools/definitions
+ */
+export * from './insertCell';
+export * from './deleteCell';
+export * from './updateCell';
+export * from './readCell';
+export * from './readAllCells';
+export * from './runCell';
+export * from './executeCode';
+//# sourceMappingURL=index.js.map

package/lib/tools/definitions/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/tools/definitions/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;GAIG;AAEH,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,YAAY,CAAC;AAC3B,cAAc,gBAAgB,CAAC;AAC/B,cAAc,WAAW,CAAC;AAC1B,cAAc,eAAe,CAAC"}

package/lib/tools/definitions/insertCell.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Insert cell tool definition.
+ *
+ * @module tools/definitions/insertCell
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const insertCellTool: ToolDefinition;

package/lib/tools/definitions/insertCell.js

@@ -0,0 +1,24 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { insertCellParamsSchema } from '../schemas/insertCell';
+export const insertCellTool = {
+    name: 'datalayer_insertCell',
+    displayName: 'Insert Notebook Cell',
+    toolReferenceName: 'insertCell',
+    description: 'IMPORTANT: Call readAllCells first to see the current notebook structure and determine the correct insertion point. Then, insert a code or markdown cell into a Jupyter notebook at a specified position or at the end',
+    parameters: zodToToolParameters(insertCellParamsSchema),
+    operation: 'insertCell',
+    config: {
+        confirmationMessage: (params) => `Insert ${params.type} cell into notebook?\n\n${params.source}`,
+        invocationMessage: (params) => `Inserting ${params.type} cell into notebook`,
+        requiresConfirmation: true,
+        canBeReferencedInPrompt: true,
+        priority: 'high',
+    },
+    tags: ['cell', 'notebook', 'manipulation', 'create'],
+};
+//# sourceMappingURL=insertCell.js.map

package/lib/tools/definitions/insertCell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"insertCell.js","sourceRoot":"","sources":["../../../src/tools/definitions/insertCell.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,MAAM,CAAC,MAAM,cAAc,GAAmB;IAC5C,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE,sBAAsB;IACnC,iBAAiB,EAAE,YAAY;IAC/B,WAAW,EACT,wNAAwN;IAE1N,UAAU,EAAE,mBAAmB,CAAC,sBAAsB,CAAC;IAEvD,SAAS,EAAE,YAAY;IAEvB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAAwC,EAAE,EAAE,CAChE,UAAU,MAAM,CAAC,IAAI,2BAA2B,MAAM,CAAC,MAAM,EAAE;QACjE,iBAAiB,EAAE,CAAC,MAAwB,EAAE,EAAE,CAC9C,aAAa,MAAM,CAAC,IAAI,qBAAqB;QAC/C,oBAAoB,EAAE,IAAI;QAC1B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,MAAM;KACjB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,QAAQ,CAAC;CACrD,CAAC"}

package/lib/tools/definitions/readAllCells.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Read all cells tool definition.
+ *
+ * @module tools/definitions/readAllCells
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const readAllCellsTool: ToolDefinition;

package/lib/tools/definitions/readAllCells.js

@@ -0,0 +1,24 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { readAllCellsParamsSchema } from '../schemas/readAllCells';
+export const readAllCellsTool = {
+    name: 'datalayer_readAllCells',
+    displayName: 'Read All Notebook Cells',
+    toolReferenceName: 'readAllCells',
+    description: "Read all cells from the Jupyter notebook. Supports two response formats: 'brief' (default) returns index, type, and 40-char content preview for structure queries and counting cells; 'detailed' returns full content with source, execution_count, and outputs. Use brief when you need to see notebook structure, count cells, or quickly scan content. Use detailed when you need to read full cell content or outputs. Brief format preview shows first 40 characters of cell source. Returns array of cells with: index (cell position), type (code, markdown, raw), preview (brief only), and optionally source/execution_count/outputs (detailed only). Works on active notebook.",
+    parameters: zodToToolParameters(readAllCellsParamsSchema),
+    operation: 'readAllCells',
+    config: {
+        confirmationMessage: () => 'Read all cells from notebook?',
+        invocationMessage: () => 'Reading all cells',
+        requiresConfirmation: false,
+        canBeReferencedInPrompt: true,
+        priority: 'high',
+    },
+    tags: ['cell', 'notebook', 'read', 'inspect', 'all'],
+};
+//# sourceMappingURL=readAllCells.js.map

package/lib/tools/definitions/readAllCells.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"readAllCells.js","sourceRoot":"","sources":["../../../src/tools/definitions/readAllCells.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AAEnE,MAAM,CAAC,MAAM,gBAAgB,GAAmB;IAC9C,IAAI,EAAE,wBAAwB;IAC9B,WAAW,EAAE,yBAAyB;IACtC,iBAAiB,EAAE,cAAc;IACjC,WAAW,EACT,0pBAA0pB;IAE5pB,UAAU,EAAE,mBAAmB,CAAC,wBAAwB,CAAC;IAEzD,SAAS,EAAE,cAAc;IAEzB,MAAM,EAAE;QACN,mBAAmB,EAAE,GAAG,EAAE,CAAC,+BAA+B;QAC1D,iBAAiB,EAAE,GAAG,EAAE,CAAC,mBAAmB;QAC5C,oBAAoB,EAAE,KAAK;QAC3B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,MAAM;KACjB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC;CACrD,CAAC"}

package/lib/tools/definitions/readCell.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Read cell tool definition.
+ *
+ * @module tools/definitions/readCell
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const readCellTool: ToolDefinition;

package/lib/tools/definitions/readCell.js

@@ -0,0 +1,24 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { readCellParamsSchema } from '../schemas/readCell';
+export const readCellTool = {
+    name: 'datalayer_readCell',
+    displayName: 'Read Notebook Cell',
+    toolReferenceName: 'readCell',
+    description: 'Reads a specific cell from a Jupyter notebook by index, including source code and outputs',
+    parameters: zodToToolParameters(readCellParamsSchema),
+    operation: 'readCell',
+    config: {
+        confirmationMessage: (params) => `Read cell at index ${params.index}?`,
+        invocationMessage: (params) => `Reading cell ${params.index}`,
+        requiresConfirmation: false,
+        canBeReferencedInPrompt: true,
+        priority: 'high',
+    },
+    tags: ['cell', 'notebook', 'read', 'inspect'],
+};
+//# sourceMappingURL=readCell.js.map

package/lib/tools/definitions/readCell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"readCell.js","sourceRoot":"","sources":["../../../src/tools/definitions/readCell.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAE3D,MAAM,CAAC,MAAM,YAAY,GAAmB;IAC1C,IAAI,EAAE,oBAAoB;IAC1B,WAAW,EAAE,oBAAoB;IACjC,iBAAiB,EAAE,UAAU;IAC7B,WAAW,EACT,2FAA2F;IAE7F,UAAU,EAAE,mBAAmB,CAAC,oBAAoB,CAAC;IAErD,SAAS,EAAE,UAAU;IAErB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAAyB,EAAE,EAAE,CACjD,sBAAsB,MAAM,CAAC,KAAK,GAAG;QACvC,iBAAiB,EAAE,CAAC,MAAyB,EAAE,EAAE,CAC/C,gBAAgB,MAAM,CAAC,KAAK,EAAE;QAChC,oBAAoB,EAAE,KAAK;QAC3B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,MAAM;KACjB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,CAAC;CAC9C,CAAC"}

package/lib/tools/definitions/runCell.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Run cell tool definition.
+ *
+ * @module tools/definitions/runCell
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const runCellTool: ToolDefinition;

package/lib/tools/definitions/runCell.js

@@ -0,0 +1,31 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { runCellParamsSchema } from '../schemas/runCell';
+export const runCellTool = {
+    name: 'datalayer_runCell',
+    displayName: 'Run Notebook Cell',
+    toolReferenceName: 'runCell',
+    description: 'Runs a code cell in a Jupyter notebook and returns its outputs with execution metadata. ' +
+        'Supports timeout and streaming progress updates. ' +
+        'Returns execution_count, outputs, and elapsed_time. ' +
+        'If no index is provided, runs the currently active cell.',
+    parameters: zodToToolParameters(runCellParamsSchema),
+    operation: 'runCell',
+    config: {
+        confirmationMessage: (params) => params.index !== undefined
+            ? `Run cell at index ${params.index}?`
+            : 'Run cell?',
+        invocationMessage: (params) => params.index !== undefined
+            ? `Running cell ${params.index}${params.timeoutSeconds ? ` (timeout: ${params.timeoutSeconds}s)` : ''}`
+            : `Running cell${params.timeoutSeconds ? ` (timeout: ${params.timeoutSeconds}s)` : ''}`,
+        requiresConfirmation: false,
+        canBeReferencedInPrompt: true,
+        priority: 'high',
+    },
+    tags: ['cell', 'notebook', 'execute', 'run', 'outputs'],
+};
+//# sourceMappingURL=runCell.js.map

package/lib/tools/definitions/runCell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runCell.js","sourceRoot":"","sources":["../../../src/tools/definitions/runCell.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAEzD,MAAM,CAAC,MAAM,WAAW,GAAmB;IACzC,IAAI,EAAE,mBAAmB;IACzB,WAAW,EAAE,mBAAmB;IAChC,iBAAiB,EAAE,SAAS;IAC5B,WAAW,EACT,0FAA0F;QAC1F,mDAAmD;QACnD,sDAAsD;QACtD,0DAA0D;IAE5D,UAAU,EAAE,mBAAmB,CAAC,mBAAmB,CAAC;IAEpD,SAAS,EAAE,SAAS;IAEpB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAA0B,EAAE,EAAE,CAClD,MAAM,CAAC,KAAK,KAAK,SAAS;YACxB,CAAC,CAAC,qBAAqB,MAAM,CAAC,KAAK,GAAG;YACtC,CAAC,CAAC,WAAW;QACjB,iBAAiB,EAAE,CAAC,MAAmD,EAAE,EAAE,CACzE,MAAM,CAAC,KAAK,KAAK,SAAS;YACxB,CAAC,CAAC,gBAAgB,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,MAAM,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;YACvG,CAAC,CAAC,eAAe,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,cAAc,MAAM,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;QAC3F,oBAAoB,EAAE,KAAK;QAC3B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,MAAM;KACjB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,SAAS,CAAC;CACxD,CAAC"}

package/lib/tools/definitions/updateCell.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Update cell tool definition.
+ *
+ * @module tools/definitions/updateCell
+ */
+import type { ToolDefinition } from '../core/schema';
+export declare const updateCellTool: ToolDefinition;

package/lib/tools/definitions/updateCell.js

@@ -0,0 +1,24 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+import { zodToToolParameters } from '../core/zodUtils';
+import { updateCellParamsSchema } from '../schemas/updateCell';
+export const updateCellTool = {
+    name: 'datalayer_updateCell',
+    displayName: 'Update Notebook Cell',
+    toolReferenceName: 'updateCell',
+    description: "Updates (overwrites) a cell's source code at the specified index. Does NOT execute the cell.",
+    parameters: zodToToolParameters(updateCellParamsSchema),
+    operation: 'updateCell',
+    config: {
+        confirmationMessage: (params) => `Update cell at index ${params.index}?\n\n${params.source}`,
+        invocationMessage: (params) => `Updating cell ${params.index}`,
+        requiresConfirmation: false,
+        canBeReferencedInPrompt: true,
+        priority: 'medium',
+    },
+    tags: ['cell', 'notebook', 'manipulation', 'update'],
+};
+//# sourceMappingURL=updateCell.js.map

package/lib/tools/definitions/updateCell.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateCell.js","sourceRoot":"","sources":["../../../src/tools/definitions/updateCell.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,MAAM,CAAC,MAAM,cAAc,GAAmB;IAC5C,IAAI,EAAE,sBAAsB;IAC5B,WAAW,EAAE,sBAAsB;IACnC,iBAAiB,EAAE,YAAY;IAC/B,WAAW,EACT,8FAA8F;IAEhG,UAAU,EAAE,mBAAmB,CAAC,sBAAsB,CAAC;IAEvD,SAAS,EAAE,YAAY;IAEvB,MAAM,EAAE;QACN,mBAAmB,EAAE,CAAC,MAAyC,EAAE,EAAE,CACjE,wBAAwB,MAAM,CAAC,KAAK,QAAQ,MAAM,CAAC,MAAM,EAAE;QAC7D,iBAAiB,EAAE,CAAC,MAAyB,EAAE,EAAE,CAC/C,iBAAiB,MAAM,CAAC,KAAK,EAAE;QACjC,oBAAoB,EAAE,KAAK;QAC3B,uBAAuB,EAAE,IAAI;QAC7B,QAAQ,EAAE,QAAQ;KACnB;IAED,IAAI,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,QAAQ,CAAC;CACrD,CAAC"}

package/lib/tools/index.d.ts

@@ -0,0 +1,20 @@
+import type { ToolDefinition } from './core/schema';
+import type { ToolOperation } from './core/interfaces';
+/**
+ * Array of all notebook tool definitions
+ */
+export declare const notebookToolDefinitions: ToolDefinition[];
+/**
+ * Registry of all notebook tool operations
+ * Maps operation names to their implementations
+ */
+export declare const notebookToolOperations: Record<string, ToolOperation<unknown, unknown>>;
+/**
+ * Complete notebook tools bundle for easy iteration and registration
+ */
+export declare const notebookTools: {
+    definitions: ToolDefinition<unknown>[];
+    operations: Record<string, ToolOperation<unknown, unknown>>;
+};
+export * from './core';
+export * from './definitions';

package/lib/tools/index.js

@@ -0,0 +1,62 @@
+/*
+ * Copyright (c) 2021-2023 Datalayer, Inc.
+ *
+ * MIT License
+ */
+/**
+ * Platform-agnostic notebook tools.
+ *
+ * @module tools
+ */
+// Import all tool definitions
+import { insertCellTool } from './definitions/insertCell';
+import { deleteCellTool } from './definitions/deleteCell';
+import { updateCellTool } from './definitions/updateCell';
+import { readCellTool } from './definitions/readCell';
+import { readAllCellsTool } from './definitions/readAllCells';
+import { runCellTool } from './definitions/runCell';
+import { executeCodeTool } from './definitions/executeCode';
+// Import all operations
+import { insertCellOperation } from './operations/insertCell';
+import { deleteCellOperation } from './operations/deleteCell';
+import { updateCellOperation } from './operations/updateCell';
+import { readCellOperation } from './operations/readCell';
+import { readAllCellsOperation } from './operations/readAllCells';
+import { runCellOperation } from './operations/runCell';
+import { executeCodeOperation } from './operations/executeCode';
+/**
+ * Array of all notebook tool definitions
+ */
+export const notebookToolDefinitions = [
+    insertCellTool,
+    deleteCellTool,
+    updateCellTool,
+    readCellTool,
+    readAllCellsTool,
+    runCellTool,
+    executeCodeTool,
+];
+/**
+ * Registry of all notebook tool operations
+ * Maps operation names to their implementations
+ */
+export const notebookToolOperations = {
+    insertCell: insertCellOperation,
+    deleteCell: deleteCellOperation,
+    updateCell: updateCellOperation,
+    readCell: readCellOperation,
+    readAllCells: readAllCellsOperation,
+    runCell: runCellOperation,
+    executeCode: executeCodeOperation,
+};
+/**
+ * Complete notebook tools bundle for easy iteration and registration
+ */
+export const notebookTools = {
+    definitions: notebookToolDefinitions,
+    operations: notebookToolOperations,
+};
+// Re-export everything for convenience
+export * from './core';
+export * from './definitions';
+//# sourceMappingURL=index.js.map

package/lib/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;GAIG;AAEH,8BAA8B;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,uBAAuB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAE5D,wBAAwB;AACxB,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAClE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAMhE;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAqB;IACvD,cAAc;IACd,cAAc;IACd,cAAc;IACd,YAAY;IACZ,gBAAgB;IAChB,WAAW;IACX,eAAe;CAChB,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAG/B;IACF,UAAU,EAAE,mBAAmB;IAC/B,UAAU,EAAE,mBAAmB;IAC/B,UAAU,EAAE,mBAAmB;IAC/B,QAAQ,EAAE,iBAAiB;IAC3B,YAAY,EAAE,qBAAqB;IACnC,OAAO,EAAE,gBAAgB;IACzB,WAAW,EAAE,oBAAoB;CAClC,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,WAAW,EAAE,uBAAuB;IACpC,UAAU,EAAE,sBAAsB;CACnC,CAAC;AAEF,uCAAuC;AACvC,cAAc,QAAQ,CAAC;AACvB,cAAc,eAAe,CAAC"}