@e2b/code-interpreter 0.0.9-beta.6 → 0.0.9-beta.8

package/dist/index.js

@@ -80,6 +80,7 @@ __export(src_exports, {
   default: () => src_default
 });
 module.exports = __toCommonJS(src_exports);
+__reExport(src_exports, require("e2b"), module.exports);
 
 // src/codeInterpreter.ts
 var import_e2b2 = require("e2b");
@@ -453,12 +454,11 @@ var _CodeInterpreter = class _CodeInterpreter extends import_e2b2.Sandbox {
     );
   }
 };
-_CodeInterpreter.defaultTemplate = "ci-no-ws";
+_CodeInterpreter.defaultTemplate = "code-interpreter-stateful";
 _CodeInterpreter.jupyterPort = 49999;
 var CodeInterpreter = _CodeInterpreter;
 
 // src/index.ts
-__reExport(src_exports, require("e2b"), module.exports);
 var src_default = CodeInterpreter;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/codeInterpreter.ts","../src/messaging.ts"],"sourcesContent":["export { CodeInterpreter, JupyterExtension } from './codeInterpreter'\n\nexport type { Logs, ExecutionError, Result, Execution, MIMEType, RawData, OutputMessage } from './messaging'\n\nimport { CodeInterpreter } from './codeInterpreter'\n\nexport * from 'e2b'\n\nexport default CodeInterpreter\n","import { ConnectionConfig, Sandbox } from 'e2b'\n\nimport { Result, Execution, OutputMessage, parseOutput, extractError } from './messaging'\n\nasync function* readLines(stream: ReadableStream<Uint8Array>) {\n  const reader = stream.getReader();\n  let buffer = ''\n\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (value !== undefined) {\n        buffer += new TextDecoder().decode(value)\n      }\n\n      if (done) {\n        if (buffer.length > 0) {\n          yield buffer\n        }\n        break\n      }\n\n      let newlineIdx = -1\n\n      do {\n        newlineIdx = buffer.indexOf('\\n')\n        if (newlineIdx !== -1) {\n          yield buffer.slice(0, newlineIdx)\n          buffer = buffer.slice(newlineIdx + 1)\n        }\n      } while (newlineIdx !== -1)\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport class JupyterExtension {\n  private static readonly execTimeoutMs = 300_000\n  private static readonly defaultKernelID = 'default'\n\n  constructor(private readonly url: string, private readonly connectionConfig: ConnectionConfig) { }\n\n  async execCell(\n    code: string,\n    opts?: {\n      kernelID?: string,\n      onStdout?: (output: OutputMessage) => (Promise<any> | any),\n      onStderr?: (output: OutputMessage) => (Promise<any> | any),\n      onResult?: (data: Result) => (Promise<any> | any),\n      timeoutMs?: number,\n      requestTimeoutMs?: number,\n    },\n  ): Promise<Execution> {\n    const controller = new AbortController()\n\n    const requestTimeout = opts?.requestTimeoutMs ?? this.connectionConfig.requestTimeoutMs\n\n    const reqTimer = requestTimeout ? setTimeout(() => {\n      controller.abort()\n    }, requestTimeout)\n      : undefined\n\n    const res = await fetch(`${this.url}/execute`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        code,\n        context_id: opts?.kernelID,\n      }),\n      keepalive: true,\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    if (!res.body) {\n      throw new Error(`Not response body: ${res.statusText} ${await res?.text()}`)\n    }\n\n    clearTimeout(reqTimer)\n\n    const bodyTimeout = opts?.timeoutMs ?? JupyterExtension.execTimeoutMs\n\n    const bodyTimer = bodyTimeout\n      ? setTimeout(() => {\n        controller.abort()\n      }, bodyTimeout)\n      : undefined\n\n    const execution = new Execution()\n\n    try {\n      for await (const chunk of readLines(res.body)) {\n        await parseOutput(execution, chunk, opts?.onStdout, opts?.onStderr, opts?.onResult)\n      }\n    } finally {\n      clearTimeout(bodyTimer)\n    }\n\n    return execution\n  }\n\n  async createKernel({\n    cwd,\n    kernelName,\n    requestTimeoutMs,\n  }: {\n    cwd?: string,\n    kernelName?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<string> {\n    const res = await fetch(`${this.url}/contexts`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        name: kernelName,\n        cwd,\n      }),\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    const data = await res.json()\n    return data.id\n  }\n\n  async restartKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n    const res = await fetch(`${this.url}/contexts/${kernelID}/restart`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async shutdownKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n\n    const res = await fetch(`${this.url}/contexts/${kernelID}`, {\n      method: 'DELETE',\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async listKernels({\n    requestTimeoutMs,\n  }: {\n    requestTimeoutMs?: number,\n  } = {}): Promise<{ kernelID: string, name: string }[]> {\n    const res = await fetch(`${this.url}/contexts`, {\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    return (await res.json()).map((kernel: any) => ({ kernelID: kernel.id, name: kernel.name }))\n  }\n}\n\nexport class CodeInterpreter extends Sandbox {\n  protected static override readonly defaultTemplate: string = 'ci-no-ws'\n  protected static readonly jupyterPort = 49999\n\n  readonly notebook = new JupyterExtension(\n    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(CodeInterpreter.jupyterPort)}`,\n    this.connectionConfig,\n  )\n}\n","import { NotFoundError, SandboxError, TimeoutError } from 'e2b'\n\nexport async function extractError(res: Response) {\n  if (res.ok) {\n    return\n  }\n\n  switch (res.status) {\n    case 502:\n      return new TimeoutError(\n        `${await res.text()}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeoutMs' when starting the sandbox or calling '.setTimeout' on the sandbox with the desired timeout.`\n      )\n    case 404:\n      return new NotFoundError(await res.text())\n    default:\n      return new SandboxError(`${res.status} ${res.statusText}`)\n  }\n}\n\nexport class OutputMessage {\n  constructor(\n    public readonly line: string,\n    /**\n     * Unix epoch in nanoseconds\n     */\n    public readonly timestamp: number,\n    public readonly error: boolean,\n  ) {\n  }\n\n  public toString() {\n    return this.line\n  }\n}\n\n/**\n * Represents an error that occurred during the execution of a cell.\n * The error contains the name of the error, the value of the error, and the traceback.\n */\nexport class ExecutionError {\n  constructor(\n    /**\n     * Name of the error.\n     **/\n    public name: string,\n    /**\n     * Value of the error.\n     **/\n    public value: string,\n    /**\n     * The raw traceback of the error.\n     **/\n    public traceback: string,\n  ) { }\n}\n\n/**\n * Represents a MIME type.\n */\nexport type MIMEType = string\n\n/**\n * Dictionary that maps MIME types to their corresponding string representations of the data.\n */\nexport type RawData = {\n  [key: MIMEType]: string\n}\n\n/**\n * Represents the data to be displayed as a result of executing a cell in a Jupyter notebook.\n * The result is similar to the structure returned by ipython kernel: https://ipython.readthedocs.io/en/stable/development/execution.html#execution-semantics\n *\n *\n * The result can contain multiple types of data, such as text, images, plots, etc. Each type of data is represented\n * as a string, and the result can contain multiple types of data. The display calls don't have to have text representation,\n * for the actual result the representation is always present for the result, the other representations are always optional.\n */\nexport class Result {\n  /**\n   * Text representation of the result.\n   */\n  readonly text?: string\n  /**\n   * HTML representation of the data.\n   */\n  readonly html?: string\n  /**\n   * Markdown representation of the data.\n   */\n  readonly markdown?: string\n  /**\n   * SVG representation of the data.\n   */\n  readonly svg?: string\n  /**\n   * PNG representation of the data.\n   */\n  readonly png?: string\n  /**\n   * JPEG representation of the data.\n   */\n  readonly jpeg?: string\n  /**\n   * PDF representation of the data.\n   */\n  readonly pdf?: string\n  /**\n   * LaTeX representation of the data.\n   */\n  readonly latex?: string\n  /**\n   * JSON representation of the data.\n   */\n  readonly json?: string\n  /**\n   * JavaScript representation of the data.\n   */\n  readonly javascript?: string\n  /**\n   * Extra data that can be included. Not part of the standard types.\n   */\n  readonly extra?: any\n\n  readonly raw: RawData\n\n  constructor(rawData: RawData, public readonly isMainResult: boolean) {\n    const data = { ...rawData }\n    delete data['type']\n    delete data['is_main_result']\n\n    this.text = data['text']\n    this.html = data['html']\n    this.markdown = data['markdown']\n    this.svg = data['svg']\n    this.png = data['png']\n    this.jpeg = data['jpeg']\n    this.pdf = data['pdf']\n    this.latex = data['latex']\n    this.json = data['json']\n    this.javascript = data['javascript']\n    this.isMainResult = isMainResult\n    this.raw = data\n\n    this.extra = {}\n\n    for (const key of Object.keys(data)) {\n      if (\n        ![\n          'plain',\n          'html',\n          'markdown',\n          'svg',\n          'png',\n          'jpeg',\n          'pdf',\n          'latex',\n          'json',\n          'javascript'\n        ].includes(key)\n      ) {\n        this.extra[key] = data[key]\n      }\n    }\n  }\n\n  /**\n   * Returns all the formats available for the result.\n   *\n   * @returns Array of strings representing the formats available for the result.\n   */\n  formats(): string[] {\n    const formats = []\n    if (this.html) {\n      formats.push('html')\n    }\n    if (this.markdown) {\n      formats.push('markdown')\n    }\n    if (this.svg) {\n      formats.push('svg')\n    }\n    if (this.png) {\n      formats.push('png')\n    }\n    if (this.jpeg) {\n      formats.push('jpeg')\n    }\n    if (this.pdf) {\n      formats.push('pdf')\n    }\n    if (this.latex) {\n      formats.push('latex')\n    }\n    if (this.json) {\n      formats.push('json')\n    }\n    if (this.javascript) {\n      formats.push('javascript')\n    }\n\n    for (const key of Object.keys(this.extra)) {\n      formats.push(key)\n    }\n\n    return formats\n  }\n\n  /**\n   * Returns the serializable representation of the result.\n   */\n  toJSON() {\n    return {\n      text: this.text,\n      html: this.html,\n      markdown: this.markdown,\n      svg: this.svg,\n      png: this.png,\n      jpeg: this.jpeg,\n      pdf: this.pdf,\n      latex: this.latex,\n      json: this.json,\n      javascript: this.javascript,\n      ...(Object.keys(this.extra).length > 0 ? { extra: this.extra } : {})\n    }\n  }\n}\n\n/**\n * Data printed to stdout and stderr during execution, usually by print statements, logs, warnings, subprocesses, etc.\n */\nexport type Logs = {\n  /**\n   * List of strings printed to stdout by prints, subprocesses, etc.\n   */\n  stdout: string[]\n  /**\n   * List of strings printed to stderr by prints, subprocesses, etc.\n   */\n  stderr: string[]\n}\n\n/**\n * Represents the result of a cell execution.\n */\nexport class Execution {\n  constructor(\n    /**\n     * List of result of the cell (interactively interpreted last line), display calls (e.g. matplotlib plots).\n     */\n    public results: Result[] = [],\n    /**\n     * Logs printed to stdout and stderr during execution.\n     */\n    public logs: Logs = { stdout: [], stderr: [] },\n    /**\n     * An Error object if an error occurred, null otherwise.\n     */\n    public error?: ExecutionError,\n    /**\n     * Execution count of the cell.\n     */\n    public executionCount?: number\n  ) { }\n\n  /**\n   * Returns the text representation of the main result of the cell.\n   */\n  get text(): string | undefined {\n    for (const data of this.results) {\n      if (data.isMainResult) {\n        return data.text\n      }\n    }\n  }\n\n  /**\n   * Returns the serializable representation of the execution result.\n   */\n  toJSON() {\n    return {\n      results: this.results,\n      logs: this.logs,\n      error: this.error\n    }\n  }\n}\n\nexport async function parseOutput(\n  execution: Execution,\n  line: string,\n  onStdout?: (output: OutputMessage) => (Promise<any> | any),\n  onStderr?: (output: OutputMessage) => (Promise<any> | any),\n  onResult?: (data: Result) => (Promise<any> | any),\n) {\n  const msg = JSON.parse(line)\n\n  switch (msg.type) {\n    case 'result':\n      const result = new Result({ ...msg, type: undefined, is_main_result: undefined }, msg.is_main_result)\n      execution.results.push(result)\n      if (onResult) {\n        await onResult(result)\n      }\n      break\n    case 'stdout':\n      execution.logs.stdout.push(msg.text)\n      if (onStdout) {\n        await onStdout({\n          error: false,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'stderr':\n      execution.logs.stderr.push(msg.text)\n      if (onStderr) {\n        await onStderr({\n          error: true,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'error':\n      execution.error = new ExecutionError(msg.name, msg.value, msg.traceback)\n      break\n    case 'number_of_executions':\n      execution.executionCount = msg.execution_count\n      break\n  }\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,IAAAA,cAA0C;;;ACA1C,iBAA0D;AAE1D,SAAsB,aAAa,KAAe;AAAA;AAChD,QAAI,IAAI,IAAI;AACV;AAAA,IACF;AAEA,YAAQ,IAAI,QAAQ;AAAA,MAClB,KAAK;AACH,eAAO,IAAI;AAAA,UACT,GAAG,MAAM,IAAI,KAAK,CAAC;AAAA,QACrB;AAAA,MACF,KAAK;AACH,eAAO,IAAI,yBAAc,MAAM,IAAI,KAAK,CAAC;AAAA,MAC3C;AACE,eAAO,IAAI,wBAAa,GAAG,IAAI,MAAM,IAAI,IAAI,UAAU,EAAE;AAAA,IAC7D;AAAA,EACF;AAAA;AAsBO,IAAM,iBAAN,MAAqB;AAAA,EAC1B,YAIS,MAIA,OAIA,WACP;AATO;AAIA;AAIA;AAAA,EACL;AACN;AAuBO,IAAM,SAAN,MAAa;AAAA,EAgDlB,YAAY,SAAkC,cAAuB;AAAvB;AAC5C,UAAM,OAAO,mBAAK;AAClB,WAAO,KAAK,MAAM;AAClB,WAAO,KAAK,gBAAgB;AAE5B,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,WAAW,KAAK,UAAU;AAC/B,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,QAAQ,KAAK,OAAO;AACzB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,aAAa,KAAK,YAAY;AACnC,SAAK,eAAe;AACpB,SAAK,MAAM;AAEX,SAAK,QAAQ,CAAC;AAEd,eAAW,OAAO,OAAO,KAAK,IAAI,GAAG;AACnC,UACE,CAAC;AAAA,QACC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,SAAS,GAAG,GACd;AACA,aAAK,MAAM,GAAG,IAAI,KAAK,GAAG;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAoB;AAClB,UAAM,UAAU,CAAC;AACjB,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,UAAU;AACjB,cAAQ,KAAK,UAAU;AAAA,IACzB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,OAAO;AACd,cAAQ,KAAK,OAAO;AAAA,IACtB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,YAAY;AACnB,cAAQ,KAAK,YAAY;AAAA,IAC3B;AAEA,eAAW,OAAO,OAAO,KAAK,KAAK,KAAK,GAAG;AACzC,cAAQ,KAAK,GAAG;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,MACV,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,OAAO,KAAK;AAAA,MACZ,MAAM,KAAK;AAAA,MACX,YAAY,KAAK;AAAA,OACb,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,IAAI,EAAE,OAAO,KAAK,MAAM,IAAI,CAAC;AAAA,EAEtE;AACF;AAmBO,IAAM,YAAN,MAAgB;AAAA,EACrB,YAIS,UAAoB,CAAC,GAIrB,OAAa,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,GAItC,OAIA,gBACP;AAbO;AAIA;AAIA;AAIA;AAAA,EACL;AAAA;AAAA;AAAA;AAAA,EAKJ,IAAI,OAA2B;AAC7B,eAAW,QAAQ,KAAK,SAAS;AAC/B,UAAI,KAAK,cAAc;AACrB,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,SAAS,KAAK;AAAA,MACd,MAAM,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,IACd;AAAA,EACF;AACF;AAEA,SAAsB,YACpB,WACA,MACA,UACA,UACA,UACA;AAAA;AACA,UAAM,MAAM,KAAK,MAAM,IAAI;AAE3B,YAAQ,IAAI,MAAM;AAAA,MAChB,KAAK;AACH,cAAM,SAAS,IAAI,OAAO,iCAAK,MAAL,EAAU,MAAM,QAAW,gBAAgB,OAAU,IAAG,IAAI,cAAc;AACpG,kBAAU,QAAQ,KAAK,MAAM;AAC7B,YAAI,UAAU;AACZ,gBAAM,SAAS,MAAM;AAAA,QACvB;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,QAAQ,IAAI,eAAe,IAAI,MAAM,IAAI,OAAO,IAAI,SAAS;AACvE;AAAA,MACF,KAAK;AACH,kBAAU,iBAAiB,IAAI;AAC/B;AAAA,IACJ;AAAA,EACF;AAAA;;;ADvUA,SAAgB,UAAU,QAAoC;AAAA;AAC5D,UAAM,SAAS,OAAO,UAAU;AAChC,QAAI,SAAS;AAEb,QAAI;AACF,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,kBAAM,OAAO,KAAK;AAE1C,YAAI,UAAU,QAAW;AACvB,oBAAU,IAAI,YAAY,EAAE,OAAO,KAAK;AAAA,QAC1C;AAEA,YAAI,MAAM;AACR,cAAI,OAAO,SAAS,GAAG;AACrB,kBAAM;AAAA,UACR;AACA;AAAA,QACF;AAEA,YAAI,aAAa;AAEjB,WAAG;AACD,uBAAa,OAAO,QAAQ,IAAI;AAChC,cAAI,eAAe,IAAI;AACrB,kBAAM,OAAO,MAAM,GAAG,UAAU;AAChC,qBAAS,OAAO,MAAM,aAAa,CAAC;AAAA,UACtC;AAAA,QACF,SAAS,eAAe;AAAA,MAC1B;AAAA,IACF,UAAE;AACA,aAAO,YAAY;AAAA,IACrB;AAAA,EACF;AAAA;AAEO,IAAM,oBAAN,MAAM,kBAAiB;AAAA,EAI5B,YAA6B,KAA8B,kBAAoC;AAAlE;AAA8B;AAAA,EAAsC;AAAA,EAE3F,SACJ,MACA,MAQoB;AAAA;AAtDxB;AAuDI,YAAM,aAAa,IAAI,gBAAgB;AAEvC,YAAM,kBAAiB,kCAAM,qBAAN,YAA0B,KAAK,iBAAiB;AAEvE,YAAM,WAAW,iBAAiB,WAAW,MAAM;AACjD,mBAAW,MAAM;AAAA,MACnB,GAAG,cAAc,IACb;AAEJ,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,YAAY;AAAA,QAC7C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB;AAAA,UACA,YAAY,6BAAM;AAAA,QACpB,CAAC;AAAA,QACD,WAAW;AAAA,MACb,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,UAAI,CAAC,IAAI,MAAM;AACb,cAAM,IAAI,MAAM,sBAAsB,IAAI,UAAU,IAAI,MAAM,2BAAK,MAAM,EAAE;AAAA,MAC7E;AAEA,mBAAa,QAAQ;AAErB,YAAM,eAAc,kCAAM,cAAN,YAAmB,kBAAiB;AAExD,YAAM,YAAY,cACd,WAAW,MAAM;AACjB,mBAAW,MAAM;AAAA,MACnB,GAAG,WAAW,IACZ;AAEJ,YAAM,YAAY,IAAI,UAAU;AAEhC,UAAI;AACF;AAAA,qCAA0B,UAAU,IAAI,IAAI,IAA5C,YAAAC,QAAA,uDAA+C;AAApC,kBAAM,QAAjB;AACE,kBAAM,YAAY,WAAW,OAAO,6BAAM,UAAU,6BAAM,UAAU,6BAAM,QAAQ;AAAA,UACpF;AAAA,iBAFA,MAlGN;AAkGM,UAAAA,SAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAAAA;AAAA,oBAAAA,OAAA;AAAA;AAAA;AAAA,MAGF,UAAE;AACA,qBAAa,SAAS;AAAA,MACxB;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,EAEM,eAQmB;AAAA,+CARN;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAII,CAAC,GAAoB;AACvB,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAAA,QACD,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,YAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,aAAO,KAAK;AAAA,IACd;AAAA;AAAA,EAEM,gBAMiB;AAAA,+CANH;AAAA,MAClB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AACxC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,YAAY;AAAA,QAClE,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,iBAMiB;AAAA,+CANF;AAAA,MACnB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AAExC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,IAAI;AAAA,QAC1D,QAAQ;AAAA,QACR,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,cAIiD;AAAA,+CAJrC;AAAA,MAChB;AAAA,IACF,IAEI,CAAC,GAAkD;AACrD,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,cAAQ,MAAM,IAAI,KAAK,GAAG,IAAI,CAAC,YAAiB,EAAE,UAAU,OAAO,IAAI,MAAM,OAAO,KAAK,EAAE;AAAA,IAC7F;AAAA;AACF;AAlKa,kBACa,gBAAgB;AAD7B,kBAEa,kBAAkB;AAFrC,IAAM,mBAAN;AAoKA,IAAM,mBAAN,MAAM,yBAAwB,oBAAQ;AAAA,EAAtC;AAAA;AAIL,SAAS,WAAW,IAAI;AAAA,MACtB,GAAG,KAAK,iBAAiB,QAAQ,SAAS,OAAO,MAAM,KAAK,QAAQ,iBAAgB,WAAW,CAAC;AAAA,MAChG,KAAK;AAAA,IACP;AAAA;AACF;AARa,iBACwB,kBAA0B;AADlD,iBAEe,cAAc;AAFnC,IAAM,kBAAN;;;ADpMP,wBAAc,gBANd;AAQA,IAAO,cAAQ;","names":["import_e2b","error"]}
+{"version":3,"sources":["../src/index.ts","../src/codeInterpreter.ts","../src/messaging.ts"],"sourcesContent":["export * from 'e2b'\n\nexport { CodeInterpreter, JupyterExtension } from './codeInterpreter'\n\nexport type { Logs, ExecutionError, Result, Execution, MIMEType, RawData, OutputMessage } from './messaging'\n\nimport { CodeInterpreter } from './codeInterpreter'\n\nexport default CodeInterpreter\n","import { ConnectionConfig, Sandbox } from 'e2b'\n\nimport { Result, Execution, OutputMessage, parseOutput, extractError } from './messaging'\n\nasync function* readLines(stream: ReadableStream<Uint8Array>) {\n  const reader = stream.getReader();\n  let buffer = ''\n\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (value !== undefined) {\n        buffer += new TextDecoder().decode(value)\n      }\n\n      if (done) {\n        if (buffer.length > 0) {\n          yield buffer\n        }\n        break\n      }\n\n      let newlineIdx = -1\n\n      do {\n        newlineIdx = buffer.indexOf('\\n')\n        if (newlineIdx !== -1) {\n          yield buffer.slice(0, newlineIdx)\n          buffer = buffer.slice(newlineIdx + 1)\n        }\n      } while (newlineIdx !== -1)\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport class JupyterExtension {\n  private static readonly execTimeoutMs = 300_000\n  private static readonly defaultKernelID = 'default'\n\n  constructor(private readonly url: string, private readonly connectionConfig: ConnectionConfig) { }\n\n  async execCell(\n    code: string,\n    opts?: {\n      kernelID?: string,\n      onStdout?: (output: OutputMessage) => (Promise<any> | any),\n      onStderr?: (output: OutputMessage) => (Promise<any> | any),\n      onResult?: (data: Result) => (Promise<any> | any),\n      timeoutMs?: number,\n      requestTimeoutMs?: number,\n    },\n  ): Promise<Execution> {\n    const controller = new AbortController()\n\n    const requestTimeout = opts?.requestTimeoutMs ?? this.connectionConfig.requestTimeoutMs\n\n    const reqTimer = requestTimeout ? setTimeout(() => {\n      controller.abort()\n    }, requestTimeout)\n      : undefined\n\n    const res = await fetch(`${this.url}/execute`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        code,\n        context_id: opts?.kernelID,\n      }),\n      keepalive: true,\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    if (!res.body) {\n      throw new Error(`Not response body: ${res.statusText} ${await res?.text()}`)\n    }\n\n    clearTimeout(reqTimer)\n\n    const bodyTimeout = opts?.timeoutMs ?? JupyterExtension.execTimeoutMs\n\n    const bodyTimer = bodyTimeout\n      ? setTimeout(() => {\n        controller.abort()\n      }, bodyTimeout)\n      : undefined\n\n    const execution = new Execution()\n\n    try {\n      for await (const chunk of readLines(res.body)) {\n        await parseOutput(execution, chunk, opts?.onStdout, opts?.onStderr, opts?.onResult)\n      }\n    } finally {\n      clearTimeout(bodyTimer)\n    }\n\n    return execution\n  }\n\n  async createKernel({\n    cwd,\n    kernelName,\n    requestTimeoutMs,\n  }: {\n    cwd?: string,\n    kernelName?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<string> {\n    const res = await fetch(`${this.url}/contexts`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        name: kernelName,\n        cwd,\n      }),\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    const data = await res.json()\n    return data.id\n  }\n\n  async restartKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n    const res = await fetch(`${this.url}/contexts/${kernelID}/restart`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async shutdownKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n\n    const res = await fetch(`${this.url}/contexts/${kernelID}`, {\n      method: 'DELETE',\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async listKernels({\n    requestTimeoutMs,\n  }: {\n    requestTimeoutMs?: number,\n  } = {}): Promise<{ kernelID: string, name: string }[]> {\n    const res = await fetch(`${this.url}/contexts`, {\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    return (await res.json()).map((kernel: any) => ({ kernelID: kernel.id, name: kernel.name }))\n  }\n}\n\nexport class CodeInterpreter extends Sandbox {\n  protected static override readonly defaultTemplate: string = 'code-interpreter-stateful'\n  protected static readonly jupyterPort = 49999\n\n  readonly notebook = new JupyterExtension(\n    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(CodeInterpreter.jupyterPort)}`,\n    this.connectionConfig,\n  )\n}\n","import { NotFoundError, SandboxError, TimeoutError } from 'e2b'\n\nexport async function extractError(res: Response) {\n  if (res.ok) {\n    return\n  }\n\n  switch (res.status) {\n    case 502:\n      return new TimeoutError(\n        `${await res.text()}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeoutMs' when starting the sandbox or calling '.setTimeout' on the sandbox with the desired timeout.`\n      )\n    case 404:\n      return new NotFoundError(await res.text())\n    default:\n      return new SandboxError(`${res.status} ${res.statusText}`)\n  }\n}\n\nexport class OutputMessage {\n  constructor(\n    public readonly line: string,\n    /**\n     * Unix epoch in nanoseconds\n     */\n    public readonly timestamp: number,\n    public readonly error: boolean,\n  ) {\n  }\n\n  public toString() {\n    return this.line\n  }\n}\n\n/**\n * Represents an error that occurred during the execution of a cell.\n * The error contains the name of the error, the value of the error, and the traceback.\n */\nexport class ExecutionError {\n  constructor(\n    /**\n     * Name of the error.\n     **/\n    public name: string,\n    /**\n     * Value of the error.\n     **/\n    public value: string,\n    /**\n     * The raw traceback of the error.\n     **/\n    public traceback: string,\n  ) { }\n}\n\n/**\n * Represents a MIME type.\n */\nexport type MIMEType = string\n\n/**\n * Dictionary that maps MIME types to their corresponding string representations of the data.\n */\nexport type RawData = {\n  [key: MIMEType]: string\n}\n\n/**\n * Represents the data to be displayed as a result of executing a cell in a Jupyter notebook.\n * The result is similar to the structure returned by ipython kernel: https://ipython.readthedocs.io/en/stable/development/execution.html#execution-semantics\n *\n *\n * The result can contain multiple types of data, such as text, images, plots, etc. Each type of data is represented\n * as a string, and the result can contain multiple types of data. The display calls don't have to have text representation,\n * for the actual result the representation is always present for the result, the other representations are always optional.\n */\nexport class Result {\n  /**\n   * Text representation of the result.\n   */\n  readonly text?: string\n  /**\n   * HTML representation of the data.\n   */\n  readonly html?: string\n  /**\n   * Markdown representation of the data.\n   */\n  readonly markdown?: string\n  /**\n   * SVG representation of the data.\n   */\n  readonly svg?: string\n  /**\n   * PNG representation of the data.\n   */\n  readonly png?: string\n  /**\n   * JPEG representation of the data.\n   */\n  readonly jpeg?: string\n  /**\n   * PDF representation of the data.\n   */\n  readonly pdf?: string\n  /**\n   * LaTeX representation of the data.\n   */\n  readonly latex?: string\n  /**\n   * JSON representation of the data.\n   */\n  readonly json?: string\n  /**\n   * JavaScript representation of the data.\n   */\n  readonly javascript?: string\n  /**\n   * Extra data that can be included. Not part of the standard types.\n   */\n  readonly extra?: any\n\n  readonly raw: RawData\n\n  constructor(rawData: RawData, public readonly isMainResult: boolean) {\n    const data = { ...rawData }\n    delete data['type']\n    delete data['is_main_result']\n\n    this.text = data['text']\n    this.html = data['html']\n    this.markdown = data['markdown']\n    this.svg = data['svg']\n    this.png = data['png']\n    this.jpeg = data['jpeg']\n    this.pdf = data['pdf']\n    this.latex = data['latex']\n    this.json = data['json']\n    this.javascript = data['javascript']\n    this.isMainResult = isMainResult\n    this.raw = data\n\n    this.extra = {}\n\n    for (const key of Object.keys(data)) {\n      if (\n        ![\n          'plain',\n          'html',\n          'markdown',\n          'svg',\n          'png',\n          'jpeg',\n          'pdf',\n          'latex',\n          'json',\n          'javascript'\n        ].includes(key)\n      ) {\n        this.extra[key] = data[key]\n      }\n    }\n  }\n\n  /**\n   * Returns all the formats available for the result.\n   *\n   * @returns Array of strings representing the formats available for the result.\n   */\n  formats(): string[] {\n    const formats = []\n    if (this.html) {\n      formats.push('html')\n    }\n    if (this.markdown) {\n      formats.push('markdown')\n    }\n    if (this.svg) {\n      formats.push('svg')\n    }\n    if (this.png) {\n      formats.push('png')\n    }\n    if (this.jpeg) {\n      formats.push('jpeg')\n    }\n    if (this.pdf) {\n      formats.push('pdf')\n    }\n    if (this.latex) {\n      formats.push('latex')\n    }\n    if (this.json) {\n      formats.push('json')\n    }\n    if (this.javascript) {\n      formats.push('javascript')\n    }\n\n    for (const key of Object.keys(this.extra)) {\n      formats.push(key)\n    }\n\n    return formats\n  }\n\n  /**\n   * Returns the serializable representation of the result.\n   */\n  toJSON() {\n    return {\n      text: this.text,\n      html: this.html,\n      markdown: this.markdown,\n      svg: this.svg,\n      png: this.png,\n      jpeg: this.jpeg,\n      pdf: this.pdf,\n      latex: this.latex,\n      json: this.json,\n      javascript: this.javascript,\n      ...(Object.keys(this.extra).length > 0 ? { extra: this.extra } : {})\n    }\n  }\n}\n\n/**\n * Data printed to stdout and stderr during execution, usually by print statements, logs, warnings, subprocesses, etc.\n */\nexport type Logs = {\n  /**\n   * List of strings printed to stdout by prints, subprocesses, etc.\n   */\n  stdout: string[]\n  /**\n   * List of strings printed to stderr by prints, subprocesses, etc.\n   */\n  stderr: string[]\n}\n\n/**\n * Represents the result of a cell execution.\n */\nexport class Execution {\n  constructor(\n    /**\n     * List of result of the cell (interactively interpreted last line), display calls (e.g. matplotlib plots).\n     */\n    public results: Result[] = [],\n    /**\n     * Logs printed to stdout and stderr during execution.\n     */\n    public logs: Logs = { stdout: [], stderr: [] },\n    /**\n     * An Error object if an error occurred, null otherwise.\n     */\n    public error?: ExecutionError,\n    /**\n     * Execution count of the cell.\n     */\n    public executionCount?: number\n  ) { }\n\n  /**\n   * Returns the text representation of the main result of the cell.\n   */\n  get text(): string | undefined {\n    for (const data of this.results) {\n      if (data.isMainResult) {\n        return data.text\n      }\n    }\n  }\n\n  /**\n   * Returns the serializable representation of the execution result.\n   */\n  toJSON() {\n    return {\n      results: this.results,\n      logs: this.logs,\n      error: this.error\n    }\n  }\n}\n\nexport async function parseOutput(\n  execution: Execution,\n  line: string,\n  onStdout?: (output: OutputMessage) => (Promise<any> | any),\n  onStderr?: (output: OutputMessage) => (Promise<any> | any),\n  onResult?: (data: Result) => (Promise<any> | any),\n) {\n  const msg = JSON.parse(line)\n\n  switch (msg.type) {\n    case 'result':\n      const result = new Result({ ...msg, type: undefined, is_main_result: undefined }, msg.is_main_result)\n      execution.results.push(result)\n      if (onResult) {\n        await onResult(result)\n      }\n      break\n    case 'stdout':\n      execution.logs.stdout.push(msg.text)\n      if (onStdout) {\n        await onStdout({\n          error: false,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'stderr':\n      execution.logs.stderr.push(msg.text)\n      if (onStderr) {\n        await onStderr({\n          error: true,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'error':\n      execution.error = new ExecutionError(msg.name, msg.value, msg.traceback)\n      break\n    case 'number_of_executions':\n      execution.executionCount = msg.execution_count\n      break\n  }\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,wBAAc,gBAAd;;;ACAA,IAAAA,cAA0C;;;ACA1C,iBAA0D;AAE1D,SAAsB,aAAa,KAAe;AAAA;AAChD,QAAI,IAAI,IAAI;AACV;AAAA,IACF;AAEA,YAAQ,IAAI,QAAQ;AAAA,MAClB,KAAK;AACH,eAAO,IAAI;AAAA,UACT,GAAG,MAAM,IAAI,KAAK,CAAC;AAAA,QACrB;AAAA,MACF,KAAK;AACH,eAAO,IAAI,yBAAc,MAAM,IAAI,KAAK,CAAC;AAAA,MAC3C;AACE,eAAO,IAAI,wBAAa,GAAG,IAAI,MAAM,IAAI,IAAI,UAAU,EAAE;AAAA,IAC7D;AAAA,EACF;AAAA;AAsBO,IAAM,iBAAN,MAAqB;AAAA,EAC1B,YAIS,MAIA,OAIA,WACP;AATO;AAIA;AAIA;AAAA,EACL;AACN;AAuBO,IAAM,SAAN,MAAa;AAAA,EAgDlB,YAAY,SAAkC,cAAuB;AAAvB;AAC5C,UAAM,OAAO,mBAAK;AAClB,WAAO,KAAK,MAAM;AAClB,WAAO,KAAK,gBAAgB;AAE5B,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,WAAW,KAAK,UAAU;AAC/B,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,QAAQ,KAAK,OAAO;AACzB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,aAAa,KAAK,YAAY;AACnC,SAAK,eAAe;AACpB,SAAK,MAAM;AAEX,SAAK,QAAQ,CAAC;AAEd,eAAW,OAAO,OAAO,KAAK,IAAI,GAAG;AACnC,UACE,CAAC;AAAA,QACC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,SAAS,GAAG,GACd;AACA,aAAK,MAAM,GAAG,IAAI,KAAK,GAAG;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAoB;AAClB,UAAM,UAAU,CAAC;AACjB,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,UAAU;AACjB,cAAQ,KAAK,UAAU;AAAA,IACzB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,OAAO;AACd,cAAQ,KAAK,OAAO;AAAA,IACtB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,YAAY;AACnB,cAAQ,KAAK,YAAY;AAAA,IAC3B;AAEA,eAAW,OAAO,OAAO,KAAK,KAAK,KAAK,GAAG;AACzC,cAAQ,KAAK,GAAG;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,MACV,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,OAAO,KAAK;AAAA,MACZ,MAAM,KAAK;AAAA,MACX,YAAY,KAAK;AAAA,OACb,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,IAAI,EAAE,OAAO,KAAK,MAAM,IAAI,CAAC;AAAA,EAEtE;AACF;AAmBO,IAAM,YAAN,MAAgB;AAAA,EACrB,YAIS,UAAoB,CAAC,GAIrB,OAAa,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,GAItC,OAIA,gBACP;AAbO;AAIA;AAIA;AAIA;AAAA,EACL;AAAA;AAAA;AAAA;AAAA,EAKJ,IAAI,OAA2B;AAC7B,eAAW,QAAQ,KAAK,SAAS;AAC/B,UAAI,KAAK,cAAc;AACrB,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,SAAS,KAAK;AAAA,MACd,MAAM,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,IACd;AAAA,EACF;AACF;AAEA,SAAsB,YACpB,WACA,MACA,UACA,UACA,UACA;AAAA;AACA,UAAM,MAAM,KAAK,MAAM,IAAI;AAE3B,YAAQ,IAAI,MAAM;AAAA,MAChB,KAAK;AACH,cAAM,SAAS,IAAI,OAAO,iCAAK,MAAL,EAAU,MAAM,QAAW,gBAAgB,OAAU,IAAG,IAAI,cAAc;AACpG,kBAAU,QAAQ,KAAK,MAAM;AAC7B,YAAI,UAAU;AACZ,gBAAM,SAAS,MAAM;AAAA,QACvB;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,QAAQ,IAAI,eAAe,IAAI,MAAM,IAAI,OAAO,IAAI,SAAS;AACvE;AAAA,MACF,KAAK;AACH,kBAAU,iBAAiB,IAAI;AAC/B;AAAA,IACJ;AAAA,EACF;AAAA;;;ADvUA,SAAgB,UAAU,QAAoC;AAAA;AAC5D,UAAM,SAAS,OAAO,UAAU;AAChC,QAAI,SAAS;AAEb,QAAI;AACF,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,kBAAM,OAAO,KAAK;AAE1C,YAAI,UAAU,QAAW;AACvB,oBAAU,IAAI,YAAY,EAAE,OAAO,KAAK;AAAA,QAC1C;AAEA,YAAI,MAAM;AACR,cAAI,OAAO,SAAS,GAAG;AACrB,kBAAM;AAAA,UACR;AACA;AAAA,QACF;AAEA,YAAI,aAAa;AAEjB,WAAG;AACD,uBAAa,OAAO,QAAQ,IAAI;AAChC,cAAI,eAAe,IAAI;AACrB,kBAAM,OAAO,MAAM,GAAG,UAAU;AAChC,qBAAS,OAAO,MAAM,aAAa,CAAC;AAAA,UACtC;AAAA,QACF,SAAS,eAAe;AAAA,MAC1B;AAAA,IACF,UAAE;AACA,aAAO,YAAY;AAAA,IACrB;AAAA,EACF;AAAA;AAEO,IAAM,oBAAN,MAAM,kBAAiB;AAAA,EAI5B,YAA6B,KAA8B,kBAAoC;AAAlE;AAA8B;AAAA,EAAsC;AAAA,EAE3F,SACJ,MACA,MAQoB;AAAA;AAtDxB;AAuDI,YAAM,aAAa,IAAI,gBAAgB;AAEvC,YAAM,kBAAiB,kCAAM,qBAAN,YAA0B,KAAK,iBAAiB;AAEvE,YAAM,WAAW,iBAAiB,WAAW,MAAM;AACjD,mBAAW,MAAM;AAAA,MACnB,GAAG,cAAc,IACb;AAEJ,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,YAAY;AAAA,QAC7C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB;AAAA,UACA,YAAY,6BAAM;AAAA,QACpB,CAAC;AAAA,QACD,WAAW;AAAA,MACb,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,UAAI,CAAC,IAAI,MAAM;AACb,cAAM,IAAI,MAAM,sBAAsB,IAAI,UAAU,IAAI,MAAM,2BAAK,MAAM,EAAE;AAAA,MAC7E;AAEA,mBAAa,QAAQ;AAErB,YAAM,eAAc,kCAAM,cAAN,YAAmB,kBAAiB;AAExD,YAAM,YAAY,cACd,WAAW,MAAM;AACjB,mBAAW,MAAM;AAAA,MACnB,GAAG,WAAW,IACZ;AAEJ,YAAM,YAAY,IAAI,UAAU;AAEhC,UAAI;AACF;AAAA,qCAA0B,UAAU,IAAI,IAAI,IAA5C,YAAAC,QAAA,uDAA+C;AAApC,kBAAM,QAAjB;AACE,kBAAM,YAAY,WAAW,OAAO,6BAAM,UAAU,6BAAM,UAAU,6BAAM,QAAQ;AAAA,UACpF;AAAA,iBAFA,MAlGN;AAkGM,UAAAA,SAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAAAA;AAAA,oBAAAA,OAAA;AAAA;AAAA;AAAA,MAGF,UAAE;AACA,qBAAa,SAAS;AAAA,MACxB;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,EAEM,eAQmB;AAAA,+CARN;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAII,CAAC,GAAoB;AACvB,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAAA,QACD,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,YAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,aAAO,KAAK;AAAA,IACd;AAAA;AAAA,EAEM,gBAMiB;AAAA,+CANH;AAAA,MAClB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AACxC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,YAAY;AAAA,QAClE,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,iBAMiB;AAAA,+CANF;AAAA,MACnB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AAExC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,IAAI;AAAA,QAC1D,QAAQ;AAAA,QACR,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,cAIiD;AAAA,+CAJrC;AAAA,MAChB;AAAA,IACF,IAEI,CAAC,GAAkD;AACrD,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,cAAQ,MAAM,IAAI,KAAK,GAAG,IAAI,CAAC,YAAiB,EAAE,UAAU,OAAO,IAAI,MAAM,OAAO,KAAK,EAAE;AAAA,IAC7F;AAAA;AACF;AAlKa,kBACa,gBAAgB;AAD7B,kBAEa,kBAAkB;AAFrC,IAAM,mBAAN;AAoKA,IAAM,mBAAN,MAAM,yBAAwB,oBAAQ;AAAA,EAAtC;AAAA;AAIL,SAAS,WAAW,IAAI;AAAA,MACtB,GAAG,KAAK,iBAAiB,QAAQ,SAAS,OAAO,MAAM,KAAK,QAAQ,iBAAgB,WAAW,CAAC;AAAA,MAChG,KAAK;AAAA,IACP;AAAA;AACF;AARa,iBACwB,kBAA0B;AADlD,iBAEe,cAAc;AAFnC,IAAM,kBAAN;;;ADlMP,IAAO,cAAQ;","names":["import_e2b","error"]}

package/dist/index.mjs

@@ -55,6 +55,9 @@ var __asyncGenerator = (__this, __arguments, generator) => {
 };
 var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")]) ? it.call(obj) : (obj = obj[__knownSymbol("iterator")](), it = {}, method = (key, fn) => (fn = obj[key]) && (it[key] = (arg) => new Promise((yes, no, done) => (arg = fn.call(obj, arg), done = arg.done, Promise.resolve(arg.value).then((value) => yes({ value, done }), no)))), method("next"), method("return"), it);
 
+// src/index.ts
+export * from "e2b";
+
 // src/codeInterpreter.ts
 import { Sandbox } from "e2b";
 
@@ -427,12 +430,11 @@ var _CodeInterpreter = class _CodeInterpreter extends Sandbox {
     );
   }
 };
-_CodeInterpreter.defaultTemplate = "ci-no-ws";
+_CodeInterpreter.defaultTemplate = "code-interpreter-stateful";
 _CodeInterpreter.jupyterPort = 49999;
 var CodeInterpreter = _CodeInterpreter;
 
 // src/index.ts
-export * from "e2b";
 var src_default = CodeInterpreter;
 export {
   CodeInterpreter,

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/codeInterpreter.ts","../src/messaging.ts","../src/index.ts"],"sourcesContent":["import { ConnectionConfig, Sandbox } from 'e2b'\n\nimport { Result, Execution, OutputMessage, parseOutput, extractError } from './messaging'\n\nasync function* readLines(stream: ReadableStream<Uint8Array>) {\n  const reader = stream.getReader();\n  let buffer = ''\n\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (value !== undefined) {\n        buffer += new TextDecoder().decode(value)\n      }\n\n      if (done) {\n        if (buffer.length > 0) {\n          yield buffer\n        }\n        break\n      }\n\n      let newlineIdx = -1\n\n      do {\n        newlineIdx = buffer.indexOf('\\n')\n        if (newlineIdx !== -1) {\n          yield buffer.slice(0, newlineIdx)\n          buffer = buffer.slice(newlineIdx + 1)\n        }\n      } while (newlineIdx !== -1)\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport class JupyterExtension {\n  private static readonly execTimeoutMs = 300_000\n  private static readonly defaultKernelID = 'default'\n\n  constructor(private readonly url: string, private readonly connectionConfig: ConnectionConfig) { }\n\n  async execCell(\n    code: string,\n    opts?: {\n      kernelID?: string,\n      onStdout?: (output: OutputMessage) => (Promise<any> | any),\n      onStderr?: (output: OutputMessage) => (Promise<any> | any),\n      onResult?: (data: Result) => (Promise<any> | any),\n      timeoutMs?: number,\n      requestTimeoutMs?: number,\n    },\n  ): Promise<Execution> {\n    const controller = new AbortController()\n\n    const requestTimeout = opts?.requestTimeoutMs ?? this.connectionConfig.requestTimeoutMs\n\n    const reqTimer = requestTimeout ? setTimeout(() => {\n      controller.abort()\n    }, requestTimeout)\n      : undefined\n\n    const res = await fetch(`${this.url}/execute`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        code,\n        context_id: opts?.kernelID,\n      }),\n      keepalive: true,\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    if (!res.body) {\n      throw new Error(`Not response body: ${res.statusText} ${await res?.text()}`)\n    }\n\n    clearTimeout(reqTimer)\n\n    const bodyTimeout = opts?.timeoutMs ?? JupyterExtension.execTimeoutMs\n\n    const bodyTimer = bodyTimeout\n      ? setTimeout(() => {\n        controller.abort()\n      }, bodyTimeout)\n      : undefined\n\n    const execution = new Execution()\n\n    try {\n      for await (const chunk of readLines(res.body)) {\n        await parseOutput(execution, chunk, opts?.onStdout, opts?.onStderr, opts?.onResult)\n      }\n    } finally {\n      clearTimeout(bodyTimer)\n    }\n\n    return execution\n  }\n\n  async createKernel({\n    cwd,\n    kernelName,\n    requestTimeoutMs,\n  }: {\n    cwd?: string,\n    kernelName?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<string> {\n    const res = await fetch(`${this.url}/contexts`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        name: kernelName,\n        cwd,\n      }),\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    const data = await res.json()\n    return data.id\n  }\n\n  async restartKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n    const res = await fetch(`${this.url}/contexts/${kernelID}/restart`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async shutdownKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n\n    const res = await fetch(`${this.url}/contexts/${kernelID}`, {\n      method: 'DELETE',\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async listKernels({\n    requestTimeoutMs,\n  }: {\n    requestTimeoutMs?: number,\n  } = {}): Promise<{ kernelID: string, name: string }[]> {\n    const res = await fetch(`${this.url}/contexts`, {\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    return (await res.json()).map((kernel: any) => ({ kernelID: kernel.id, name: kernel.name }))\n  }\n}\n\nexport class CodeInterpreter extends Sandbox {\n  protected static override readonly defaultTemplate: string = 'ci-no-ws'\n  protected static readonly jupyterPort = 49999\n\n  readonly notebook = new JupyterExtension(\n    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(CodeInterpreter.jupyterPort)}`,\n    this.connectionConfig,\n  )\n}\n","import { NotFoundError, SandboxError, TimeoutError } from 'e2b'\n\nexport async function extractError(res: Response) {\n  if (res.ok) {\n    return\n  }\n\n  switch (res.status) {\n    case 502:\n      return new TimeoutError(\n        `${await res.text()}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeoutMs' when starting the sandbox or calling '.setTimeout' on the sandbox with the desired timeout.`\n      )\n    case 404:\n      return new NotFoundError(await res.text())\n    default:\n      return new SandboxError(`${res.status} ${res.statusText}`)\n  }\n}\n\nexport class OutputMessage {\n  constructor(\n    public readonly line: string,\n    /**\n     * Unix epoch in nanoseconds\n     */\n    public readonly timestamp: number,\n    public readonly error: boolean,\n  ) {\n  }\n\n  public toString() {\n    return this.line\n  }\n}\n\n/**\n * Represents an error that occurred during the execution of a cell.\n * The error contains the name of the error, the value of the error, and the traceback.\n */\nexport class ExecutionError {\n  constructor(\n    /**\n     * Name of the error.\n     **/\n    public name: string,\n    /**\n     * Value of the error.\n     **/\n    public value: string,\n    /**\n     * The raw traceback of the error.\n     **/\n    public traceback: string,\n  ) { }\n}\n\n/**\n * Represents a MIME type.\n */\nexport type MIMEType = string\n\n/**\n * Dictionary that maps MIME types to their corresponding string representations of the data.\n */\nexport type RawData = {\n  [key: MIMEType]: string\n}\n\n/**\n * Represents the data to be displayed as a result of executing a cell in a Jupyter notebook.\n * The result is similar to the structure returned by ipython kernel: https://ipython.readthedocs.io/en/stable/development/execution.html#execution-semantics\n *\n *\n * The result can contain multiple types of data, such as text, images, plots, etc. Each type of data is represented\n * as a string, and the result can contain multiple types of data. The display calls don't have to have text representation,\n * for the actual result the representation is always present for the result, the other representations are always optional.\n */\nexport class Result {\n  /**\n   * Text representation of the result.\n   */\n  readonly text?: string\n  /**\n   * HTML representation of the data.\n   */\n  readonly html?: string\n  /**\n   * Markdown representation of the data.\n   */\n  readonly markdown?: string\n  /**\n   * SVG representation of the data.\n   */\n  readonly svg?: string\n  /**\n   * PNG representation of the data.\n   */\n  readonly png?: string\n  /**\n   * JPEG representation of the data.\n   */\n  readonly jpeg?: string\n  /**\n   * PDF representation of the data.\n   */\n  readonly pdf?: string\n  /**\n   * LaTeX representation of the data.\n   */\n  readonly latex?: string\n  /**\n   * JSON representation of the data.\n   */\n  readonly json?: string\n  /**\n   * JavaScript representation of the data.\n   */\n  readonly javascript?: string\n  /**\n   * Extra data that can be included. Not part of the standard types.\n   */\n  readonly extra?: any\n\n  readonly raw: RawData\n\n  constructor(rawData: RawData, public readonly isMainResult: boolean) {\n    const data = { ...rawData }\n    delete data['type']\n    delete data['is_main_result']\n\n    this.text = data['text']\n    this.html = data['html']\n    this.markdown = data['markdown']\n    this.svg = data['svg']\n    this.png = data['png']\n    this.jpeg = data['jpeg']\n    this.pdf = data['pdf']\n    this.latex = data['latex']\n    this.json = data['json']\n    this.javascript = data['javascript']\n    this.isMainResult = isMainResult\n    this.raw = data\n\n    this.extra = {}\n\n    for (const key of Object.keys(data)) {\n      if (\n        ![\n          'plain',\n          'html',\n          'markdown',\n          'svg',\n          'png',\n          'jpeg',\n          'pdf',\n          'latex',\n          'json',\n          'javascript'\n        ].includes(key)\n      ) {\n        this.extra[key] = data[key]\n      }\n    }\n  }\n\n  /**\n   * Returns all the formats available for the result.\n   *\n   * @returns Array of strings representing the formats available for the result.\n   */\n  formats(): string[] {\n    const formats = []\n    if (this.html) {\n      formats.push('html')\n    }\n    if (this.markdown) {\n      formats.push('markdown')\n    }\n    if (this.svg) {\n      formats.push('svg')\n    }\n    if (this.png) {\n      formats.push('png')\n    }\n    if (this.jpeg) {\n      formats.push('jpeg')\n    }\n    if (this.pdf) {\n      formats.push('pdf')\n    }\n    if (this.latex) {\n      formats.push('latex')\n    }\n    if (this.json) {\n      formats.push('json')\n    }\n    if (this.javascript) {\n      formats.push('javascript')\n    }\n\n    for (const key of Object.keys(this.extra)) {\n      formats.push(key)\n    }\n\n    return formats\n  }\n\n  /**\n   * Returns the serializable representation of the result.\n   */\n  toJSON() {\n    return {\n      text: this.text,\n      html: this.html,\n      markdown: this.markdown,\n      svg: this.svg,\n      png: this.png,\n      jpeg: this.jpeg,\n      pdf: this.pdf,\n      latex: this.latex,\n      json: this.json,\n      javascript: this.javascript,\n      ...(Object.keys(this.extra).length > 0 ? { extra: this.extra } : {})\n    }\n  }\n}\n\n/**\n * Data printed to stdout and stderr during execution, usually by print statements, logs, warnings, subprocesses, etc.\n */\nexport type Logs = {\n  /**\n   * List of strings printed to stdout by prints, subprocesses, etc.\n   */\n  stdout: string[]\n  /**\n   * List of strings printed to stderr by prints, subprocesses, etc.\n   */\n  stderr: string[]\n}\n\n/**\n * Represents the result of a cell execution.\n */\nexport class Execution {\n  constructor(\n    /**\n     * List of result of the cell (interactively interpreted last line), display calls (e.g. matplotlib plots).\n     */\n    public results: Result[] = [],\n    /**\n     * Logs printed to stdout and stderr during execution.\n     */\n    public logs: Logs = { stdout: [], stderr: [] },\n    /**\n     * An Error object if an error occurred, null otherwise.\n     */\n    public error?: ExecutionError,\n    /**\n     * Execution count of the cell.\n     */\n    public executionCount?: number\n  ) { }\n\n  /**\n   * Returns the text representation of the main result of the cell.\n   */\n  get text(): string | undefined {\n    for (const data of this.results) {\n      if (data.isMainResult) {\n        return data.text\n      }\n    }\n  }\n\n  /**\n   * Returns the serializable representation of the execution result.\n   */\n  toJSON() {\n    return {\n      results: this.results,\n      logs: this.logs,\n      error: this.error\n    }\n  }\n}\n\nexport async function parseOutput(\n  execution: Execution,\n  line: string,\n  onStdout?: (output: OutputMessage) => (Promise<any> | any),\n  onStderr?: (output: OutputMessage) => (Promise<any> | any),\n  onResult?: (data: Result) => (Promise<any> | any),\n) {\n  const msg = JSON.parse(line)\n\n  switch (msg.type) {\n    case 'result':\n      const result = new Result({ ...msg, type: undefined, is_main_result: undefined }, msg.is_main_result)\n      execution.results.push(result)\n      if (onResult) {\n        await onResult(result)\n      }\n      break\n    case 'stdout':\n      execution.logs.stdout.push(msg.text)\n      if (onStdout) {\n        await onStdout({\n          error: false,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'stderr':\n      execution.logs.stderr.push(msg.text)\n      if (onStderr) {\n        await onStderr({\n          error: true,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'error':\n      execution.error = new ExecutionError(msg.name, msg.value, msg.traceback)\n      break\n    case 'number_of_executions':\n      execution.executionCount = msg.execution_count\n      break\n  }\n}","export { CodeInterpreter, JupyterExtension } from './codeInterpreter'\n\nexport type { Logs, ExecutionError, Result, Execution, MIMEType, RawData, OutputMessage } from './messaging'\n\nimport { CodeInterpreter } from './codeInterpreter'\n\nexport * from 'e2b'\n\nexport default CodeInterpreter\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,SAA2B,eAAe;;;ACA1C,SAAS,eAAe,cAAc,oBAAoB;AAE1D,SAAsB,aAAa,KAAe;AAAA;AAChD,QAAI,IAAI,IAAI;AACV;AAAA,IACF;AAEA,YAAQ,IAAI,QAAQ;AAAA,MAClB,KAAK;AACH,eAAO,IAAI;AAAA,UACT,GAAG,MAAM,IAAI,KAAK,CAAC;AAAA,QACrB;AAAA,MACF,KAAK;AACH,eAAO,IAAI,cAAc,MAAM,IAAI,KAAK,CAAC;AAAA,MAC3C;AACE,eAAO,IAAI,aAAa,GAAG,IAAI,MAAM,IAAI,IAAI,UAAU,EAAE;AAAA,IAC7D;AAAA,EACF;AAAA;AAsBO,IAAM,iBAAN,MAAqB;AAAA,EAC1B,YAIS,MAIA,OAIA,WACP;AATO;AAIA;AAIA;AAAA,EACL;AACN;AAuBO,IAAM,SAAN,MAAa;AAAA,EAgDlB,YAAY,SAAkC,cAAuB;AAAvB;AAC5C,UAAM,OAAO,mBAAK;AAClB,WAAO,KAAK,MAAM;AAClB,WAAO,KAAK,gBAAgB;AAE5B,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,WAAW,KAAK,UAAU;AAC/B,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,QAAQ,KAAK,OAAO;AACzB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,aAAa,KAAK,YAAY;AACnC,SAAK,eAAe;AACpB,SAAK,MAAM;AAEX,SAAK,QAAQ,CAAC;AAEd,eAAW,OAAO,OAAO,KAAK,IAAI,GAAG;AACnC,UACE,CAAC;AAAA,QACC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,SAAS,GAAG,GACd;AACA,aAAK,MAAM,GAAG,IAAI,KAAK,GAAG;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAoB;AAClB,UAAM,UAAU,CAAC;AACjB,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,UAAU;AACjB,cAAQ,KAAK,UAAU;AAAA,IACzB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,OAAO;AACd,cAAQ,KAAK,OAAO;AAAA,IACtB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,YAAY;AACnB,cAAQ,KAAK,YAAY;AAAA,IAC3B;AAEA,eAAW,OAAO,OAAO,KAAK,KAAK,KAAK,GAAG;AACzC,cAAQ,KAAK,GAAG;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,MACV,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,OAAO,KAAK;AAAA,MACZ,MAAM,KAAK;AAAA,MACX,YAAY,KAAK;AAAA,OACb,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,IAAI,EAAE,OAAO,KAAK,MAAM,IAAI,CAAC;AAAA,EAEtE;AACF;AAmBO,IAAM,YAAN,MAAgB;AAAA,EACrB,YAIS,UAAoB,CAAC,GAIrB,OAAa,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,GAItC,OAIA,gBACP;AAbO;AAIA;AAIA;AAIA;AAAA,EACL;AAAA;AAAA;AAAA;AAAA,EAKJ,IAAI,OAA2B;AAC7B,eAAW,QAAQ,KAAK,SAAS;AAC/B,UAAI,KAAK,cAAc;AACrB,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,SAAS,KAAK;AAAA,MACd,MAAM,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,IACd;AAAA,EACF;AACF;AAEA,SAAsB,YACpB,WACA,MACA,UACA,UACA,UACA;AAAA;AACA,UAAM,MAAM,KAAK,MAAM,IAAI;AAE3B,YAAQ,IAAI,MAAM;AAAA,MAChB,KAAK;AACH,cAAM,SAAS,IAAI,OAAO,iCAAK,MAAL,EAAU,MAAM,QAAW,gBAAgB,OAAU,IAAG,IAAI,cAAc;AACpG,kBAAU,QAAQ,KAAK,MAAM;AAC7B,YAAI,UAAU;AACZ,gBAAM,SAAS,MAAM;AAAA,QACvB;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,QAAQ,IAAI,eAAe,IAAI,MAAM,IAAI,OAAO,IAAI,SAAS;AACvE;AAAA,MACF,KAAK;AACH,kBAAU,iBAAiB,IAAI;AAC/B;AAAA,IACJ;AAAA,EACF;AAAA;;;ADvUA,SAAgB,UAAU,QAAoC;AAAA;AAC5D,UAAM,SAAS,OAAO,UAAU;AAChC,QAAI,SAAS;AAEb,QAAI;AACF,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,kBAAM,OAAO,KAAK;AAE1C,YAAI,UAAU,QAAW;AACvB,oBAAU,IAAI,YAAY,EAAE,OAAO,KAAK;AAAA,QAC1C;AAEA,YAAI,MAAM;AACR,cAAI,OAAO,SAAS,GAAG;AACrB,kBAAM;AAAA,UACR;AACA;AAAA,QACF;AAEA,YAAI,aAAa;AAEjB,WAAG;AACD,uBAAa,OAAO,QAAQ,IAAI;AAChC,cAAI,eAAe,IAAI;AACrB,kBAAM,OAAO,MAAM,GAAG,UAAU;AAChC,qBAAS,OAAO,MAAM,aAAa,CAAC;AAAA,UACtC;AAAA,QACF,SAAS,eAAe;AAAA,MAC1B;AAAA,IACF,UAAE;AACA,aAAO,YAAY;AAAA,IACrB;AAAA,EACF;AAAA;AAEO,IAAM,oBAAN,MAAM,kBAAiB;AAAA,EAI5B,YAA6B,KAA8B,kBAAoC;AAAlE;AAA8B;AAAA,EAAsC;AAAA,EAE3F,SACJ,MACA,MAQoB;AAAA;AAtDxB;AAuDI,YAAM,aAAa,IAAI,gBAAgB;AAEvC,YAAM,kBAAiB,kCAAM,qBAAN,YAA0B,KAAK,iBAAiB;AAEvE,YAAM,WAAW,iBAAiB,WAAW,MAAM;AACjD,mBAAW,MAAM;AAAA,MACnB,GAAG,cAAc,IACb;AAEJ,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,YAAY;AAAA,QAC7C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB;AAAA,UACA,YAAY,6BAAM;AAAA,QACpB,CAAC;AAAA,QACD,WAAW;AAAA,MACb,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,UAAI,CAAC,IAAI,MAAM;AACb,cAAM,IAAI,MAAM,sBAAsB,IAAI,UAAU,IAAI,MAAM,2BAAK,MAAM,EAAE;AAAA,MAC7E;AAEA,mBAAa,QAAQ;AAErB,YAAM,eAAc,kCAAM,cAAN,YAAmB,kBAAiB;AAExD,YAAM,YAAY,cACd,WAAW,MAAM;AACjB,mBAAW,MAAM;AAAA,MACnB,GAAG,WAAW,IACZ;AAEJ,YAAM,YAAY,IAAI,UAAU;AAEhC,UAAI;AACF;AAAA,qCAA0B,UAAU,IAAI,IAAI,IAA5C,YAAAA,QAAA,uDAA+C;AAApC,kBAAM,QAAjB;AACE,kBAAM,YAAY,WAAW,OAAO,6BAAM,UAAU,6BAAM,UAAU,6BAAM,QAAQ;AAAA,UACpF;AAAA,iBAFA,MAlGN;AAkGM,UAAAA,SAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAAAA;AAAA,oBAAAA,OAAA;AAAA;AAAA;AAAA,MAGF,UAAE;AACA,qBAAa,SAAS;AAAA,MACxB;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,EAEM,eAQmB;AAAA,+CARN;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAII,CAAC,GAAoB;AACvB,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAAA,QACD,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,YAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,aAAO,KAAK;AAAA,IACd;AAAA;AAAA,EAEM,gBAMiB;AAAA,+CANH;AAAA,MAClB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AACxC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,YAAY;AAAA,QAClE,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,iBAMiB;AAAA,+CANF;AAAA,MACnB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AAExC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,IAAI;AAAA,QAC1D,QAAQ;AAAA,QACR,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,cAIiD;AAAA,+CAJrC;AAAA,MAChB;AAAA,IACF,IAEI,CAAC,GAAkD;AACrD,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,cAAQ,MAAM,IAAI,KAAK,GAAG,IAAI,CAAC,YAAiB,EAAE,UAAU,OAAO,IAAI,MAAM,OAAO,KAAK,EAAE;AAAA,IAC7F;AAAA;AACF;AAlKa,kBACa,gBAAgB;AAD7B,kBAEa,kBAAkB;AAFrC,IAAM,mBAAN;AAoKA,IAAM,mBAAN,MAAM,yBAAwB,QAAQ;AAAA,EAAtC;AAAA;AAIL,SAAS,WAAW,IAAI;AAAA,MACtB,GAAG,KAAK,iBAAiB,QAAQ,SAAS,OAAO,MAAM,KAAK,QAAQ,iBAAgB,WAAW,CAAC;AAAA,MAChG,KAAK;AAAA,IACP;AAAA;AACF;AARa,iBACwB,kBAA0B;AADlD,iBAEe,cAAc;AAFnC,IAAM,kBAAN;;;AEpMP,cAAc;AAEd,IAAO,cAAQ;","names":["error"]}
+{"version":3,"sources":["../src/index.ts","../src/codeInterpreter.ts","../src/messaging.ts"],"sourcesContent":["export * from 'e2b'\n\nexport { CodeInterpreter, JupyterExtension } from './codeInterpreter'\n\nexport type { Logs, ExecutionError, Result, Execution, MIMEType, RawData, OutputMessage } from './messaging'\n\nimport { CodeInterpreter } from './codeInterpreter'\n\nexport default CodeInterpreter\n","import { ConnectionConfig, Sandbox } from 'e2b'\n\nimport { Result, Execution, OutputMessage, parseOutput, extractError } from './messaging'\n\nasync function* readLines(stream: ReadableStream<Uint8Array>) {\n  const reader = stream.getReader();\n  let buffer = ''\n\n  try {\n    while (true) {\n      const { done, value } = await reader.read();\n\n      if (value !== undefined) {\n        buffer += new TextDecoder().decode(value)\n      }\n\n      if (done) {\n        if (buffer.length > 0) {\n          yield buffer\n        }\n        break\n      }\n\n      let newlineIdx = -1\n\n      do {\n        newlineIdx = buffer.indexOf('\\n')\n        if (newlineIdx !== -1) {\n          yield buffer.slice(0, newlineIdx)\n          buffer = buffer.slice(newlineIdx + 1)\n        }\n      } while (newlineIdx !== -1)\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\nexport class JupyterExtension {\n  private static readonly execTimeoutMs = 300_000\n  private static readonly defaultKernelID = 'default'\n\n  constructor(private readonly url: string, private readonly connectionConfig: ConnectionConfig) { }\n\n  async execCell(\n    code: string,\n    opts?: {\n      kernelID?: string,\n      onStdout?: (output: OutputMessage) => (Promise<any> | any),\n      onStderr?: (output: OutputMessage) => (Promise<any> | any),\n      onResult?: (data: Result) => (Promise<any> | any),\n      timeoutMs?: number,\n      requestTimeoutMs?: number,\n    },\n  ): Promise<Execution> {\n    const controller = new AbortController()\n\n    const requestTimeout = opts?.requestTimeoutMs ?? this.connectionConfig.requestTimeoutMs\n\n    const reqTimer = requestTimeout ? setTimeout(() => {\n      controller.abort()\n    }, requestTimeout)\n      : undefined\n\n    const res = await fetch(`${this.url}/execute`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        code,\n        context_id: opts?.kernelID,\n      }),\n      keepalive: true,\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    if (!res.body) {\n      throw new Error(`Not response body: ${res.statusText} ${await res?.text()}`)\n    }\n\n    clearTimeout(reqTimer)\n\n    const bodyTimeout = opts?.timeoutMs ?? JupyterExtension.execTimeoutMs\n\n    const bodyTimer = bodyTimeout\n      ? setTimeout(() => {\n        controller.abort()\n      }, bodyTimeout)\n      : undefined\n\n    const execution = new Execution()\n\n    try {\n      for await (const chunk of readLines(res.body)) {\n        await parseOutput(execution, chunk, opts?.onStdout, opts?.onStderr, opts?.onResult)\n      }\n    } finally {\n      clearTimeout(bodyTimer)\n    }\n\n    return execution\n  }\n\n  async createKernel({\n    cwd,\n    kernelName,\n    requestTimeoutMs,\n  }: {\n    cwd?: string,\n    kernelName?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<string> {\n    const res = await fetch(`${this.url}/contexts`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      body: JSON.stringify({\n        name: kernelName,\n        cwd,\n      }),\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    const data = await res.json()\n    return data.id\n  }\n\n  async restartKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n    const res = await fetch(`${this.url}/contexts/${kernelID}/restart`, {\n      method: 'POST',\n      headers: {\n        'Content-Type': 'application/json',\n      },\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async shutdownKernel({\n    kernelID,\n    requestTimeoutMs,\n  }: {\n    kernelID?: string,\n    requestTimeoutMs?: number,\n  } = {}): Promise<void> {\n    kernelID = kernelID || JupyterExtension.defaultKernelID\n\n    const res = await fetch(`${this.url}/contexts/${kernelID}`, {\n      method: 'DELETE',\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n  }\n\n  async listKernels({\n    requestTimeoutMs,\n  }: {\n    requestTimeoutMs?: number,\n  } = {}): Promise<{ kernelID: string, name: string }[]> {\n    const res = await fetch(`${this.url}/contexts`, {\n      keepalive: true,\n      signal: this.connectionConfig.getSignal(requestTimeoutMs),\n    })\n\n    const error = await extractError(res)\n    if (error) {\n      throw error\n    }\n\n    return (await res.json()).map((kernel: any) => ({ kernelID: kernel.id, name: kernel.name }))\n  }\n}\n\nexport class CodeInterpreter extends Sandbox {\n  protected static override readonly defaultTemplate: string = 'code-interpreter-stateful'\n  protected static readonly jupyterPort = 49999\n\n  readonly notebook = new JupyterExtension(\n    `${this.connectionConfig.debug ? 'http' : 'https'}://${this.getHost(CodeInterpreter.jupyterPort)}`,\n    this.connectionConfig,\n  )\n}\n","import { NotFoundError, SandboxError, TimeoutError } from 'e2b'\n\nexport async function extractError(res: Response) {\n  if (res.ok) {\n    return\n  }\n\n  switch (res.status) {\n    case 502:\n      return new TimeoutError(\n        `${await res.text()}: This error is likely due to sandbox timeout. You can modify the sandbox timeout by passing 'timeoutMs' when starting the sandbox or calling '.setTimeout' on the sandbox with the desired timeout.`\n      )\n    case 404:\n      return new NotFoundError(await res.text())\n    default:\n      return new SandboxError(`${res.status} ${res.statusText}`)\n  }\n}\n\nexport class OutputMessage {\n  constructor(\n    public readonly line: string,\n    /**\n     * Unix epoch in nanoseconds\n     */\n    public readonly timestamp: number,\n    public readonly error: boolean,\n  ) {\n  }\n\n  public toString() {\n    return this.line\n  }\n}\n\n/**\n * Represents an error that occurred during the execution of a cell.\n * The error contains the name of the error, the value of the error, and the traceback.\n */\nexport class ExecutionError {\n  constructor(\n    /**\n     * Name of the error.\n     **/\n    public name: string,\n    /**\n     * Value of the error.\n     **/\n    public value: string,\n    /**\n     * The raw traceback of the error.\n     **/\n    public traceback: string,\n  ) { }\n}\n\n/**\n * Represents a MIME type.\n */\nexport type MIMEType = string\n\n/**\n * Dictionary that maps MIME types to their corresponding string representations of the data.\n */\nexport type RawData = {\n  [key: MIMEType]: string\n}\n\n/**\n * Represents the data to be displayed as a result of executing a cell in a Jupyter notebook.\n * The result is similar to the structure returned by ipython kernel: https://ipython.readthedocs.io/en/stable/development/execution.html#execution-semantics\n *\n *\n * The result can contain multiple types of data, such as text, images, plots, etc. Each type of data is represented\n * as a string, and the result can contain multiple types of data. The display calls don't have to have text representation,\n * for the actual result the representation is always present for the result, the other representations are always optional.\n */\nexport class Result {\n  /**\n   * Text representation of the result.\n   */\n  readonly text?: string\n  /**\n   * HTML representation of the data.\n   */\n  readonly html?: string\n  /**\n   * Markdown representation of the data.\n   */\n  readonly markdown?: string\n  /**\n   * SVG representation of the data.\n   */\n  readonly svg?: string\n  /**\n   * PNG representation of the data.\n   */\n  readonly png?: string\n  /**\n   * JPEG representation of the data.\n   */\n  readonly jpeg?: string\n  /**\n   * PDF representation of the data.\n   */\n  readonly pdf?: string\n  /**\n   * LaTeX representation of the data.\n   */\n  readonly latex?: string\n  /**\n   * JSON representation of the data.\n   */\n  readonly json?: string\n  /**\n   * JavaScript representation of the data.\n   */\n  readonly javascript?: string\n  /**\n   * Extra data that can be included. Not part of the standard types.\n   */\n  readonly extra?: any\n\n  readonly raw: RawData\n\n  constructor(rawData: RawData, public readonly isMainResult: boolean) {\n    const data = { ...rawData }\n    delete data['type']\n    delete data['is_main_result']\n\n    this.text = data['text']\n    this.html = data['html']\n    this.markdown = data['markdown']\n    this.svg = data['svg']\n    this.png = data['png']\n    this.jpeg = data['jpeg']\n    this.pdf = data['pdf']\n    this.latex = data['latex']\n    this.json = data['json']\n    this.javascript = data['javascript']\n    this.isMainResult = isMainResult\n    this.raw = data\n\n    this.extra = {}\n\n    for (const key of Object.keys(data)) {\n      if (\n        ![\n          'plain',\n          'html',\n          'markdown',\n          'svg',\n          'png',\n          'jpeg',\n          'pdf',\n          'latex',\n          'json',\n          'javascript'\n        ].includes(key)\n      ) {\n        this.extra[key] = data[key]\n      }\n    }\n  }\n\n  /**\n   * Returns all the formats available for the result.\n   *\n   * @returns Array of strings representing the formats available for the result.\n   */\n  formats(): string[] {\n    const formats = []\n    if (this.html) {\n      formats.push('html')\n    }\n    if (this.markdown) {\n      formats.push('markdown')\n    }\n    if (this.svg) {\n      formats.push('svg')\n    }\n    if (this.png) {\n      formats.push('png')\n    }\n    if (this.jpeg) {\n      formats.push('jpeg')\n    }\n    if (this.pdf) {\n      formats.push('pdf')\n    }\n    if (this.latex) {\n      formats.push('latex')\n    }\n    if (this.json) {\n      formats.push('json')\n    }\n    if (this.javascript) {\n      formats.push('javascript')\n    }\n\n    for (const key of Object.keys(this.extra)) {\n      formats.push(key)\n    }\n\n    return formats\n  }\n\n  /**\n   * Returns the serializable representation of the result.\n   */\n  toJSON() {\n    return {\n      text: this.text,\n      html: this.html,\n      markdown: this.markdown,\n      svg: this.svg,\n      png: this.png,\n      jpeg: this.jpeg,\n      pdf: this.pdf,\n      latex: this.latex,\n      json: this.json,\n      javascript: this.javascript,\n      ...(Object.keys(this.extra).length > 0 ? { extra: this.extra } : {})\n    }\n  }\n}\n\n/**\n * Data printed to stdout and stderr during execution, usually by print statements, logs, warnings, subprocesses, etc.\n */\nexport type Logs = {\n  /**\n   * List of strings printed to stdout by prints, subprocesses, etc.\n   */\n  stdout: string[]\n  /**\n   * List of strings printed to stderr by prints, subprocesses, etc.\n   */\n  stderr: string[]\n}\n\n/**\n * Represents the result of a cell execution.\n */\nexport class Execution {\n  constructor(\n    /**\n     * List of result of the cell (interactively interpreted last line), display calls (e.g. matplotlib plots).\n     */\n    public results: Result[] = [],\n    /**\n     * Logs printed to stdout and stderr during execution.\n     */\n    public logs: Logs = { stdout: [], stderr: [] },\n    /**\n     * An Error object if an error occurred, null otherwise.\n     */\n    public error?: ExecutionError,\n    /**\n     * Execution count of the cell.\n     */\n    public executionCount?: number\n  ) { }\n\n  /**\n   * Returns the text representation of the main result of the cell.\n   */\n  get text(): string | undefined {\n    for (const data of this.results) {\n      if (data.isMainResult) {\n        return data.text\n      }\n    }\n  }\n\n  /**\n   * Returns the serializable representation of the execution result.\n   */\n  toJSON() {\n    return {\n      results: this.results,\n      logs: this.logs,\n      error: this.error\n    }\n  }\n}\n\nexport async function parseOutput(\n  execution: Execution,\n  line: string,\n  onStdout?: (output: OutputMessage) => (Promise<any> | any),\n  onStderr?: (output: OutputMessage) => (Promise<any> | any),\n  onResult?: (data: Result) => (Promise<any> | any),\n) {\n  const msg = JSON.parse(line)\n\n  switch (msg.type) {\n    case 'result':\n      const result = new Result({ ...msg, type: undefined, is_main_result: undefined }, msg.is_main_result)\n      execution.results.push(result)\n      if (onResult) {\n        await onResult(result)\n      }\n      break\n    case 'stdout':\n      execution.logs.stdout.push(msg.text)\n      if (onStdout) {\n        await onStdout({\n          error: false,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'stderr':\n      execution.logs.stderr.push(msg.text)\n      if (onStderr) {\n        await onStderr({\n          error: true,\n          line: msg.text,\n          timestamp: new Date().getTime() * 1000,\n        })\n      }\n      break\n    case 'error':\n      execution.error = new ExecutionError(msg.name, msg.value, msg.traceback)\n      break\n    case 'number_of_executions':\n      execution.executionCount = msg.execution_count\n      break\n  }\n}"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,cAAc;;;ACAd,SAA2B,eAAe;;;ACA1C,SAAS,eAAe,cAAc,oBAAoB;AAE1D,SAAsB,aAAa,KAAe;AAAA;AAChD,QAAI,IAAI,IAAI;AACV;AAAA,IACF;AAEA,YAAQ,IAAI,QAAQ;AAAA,MAClB,KAAK;AACH,eAAO,IAAI;AAAA,UACT,GAAG,MAAM,IAAI,KAAK,CAAC;AAAA,QACrB;AAAA,MACF,KAAK;AACH,eAAO,IAAI,cAAc,MAAM,IAAI,KAAK,CAAC;AAAA,MAC3C;AACE,eAAO,IAAI,aAAa,GAAG,IAAI,MAAM,IAAI,IAAI,UAAU,EAAE;AAAA,IAC7D;AAAA,EACF;AAAA;AAsBO,IAAM,iBAAN,MAAqB;AAAA,EAC1B,YAIS,MAIA,OAIA,WACP;AATO;AAIA;AAIA;AAAA,EACL;AACN;AAuBO,IAAM,SAAN,MAAa;AAAA,EAgDlB,YAAY,SAAkC,cAAuB;AAAvB;AAC5C,UAAM,OAAO,mBAAK;AAClB,WAAO,KAAK,MAAM;AAClB,WAAO,KAAK,gBAAgB;AAE5B,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,WAAW,KAAK,UAAU;AAC/B,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,MAAM,KAAK,KAAK;AACrB,SAAK,QAAQ,KAAK,OAAO;AACzB,SAAK,OAAO,KAAK,MAAM;AACvB,SAAK,aAAa,KAAK,YAAY;AACnC,SAAK,eAAe;AACpB,SAAK,MAAM;AAEX,SAAK,QAAQ,CAAC;AAEd,eAAW,OAAO,OAAO,KAAK,IAAI,GAAG;AACnC,UACE,CAAC;AAAA,QACC;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,QACA;AAAA,MACF,EAAE,SAAS,GAAG,GACd;AACA,aAAK,MAAM,GAAG,IAAI,KAAK,GAAG;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,UAAoB;AAClB,UAAM,UAAU,CAAC;AACjB,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,UAAU;AACjB,cAAQ,KAAK,UAAU;AAAA,IACzB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,KAAK;AACZ,cAAQ,KAAK,KAAK;AAAA,IACpB;AACA,QAAI,KAAK,OAAO;AACd,cAAQ,KAAK,OAAO;AAAA,IACtB;AACA,QAAI,KAAK,MAAM;AACb,cAAQ,KAAK,MAAM;AAAA,IACrB;AACA,QAAI,KAAK,YAAY;AACnB,cAAQ,KAAK,YAAY;AAAA,IAC3B;AAEA,eAAW,OAAO,OAAO,KAAK,KAAK,KAAK,GAAG;AACzC,cAAQ,KAAK,GAAG;AAAA,IAClB;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,MAAM,KAAK;AAAA,MACX,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,KAAK,KAAK;AAAA,MACV,KAAK,KAAK;AAAA,MACV,MAAM,KAAK;AAAA,MACX,KAAK,KAAK;AAAA,MACV,OAAO,KAAK;AAAA,MACZ,MAAM,KAAK;AAAA,MACX,YAAY,KAAK;AAAA,OACb,OAAO,KAAK,KAAK,KAAK,EAAE,SAAS,IAAI,EAAE,OAAO,KAAK,MAAM,IAAI,CAAC;AAAA,EAEtE;AACF;AAmBO,IAAM,YAAN,MAAgB;AAAA,EACrB,YAIS,UAAoB,CAAC,GAIrB,OAAa,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAC,EAAE,GAItC,OAIA,gBACP;AAbO;AAIA;AAIA;AAIA;AAAA,EACL;AAAA;AAAA;AAAA;AAAA,EAKJ,IAAI,OAA2B;AAC7B,eAAW,QAAQ,KAAK,SAAS;AAC/B,UAAI,KAAK,cAAc;AACrB,eAAO,KAAK;AAAA,MACd;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,WAAO;AAAA,MACL,SAAS,KAAK;AAAA,MACd,MAAM,KAAK;AAAA,MACX,OAAO,KAAK;AAAA,IACd;AAAA,EACF;AACF;AAEA,SAAsB,YACpB,WACA,MACA,UACA,UACA,UACA;AAAA;AACA,UAAM,MAAM,KAAK,MAAM,IAAI;AAE3B,YAAQ,IAAI,MAAM;AAAA,MAChB,KAAK;AACH,cAAM,SAAS,IAAI,OAAO,iCAAK,MAAL,EAAU,MAAM,QAAW,gBAAgB,OAAU,IAAG,IAAI,cAAc;AACpG,kBAAU,QAAQ,KAAK,MAAM;AAC7B,YAAI,UAAU;AACZ,gBAAM,SAAS,MAAM;AAAA,QACvB;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,KAAK,OAAO,KAAK,IAAI,IAAI;AACnC,YAAI,UAAU;AACZ,gBAAM,SAAS;AAAA,YACb,OAAO;AAAA,YACP,MAAM,IAAI;AAAA,YACV,YAAW,oBAAI,KAAK,GAAE,QAAQ,IAAI;AAAA,UACpC,CAAC;AAAA,QACH;AACA;AAAA,MACF,KAAK;AACH,kBAAU,QAAQ,IAAI,eAAe,IAAI,MAAM,IAAI,OAAO,IAAI,SAAS;AACvE;AAAA,MACF,KAAK;AACH,kBAAU,iBAAiB,IAAI;AAC/B;AAAA,IACJ;AAAA,EACF;AAAA;;;ADvUA,SAAgB,UAAU,QAAoC;AAAA;AAC5D,UAAM,SAAS,OAAO,UAAU;AAChC,QAAI,SAAS;AAEb,QAAI;AACF,aAAO,MAAM;AACX,cAAM,EAAE,MAAM,MAAM,IAAI,kBAAM,OAAO,KAAK;AAE1C,YAAI,UAAU,QAAW;AACvB,oBAAU,IAAI,YAAY,EAAE,OAAO,KAAK;AAAA,QAC1C;AAEA,YAAI,MAAM;AACR,cAAI,OAAO,SAAS,GAAG;AACrB,kBAAM;AAAA,UACR;AACA;AAAA,QACF;AAEA,YAAI,aAAa;AAEjB,WAAG;AACD,uBAAa,OAAO,QAAQ,IAAI;AAChC,cAAI,eAAe,IAAI;AACrB,kBAAM,OAAO,MAAM,GAAG,UAAU;AAChC,qBAAS,OAAO,MAAM,aAAa,CAAC;AAAA,UACtC;AAAA,QACF,SAAS,eAAe;AAAA,MAC1B;AAAA,IACF,UAAE;AACA,aAAO,YAAY;AAAA,IACrB;AAAA,EACF;AAAA;AAEO,IAAM,oBAAN,MAAM,kBAAiB;AAAA,EAI5B,YAA6B,KAA8B,kBAAoC;AAAlE;AAA8B;AAAA,EAAsC;AAAA,EAE3F,SACJ,MACA,MAQoB;AAAA;AAtDxB;AAuDI,YAAM,aAAa,IAAI,gBAAgB;AAEvC,YAAM,kBAAiB,kCAAM,qBAAN,YAA0B,KAAK,iBAAiB;AAEvE,YAAM,WAAW,iBAAiB,WAAW,MAAM;AACjD,mBAAW,MAAM;AAAA,MACnB,GAAG,cAAc,IACb;AAEJ,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,YAAY;AAAA,QAC7C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB;AAAA,UACA,YAAY,6BAAM;AAAA,QACpB,CAAC;AAAA,QACD,WAAW;AAAA,MACb,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,UAAI,CAAC,IAAI,MAAM;AACb,cAAM,IAAI,MAAM,sBAAsB,IAAI,UAAU,IAAI,MAAM,2BAAK,MAAM,EAAE;AAAA,MAC7E;AAEA,mBAAa,QAAQ;AAErB,YAAM,eAAc,kCAAM,cAAN,YAAmB,kBAAiB;AAExD,YAAM,YAAY,cACd,WAAW,MAAM;AACjB,mBAAW,MAAM;AAAA,MACnB,GAAG,WAAW,IACZ;AAEJ,YAAM,YAAY,IAAI,UAAU;AAEhC,UAAI;AACF;AAAA,qCAA0B,UAAU,IAAI,IAAI,IAA5C,YAAAA,QAAA,uDAA+C;AAApC,kBAAM,QAAjB;AACE,kBAAM,YAAY,WAAW,OAAO,6BAAM,UAAU,6BAAM,UAAU,6BAAM,QAAQ;AAAA,UACpF;AAAA,iBAFA,MAlGN;AAkGM,UAAAA,SAAA;AAAA;AAAA;AAAA;AAAA;AAAA,gBAAAA;AAAA,oBAAAA,OAAA;AAAA;AAAA;AAAA,MAGF,UAAE;AACA,qBAAa,SAAS;AAAA,MACxB;AAEA,aAAO;AAAA,IACT;AAAA;AAAA,EAEM,eAQmB;AAAA,+CARN;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,IACF,IAII,CAAC,GAAoB;AACvB,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,MAAM,KAAK,UAAU;AAAA,UACnB,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAAA,QACD,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,YAAM,OAAO,MAAM,IAAI,KAAK;AAC5B,aAAO,KAAK;AAAA,IACd;AAAA;AAAA,EAEM,gBAMiB;AAAA,+CANH;AAAA,MAClB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AACxC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,YAAY;AAAA,QAClE,QAAQ;AAAA,QACR,SAAS;AAAA,UACP,gBAAgB;AAAA,QAClB;AAAA,QACA,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,iBAMiB;AAAA,+CANF;AAAA,MACnB;AAAA,MACA;AAAA,IACF,IAGI,CAAC,GAAkB;AACrB,iBAAW,YAAY,kBAAiB;AAExC,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa,QAAQ,IAAI;AAAA,QAC1D,QAAQ;AAAA,QACR,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAAA,IACF;AAAA;AAAA,EAEM,cAIiD;AAAA,+CAJrC;AAAA,MAChB;AAAA,IACF,IAEI,CAAC,GAAkD;AACrD,YAAM,MAAM,MAAM,MAAM,GAAG,KAAK,GAAG,aAAa;AAAA,QAC9C,WAAW;AAAA,QACX,QAAQ,KAAK,iBAAiB,UAAU,gBAAgB;AAAA,MAC1D,CAAC;AAED,YAAM,QAAQ,MAAM,aAAa,GAAG;AACpC,UAAI,OAAO;AACT,cAAM;AAAA,MACR;AAEA,cAAQ,MAAM,IAAI,KAAK,GAAG,IAAI,CAAC,YAAiB,EAAE,UAAU,OAAO,IAAI,MAAM,OAAO,KAAK,EAAE;AAAA,IAC7F;AAAA;AACF;AAlKa,kBACa,gBAAgB;AAD7B,kBAEa,kBAAkB;AAFrC,IAAM,mBAAN;AAoKA,IAAM,mBAAN,MAAM,yBAAwB,QAAQ;AAAA,EAAtC;AAAA;AAIL,SAAS,WAAW,IAAI;AAAA,MACtB,GAAG,KAAK,iBAAiB,QAAQ,SAAS,OAAO,MAAM,KAAK,QAAQ,iBAAgB,WAAW,CAAC;AAAA,MAChG,KAAK;AAAA,IACP;AAAA;AACF;AARa,iBACwB,kBAA0B;AADlD,iBAEe,cAAc;AAFnC,IAAM,kBAAN;;;ADlMP,IAAO,cAAQ;","names":["error"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@e2b/code-interpreter",
-  "version": "0.0.9-beta.6",
+  "version": "0.0.9-beta.8",
   "description": "E2B Code Interpreter - Stateful code execution",
   "homepage": "https://e2b.dev",
   "license": "MIT",