@shareai-lab/kode 1.1.23 → 2.0.1

package/dist/tools/MultiEditTool/MultiEditTool.js.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/tools/MultiEditTool/MultiEditTool.tsx"],
-  "sourcesContent": ["import { existsSync, mkdirSync, readFileSync, statSync } from 'fs'\nimport { Box, Text } from 'ink'\nimport { dirname, isAbsolute, relative, resolve, sep } from 'path'\nimport * as React from 'react'\nimport { z } from 'zod'\nimport { FileEditToolUpdatedMessage } from '../../components/FileEditToolUpdatedMessage'\nimport { StructuredDiff } from '../../components/StructuredDiff'\nimport { logEvent } from '../../services/statsig'\nimport { Tool, ValidationResult } from '../../Tool'\nimport { intersperse } from '../../utils/array'\nimport {\n  addLineNumbers,\n  detectFileEncoding,\n  detectLineEndings,\n  findSimilarFile,\n  writeTextContent,\n} from '../../utils/file.js'\nimport { logError } from '../../utils/log'\nimport { getCwd } from '../../utils/state'\nimport { getTheme } from '../../utils/theme'\nimport { NotebookEditTool } from '../NotebookEditTool/NotebookEditTool'\n// Local content-based edit function for MultiEditTool\nfunction applyContentEdit(\n  content: string,\n  oldString: string,\n  newString: string,\n  replaceAll: boolean = false\n): { newContent: string; occurrences: number } {\n  if (replaceAll) {\n    const regex = new RegExp(oldString.replace(/[.*+?^${}()|[\\]\\\\]/g, '\\\\$&'), 'g')\n    const matches = content.match(regex)\n    const occurrences = matches ? matches.length : 0\n    const newContent = content.replace(regex, newString)\n    return { newContent, occurrences }\n  } else {\n    if (content.includes(oldString)) {\n      const newContent = content.replace(oldString, newString)\n      return { newContent, occurrences: 1 }\n    } else {\n      throw new Error(`String not found: ${oldString.substring(0, 50)}...`)\n    }\n  }\n}\nimport { hasWritePermission } from '../../utils/permissions/filesystem'\nimport { PROJECT_FILE } from '../../constants/product'\nimport { DESCRIPTION, PROMPT } from './prompt'\nimport { emitReminderEvent } from '../../services/systemReminder'\nimport { recordFileEdit } from '../../services/fileFreshness'\n\nconst EditSchema = z.object({\n  old_string: z.string().describe('The text to replace'),\n  new_string: z.string().describe('The text to replace it with'),\n  replace_all: z\n    .boolean()\n    .optional()\n    .default(false)\n    .describe('Replace all occurences of old_string (default false)'),\n})\n\nconst inputSchema = z.strictObject({\n  file_path: z.string().describe('The absolute path to the file to modify'),\n  edits: z\n    .array(EditSchema)\n    .min(1)\n    .describe('Array of edit operations to perform sequentially on the file'),\n})\n\nexport type In = typeof inputSchema\n\n// Number of lines of context to include before/after the change in our result message\nconst N_LINES_SNIPPET = 4\n\nexport const MultiEditTool = {\n  name: 'MultiEdit',\n  async description() {\n    return 'A tool for making multiple edits to a single file atomically'\n  },\n  async prompt() {\n    return PROMPT\n  },\n  inputSchema,\n  userFacingName() {\n    return 'Multi-Edit'\n  },\n  async isEnabled() {\n    return true\n  },\n  isReadOnly() {\n    return false\n  },\n  isConcurrencySafe() {\n    return false // MultiEdit modifies files, not safe for concurrent execution\n  },\n  needsPermissions(input?: z.infer<typeof inputSchema>) {\n    if (!input) return true\n    return !hasWritePermission(input.file_path)\n  },\n  renderResultForAssistant(content) {\n    return content\n  },\n  renderToolUseMessage(input, { verbose }) {\n    const { file_path, edits } = input\n    const workingDir = getCwd()\n    const relativePath = isAbsolute(file_path)\n      ? relative(workingDir, file_path)\n      : file_path\n\n    if (verbose) {\n      const editSummary = edits\n        .map(\n          (edit, index) =>\n            `${index + 1}. Replace \"${edit.old_string.substring(0, 50)}${edit.old_string.length > 50 ? '...' : ''}\" with \"${edit.new_string.substring(0, 50)}${edit.new_string.length > 50 ? '...' : ''}\"`,\n        )\n        .join('\\n')\n      return `Multiple edits to ${relativePath}:\\n${editSummary}`\n    }\n\n    return `Making ${edits.length} edits to ${relativePath}`\n  },\n  renderToolUseRejectedMessage() {\n    return (\n      <Box>\n        <Text color={getTheme().error}>\u26A0 Edit request rejected</Text>\n      </Box>\n    )\n  },\n  renderToolResultMessage(output) {\n    if (typeof output === 'string') {\n      const isError = output.includes('Error:')\n      return (\n        <Box flexDirection=\"column\">\n          <Text color={isError ? getTheme().error : getTheme().success}>\n            {output}\n          </Text>\n        </Box>\n      )\n    }\n\n    return <FileEditToolUpdatedMessage {...output} />\n  },\n  async validateInput(\n    { file_path, edits }: z.infer<typeof inputSchema>,\n    context?: { readFileTimestamps?: Record<string, number> },\n  ): Promise<ValidationResult> {\n    const workingDir = getCwd()\n    const normalizedPath = isAbsolute(file_path)\n      ? resolve(file_path)\n      : resolve(workingDir, file_path)\n\n    // Check if it's a notebook file\n    if (normalizedPath.endsWith('.ipynb')) {\n      return {\n        result: false,\n        errorCode: 1,\n        message: `For Jupyter notebooks (.ipynb files), use the ${NotebookEditTool.name} tool instead.`,\n      }\n    }\n\n    // For new files, check parent directory exists\n    if (!existsSync(normalizedPath)) {\n      const parentDir = dirname(normalizedPath)\n      if (!existsSync(parentDir)) {\n        return {\n          result: false,\n          errorCode: 2,\n          message: `Parent directory does not exist: ${parentDir}`,\n        }\n      }\n\n      // For new files, ensure first edit creates the file (empty old_string)\n      if (edits.length === 0 || edits[0].old_string !== '') {\n        return {\n          result: false,\n          errorCode: 6,\n          message:\n            'For new files, the first edit must have an empty old_string to create the file content.',\n        }\n      }\n    } else {\n      // For existing files, apply file protection mechanisms\n      const readFileTimestamps = context?.readFileTimestamps || {}\n      const readTimestamp = readFileTimestamps[normalizedPath]\n\n      if (!readTimestamp) {\n        return {\n          result: false,\n          errorCode: 7,\n          message:\n            'File has not been read yet. Read it first before editing it.',\n          meta: {\n            filePath: normalizedPath,\n            isFilePathAbsolute: String(isAbsolute(file_path)),\n          },\n        }\n      }\n\n      // Check if file has been modified since last read\n      const stats = statSync(normalizedPath)\n      const lastWriteTime = stats.mtimeMs\n      if (lastWriteTime > readTimestamp) {\n        return {\n          result: false,\n          errorCode: 8,\n          message:\n            'File has been modified since read, either by the user or by a linter. Read it again before attempting to edit it.',\n          meta: {\n            filePath: normalizedPath,\n            lastWriteTime,\n            readTimestamp,\n          },\n        }\n      }\n\n      // Pre-validate that all old_strings exist in the file\n      const encoding = detectFileEncoding(normalizedPath)\n      if (encoding === 'binary') {\n        return {\n          result: false,\n          errorCode: 9,\n          message: 'Cannot edit binary files.',\n        }\n      }\n\n      const currentContent = readFileSync(normalizedPath, 'utf-8')\n      for (let i = 0; i < edits.length; i++) {\n        const edit = edits[i]\n        if (\n          edit.old_string !== '' &&\n          !currentContent.includes(edit.old_string)\n        ) {\n          return {\n            result: false,\n            errorCode: 10,\n            message: `Edit ${i + 1}: String to replace not found in file: \"${edit.old_string.substring(0, 100)}${edit.old_string.length > 100 ? '...' : ''}\"`,\n            meta: {\n              editIndex: i + 1,\n              oldString: edit.old_string.substring(0, 200),\n            },\n          }\n        }\n      }\n    }\n\n    // Validate each edit\n    for (let i = 0; i < edits.length; i++) {\n      const edit = edits[i]\n      if (edit.old_string === edit.new_string) {\n        return {\n          result: false,\n          errorCode: 3,\n          message: `Edit ${i + 1}: old_string and new_string cannot be the same`,\n        }\n      }\n    }\n\n    return { result: true }\n  },\n  async *call({ file_path, edits }, { readFileTimestamps }) {\n    const startTime = Date.now()\n    const workingDir = getCwd()\n    const filePath = isAbsolute(file_path)\n      ? resolve(file_path)\n      : resolve(workingDir, file_path)\n\n    try {\n      // Read current file content (or empty for new files)\n      let currentContent = ''\n      let fileExists = existsSync(filePath)\n\n      if (fileExists) {\n        const encoding = detectFileEncoding(filePath)\n        if (encoding === 'binary') {\n          yield {\n            type: 'result',\n            data: 'Error: Cannot edit binary files',\n            resultForAssistant: 'Error: Cannot edit binary files',\n          }\n          return\n        }\n        currentContent = readFileSync(filePath, 'utf-8')\n      } else {\n        // For new files, ensure parent directory exists\n        const parentDir = dirname(filePath)\n        if (!existsSync(parentDir)) {\n          mkdirSync(parentDir, { recursive: true })\n        }\n      }\n\n      // Apply all edits sequentially\n      let modifiedContent = currentContent\n      const appliedEdits = []\n\n      for (let i = 0; i < edits.length; i++) {\n        const edit = edits[i]\n        const { old_string, new_string, replace_all } = edit\n\n        try {\n          const result = applyContentEdit(\n            modifiedContent,\n            old_string,\n            new_string,\n            replace_all,\n          )\n          modifiedContent = result.newContent\n          appliedEdits.push({\n            editIndex: i + 1,\n            success: true,\n            old_string: old_string.substring(0, 100),\n            new_string: new_string.substring(0, 100),\n            occurrences: result.occurrences,\n          })\n        } catch (error) {\n          // If any edit fails, abort the entire operation\n          const errorMessage =\n            error instanceof Error ? error.message : 'Unknown error'\n          yield {\n            type: 'result',\n            data: `Error in edit ${i + 1}: ${errorMessage}`,\n            resultForAssistant: `Error in edit ${i + 1}: ${errorMessage}`,\n          }\n          return\n        }\n      }\n\n      // Write the modified content\n      const lineEndings = fileExists ? detectLineEndings(currentContent) : 'LF'\n      const encoding = fileExists ? detectFileEncoding(filePath) : 'utf8'\n      writeTextContent(filePath, modifiedContent, encoding, lineEndings)\n\n      // Record Agent edit operation for file freshness tracking\n      recordFileEdit(filePath, modifiedContent)\n\n      // Update readFileTimestamps to prevent stale file warnings\n      readFileTimestamps[filePath] = Date.now()\n\n      // Emit file edited event for system reminders\n      emitReminderEvent('file:edited', {\n        filePath,\n        edits: edits.map(e => ({\n          oldString: e.old_string,\n          newString: e.new_string,\n        })),\n        originalContent: currentContent,\n        newContent: modifiedContent,\n        timestamp: Date.now(),\n        operation: fileExists ? 'update' : 'create',\n      })\n\n      // Generate result data\n      const relativePath = relative(workingDir, filePath)\n      const summary = `Successfully applied ${edits.length} edits to ${relativePath}`\n\n      const resultData = {\n        filePath: relativePath,\n        wasNewFile: !fileExists,\n        editsApplied: appliedEdits,\n        totalEdits: edits.length,\n        summary,\n      }\n\n      // Log the operation\n      logEvent('multi_edit_tool_used', {\n        file_path: relativePath,\n        edits_count: String(edits.length),\n        was_new_file: String(!fileExists),\n        duration_ms: String(Date.now() - startTime),\n      })\n\n      yield {\n        type: 'result',\n        data: resultData,\n        resultForAssistant: summary,\n      }\n    } catch (error) {\n      const errorMessage =\n        error instanceof Error ? error.message : 'Unknown error occurred'\n      const errorResult = `Error applying multi-edit: ${errorMessage}`\n\n      logError(error)\n\n      yield {\n        type: 'result',\n        data: errorResult,\n        resultForAssistant: errorResult,\n      }\n    }\n  },\n} satisfies Tool<typeof inputSchema, any>\n"],
-  "mappings": "AAAA,SAAS,YAAY,WAAW,cAAc,gBAAgB;AAC9D,SAAS,KAAK,YAAY;AAC1B,SAAS,SAAS,YAAY,UAAU,eAAoB;AAC5D,YAAY,WAAW;AACvB,SAAS,SAAS;AAClB,SAAS,kCAAkC;AAE3C,SAAS,gBAAgB;AAGzB;AAAA,EAEE;AAAA,EACA;AAAA,EAEA;AAAA,OACK;AACP,SAAS,gBAAgB;AACzB,SAAS,cAAc;AACvB,SAAS,gBAAgB;AACzB,SAAS,wBAAwB;AAEjC,SAAS,iBACP,SACA,WACA,WACA,aAAsB,OACuB;AAC7C,MAAI,YAAY;AACd,UAAM,QAAQ,IAAI,OAAO,UAAU,QAAQ,uBAAuB,MAAM,GAAG,GAAG;AAC9E,UAAM,UAAU,QAAQ,MAAM,KAAK;AACnC,UAAM,cAAc,UAAU,QAAQ,SAAS;AAC/C,UAAM,aAAa,QAAQ,QAAQ,OAAO,SAAS;AACnD,WAAO,EAAE,YAAY,YAAY;AAAA,EACnC,OAAO;AACL,QAAI,QAAQ,SAAS,SAAS,GAAG;AAC/B,YAAM,aAAa,QAAQ,QAAQ,WAAW,SAAS;AACvD,aAAO,EAAE,YAAY,aAAa,EAAE;AAAA,IACtC,OAAO;AACL,YAAM,IAAI,MAAM,qBAAqB,UAAU,UAAU,GAAG,EAAE,CAAC,KAAK;AAAA,IACtE;AAAA,EACF;AACF;AACA,SAAS,0BAA0B;AAEnC,SAAsB,cAAc;AACpC,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAE/B,MAAM,aAAa,EAAE,OAAO;AAAA,EAC1B,YAAY,EAAE,OAAO,EAAE,SAAS,qBAAqB;AAAA,EACrD,YAAY,EAAE,OAAO,EAAE,SAAS,6BAA6B;AAAA,EAC7D,aAAa,EACV,QAAQ,EACR,SAAS,EACT,QAAQ,KAAK,EACb,SAAS,sDAAsD;AACpE,CAAC;AAED,MAAM,cAAc,EAAE,aAAa;AAAA,EACjC,WAAW,EAAE,OAAO,EAAE,SAAS,yCAAyC;AAAA,EACxE,OAAO,EACJ,MAAM,UAAU,EAChB,IAAI,CAAC,EACL,SAAS,8DAA8D;AAC5E,CAAC;AAKD,MAAM,kBAAkB;AAEjB,MAAM,gBAAgB;AAAA,EAC3B,MAAM;AAAA,EACN,MAAM,cAAc;AAClB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,SAAS;AACb,WAAO;AAAA,EACT;AAAA,EACA;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,MAAM,YAAY;AAChB,WAAO;AAAA,EACT;AAAA,EACA,aAAa;AACX,WAAO;AAAA,EACT;AAAA,EACA,oBAAoB;AAClB,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB,OAAqC;AACpD,QAAI,CAAC,MAAO,QAAO;AACnB,WAAO,CAAC,mBAAmB,MAAM,SAAS;AAAA,EAC5C;AAAA,EACA,yBAAyB,SAAS;AAChC,WAAO;AAAA,EACT;AAAA,EACA,qBAAqB,OAAO,EAAE,QAAQ,GAAG;AACvC,UAAM,EAAE,WAAW,MAAM,IAAI;AAC7B,UAAM,aAAa,OAAO;AAC1B,UAAM,eAAe,WAAW,SAAS,IACrC,SAAS,YAAY,SAAS,IAC9B;AAEJ,QAAI,SAAS;AACX,YAAM,cAAc,MACjB;AAAA,QACC,CAAC,MAAM,UACL,GAAG,QAAQ,CAAC,cAAc,KAAK,WAAW,UAAU,GAAG,EAAE,CAAC,GAAG,KAAK,WAAW,SAAS,KAAK,QAAQ,EAAE,WAAW,KAAK,WAAW,UAAU,GAAG,EAAE,CAAC,GAAG,KAAK,WAAW,SAAS,KAAK,QAAQ,EAAE;AAAA,MAC/L,EACC,KAAK,IAAI;AACZ,aAAO,qBAAqB,YAAY;AAAA,EAAM,WAAW;AAAA,IAC3D;AAEA,WAAO,UAAU,MAAM,MAAM,aAAa,YAAY;AAAA,EACxD;AAAA,EACA,+BAA+B;AAC7B,WACE,oCAAC,WACC,oCAAC,QAAK,OAAO,SAAS,EAAE,SAAO,8BAAuB,CACxD;AAAA,EAEJ;AAAA,EACA,wBAAwB,QAAQ;AAC9B,QAAI,OAAO,WAAW,UAAU;AAC9B,YAAM,UAAU,OAAO,SAAS,QAAQ;AACxC,aACE,oCAAC,OAAI,eAAc,YACjB,oCAAC,QAAK,OAAO,UAAU,SAAS,EAAE,QAAQ,SAAS,EAAE,WAClD,MACH,CACF;AAAA,IAEJ;AAEA,WAAO,oCAAC,8BAA4B,GAAG,QAAQ;AAAA,EACjD;AAAA,EACA,MAAM,cACJ,EAAE,WAAW,MAAM,GACnB,SAC2B;AAC3B,UAAM,aAAa,OAAO;AAC1B,UAAM,iBAAiB,WAAW,SAAS,IACvC,QAAQ,SAAS,IACjB,QAAQ,YAAY,SAAS;AAGjC,QAAI,eAAe,SAAS,QAAQ,GAAG;AACrC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,WAAW;AAAA,QACX,SAAS,iDAAiD,iBAAiB,IAAI;AAAA,MACjF;AAAA,IACF;AAGA,QAAI,CAAC,WAAW,cAAc,GAAG;AAC/B,YAAM,YAAY,QAAQ,cAAc;AACxC,UAAI,CAAC,WAAW,SAAS,GAAG;AAC1B,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SAAS,oCAAoC,SAAS;AAAA,QACxD;AAAA,MACF;AAGA,UAAI,MAAM,WAAW,KAAK,MAAM,CAAC,EAAE,eAAe,IAAI;AACpD,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SACE;AAAA,QACJ;AAAA,MACF;AAAA,IACF,OAAO;AAEL,YAAM,qBAAqB,SAAS,sBAAsB,CAAC;AAC3D,YAAM,gBAAgB,mBAAmB,cAAc;AAEvD,UAAI,CAAC,eAAe;AAClB,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SACE;AAAA,UACF,MAAM;AAAA,YACJ,UAAU;AAAA,YACV,oBAAoB,OAAO,WAAW,SAAS,CAAC;AAAA,UAClD;AAAA,QACF;AAAA,MACF;AAGA,YAAM,QAAQ,SAAS,cAAc;AACrC,YAAM,gBAAgB,MAAM;AAC5B,UAAI,gBAAgB,eAAe;AACjC,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SACE;AAAA,UACF,MAAM;AAAA,YACJ,UAAU;AAAA,YACV;AAAA,YACA;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAGA,YAAM,WAAW,mBAAmB,cAAc;AAClD,UAAI,aAAa,UAAU;AACzB,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SAAS;AAAA,QACX;AAAA,MACF;AAEA,YAAM,iBAAiB,aAAa,gBAAgB,OAAO;AAC3D,eAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,cAAM,OAAO,MAAM,CAAC;AACpB,YACE,KAAK,eAAe,MACpB,CAAC,eAAe,SAAS,KAAK,UAAU,GACxC;AACA,iBAAO;AAAA,YACL,QAAQ;AAAA,YACR,WAAW;AAAA,YACX,SAAS,QAAQ,IAAI,CAAC,2CAA2C,KAAK,WAAW,UAAU,GAAG,GAAG,CAAC,GAAG,KAAK,WAAW,SAAS,MAAM,QAAQ,EAAE;AAAA,YAC9I,MAAM;AAAA,cACJ,WAAW,IAAI;AAAA,cACf,WAAW,KAAK,WAAW,UAAU,GAAG,GAAG;AAAA,YAC7C;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAGA,aAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,YAAM,OAAO,MAAM,CAAC;AACpB,UAAI,KAAK,eAAe,KAAK,YAAY;AACvC,eAAO;AAAA,UACL,QAAQ;AAAA,UACR,WAAW;AAAA,UACX,SAAS,QAAQ,IAAI,CAAC;AAAA,QACxB;AAAA,MACF;AAAA,IACF;AAEA,WAAO,EAAE,QAAQ,KAAK;AAAA,EACxB;AAAA,EACA,OAAO,KAAK,EAAE,WAAW,MAAM,GAAG,EAAE,mBAAmB,GAAG;AACxD,UAAM,YAAY,KAAK,IAAI;AAC3B,UAAM,aAAa,OAAO;AAC1B,UAAM,WAAW,WAAW,SAAS,IACjC,QAAQ,SAAS,IACjB,QAAQ,YAAY,SAAS;AAEjC,QAAI;AAEF,UAAI,iBAAiB;AACrB,UAAI,aAAa,WAAW,QAAQ;AAEpC,UAAI,YAAY;AACd,cAAMA,YAAW,mBAAmB,QAAQ;AAC5C,YAAIA,cAAa,UAAU;AACzB,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,MAAM;AAAA,YACN,oBAAoB;AAAA,UACtB;AACA;AAAA,QACF;AACA,yBAAiB,aAAa,UAAU,OAAO;AAAA,MACjD,OAAO;AAEL,cAAM,YAAY,QAAQ,QAAQ;AAClC,YAAI,CAAC,WAAW,SAAS,GAAG;AAC1B,oBAAU,WAAW,EAAE,WAAW,KAAK,CAAC;AAAA,QAC1C;AAAA,MACF;AAGA,UAAI,kBAAkB;AACtB,YAAM,eAAe,CAAC;AAEtB,eAAS,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK;AACrC,cAAM,OAAO,MAAM,CAAC;AACpB,cAAM,EAAE,YAAY,YAAY,YAAY,IAAI;AAEhD,YAAI;AACF,gBAAM,SAAS;AAAA,YACb;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF;AACA,4BAAkB,OAAO;AACzB,uBAAa,KAAK;AAAA,YAChB,WAAW,IAAI;AAAA,YACf,SAAS;AAAA,YACT,YAAY,WAAW,UAAU,GAAG,GAAG;AAAA,YACvC,YAAY,WAAW,UAAU,GAAG,GAAG;AAAA,YACvC,aAAa,OAAO;AAAA,UACtB,CAAC;AAAA,QACH,SAAS,OAAO;AAEd,gBAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU;AAC3C,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,MAAM,iBAAiB,IAAI,CAAC,KAAK,YAAY;AAAA,YAC7C,oBAAoB,iBAAiB,IAAI,CAAC,KAAK,YAAY;AAAA,UAC7D;AACA;AAAA,QACF;AAAA,MACF;AAGA,YAAM,cAAc,aAAa,kBAAkB,cAAc,IAAI;AACrE,YAAM,WAAW,aAAa,mBAAmB,QAAQ,IAAI;AAC7D,uBAAiB,UAAU,iBAAiB,UAAU,WAAW;AAGjE,qBAAe,UAAU,eAAe;AAGxC,yBAAmB,QAAQ,IAAI,KAAK,IAAI;AAGxC,wBAAkB,eAAe;AAAA,QAC/B;AAAA,QACA,OAAO,MAAM,IAAI,QAAM;AAAA,UACrB,WAAW,EAAE;AAAA,UACb,WAAW,EAAE;AAAA,QACf,EAAE;AAAA,QACF,iBAAiB;AAAA,QACjB,YAAY;AAAA,QACZ,WAAW,KAAK,IAAI;AAAA,QACpB,WAAW,aAAa,WAAW;AAAA,MACrC,CAAC;AAGD,YAAM,eAAe,SAAS,YAAY,QAAQ;AAClD,YAAM,UAAU,wBAAwB,MAAM,MAAM,aAAa,YAAY;AAE7E,YAAM,aAAa;AAAA,QACjB,UAAU;AAAA,QACV,YAAY,CAAC;AAAA,QACb,cAAc;AAAA,QACd,YAAY,MAAM;AAAA,QAClB;AAAA,MACF;AAGA,eAAS,wBAAwB;AAAA,QAC/B,WAAW;AAAA,QACX,aAAa,OAAO,MAAM,MAAM;AAAA,QAChC,cAAc,OAAO,CAAC,UAAU;AAAA,QAChC,aAAa,OAAO,KAAK,IAAI,IAAI,SAAS;AAAA,MAC5C,CAAC;AAED,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAM;AAAA,QACN,oBAAoB;AAAA,MACtB;AAAA,IACF,SAAS,OAAO;AACd,YAAM,eACJ,iBAAiB,QAAQ,MAAM,UAAU;AAC3C,YAAM,cAAc,8BAA8B,YAAY;AAE9D,eAAS,KAAK;AAEd,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,MAAM;AAAA,QACN,oBAAoB;AAAA,MACtB;AAAA,IACF;AAAA,EACF;AACF;",
-  "names": ["encoding"]
-}

package/dist/tools/MultiEditTool/prompt.js

@@ -1,48 +0,0 @@
-import { NotebookEditTool } from "../NotebookEditTool/NotebookEditTool.js";
-const DESCRIPTION = `This is a tool for making multiple edits to a single file in one operation. It is built on top of the Edit tool and allows you to perform multiple find-and-replace operations efficiently. Prefer this tool over the Edit tool when you need to make multiple edits to the same file.
-
-Before using this tool:
-
-1. Use the Read tool to understand the file's contents and context
-2. Verify the directory path is correct
-
-To make multiple file edits, provide the following:
-1. file_path: The absolute path to the file to modify (must be absolute, not relative)
-2. edits: An array of edit operations to perform, where each edit contains:
-   - old_string: The text to replace (must match the file contents exactly, including all whitespace and indentation)
-   - new_string: The edited text to replace the old_string
-   - replace_all: Replace all occurences of old_string. This parameter is optional and defaults to false.
-
-IMPORTANT:
-- All edits are applied in sequence, in the order they are provided
-- Each edit operates on the result of the previous edit
-- All edits must be valid for the operation to succeed - if any edit fails, none will be applied
-- This tool is ideal when you need to make several changes to different parts of the same file
-- For Jupyter notebooks (.ipynb files), use the ${NotebookEditTool.name} instead
-
-CRITICAL REQUIREMENTS:
-1. All edits follow the same requirements as the single Edit tool
-2. The edits are atomic - either all succeed or none are applied
-3. Plan your edits carefully to avoid conflicts between sequential operations
-
-WARNING:
-- The tool will fail if edits.old_string doesn't match the file contents exactly (including whitespace)
-- The tool will fail if edits.old_string and edits.new_string are the same
-- Since edits are applied in sequence, ensure that earlier edits don't affect the text that later edits are trying to find
-
-When making edits:
-- Ensure all edits result in idiomatic, correct code
-- Do not leave the code in a broken state
-- Always use absolute file paths (starting with /)
-- Use replace_all for replacing and renaming strings across the file. This parameter is useful if you want to rename a variable for instance.
-
-If you want to create a new file, use:
-- A new file path, including dir name if needed
-- First edit: empty old_string and the new file's contents as new_string
-- Subsequent edits: normal edit operations on the created content`;
-const PROMPT = DESCRIPTION;
-export {
-  DESCRIPTION,
-  PROMPT
-};
-//# sourceMappingURL=prompt.js.map

package/dist/tools/MultiEditTool/prompt.js.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/tools/MultiEditTool/prompt.ts"],
-  "sourcesContent": ["import { NotebookEditTool } from '../NotebookEditTool/NotebookEditTool'\n\nexport const DESCRIPTION = `This is a tool for making multiple edits to a single file in one operation. It is built on top of the Edit tool and allows you to perform multiple find-and-replace operations efficiently. Prefer this tool over the Edit tool when you need to make multiple edits to the same file.\n\nBefore using this tool:\n\n1. Use the Read tool to understand the file's contents and context\n2. Verify the directory path is correct\n\nTo make multiple file edits, provide the following:\n1. file_path: The absolute path to the file to modify (must be absolute, not relative)\n2. edits: An array of edit operations to perform, where each edit contains:\n   - old_string: The text to replace (must match the file contents exactly, including all whitespace and indentation)\n   - new_string: The edited text to replace the old_string\n   - replace_all: Replace all occurences of old_string. This parameter is optional and defaults to false.\n\nIMPORTANT:\n- All edits are applied in sequence, in the order they are provided\n- Each edit operates on the result of the previous edit\n- All edits must be valid for the operation to succeed - if any edit fails, none will be applied\n- This tool is ideal when you need to make several changes to different parts of the same file\n- For Jupyter notebooks (.ipynb files), use the ${NotebookEditTool.name} instead\n\nCRITICAL REQUIREMENTS:\n1. All edits follow the same requirements as the single Edit tool\n2. The edits are atomic - either all succeed or none are applied\n3. Plan your edits carefully to avoid conflicts between sequential operations\n\nWARNING:\n- The tool will fail if edits.old_string doesn't match the file contents exactly (including whitespace)\n- The tool will fail if edits.old_string and edits.new_string are the same\n- Since edits are applied in sequence, ensure that earlier edits don't affect the text that later edits are trying to find\n\nWhen making edits:\n- Ensure all edits result in idiomatic, correct code\n- Do not leave the code in a broken state\n- Always use absolute file paths (starting with /)\n- Use replace_all for replacing and renaming strings across the file. This parameter is useful if you want to rename a variable for instance.\n\nIf you want to create a new file, use:\n- A new file path, including dir name if needed\n- First edit: empty old_string and the new file's contents as new_string\n- Subsequent edits: normal edit operations on the created content`\n\nexport const PROMPT = DESCRIPTION\n"],
-  "mappings": "AAAA,SAAS,wBAAwB;AAE1B,MAAM,cAAc;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,kDAmBuB,iBAAiB,IAAI;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAuBhE,MAAM,SAAS;",
-  "names": []
-}

package/dist/tools/NotebookEditTool/NotebookEditTool.js

@@ -1,238 +0,0 @@
-import { existsSync, readFileSync } from "fs";
-import { Box, Text } from "ink";
-import { extname, isAbsolute, relative, resolve } from "path";
-import * as React from "react";
-import { z } from "zod";
-import { FallbackToolUseRejectedMessage } from "../../components/FallbackToolUseRejectedMessage.js";
-import { HighlightedCode } from "../../components/HighlightedCode.js";
-import {
-  detectFileEncoding,
-  detectLineEndings,
-  writeTextContent
-} from "../../utils/file.js";
-import { safeParseJSON } from "../../utils/json.js";
-import { getCwd } from "../../utils/state.js";
-import { DESCRIPTION, PROMPT } from "./prompt.js";
-import { hasWritePermission } from "../../utils/permissions/filesystem.js";
-import { emitReminderEvent } from "../../services/systemReminder.js";
-import { recordFileEdit } from "../../services/fileFreshness.js";
-const inputSchema = z.strictObject({
-  notebook_path: z.string().describe(
-    "The absolute path to the Jupyter notebook file to edit (must be absolute, not relative)"
-  ),
-  cell_number: z.number().describe("The index of the cell to edit (0-based)"),
-  new_source: z.string().describe("The new source for the cell"),
-  cell_type: z.enum(["code", "markdown"]).optional().describe(
-    "The type of the cell (code or markdown). If not specified, it defaults to the current cell type. If using edit_mode=insert, this is required."
-  ),
-  edit_mode: z.string().optional().describe(
-    "The type of edit to make (replace, insert, delete). Defaults to replace."
-  )
-});
-const NotebookEditTool = {
-  name: "NotebookEditCell",
-  async description() {
-    return DESCRIPTION;
-  },
-  async prompt() {
-    return PROMPT;
-  },
-  inputSchema,
-  userFacingName() {
-    return "Edit Notebook";
-  },
-  async isEnabled() {
-    return true;
-  },
-  isReadOnly() {
-    return false;
-  },
-  isConcurrencySafe() {
-    return false;
-  },
-  needsPermissions({ notebook_path }) {
-    return !hasWritePermission(notebook_path);
-  },
-  renderResultForAssistant({ cell_number, edit_mode, new_source, error }) {
-    if (error) {
-      return error;
-    }
-    switch (edit_mode) {
-      case "replace":
-        return `Updated cell ${cell_number} with ${new_source}`;
-      case "insert":
-        return `Inserted cell ${cell_number} with ${new_source}`;
-      case "delete":
-        return `Deleted cell ${cell_number}`;
-    }
-  },
-  renderToolUseMessage(input, { verbose }) {
-    return `notebook_path: ${verbose ? input.notebook_path : relative(getCwd(), input.notebook_path)}, cell: ${input.cell_number}, content: ${input.new_source.slice(0, 30)}\u2026, cell_type: ${input.cell_type}, edit_mode: ${input.edit_mode ?? "replace"}`;
-  },
-  renderToolUseRejectedMessage() {
-    return /* @__PURE__ */ React.createElement(FallbackToolUseRejectedMessage, null);
-  },
-  renderToolResultMessage({ cell_number, new_source, language, error }) {
-    if (error) {
-      return /* @__PURE__ */ React.createElement(Box, { flexDirection: "column" }, /* @__PURE__ */ React.createElement(Text, { color: "red" }, error));
-    }
-    return /* @__PURE__ */ React.createElement(Box, { flexDirection: "column" }, /* @__PURE__ */ React.createElement(Text, null, "Updated cell ", cell_number, ":"), /* @__PURE__ */ React.createElement(Box, { marginLeft: 2 }, /* @__PURE__ */ React.createElement(HighlightedCode, { code: new_source, language })));
-  },
-  async validateInput({
-    notebook_path,
-    cell_number,
-    cell_type,
-    edit_mode = "replace"
-  }) {
-    const fullPath = isAbsolute(notebook_path) ? notebook_path : resolve(getCwd(), notebook_path);
-    if (!existsSync(fullPath)) {
-      return {
-        result: false,
-        message: "Notebook file does not exist."
-      };
-    }
-    if (extname(fullPath) !== ".ipynb") {
-      return {
-        result: false,
-        message: "File must be a Jupyter notebook (.ipynb file). For editing other file types, use the FileEdit tool."
-      };
-    }
-    if (cell_number < 0) {
-      return {
-        result: false,
-        message: "Cell number must be non-negative."
-      };
-    }
-    if (edit_mode !== "replace" && edit_mode !== "insert" && edit_mode !== "delete") {
-      return {
-        result: false,
-        message: "Edit mode must be replace, insert, or delete."
-      };
-    }
-    if (edit_mode === "insert" && !cell_type) {
-      return {
-        result: false,
-        message: "Cell type is required when using edit_mode=insert."
-      };
-    }
-    const enc = detectFileEncoding(fullPath);
-    const content = readFileSync(fullPath, enc);
-    const notebook = safeParseJSON(content);
-    if (!notebook) {
-      return {
-        result: false,
-        message: "Notebook is not valid JSON."
-      };
-    }
-    if (edit_mode === "insert" && cell_number > notebook.cells.length) {
-      return {
-        result: false,
-        message: `Cell number is out of bounds. For insert mode, the maximum value is ${notebook.cells.length} (to append at the end).`
-      };
-    } else if ((edit_mode === "replace" || edit_mode === "delete") && (cell_number >= notebook.cells.length || !notebook.cells[cell_number])) {
-      return {
-        result: false,
-        message: `Cell number is out of bounds. Notebook has ${notebook.cells.length} cells.`
-      };
-    }
-    return { result: true };
-  },
-  async *call({
-    notebook_path,
-    cell_number,
-    new_source,
-    cell_type,
-    edit_mode
-  }) {
-    const fullPath = isAbsolute(notebook_path) ? notebook_path : resolve(getCwd(), notebook_path);
-    try {
-      const enc = detectFileEncoding(fullPath);
-      const content = readFileSync(fullPath, enc);
-      const notebook = JSON.parse(content);
-      const language = notebook.metadata.language_info?.name ?? "python";
-      if (edit_mode === "delete") {
-        notebook.cells.splice(cell_number, 1);
-      } else if (edit_mode === "insert") {
-        const new_cell = {
-          cell_type,
-          // validateInput ensures cell_type is not undefined
-          source: new_source,
-          metadata: {}
-        };
-        notebook.cells.splice(
-          cell_number,
-          0,
-          cell_type == "markdown" ? new_cell : { ...new_cell, outputs: [] }
-        );
-      } else {
-        const targetCell = notebook.cells[cell_number];
-        targetCell.source = new_source;
-        targetCell.execution_count = void 0;
-        targetCell.outputs = [];
-        if (cell_type && cell_type !== targetCell.cell_type) {
-          targetCell.cell_type = cell_type;
-        }
-      }
-      const endings = detectLineEndings(fullPath);
-      const updatedNotebook = JSON.stringify(notebook, null, 1);
-      writeTextContent(fullPath, updatedNotebook, enc, endings);
-      recordFileEdit(fullPath, updatedNotebook);
-      emitReminderEvent("file:edited", {
-        filePath: fullPath,
-        cellNumber: cell_number,
-        newSource: new_source,
-        cellType: cell_type,
-        editMode: edit_mode || "replace",
-        timestamp: Date.now(),
-        operation: "notebook_edit"
-      });
-      const data = {
-        cell_number,
-        new_source,
-        cell_type: cell_type ?? "code",
-        language,
-        edit_mode: edit_mode ?? "replace",
-        error: ""
-      };
-      yield {
-        type: "result",
-        data,
-        resultForAssistant: this.renderResultForAssistant(data)
-      };
-    } catch (error) {
-      if (error instanceof Error) {
-        const data2 = {
-          cell_number,
-          new_source,
-          cell_type: cell_type ?? "code",
-          language: "python",
-          edit_mode: "replace",
-          error: error.message
-        };
-        yield {
-          type: "result",
-          data: data2,
-          resultForAssistant: this.renderResultForAssistant(data2)
-        };
-        return;
-      }
-      const data = {
-        cell_number,
-        new_source,
-        cell_type: cell_type ?? "code",
-        language: "python",
-        edit_mode: "replace",
-        error: "Unknown error occurred while editing notebook"
-      };
-      yield {
-        type: "result",
-        data,
-        resultForAssistant: this.renderResultForAssistant(data)
-      };
-    }
-  }
-};
-export {
-  NotebookEditTool
-};
-//# sourceMappingURL=NotebookEditTool.js.map

package/dist/tools/NotebookEditTool/NotebookEditTool.js.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/tools/NotebookEditTool/NotebookEditTool.tsx"],
-  "sourcesContent": ["import { existsSync, readFileSync } from 'fs'\nimport { Box, Text } from 'ink'\nimport { extname, isAbsolute, relative, resolve } from 'path'\nimport * as React from 'react'\nimport { z } from 'zod'\nimport { FallbackToolUseRejectedMessage } from '../../components/FallbackToolUseRejectedMessage'\nimport { HighlightedCode } from '../../components/HighlightedCode'\nimport type { Tool } from '../../Tool'\nimport { NotebookCellType, NotebookContent } from '../../types/notebook'\nimport {\n  detectFileEncoding,\n  detectLineEndings,\n  writeTextContent,\n} from '../../utils/file.js'\nimport { safeParseJSON } from '../../utils/json'\nimport { getCwd } from '../../utils/state'\nimport { DESCRIPTION, PROMPT } from './prompt'\nimport { hasWritePermission } from '../../utils/permissions/filesystem'\nimport { emitReminderEvent } from '../../services/systemReminder'\nimport { recordFileEdit } from '../../services/fileFreshness'\n\nconst inputSchema = z.strictObject({\n  notebook_path: z\n    .string()\n    .describe(\n      'The absolute path to the Jupyter notebook file to edit (must be absolute, not relative)',\n    ),\n  cell_number: z.number().describe('The index of the cell to edit (0-based)'),\n  new_source: z.string().describe('The new source for the cell'),\n  cell_type: z\n    .enum(['code', 'markdown'])\n    .optional()\n    .describe(\n      'The type of the cell (code or markdown). If not specified, it defaults to the current cell type. If using edit_mode=insert, this is required.',\n    ),\n  edit_mode: z\n    .string()\n    .optional()\n    .describe(\n      'The type of edit to make (replace, insert, delete). Defaults to replace.',\n    ),\n})\n\nexport const NotebookEditTool = {\n  name: 'NotebookEditCell',\n  async description() {\n    return DESCRIPTION\n  },\n  async prompt() {\n    return PROMPT\n  },\n  inputSchema,\n  userFacingName() {\n    return 'Edit Notebook'\n  },\n  async isEnabled() {\n    return true\n  },\n  isReadOnly() {\n    return false\n  },\n  isConcurrencySafe() {\n    return false // NotebookEditTool modifies state/files, not safe for concurrent execution\n  },\n  needsPermissions({ notebook_path }) {\n    return !hasWritePermission(notebook_path)\n  },\n  renderResultForAssistant({ cell_number, edit_mode, new_source, error }) {\n    if (error) {\n      return error\n    }\n    switch (edit_mode) {\n      case 'replace':\n        return `Updated cell ${cell_number} with ${new_source}`\n      case 'insert':\n        return `Inserted cell ${cell_number} with ${new_source}`\n      case 'delete':\n        return `Deleted cell ${cell_number}`\n    }\n  },\n  renderToolUseMessage(input, { verbose }) {\n    return `notebook_path: ${verbose ? input.notebook_path : relative(getCwd(), input.notebook_path)}, cell: ${input.cell_number}, content: ${input.new_source.slice(0, 30)}\u2026, cell_type: ${input.cell_type}, edit_mode: ${input.edit_mode ?? 'replace'}`\n  },\n  renderToolUseRejectedMessage() {\n    return <FallbackToolUseRejectedMessage />\n  },\n  renderToolResultMessage({ cell_number, new_source, language, error }) {\n    if (error) {\n      return (\n        <Box flexDirection=\"column\">\n          <Text color=\"red\">{error}</Text>\n        </Box>\n      )\n    }\n\n    return (\n      <Box flexDirection=\"column\">\n        <Text>Updated cell {cell_number}:</Text>\n        <Box marginLeft={2}>\n          <HighlightedCode code={new_source} language={language} />\n        </Box>\n      </Box>\n    )\n  },\n  async validateInput({\n    notebook_path,\n    cell_number,\n    cell_type,\n    edit_mode = 'replace',\n  }) {\n    const fullPath = isAbsolute(notebook_path)\n      ? notebook_path\n      : resolve(getCwd(), notebook_path)\n\n    if (!existsSync(fullPath)) {\n      return {\n        result: false,\n        message: 'Notebook file does not exist.',\n      }\n    }\n\n    if (extname(fullPath) !== '.ipynb') {\n      return {\n        result: false,\n        message:\n          'File must be a Jupyter notebook (.ipynb file). For editing other file types, use the FileEdit tool.',\n      }\n    }\n\n    if (cell_number < 0) {\n      return {\n        result: false,\n        message: 'Cell number must be non-negative.',\n      }\n    }\n\n    if (\n      edit_mode !== 'replace' &&\n      edit_mode !== 'insert' &&\n      edit_mode !== 'delete'\n    ) {\n      return {\n        result: false,\n        message: 'Edit mode must be replace, insert, or delete.',\n      }\n    }\n\n    if (edit_mode === 'insert' && !cell_type) {\n      return {\n        result: false,\n        message: 'Cell type is required when using edit_mode=insert.',\n      }\n    }\n\n    const enc = detectFileEncoding(fullPath)\n    const content = readFileSync(fullPath, enc)\n    const notebook = safeParseJSON(content) as NotebookContent | null\n    if (!notebook) {\n      return {\n        result: false,\n        message: 'Notebook is not valid JSON.',\n      }\n    }\n\n    if (edit_mode === 'insert' && cell_number > notebook.cells.length) {\n      return {\n        result: false,\n        message: `Cell number is out of bounds. For insert mode, the maximum value is ${notebook.cells.length} (to append at the end).`,\n      }\n    } else if (\n      (edit_mode === 'replace' || edit_mode === 'delete') &&\n      (cell_number >= notebook.cells.length || !notebook.cells[cell_number])\n    ) {\n      return {\n        result: false,\n        message: `Cell number is out of bounds. Notebook has ${notebook.cells.length} cells.`,\n      }\n    }\n\n    return { result: true }\n  },\n  async *call({\n    notebook_path,\n    cell_number,\n    new_source,\n    cell_type,\n    edit_mode,\n  }) {\n    const fullPath = isAbsolute(notebook_path)\n      ? notebook_path\n      : resolve(getCwd(), notebook_path)\n\n    try {\n      const enc = detectFileEncoding(fullPath)\n      const content = readFileSync(fullPath, enc)\n      const notebook = JSON.parse(content) as NotebookContent\n      const language = notebook.metadata.language_info?.name ?? 'python'\n\n      if (edit_mode === 'delete') {\n        // Delete the specified cell\n        notebook.cells.splice(cell_number, 1)\n      } else if (edit_mode === 'insert') {\n        // Insert the new cell\n        const new_cell = {\n          cell_type: cell_type!, // validateInput ensures cell_type is not undefined\n          source: new_source,\n          metadata: {},\n        }\n        notebook.cells.splice(\n          cell_number,\n          0,\n          cell_type == 'markdown' ? new_cell : { ...new_cell, outputs: [] },\n        )\n      } else {\n        // Find the specified cell\n        const targetCell = notebook.cells[cell_number]! // validateInput ensures cell_number is in bounds\n        targetCell.source = new_source\n        // Reset execution count and clear outputs since cell was modified\n        targetCell.execution_count = undefined\n        targetCell.outputs = []\n        if (cell_type && cell_type !== targetCell.cell_type) {\n          targetCell.cell_type = cell_type\n        }\n      }\n      // Write back to file\n      const endings = detectLineEndings(fullPath)\n      const updatedNotebook = JSON.stringify(notebook, null, 1)\n      writeTextContent(fullPath, updatedNotebook, enc, endings!)\n\n      // Record Agent edit operation for file freshness tracking\n      recordFileEdit(fullPath, updatedNotebook)\n\n      // Emit file edited event for system reminders\n      emitReminderEvent('file:edited', {\n        filePath: fullPath,\n        cellNumber: cell_number,\n        newSource: new_source,\n        cellType: cell_type,\n        editMode: edit_mode || 'replace',\n        timestamp: Date.now(),\n        operation: 'notebook_edit',\n      })\n      const data = {\n        cell_number,\n        new_source,\n        cell_type: cell_type ?? 'code',\n        language,\n        edit_mode: edit_mode ?? 'replace',\n        error: '',\n      }\n      yield {\n        type: 'result',\n        data,\n        resultForAssistant: this.renderResultForAssistant(data),\n      }\n    } catch (error) {\n      if (error instanceof Error) {\n        const data = {\n          cell_number,\n          new_source,\n          cell_type: cell_type ?? 'code',\n          language: 'python',\n          edit_mode: 'replace',\n          error: error.message,\n        }\n        yield {\n          type: 'result',\n          data,\n          resultForAssistant: this.renderResultForAssistant(data),\n        }\n        return\n      }\n      const data = {\n        cell_number,\n        new_source,\n        cell_type: cell_type ?? 'code',\n        language: 'python',\n        edit_mode: 'replace',\n        error: 'Unknown error occurred while editing notebook',\n      }\n      yield {\n        type: 'result',\n        data,\n        resultForAssistant: this.renderResultForAssistant(data),\n      }\n    }\n  },\n} satisfies Tool<\n  typeof inputSchema,\n  {\n    cell_number: number\n    new_source: string\n    cell_type: NotebookCellType\n    language: string\n    edit_mode: string\n    error?: string\n  }\n>\n"],
-  "mappings": "AAAA,SAAS,YAAY,oBAAoB;AACzC,SAAS,KAAK,YAAY;AAC1B,SAAS,SAAS,YAAY,UAAU,eAAe;AACvD,YAAY,WAAW;AACvB,SAAS,SAAS;AAClB,SAAS,sCAAsC;AAC/C,SAAS,uBAAuB;AAGhC;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,qBAAqB;AAC9B,SAAS,cAAc;AACvB,SAAS,aAAa,cAAc;AACpC,SAAS,0BAA0B;AACnC,SAAS,yBAAyB;AAClC,SAAS,sBAAsB;AAE/B,MAAM,cAAc,EAAE,aAAa;AAAA,EACjC,eAAe,EACZ,OAAO,EACP;AAAA,IACC;AAAA,EACF;AAAA,EACF,aAAa,EAAE,OAAO,EAAE,SAAS,yCAAyC;AAAA,EAC1E,YAAY,EAAE,OAAO,EAAE,SAAS,6BAA6B;AAAA,EAC7D,WAAW,EACR,KAAK,CAAC,QAAQ,UAAU,CAAC,EACzB,SAAS,EACT;AAAA,IACC;AAAA,EACF;AAAA,EACF,WAAW,EACR,OAAO,EACP,SAAS,EACT;AAAA,IACC;AAAA,EACF;AACJ,CAAC;AAEM,MAAM,mBAAmB;AAAA,EAC9B,MAAM;AAAA,EACN,MAAM,cAAc;AAClB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,SAAS;AACb,WAAO;AAAA,EACT;AAAA,EACA;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,MAAM,YAAY;AAChB,WAAO;AAAA,EACT;AAAA,EACA,aAAa;AACX,WAAO;AAAA,EACT;AAAA,EACA,oBAAoB;AAClB,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB,EAAE,cAAc,GAAG;AAClC,WAAO,CAAC,mBAAmB,aAAa;AAAA,EAC1C;AAAA,EACA,yBAAyB,EAAE,aAAa,WAAW,YAAY,MAAM,GAAG;AACtE,QAAI,OAAO;AACT,aAAO;AAAA,IACT;AACA,YAAQ,WAAW;AAAA,MACjB,KAAK;AACH,eAAO,gBAAgB,WAAW,SAAS,UAAU;AAAA,MACvD,KAAK;AACH,eAAO,iBAAiB,WAAW,SAAS,UAAU;AAAA,MACxD,KAAK;AACH,eAAO,gBAAgB,WAAW;AAAA,IACtC;AAAA,EACF;AAAA,EACA,qBAAqB,OAAO,EAAE,QAAQ,GAAG;AACvC,WAAO,kBAAkB,UAAU,MAAM,gBAAgB,SAAS,OAAO,GAAG,MAAM,aAAa,CAAC,WAAW,MAAM,WAAW,cAAc,MAAM,WAAW,MAAM,GAAG,EAAE,CAAC,sBAAiB,MAAM,SAAS,gBAAgB,MAAM,aAAa,SAAS;AAAA,EACrP;AAAA,EACA,+BAA+B;AAC7B,WAAO,oCAAC,oCAA+B;AAAA,EACzC;AAAA,EACA,wBAAwB,EAAE,aAAa,YAAY,UAAU,MAAM,GAAG;AACpE,QAAI,OAAO;AACT,aACE,oCAAC,OAAI,eAAc,YACjB,oCAAC,QAAK,OAAM,SAAO,KAAM,CAC3B;AAAA,IAEJ;AAEA,WACE,oCAAC,OAAI,eAAc,YACjB,oCAAC,YAAK,iBAAc,aAAY,GAAC,GACjC,oCAAC,OAAI,YAAY,KACf,oCAAC,mBAAgB,MAAM,YAAY,UAAoB,CACzD,CACF;AAAA,EAEJ;AAAA,EACA,MAAM,cAAc;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,IACA,YAAY;AAAA,EACd,GAAG;AACD,UAAM,WAAW,WAAW,aAAa,IACrC,gBACA,QAAQ,OAAO,GAAG,aAAa;AAEnC,QAAI,CAAC,WAAW,QAAQ,GAAG;AACzB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,QAAI,QAAQ,QAAQ,MAAM,UAAU;AAClC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SACE;AAAA,MACJ;AAAA,IACF;AAEA,QAAI,cAAc,GAAG;AACnB,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,QACE,cAAc,aACd,cAAc,YACd,cAAc,UACd;AACA,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,QAAI,cAAc,YAAY,CAAC,WAAW;AACxC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,UAAM,MAAM,mBAAmB,QAAQ;AACvC,UAAM,UAAU,aAAa,UAAU,GAAG;AAC1C,UAAM,WAAW,cAAc,OAAO;AACtC,QAAI,CAAC,UAAU;AACb,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,QAAI,cAAc,YAAY,cAAc,SAAS,MAAM,QAAQ;AACjE,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS,uEAAuE,SAAS,MAAM,MAAM;AAAA,MACvG;AAAA,IACF,YACG,cAAc,aAAa,cAAc,cACzC,eAAe,SAAS,MAAM,UAAU,CAAC,SAAS,MAAM,WAAW,IACpE;AACA,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS,8CAA8C,SAAS,MAAM,MAAM;AAAA,MAC9E;AAAA,IACF;AAEA,WAAO,EAAE,QAAQ,KAAK;AAAA,EACxB;AAAA,EACA,OAAO,KAAK;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAAG;AACD,UAAM,WAAW,WAAW,aAAa,IACrC,gBACA,QAAQ,OAAO,GAAG,aAAa;AAEnC,QAAI;AACF,YAAM,MAAM,mBAAmB,QAAQ;AACvC,YAAM,UAAU,aAAa,UAAU,GAAG;AAC1C,YAAM,WAAW,KAAK,MAAM,OAAO;AACnC,YAAM,WAAW,SAAS,SAAS,eAAe,QAAQ;AAE1D,UAAI,cAAc,UAAU;AAE1B,iBAAS,MAAM,OAAO,aAAa,CAAC;AAAA,MACtC,WAAW,cAAc,UAAU;AAEjC,cAAM,WAAW;AAAA,UACf;AAAA;AAAA,UACA,QAAQ;AAAA,UACR,UAAU,CAAC;AAAA,QACb;AACA,iBAAS,MAAM;AAAA,UACb;AAAA,UACA;AAAA,UACA,aAAa,aAAa,WAAW,EAAE,GAAG,UAAU,SAAS,CAAC,EAAE;AAAA,QAClE;AAAA,MACF,OAAO;AAEL,cAAM,aAAa,SAAS,MAAM,WAAW;AAC7C,mBAAW,SAAS;AAEpB,mBAAW,kBAAkB;AAC7B,mBAAW,UAAU,CAAC;AACtB,YAAI,aAAa,cAAc,WAAW,WAAW;AACnD,qBAAW,YAAY;AAAA,QACzB;AAAA,MACF;AAEA,YAAM,UAAU,kBAAkB,QAAQ;AAC1C,YAAM,kBAAkB,KAAK,UAAU,UAAU,MAAM,CAAC;AACxD,uBAAiB,UAAU,iBAAiB,KAAK,OAAQ;AAGzD,qBAAe,UAAU,eAAe;AAGxC,wBAAkB,eAAe;AAAA,QAC/B,UAAU;AAAA,QACV,YAAY;AAAA,QACZ,WAAW;AAAA,QACX,UAAU;AAAA,QACV,UAAU,aAAa;AAAA,QACvB,WAAW,KAAK,IAAI;AAAA,QACpB,WAAW;AAAA,MACb,CAAC;AACD,YAAM,OAAO;AAAA,QACX;AAAA,QACA;AAAA,QACA,WAAW,aAAa;AAAA,QACxB;AAAA,QACA,WAAW,aAAa;AAAA,QACxB,OAAO;AAAA,MACT;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN;AAAA,QACA,oBAAoB,KAAK,yBAAyB,IAAI;AAAA,MACxD;AAAA,IACF,SAAS,OAAO;AACd,UAAI,iBAAiB,OAAO;AAC1B,cAAMA,QAAO;AAAA,UACX;AAAA,UACA;AAAA,UACA,WAAW,aAAa;AAAA,UACxB,UAAU;AAAA,UACV,WAAW;AAAA,UACX,OAAO,MAAM;AAAA,QACf;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,MAAAA;AAAA,UACA,oBAAoB,KAAK,yBAAyBA,KAAI;AAAA,QACxD;AACA;AAAA,MACF;AACA,YAAM,OAAO;AAAA,QACX;AAAA,QACA;AAAA,QACA,WAAW,aAAa;AAAA,QACxB,UAAU;AAAA,QACV,WAAW;AAAA,QACX,OAAO;AAAA,MACT;AACA,YAAM;AAAA,QACJ,MAAM;AAAA,QACN;AAAA,QACA,oBAAoB,KAAK,yBAAyB,IAAI;AAAA,MACxD;AAAA,IACF;AAAA,EACF;AACF;",
-  "names": ["data"]
-}

package/dist/tools/NotebookEditTool/prompt.js

@@ -1,7 +0,0 @@
-const DESCRIPTION = "Replace the contents of a specific cell in a Jupyter notebook.";
-const PROMPT = `Completely replaces the contents of a specific cell in a Jupyter notebook (.ipynb file) with new source. Jupyter notebooks are interactive documents that combine code, text, and visualizations, commonly used for data analysis and scientific computing. The notebook_path parameter must be an absolute path, not a relative path. The cell_number is 0-indexed. Use edit_mode=insert to add a new cell at the index specified by cell_number. Use edit_mode=delete to delete the cell at the index specified by cell_number.`;
-export {
-  DESCRIPTION,
-  PROMPT
-};
-//# sourceMappingURL=prompt.js.map

package/dist/tools/NotebookEditTool/prompt.js.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/tools/NotebookEditTool/prompt.ts"],
-  "sourcesContent": ["export const DESCRIPTION =\n  'Replace the contents of a specific cell in a Jupyter notebook.'\nexport const PROMPT = `Completely replaces the contents of a specific cell in a Jupyter notebook (.ipynb file) with new source. Jupyter notebooks are interactive documents that combine code, text, and visualizations, commonly used for data analysis and scientific computing. The notebook_path parameter must be an absolute path, not a relative path. The cell_number is 0-indexed. Use edit_mode=insert to add a new cell at the index specified by cell_number. Use edit_mode=delete to delete the cell at the index specified by cell_number.`\n"],
-  "mappings": "AAAO,MAAM,cACX;AACK,MAAM,SAAS;",
-  "names": []
-}

package/dist/tools/NotebookReadTool/NotebookReadTool.js

@@ -1,212 +0,0 @@
-import { existsSync, readFileSync } from "fs";
-import { Text } from "ink";
-import { extname, isAbsolute, relative, resolve } from "path";
-import * as React from "react";
-import { z } from "zod";
-import { FallbackToolUseRejectedMessage } from "../../components/FallbackToolUseRejectedMessage.js";
-import { formatOutput } from "../BashTool/utils.js";
-import { getCwd } from "../../utils/state.js";
-import { findSimilarFile } from "../../utils/file.js";
-import { DESCRIPTION, PROMPT } from "./prompt.js";
-import { hasReadPermission } from "../../utils/permissions/filesystem.js";
-const inputSchema = z.strictObject({
-  notebook_path: z.string().describe(
-    "The absolute path to the Jupyter notebook file to read (must be absolute, not relative)"
-  )
-});
-const NotebookReadTool = {
-  name: "ReadNotebook",
-  async description() {
-    return DESCRIPTION;
-  },
-  async prompt() {
-    return PROMPT;
-  },
-  isReadOnly() {
-    return true;
-  },
-  isConcurrencySafe() {
-    return true;
-  },
-  inputSchema,
-  userFacingName() {
-    return "Read Notebook";
-  },
-  async isEnabled() {
-    return true;
-  },
-  needsPermissions({ notebook_path }) {
-    return !hasReadPermission(notebook_path);
-  },
-  async validateInput({ notebook_path }) {
-    const fullFilePath = isAbsolute(notebook_path) ? notebook_path : resolve(getCwd(), notebook_path);
-    if (!existsSync(fullFilePath)) {
-      const similarFilename = findSimilarFile(fullFilePath);
-      let message = "File does not exist.";
-      if (similarFilename) {
-        message += ` Did you mean ${similarFilename}?`;
-      }
-      return {
-        result: false,
-        message
-      };
-    }
-    if (extname(fullFilePath) !== ".ipynb") {
-      return {
-        result: false,
-        message: "File must be a Jupyter notebook (.ipynb file)."
-      };
-    }
-    return { result: true };
-  },
-  renderToolUseMessage(input, { verbose }) {
-    return `notebook_path: ${verbose ? input.notebook_path : relative(getCwd(), input.notebook_path)}`;
-  },
-  renderToolUseRejectedMessage() {
-    return /* @__PURE__ */ React.createElement(FallbackToolUseRejectedMessage, null);
-  },
-  renderToolResultMessage(content) {
-    if (!content) {
-      return /* @__PURE__ */ React.createElement(Text, null, "No cells found in notebook");
-    }
-    if (content.length < 1 || !content[0]) {
-      return /* @__PURE__ */ React.createElement(Text, null, "No cells found in notebook");
-    }
-    return /* @__PURE__ */ React.createElement(Text, null, "Read ", content.length, " cells");
-  },
-  async *call({ notebook_path }) {
-    const fullPath = isAbsolute(notebook_path) ? notebook_path : resolve(getCwd(), notebook_path);
-    const content = readFileSync(fullPath, "utf-8");
-    const notebook = JSON.parse(content);
-    const language = notebook.metadata.language_info?.name ?? "python";
-    const cells = notebook.cells.map(
-      (cell, index) => processCell(cell, index, language)
-    );
-    yield {
-      type: "result",
-      resultForAssistant: this.renderResultForAssistant(cells),
-      data: cells
-    };
-  },
-  renderResultForAssistant(data) {
-    return data.map((cell, index) => {
-      let content = `Cell ${index + 1} (${cell.cellType}):
-${cell.source}`;
-      if (cell.outputs && cell.outputs.length > 0) {
-        const outputText = cell.outputs.map((output) => output.text).filter(Boolean).join("\n");
-        if (outputText) {
-          content += `
-Output:
-${outputText}`;
-        }
-      }
-      return content;
-    }).join("\n\n");
-  }
-};
-function processOutputText(text) {
-  if (!text) return "";
-  const rawText = Array.isArray(text) ? text.join("") : text;
-  const { truncatedContent } = formatOutput(rawText);
-  return truncatedContent;
-}
-function extractImage(data) {
-  if (typeof data["image/png"] === "string") {
-    return {
-      image_data: data["image/png"],
-      media_type: "image/png"
-    };
-  }
-  if (typeof data["image/jpeg"] === "string") {
-    return {
-      image_data: data["image/jpeg"],
-      media_type: "image/jpeg"
-    };
-  }
-  return void 0;
-}
-function processOutput(output) {
-  switch (output.output_type) {
-    case "stream":
-      return {
-        output_type: output.output_type,
-        text: processOutputText(output.text)
-      };
-    case "execute_result":
-    case "display_data":
-      return {
-        output_type: output.output_type,
-        text: processOutputText(output.data?.["text/plain"]),
-        image: output.data && extractImage(output.data)
-      };
-    case "error":
-      return {
-        output_type: output.output_type,
-        text: processOutputText(
-          `${output.ename}: ${output.evalue}
-${output.traceback.join("\n")}`
-        )
-      };
-  }
-}
-function processCell(cell, index, language) {
-  const cellData = {
-    cell: index,
-    cellType: cell.cell_type,
-    source: Array.isArray(cell.source) ? cell.source.join("") : cell.source,
-    language,
-    execution_count: cell.execution_count
-  };
-  if (cell.outputs?.length) {
-    cellData.outputs = cell.outputs.map(processOutput);
-  }
-  return cellData;
-}
-function cellContentToToolResult(cell) {
-  const metadata = [];
-  if (cell.cellType !== "code") {
-    metadata.push(`<cell_type>${cell.cellType}</cell_type>`);
-  }
-  if (cell.language !== "python" && cell.cellType === "code") {
-    metadata.push(`<language>${cell.language}</language>`);
-  }
-  const cellContent = `<cell ${cell.cell}>${metadata.join("")}${cell.source}</cell ${cell.cell}>`;
-  return {
-    text: cellContent,
-    type: "text"
-  };
-}
-function cellOutputToToolResult(output) {
-  const outputs = [];
-  if (output.text) {
-    outputs.push({
-      text: `
-${output.text}`,
-      type: "text"
-    });
-  }
-  if (output.image) {
-    outputs.push({
-      type: "image",
-      source: {
-        data: output.image.image_data,
-        media_type: output.image.media_type,
-        type: "base64"
-      }
-    });
-  }
-  return outputs;
-}
-function getToolResultFromCell(cell) {
-  const contentResult = cellContentToToolResult(cell);
-  const outputResults = cell.outputs?.flatMap(cellOutputToToolResult);
-  return [contentResult, ...outputResults ?? []];
-}
-function isNotebookCellType(value) {
-  return value === "code" || value === "markdown";
-}
-export {
-  NotebookReadTool,
-  isNotebookCellType
-};
-//# sourceMappingURL=NotebookReadTool.js.map

package/dist/tools/NotebookReadTool/NotebookReadTool.js.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/tools/NotebookReadTool/NotebookReadTool.tsx"],
-  "sourcesContent": ["import type {\n  ImageBlockParam,\n  TextBlockParam,\n} from '@anthropic-ai/sdk/resources/index.mjs'\n\nimport { existsSync, readFileSync } from 'fs'\nimport { Text } from 'ink'\nimport { extname, isAbsolute, relative, resolve } from 'path'\nimport * as React from 'react'\nimport { z } from 'zod'\nimport { FallbackToolUseRejectedMessage } from '../../components/FallbackToolUseRejectedMessage'\nimport { Tool } from '../../Tool'\nimport {\n  NotebookCellSource,\n  NotebookContent,\n  NotebookCell,\n  NotebookOutputImage,\n  NotebookCellSourceOutput,\n  NotebookCellOutput,\n  NotebookCellType,\n} from '../../types/notebook'\nimport { formatOutput } from '../BashTool/utils'\nimport { getCwd } from '../../utils/state'\nimport { findSimilarFile } from '../../utils/file'\nimport { DESCRIPTION, PROMPT } from './prompt'\nimport { hasReadPermission } from '../../utils/permissions/filesystem'\n\nconst inputSchema = z.strictObject({\n  notebook_path: z\n    .string()\n    .describe(\n      'The absolute path to the Jupyter notebook file to read (must be absolute, not relative)',\n    ),\n})\n\ntype In = typeof inputSchema\ntype Out = NotebookCellSource[]\n\n\nexport const NotebookReadTool = {\n  name: 'ReadNotebook',\n  async description() {\n    return DESCRIPTION\n  },\n  async prompt() {\n    return PROMPT\n  },\n  isReadOnly() {\n    return true\n  },\n  isConcurrencySafe() {\n    return true // NotebookReadTool is read-only, safe for concurrent execution\n  },\n  inputSchema,\n  userFacingName() {\n    return 'Read Notebook'\n  },\n  async isEnabled() {\n    return true\n  },\n  needsPermissions({ notebook_path }) {\n    return !hasReadPermission(notebook_path)\n  },\n  async validateInput({ notebook_path }) {\n    const fullFilePath = isAbsolute(notebook_path)\n      ? notebook_path\n      : resolve(getCwd(), notebook_path)\n\n    if (!existsSync(fullFilePath)) {\n      // Try to find a similar file with a different extension\n      const similarFilename = findSimilarFile(fullFilePath)\n      let message = 'File does not exist.'\n\n      // If we found a similar file, suggest it to the assistant\n      if (similarFilename) {\n        message += ` Did you mean ${similarFilename}?`\n      }\n\n      return {\n        result: false,\n        message,\n      }\n    }\n\n    if (extname(fullFilePath) !== '.ipynb') {\n      return {\n        result: false,\n        message: 'File must be a Jupyter notebook (.ipynb file).',\n      }\n    }\n\n    return { result: true }\n  },\n  renderToolUseMessage(input, { verbose }) {\n    return `notebook_path: ${verbose ? input.notebook_path : relative(getCwd(), input.notebook_path)}`\n  },\n  renderToolUseRejectedMessage() {\n    return <FallbackToolUseRejectedMessage />\n  },\n\n  renderToolResultMessage(content) {\n    if (!content) {\n      return <Text>No cells found in notebook</Text>\n    }\n    if (content.length < 1 || !content[0]) {\n      return <Text>No cells found in notebook</Text>\n    }\n    return <Text>Read {content.length} cells</Text>\n  },\n  async *call({ notebook_path }) {\n    const fullPath = isAbsolute(notebook_path)\n      ? notebook_path\n      : resolve(getCwd(), notebook_path)\n\n    const content = readFileSync(fullPath, 'utf-8')\n    const notebook = JSON.parse(content) as NotebookContent\n    const language = notebook.metadata.language_info?.name ?? 'python'\n    const cells = notebook.cells.map((cell, index) =>\n      processCell(cell, index, language),\n    )\n\n    yield {\n      type: 'result',\n      resultForAssistant: this.renderResultForAssistant(cells),\n      data: cells,\n    }\n  },\n  renderResultForAssistant(data: NotebookCellSource[]) {\n    // Convert the complex structure to a string representation for the assistant\n    return data.map((cell, index) => {\n      let content = `Cell ${index + 1} (${cell.cellType}):\\n${cell.source}`\n      if (cell.outputs && cell.outputs.length > 0) {\n        const outputText = cell.outputs.map(output => output.text).filter(Boolean).join('\\n')\n        if (outputText) {\n          content += `\\nOutput:\\n${outputText}`\n        }\n      }\n      return content\n    }).join('\\n\\n')\n  },\n} satisfies Tool<In, Out>\n\nfunction processOutputText(text: string | string[] | undefined): string {\n  if (!text) return ''\n  const rawText = Array.isArray(text) ? text.join('') : text\n  const { truncatedContent } = formatOutput(rawText)\n  return truncatedContent\n}\n\nfunction extractImage(\n  data: Record<string, unknown>,\n): NotebookOutputImage | undefined {\n  if (typeof data['image/png'] === 'string') {\n    return {\n      image_data: data['image/png'] as string,\n      media_type: 'image/png',\n    }\n  }\n  if (typeof data['image/jpeg'] === 'string') {\n    return {\n      image_data: data['image/jpeg'] as string,\n      media_type: 'image/jpeg',\n    }\n  }\n  return undefined\n}\n\nfunction processOutput(output: NotebookCellOutput) {\n  switch (output.output_type) {\n    case 'stream':\n      return {\n        output_type: output.output_type,\n        text: processOutputText(output.text),\n      }\n    case 'execute_result':\n    case 'display_data':\n      return {\n        output_type: output.output_type,\n        text: processOutputText(output.data?.['text/plain'] as string | string[] | undefined),\n        image: output.data && extractImage(output.data),\n      }\n    case 'error':\n      return {\n        output_type: output.output_type,\n        text: processOutputText(\n          `${output.ename}: ${output.evalue}\\n${output.traceback.join('\\n')}`,\n        ),\n      }\n  }\n}\n\nfunction processCell(\n  cell: NotebookCell,\n  index: number,\n  language: string,\n): NotebookCellSource {\n  const cellData: NotebookCellSource = {\n    cell: index,\n    cellType: cell.cell_type,\n    source: Array.isArray(cell.source) ? cell.source.join('') : cell.source,\n    language,\n    execution_count: cell.execution_count,\n  }\n\n  if (cell.outputs?.length) {\n    cellData.outputs = cell.outputs.map(processOutput)\n  }\n\n  return cellData\n}\n\nfunction cellContentToToolResult(cell: NotebookCellSource): TextBlockParam {\n  const metadata = []\n  if (cell.cellType !== 'code') {\n    metadata.push(`<cell_type>${cell.cellType}</cell_type>`)\n  }\n  if (cell.language !== 'python' && cell.cellType === 'code') {\n    metadata.push(`<language>${cell.language}</language>`)\n  }\n  const cellContent = `<cell ${cell.cell}>${metadata.join('')}${cell.source}</cell ${cell.cell}>`\n  return {\n    text: cellContent,\n    type: 'text',\n  }\n}\n\nfunction cellOutputToToolResult(output: NotebookCellSourceOutput) {\n  const outputs: (TextBlockParam | ImageBlockParam)[] = []\n  if (output.text) {\n    outputs.push({\n      text: `\\n${output.text}`,\n      type: 'text',\n    })\n  }\n  if (output.image) {\n    outputs.push({\n      type: 'image',\n      source: {\n        data: output.image.image_data,\n        media_type: output.image.media_type,\n        type: 'base64',\n      },\n    })\n  }\n  return outputs\n}\n\nfunction getToolResultFromCell(cell: NotebookCellSource) {\n  const contentResult = cellContentToToolResult(cell)\n  const outputResults = cell.outputs?.flatMap(cellOutputToToolResult)\n  return [contentResult, ...(outputResults ?? [])]\n}\n\nexport function isNotebookCellType(\n  value: string | null,\n): value is NotebookCellType {\n  return value === 'code' || value === 'markdown'\n}\n"],
-  "mappings": "AAKA,SAAS,YAAY,oBAAoB;AACzC,SAAS,YAAY;AACrB,SAAS,SAAS,YAAY,UAAU,eAAe;AACvD,YAAY,WAAW;AACvB,SAAS,SAAS;AAClB,SAAS,sCAAsC;AAW/C,SAAS,oBAAoB;AAC7B,SAAS,cAAc;AACvB,SAAS,uBAAuB;AAChC,SAAS,aAAa,cAAc;AACpC,SAAS,yBAAyB;AAElC,MAAM,cAAc,EAAE,aAAa;AAAA,EACjC,eAAe,EACZ,OAAO,EACP;AAAA,IACC;AAAA,EACF;AACJ,CAAC;AAMM,MAAM,mBAAmB;AAAA,EAC9B,MAAM;AAAA,EACN,MAAM,cAAc;AAClB,WAAO;AAAA,EACT;AAAA,EACA,MAAM,SAAS;AACb,WAAO;AAAA,EACT;AAAA,EACA,aAAa;AACX,WAAO;AAAA,EACT;AAAA,EACA,oBAAoB;AAClB,WAAO;AAAA,EACT;AAAA,EACA;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,MAAM,YAAY;AAChB,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB,EAAE,cAAc,GAAG;AAClC,WAAO,CAAC,kBAAkB,aAAa;AAAA,EACzC;AAAA,EACA,MAAM,cAAc,EAAE,cAAc,GAAG;AACrC,UAAM,eAAe,WAAW,aAAa,IACzC,gBACA,QAAQ,OAAO,GAAG,aAAa;AAEnC,QAAI,CAAC,WAAW,YAAY,GAAG;AAE7B,YAAM,kBAAkB,gBAAgB,YAAY;AACpD,UAAI,UAAU;AAGd,UAAI,iBAAiB;AACnB,mBAAW,iBAAiB,eAAe;AAAA,MAC7C;AAEA,aAAO;AAAA,QACL,QAAQ;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,QAAI,QAAQ,YAAY,MAAM,UAAU;AACtC,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AAAA,IACF;AAEA,WAAO,EAAE,QAAQ,KAAK;AAAA,EACxB;AAAA,EACA,qBAAqB,OAAO,EAAE,QAAQ,GAAG;AACvC,WAAO,kBAAkB,UAAU,MAAM,gBAAgB,SAAS,OAAO,GAAG,MAAM,aAAa,CAAC;AAAA,EAClG;AAAA,EACA,+BAA+B;AAC7B,WAAO,oCAAC,oCAA+B;AAAA,EACzC;AAAA,EAEA,wBAAwB,SAAS;AAC/B,QAAI,CAAC,SAAS;AACZ,aAAO,oCAAC,YAAK,4BAA0B;AAAA,IACzC;AACA,QAAI,QAAQ,SAAS,KAAK,CAAC,QAAQ,CAAC,GAAG;AACrC,aAAO,oCAAC,YAAK,4BAA0B;AAAA,IACzC;AACA,WAAO,oCAAC,YAAK,SAAM,QAAQ,QAAO,QAAM;AAAA,EAC1C;AAAA,EACA,OAAO,KAAK,EAAE,cAAc,GAAG;AAC7B,UAAM,WAAW,WAAW,aAAa,IACrC,gBACA,QAAQ,OAAO,GAAG,aAAa;AAEnC,UAAM,UAAU,aAAa,UAAU,OAAO;AAC9C,UAAM,WAAW,KAAK,MAAM,OAAO;AACnC,UAAM,WAAW,SAAS,SAAS,eAAe,QAAQ;AAC1D,UAAM,QAAQ,SAAS,MAAM;AAAA,MAAI,CAAC,MAAM,UACtC,YAAY,MAAM,OAAO,QAAQ;AAAA,IACnC;AAEA,UAAM;AAAA,MACJ,MAAM;AAAA,MACN,oBAAoB,KAAK,yBAAyB,KAAK;AAAA,MACvD,MAAM;AAAA,IACR;AAAA,EACF;AAAA,EACA,yBAAyB,MAA4B;AAEnD,WAAO,KAAK,IAAI,CAAC,MAAM,UAAU;AAC/B,UAAI,UAAU,QAAQ,QAAQ,CAAC,KAAK,KAAK,QAAQ;AAAA,EAAO,KAAK,MAAM;AACnE,UAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;AAC3C,cAAM,aAAa,KAAK,QAAQ,IAAI,YAAU,OAAO,IAAI,EAAE,OAAO,OAAO,EAAE,KAAK,IAAI;AACpF,YAAI,YAAY;AACd,qBAAW;AAAA;AAAA,EAAc,UAAU;AAAA,QACrC;AAAA,MACF;AACA,aAAO;AAAA,IACT,CAAC,EAAE,KAAK,MAAM;AAAA,EAChB;AACF;AAEA,SAAS,kBAAkB,MAA6C;AACtE,MAAI,CAAC,KAAM,QAAO;AAClB,QAAM,UAAU,MAAM,QAAQ,IAAI,IAAI,KAAK,KAAK,EAAE,IAAI;AACtD,QAAM,EAAE,iBAAiB,IAAI,aAAa,OAAO;AACjD,SAAO;AACT;AAEA,SAAS,aACP,MACiC;AACjC,MAAI,OAAO,KAAK,WAAW,MAAM,UAAU;AACzC,WAAO;AAAA,MACL,YAAY,KAAK,WAAW;AAAA,MAC5B,YAAY;AAAA,IACd;AAAA,EACF;AACA,MAAI,OAAO,KAAK,YAAY,MAAM,UAAU;AAC1C,WAAO;AAAA,MACL,YAAY,KAAK,YAAY;AAAA,MAC7B,YAAY;AAAA,IACd;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,cAAc,QAA4B;AACjD,UAAQ,OAAO,aAAa;AAAA,IAC1B,KAAK;AACH,aAAO;AAAA,QACL,aAAa,OAAO;AAAA,QACpB,MAAM,kBAAkB,OAAO,IAAI;AAAA,MACrC;AAAA,IACF,KAAK;AAAA,IACL,KAAK;AACH,aAAO;AAAA,QACL,aAAa,OAAO;AAAA,QACpB,MAAM,kBAAkB,OAAO,OAAO,YAAY,CAAkC;AAAA,QACpF,OAAO,OAAO,QAAQ,aAAa,OAAO,IAAI;AAAA,MAChD;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,aAAa,OAAO;AAAA,QACpB,MAAM;AAAA,UACJ,GAAG,OAAO,KAAK,KAAK,OAAO,MAAM;AAAA,EAAK,OAAO,UAAU,KAAK,IAAI,CAAC;AAAA,QACnE;AAAA,MACF;AAAA,EACJ;AACF;AAEA,SAAS,YACP,MACA,OACA,UACoB;AACpB,QAAM,WAA+B;AAAA,IACnC,MAAM;AAAA,IACN,UAAU,KAAK;AAAA,IACf,QAAQ,MAAM,QAAQ,KAAK,MAAM,IAAI,KAAK,OAAO,KAAK,EAAE,IAAI,KAAK;AAAA,IACjE;AAAA,IACA,iBAAiB,KAAK;AAAA,EACxB;AAEA,MAAI,KAAK,SAAS,QAAQ;AACxB,aAAS,UAAU,KAAK,QAAQ,IAAI,aAAa;AAAA,EACnD;AAEA,SAAO;AACT;AAEA,SAAS,wBAAwB,MAA0C;AACzE,QAAM,WAAW,CAAC;AAClB,MAAI,KAAK,aAAa,QAAQ;AAC5B,aAAS,KAAK,cAAc,KAAK,QAAQ,cAAc;AAAA,EACzD;AACA,MAAI,KAAK,aAAa,YAAY,KAAK,aAAa,QAAQ;AAC1D,aAAS,KAAK,aAAa,KAAK,QAAQ,aAAa;AAAA,EACvD;AACA,QAAM,cAAc,SAAS,KAAK,IAAI,IAAI,SAAS,KAAK,EAAE,CAAC,GAAG,KAAK,MAAM,UAAU,KAAK,IAAI;AAC5F,SAAO;AAAA,IACL,MAAM;AAAA,IACN,MAAM;AAAA,EACR;AACF;AAEA,SAAS,uBAAuB,QAAkC;AAChE,QAAM,UAAgD,CAAC;AACvD,MAAI,OAAO,MAAM;AACf,YAAQ,KAAK;AAAA,MACX,MAAM;AAAA,EAAK,OAAO,IAAI;AAAA,MACtB,MAAM;AAAA,IACR,CAAC;AAAA,EACH;AACA,MAAI,OAAO,OAAO;AAChB,YAAQ,KAAK;AAAA,MACX,MAAM;AAAA,MACN,QAAQ;AAAA,QACN,MAAM,OAAO,MAAM;AAAA,QACnB,YAAY,OAAO,MAAM;AAAA,QACzB,MAAM;AAAA,MACR;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO;AACT;AAEA,SAAS,sBAAsB,MAA0B;AACvD,QAAM,gBAAgB,wBAAwB,IAAI;AAClD,QAAM,gBAAgB,KAAK,SAAS,QAAQ,sBAAsB;AAClE,SAAO,CAAC,eAAe,GAAI,iBAAiB,CAAC,CAAE;AACjD;AAEO,SAAS,mBACd,OAC2B;AAC3B,SAAO,UAAU,UAAU,UAAU;AACvC;",
-  "names": []
-}