beeai-framework 0.1.2

package/dist/tools/python/python.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/python/python.ts"],"names":["PythonTool","Tool","name","description","join","storage","files","emitter","Emitter","root","child","namespace","creator","inputSchema","list","fileNames","unique","map","filename","z","object","language","enum","describe","code","string","hasMinLength","inputFiles","array","preprocess","options","codeInterpreter","url","ValidationError","message","data","propertyName","register","_run","input","_options","run","pipe","uniqueBy","f","filter","file","isIncludedIn","upload","getSourceCode","llm","promptTemplate","response","create","messages","UserMessage","render","abortSignal","signal","getTextContent","trim","prefix","result","callCodeInterpreter","body","source_code","Object","fromEntries","pythonId","filesOutput","download","entries","path","_","startsWith","String","slice","length","every","PythonToolOutput","stdout","stderr","exit_code","createSnapshot","shallowCopy","loadSnapshot","snapshot","fetch","method","headers","JSON","stringify","catch","error","cause","ToolError","ok","status","Error","text","json"],"mappings":";;;;;;;;;;;;;;AAoDO,MAAMA,mBAAmBC,aAAAA,CAAAA;EApDhC;;;EAqDEC,IAAO,GAAA,QAAA;EACPC,WAAc,GAAA;AACZ,IAAA,0OAAA;AACA,IAAA,yKAAA;AACA,IAAA,6CAAA;AACA,IAAA,+BAAA;AACA,IAAA,yJAAA;AACA,IAAA,iKAAA;AACA,IAAA,iDAAA;AACA,IAAA,2DAAA;AACA,IAAA,gCAAA;AACA,IAAA,0DAAA;AACA,IAAA,oDAAA;AACA,IAAA,kDAAA;AACA,IAAA,yCAAA;AACA,IAAA,oFAAA;AACA,IAAA,6GAAA;AACA,IAAA,mEAAA;AACA,IAAA,uGAAA;AACA,IAAA;AACAC,GAAAA,CAAAA,IAAAA,CAAK,GAAA,CAAA;AAESC,EAAAA,OAAAA;AACNC,EAAAA,KAAAA,GAAsB,EAAA;EAEhBC,OAA0DC,GAAAA,mBAAAA,CAAQC,KAAKC,KAAM,CAAA;IAC3FC,SAAW,EAAA;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAS,EAAA;GACX,CAAA;AAEA,EAAA,MAAMC,WAAc,GAAA;AAClB,IAAA,IAAA,CAAKP,KAAQ,GAAA,MAAM,IAAKD,CAAAA,OAAAA,CAAQS,IAAI,EAAA;AACpC,IAAMC,MAAAA,SAAAA,GAAYC,aAAOC,CAAAA,UAAAA,CAAI,IAAKX,CAAAA,KAAAA,EAAO,CAAC,EAAEY,QAAAA,EAAeA,KAAAA,QAAAA,CAAAA,CAAAA;AAC3D,IAAA,OAAOC,MAAEC,MAAO,CAAA;AACdC,MAAAA,QAAAA,EAAUF,MAAEG,IAAK,CAAA;AAAC,QAAA,QAAA;AAAU,QAAA;AAAQ,OAAA,CAAA,CAAEC,SAAS,sCAAA,CAAA;AAC/CC,MAAAA,IAAAA,EAAML,KAAEM,CAAAA,MAAAA,EAASF,CAAAA,QAAAA,CAAS,6CAAA,CAAA;MAC1B,GAAIG,sBAAAA,CAAaX,SAAW,EAAA,CAAA,CACxB,GAAA;QACEY,UAAYR,EAAAA,KAAAA,CACTS,MAAMT,KAAEG,CAAAA,IAAAA,CAAKP,SAAAA,CAAAA,CAAAA,CACbQ,SACC,8KAAA;AAEN,OAAA,GACA;KACN,CAAA;AACF;AAEmBM,EAAAA,UAAAA;AAEnB,EAAA,WAAA,CAAmBC,OAA4B,EAAA;AAC7C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAI,IAAA,CAACA,OAAQC,CAAAA,eAAAA,CAAgBC,GAAK,EAAA;AAChC,MAAA,MAAM,IAAIC,mBAAgB,CAAA;AACxB,QAAA;UACEC,OAAS,EAAA,+BAAA;UACTC,IAAML,EAAAA,OAAAA;UACNM,YAAc,EAAA;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAQD,CAAAA,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAQzB,CAAAA,OAAAA;AACzB;EAEA;AACE,IAAA,IAAA,CAAKgC,QAAQ,EAAA;AACf;EAEA,MAAgBC,IAAAA,CACdC,KACAC,EAAAA,QAAAA,EACAC,GACA,EAAA;AACA,IAAA,MAAMd,UAAa,GAAA,MAAMe,WACvB,CAAA,IAAA,CAAKpC,SAAU,MAAM,IAAA,CAAKD,OAAQS,CAAAA,IAAAA,IAClC6B,eAAS,CAAA,CAACC,CAAMA,KAAAA,CAAAA,CAAE1B,QAAQ,CAC1B2B,EAAAA,aAAAA,CAAO,CAACC,IAAAA,KAASC,mBAAaD,CAAAA,IAAAA,CAAK5B,QAAWqB,EAAAA,KAAAA,CAAMZ,cAAc,EAAE,CAAA,CAAA,EACpE,CAACrB,KAAU,KAAA,IAAA,CAAKD,OAAQ2C,CAAAA,MAAAA,CAAO1C,KAAAA,CAAAA,CAAAA;AAGjC,IAAA,MAAM2C,gCAAgB,MAAA,CAAA,YAAA;AACpB,MAAA,IAAI,KAAKpB,UAAY,EAAA;AACnB,QAAA,MAAM,EAAEqB,GAAAA,EAAKC,cAAc,EAAA,GAAK,IAAKtB,CAAAA,UAAAA;AACrC,QAAMuB,MAAAA,QAAAA,GAAW,MAAMF,GAAAA,CAAIG,MAAO,CAAA;UAChCC,QAAU,EAAA;YAAC,IAAIC,uBAAAA,CAAYJ,eAAeK,MAAO,CAAA;AAAEjB,cAAAA,KAAAA,EAAOA,KAAMf,CAAAA;AAAK,aAAA,CAAA;;AACrEiC,UAAAA,WAAAA,EAAahB,GAAIiB,CAAAA;SACnB,CAAA;AACA,QAAON,OAAAA,QAAAA,CAASO,cAAc,EAAA,CAAGC,IAAI,EAAA;AACvC;AACA,MAAA,OAAOrB,KAAMf,CAAAA,IAAAA;KATO,EAAA,eAAA,CAAA;AAYtB,IAAA,MAAMqC,MAAS,GAAA,aAAA;AAEf,IAAMC,MAAAA,MAAAA,GAAS,MAAMC,mBAAoB,CAAA;AACvC/B,MAAAA,GAAAA,EAAK,CAAG,EAAA,IAAA,CAAKF,OAAQC,CAAAA,eAAAA,CAAgBC,GAAG,CAAA,WAAA,CAAA;MACxCgC,IAAM,EAAA;AACJC,QAAAA,WAAAA,EAAa,MAAMhB,aAAAA,EAAAA;AACnB3C,QAAAA,KAAAA,EAAO4D,MAAOC,CAAAA,WAAAA,CACZxC,UAAWV,CAAAA,GAAAA,CAAI,CAAC6B,IAAS,KAAA;UAAC,CAAGe,EAAAA,MAAAA,CAASf,EAAAA,IAAAA,CAAK5B,QAAQ,CAAA,CAAA;UAAI4B,IAAKsB,CAAAA;SAAS,CAAA;AAEzE,OAAA;AACAV,MAAAA,MAAAA,EAAQjB,GAAIiB,CAAAA;KACd,CAAA;AAEA,IAAMW,MAAAA,WAAAA,GAAc,MAAM,IAAA,CAAKhE,OAAQiE,CAAAA,QAAAA,CACrCJ,MAAOK,CAAAA,OAAAA,CAAQT,MAAOxD,CAAAA,KAAK,CACxBuC,CAAAA,MAAAA,CAAO,CAAC,CAAC2B,MAAMC,CAAAA,CAAAA,KAAOD,IAAKE,CAAAA,UAAAA,CAAWb,MAAAA,CAAAA,CACtC5C,CAAAA,GAAAA,CAAI,CAAC,CAACuD,IAAMJ,EAAAA,QAAAA,CAAe,MAAA;AAAEI,MAAAA,IAAAA;AAAYJ,MAAAA,QAAAA,EAAUO,OAAOP,QAAAA;KAAU,CAAA,CAAA,CACpEnD,GAAI,CAAA,CAAC6B,IAAU,MAAA;AAAE5B,MAAAA,QAAAA,EAAU4B,IAAK0B,CAAAA,IAAAA,CAAKI,KAAMf,CAAAA,MAAAA,CAAOgB,MAAM,CAAA;AAAGT,MAAAA,QAAAA,EAAUtB,IAAKsB,CAAAA;AAAS,KAAA,CAAA,EACnFvB,MAAO,CAAA,CAACC,IACPnB,KAAAA,UAAAA,CAAWmD,MACT,CAACvC,MAAAA,KAAUA,MAAMrB,CAAAA,QAAAA,KAAa4B,KAAK5B,QAAYqB,IAAAA,MAAAA,CAAM6B,aAAatB,IAAKsB,CAAAA,QAAQ,CAAA,CAAA,CAAA;AAKvF,IAAO,OAAA,IAAIW,4BAAiBjB,MAAOkB,CAAAA,MAAAA,EAAQlB,OAAOmB,MAAQnB,EAAAA,MAAAA,CAAOoB,WAAWb,WAAAA,CAAAA;AAC9E;EAEAc,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7E,KAAO8E,EAAAA,qBAAAA,CAAY,KAAK9E,KAAK,CAAA;AAC7BD,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACdwB,MAAAA,UAAAA,EAAY,IAAKA,CAAAA;AACnB,KAAA;AACF;AAEAwD,EAAAA,YAAAA,CAAaC,QAAwD,EAAA;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB;AACF;AAEA,eAAsBvB,mBAAoB,CAAA,EACxC/B,GACAgC,EAAAA,IAAAA,EACAN,QAKD,EAAA;AACC,EAAMN,MAAAA,QAAAA,GAAW,MAAMmC,KAAAA,CAAMvD,GAAK,EAAA;IAChCwD,MAAQ,EAAA,MAAA;IACRC,OAAS,EAAA;MACP,QAAU,EAAA,kBAAA;MACV,cAAgB,EAAA;AAClB,KAAA;IACAzB,IAAM0B,EAAAA,IAAAA,CAAKC,UAAU3B,IAAAA,CAAAA;AACrBN,IAAAA;GACF,CAAA,CAAGkC,KAAM,CAAA,CAACC,KAAAA,KAAAA;AACR,IAAIA,IAAAA,KAAAA,CAAMC,KAAM5F,CAAAA,IAAAA,IAAQ,iBAAmB,EAAA;AACzC,MAAM,MAAA,IAAI6F,mBACR,uIACA,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEJ,MAAA;AACL,MAAM,MAAA,IAAIE,mBAAU,6CAA+C,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;AAC5E;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAM5C,IAAAA,QAAAA,CAAS6C,SAAS,GAAK,EAAA;AAC1C,IAAA,MAAM,IAAIF,kBAAAA,CACR,CAAoE3C,iEAAAA,EAAAA,QAAAA,CAAS6C,MAAM,CACnF,CAAA,CAAA,EAAA;AAAC,MAAA,IAAIC,KAAM,CAAA,MAAM9C,QAAS+C,CAAAA,IAAAA,EAAI;AAAI,KAAA,CAAA;AAEtC;AAEA,EAAO,OAAA,MAAM/C,SAASgD,IAAI,EAAA;AAC5B;AApCsBrC,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"python.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  ToolEmitter,\n  Tool,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { filter, isIncludedIn, map, pipe, unique, uniqueBy } from \"remeda\";\nimport { PythonFile, PythonStorage } from \"@/tools/python/storage.js\";\nimport { PythonToolOutput } from \"@/tools/python/output.js\";\nimport { ValidationError } from \"ajv\";\nimport { ConnectionOptions } from \"node:tls\";\nimport { RunContext } from \"@/context.js\";\nimport { hasMinLength } from \"@/internals/helpers/array.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { UserMessage } from \"@/backend/message.js\";\n\nexport interface CodeInterpreterOptions {\n  url: string;\n  connectionOptions?: ConnectionOptions;\n}\n\nexport interface PythonToolOptions extends BaseToolOptions {\n  codeInterpreter: CodeInterpreterOptions;\n  preprocess?: {\n    llm: ChatModel;\n    promptTemplate: PromptTemplate.infer<{ input: string }>;\n  };\n  storage: PythonStorage;\n}\n\nexport class PythonTool extends Tool<PythonToolOutput, PythonToolOptions> {\n  name = \"Python\";\n  description = [\n    \"Run Python and/or shell code and return the console output. Use for isolated calculations, computations, data or file manipulation but still prefer assistant's capabilities (IMPORTANT: Do not use for text analysis or summarization).\",\n    \"Files provided by the user, or created in a previous run, will be accessible if and only if they are specified in the input. It is necessary to always print() results.\",\n    \"The following shell commands are available:\",\n    \"Use ffmpeg to convert videos.\",\n    \"Use yt-dlp to download videos, and unless specified otherwise use `-S vcodec:h264,res,acodec:m4a` for video and `-x --audio-format mp3` for audio-only.\",\n    \"Use pandoc to convert documents between formats (like MD, DOC, DOCX, PDF) -- and don't forget that you can create PDFs by writing markdown and then converting.\",\n    \"In Python, the following modules are available:\",\n    \"Use numpy, pandas, scipy and sympy for working with data.\",\n    \"Use matplotlib to plot charts.\",\n    \"Use pillow (import PIL) to create and manipulate images.\",\n    \"Use moviepy for complex manipulations with videos.\",\n    \"Use PyPDF2, pikepdf, or fitz to manipulate PDFs.\",\n    \"Use pdf2image to convert PDF to images.\",\n    \"Other Python libraries are also available -- however, prefer using the ones above.\",\n    \"Prefer using qualified imports -- `import library; library.thing()` instead of `import thing from library`.\",\n    \"Do not attempt to install libraries manually -- it will not work.\",\n    \"Each invocation of Python runs in a completely fresh VM -- it will not remember anything from before.\",\n    \"Do not use this tool multiple times in a row, always write the full code you want to run in a single invocation.\",\n  ].join(\" \");\n\n  public readonly storage: PythonStorage;\n  protected files: PythonFile[] = [];\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, PythonToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"python\"],\n    creator: this,\n  });\n\n  async inputSchema() {\n    this.files = await this.storage.list();\n    const fileNames = unique(map(this.files, ({ filename }) => filename));\n    return z.object({\n      language: z.enum([\"python\", \"shell\"]).describe(\"Use shell for ffmpeg, pandoc, yt-dlp\"),\n      code: z.string().describe(\"full source code file that will be executed\"),\n      ...(hasMinLength(fileNames, 1)\n        ? {\n            inputFiles: z\n              .array(z.enum(fileNames))\n              .describe(\n                \"To access an existing file, you must specify it; otherwise, the file will not be accessible. IMPORTANT: If the file is not provided in the input, it will not be accessible.\",\n              ),\n          }\n        : {}),\n    });\n  }\n\n  protected readonly preprocess;\n\n  public constructor(options: PythonToolOptions) {\n    super(options);\n    if (!options.codeInterpreter.url) {\n      throw new ValidationError([\n        {\n          message: \"Property must be a valid URL!\",\n          data: options,\n          propertyName: \"codeInterpreter.url\",\n        },\n      ]);\n    }\n    this.preprocess = options.preprocess;\n    this.storage = options.storage;\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: RunContext<this>,\n  ) {\n    const inputFiles = await pipe(\n      this.files ?? (await this.storage.list()),\n      uniqueBy((f) => f.filename),\n      filter((file) => isIncludedIn(file.filename, (input.inputFiles ?? []) as string[])),\n      (files) => this.storage.upload(files),\n    );\n\n    const getSourceCode = async () => {\n      if (this.preprocess) {\n        const { llm, promptTemplate } = this.preprocess;\n        const response = await llm.create({\n          messages: [new UserMessage(promptTemplate.render({ input: input.code }))],\n          abortSignal: run.signal,\n        });\n        return response.getTextContent().trim();\n      }\n      return input.code;\n    };\n\n    const prefix = \"/workspace/\";\n\n    const result = await callCodeInterpreter({\n      url: `${this.options.codeInterpreter.url}/v1/execute`,\n      body: {\n        source_code: await getSourceCode(),\n        files: Object.fromEntries(\n          inputFiles.map((file) => [`${prefix}${file.filename}`, file.pythonId]),\n        ),\n      },\n      signal: run.signal,\n    });\n\n    const filesOutput = await this.storage.download(\n      Object.entries(result.files)\n        .filter(([path, _]) => path.startsWith(prefix))\n        .map(([path, pythonId]) => ({ path: path, pythonId: String(pythonId) }))\n        .map((file) => ({ filename: file.path.slice(prefix.length), pythonId: file.pythonId }))\n        .filter((file) =>\n          inputFiles.every(\n            (input) => input.filename !== file.filename || input.pythonId !== file.pythonId,\n          ),\n        ),\n    );\n\n    return new PythonToolOutput(result.stdout, result.stderr, result.exit_code, filesOutput);\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      files: shallowCopy(this.files),\n      storage: this.storage,\n      preprocess: this.preprocess,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void {\n    super.loadSnapshot(snapshot);\n  }\n}\n\nexport async function callCodeInterpreter({\n  url,\n  body,\n  signal,\n}: {\n  url: string;\n  body: unknown;\n  signal?: AbortSignal;\n}): Promise<Record<string, any>> {\n  const response = await fetch(url, {\n    method: \"POST\",\n    headers: {\n      \"Accept\": \"application/json\",\n      \"Content-Type\": \"application/json\",\n    },\n    body: JSON.stringify(body),\n    signal,\n  }).catch((error) => {\n    if (error.cause.name == \"HTTPParserError\") {\n      throw new ToolError(\n        \"Request to bee-code-interpreter has failed -- ensure that CODE_INTERPRETER_URL points to the new HTTP endpoint (default port: 50081).\",\n        [error],\n      );\n    } else {\n      throw new ToolError(\"Request to bee-code-interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to bee-code-interpreter has failed with HTTP status code ${response.status}.`,\n      [new Error(await response.text())],\n    );\n  }\n\n  return await response.json();\n}\n"]}

package/dist/tools/python/python.d.cts

@@ -0,0 +1,91 @@
+import { BaseCache } from '../../cache/base.cjs';
+import * as promise_based_task from 'promise-based-task';
+import { BaseToolOptions, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.cjs';
+import { z } from 'zod';
+import { PromptTemplate } from '../../template.cjs';
+import { PythonStorage, PythonFile } from './storage.cjs';
+import { PythonToolOutput } from './output.cjs';
+import { ConnectionOptions } from 'node:tls';
+import { RunContext } from '../../context.cjs';
+import { E as Emitter } from '../../emitter-BWtGHYn0.cjs';
+import { ChatModel } from '../../backend/chat.cjs';
+import '../../internals/serializable.cjs';
+import '../../internals/types.cjs';
+import '../../internals/helpers/guards.cjs';
+import 'ajv';
+import '../../errors.cjs';
+import '../../internals/helpers/schema.cjs';
+import 'zod-to-json-schema';
+import '../../internals/helpers/promise.cjs';
+import 'fs';
+import '../../backend/message.cjs';
+import 'ai';
+import '../../backend/utils.cjs';
+import '../../backend/constants.cjs';
+import '@ai-sdk/provider';
+
+interface CodeInterpreterOptions {
+    url: string;
+    connectionOptions?: ConnectionOptions;
+}
+interface PythonToolOptions extends BaseToolOptions {
+    codeInterpreter: CodeInterpreterOptions;
+    preprocess?: {
+        llm: ChatModel;
+        promptTemplate: PromptTemplate.infer<{
+            input: string;
+        }>;
+    };
+    storage: PythonStorage;
+}
+declare class PythonTool extends Tool<PythonToolOutput, PythonToolOptions> {
+    name: string;
+    description: string;
+    readonly storage: PythonStorage;
+    protected files: PythonFile[];
+    readonly emitter: ToolEmitter<ToolInput<this>, PythonToolOutput>;
+    inputSchema(): Promise<z.ZodObject<{
+        inputFiles?: z.ZodArray<z.ZodEnum<[any, ...string[]]>, "many"> | undefined;
+        language: z.ZodEnum<["python", "shell"]>;
+        code: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        code: string;
+        language: "python" | "shell";
+        inputFiles?: unknown;
+    }, {
+        code: string;
+        language: "python" | "shell";
+        inputFiles?: unknown;
+    }>>;
+    protected readonly preprocess: {
+        llm: ChatModel;
+        promptTemplate: PromptTemplate.infer<{
+            input: string;
+        }>;
+    } | undefined;
+    constructor(options: PythonToolOptions);
+    protected _run(input: ToolInput<this>, _options: Partial<BaseToolRunOptions>, run: RunContext<this>): Promise<PythonToolOutput>;
+    createSnapshot(): {
+        files: PythonFile[];
+        storage: PythonStorage;
+        preprocess: {
+            llm: ChatModel;
+            promptTemplate: PromptTemplate.infer<{
+                input: string;
+            }>;
+        } | undefined;
+        name: string;
+        description: string;
+        options: PythonToolOptions;
+        cache: BaseCache<promise_based_task.Task<PythonToolOutput, any>>;
+        emitter: Emitter<any>;
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+declare function callCodeInterpreter({ url, body, signal, }: {
+    url: string;
+    body: unknown;
+    signal?: AbortSignal;
+}): Promise<Record<string, any>>;
+
+export { type CodeInterpreterOptions, PythonTool, type PythonToolOptions, callCodeInterpreter };

package/dist/tools/python/python.d.ts

@@ -0,0 +1,91 @@
+import { BaseCache } from '../../cache/base.js';
+import * as promise_based_task from 'promise-based-task';
+import { BaseToolOptions, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.js';
+import { z } from 'zod';
+import { PromptTemplate } from '../../template.js';
+import { PythonStorage, PythonFile } from './storage.js';
+import { PythonToolOutput } from './output.js';
+import { ConnectionOptions } from 'node:tls';
+import { RunContext } from '../../context.js';
+import { E as Emitter } from '../../emitter-l0W9gC1A.js';
+import { ChatModel } from '../../backend/chat.js';
+import '../../internals/serializable.js';
+import '../../internals/types.js';
+import '../../internals/helpers/guards.js';
+import 'ajv';
+import '../../errors.js';
+import '../../internals/helpers/schema.js';
+import 'zod-to-json-schema';
+import '../../internals/helpers/promise.js';
+import 'fs';
+import '../../backend/message.js';
+import 'ai';
+import '../../backend/utils.js';
+import '../../backend/constants.js';
+import '@ai-sdk/provider';
+
+interface CodeInterpreterOptions {
+    url: string;
+    connectionOptions?: ConnectionOptions;
+}
+interface PythonToolOptions extends BaseToolOptions {
+    codeInterpreter: CodeInterpreterOptions;
+    preprocess?: {
+        llm: ChatModel;
+        promptTemplate: PromptTemplate.infer<{
+            input: string;
+        }>;
+    };
+    storage: PythonStorage;
+}
+declare class PythonTool extends Tool<PythonToolOutput, PythonToolOptions> {
+    name: string;
+    description: string;
+    readonly storage: PythonStorage;
+    protected files: PythonFile[];
+    readonly emitter: ToolEmitter<ToolInput<this>, PythonToolOutput>;
+    inputSchema(): Promise<z.ZodObject<{
+        inputFiles?: z.ZodArray<z.ZodEnum<[any, ...string[]]>, "many"> | undefined;
+        language: z.ZodEnum<["python", "shell"]>;
+        code: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        code: string;
+        language: "python" | "shell";
+        inputFiles?: unknown;
+    }, {
+        code: string;
+        language: "python" | "shell";
+        inputFiles?: unknown;
+    }>>;
+    protected readonly preprocess: {
+        llm: ChatModel;
+        promptTemplate: PromptTemplate.infer<{
+            input: string;
+        }>;
+    } | undefined;
+    constructor(options: PythonToolOptions);
+    protected _run(input: ToolInput<this>, _options: Partial<BaseToolRunOptions>, run: RunContext<this>): Promise<PythonToolOutput>;
+    createSnapshot(): {
+        files: PythonFile[];
+        storage: PythonStorage;
+        preprocess: {
+            llm: ChatModel;
+            promptTemplate: PromptTemplate.infer<{
+                input: string;
+            }>;
+        } | undefined;
+        name: string;
+        description: string;
+        options: PythonToolOptions;
+        cache: BaseCache<promise_based_task.Task<PythonToolOutput, any>>;
+        emitter: Emitter<any>;
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+declare function callCodeInterpreter({ url, body, signal, }: {
+    url: string;
+    body: unknown;
+    signal?: AbortSignal;
+}): Promise<Record<string, any>>;
+
+export { type CodeInterpreterOptions, PythonTool, type PythonToolOptions, callCodeInterpreter };

package/dist/tools/python/python.js

@@ -0,0 +1,160 @@
+import { Tool, ToolError } from '../base.js';
+import { z } from 'zod';
+import { unique, map, pipe, uniqueBy, filter, isIncludedIn } from 'remeda';
+import { PythonToolOutput } from './output.js';
+import { ValidationError } from 'ajv';
+import { hasMinLength } from '../../internals/helpers/array.js';
+import { Emitter } from '../../emitter/emitter.js';
+import { shallowCopy } from '../../serializer/utils.js';
+import { UserMessage } from '../../backend/message.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class PythonTool extends Tool {
+  static {
+    __name(this, "PythonTool");
+  }
+  name = "Python";
+  description = [
+    "Run Python and/or shell code and return the console output. Use for isolated calculations, computations, data or file manipulation but still prefer assistant's capabilities (IMPORTANT: Do not use for text analysis or summarization).",
+    "Files provided by the user, or created in a previous run, will be accessible if and only if they are specified in the input. It is necessary to always print() results.",
+    "The following shell commands are available:",
+    "Use ffmpeg to convert videos.",
+    "Use yt-dlp to download videos, and unless specified otherwise use `-S vcodec:h264,res,acodec:m4a` for video and `-x --audio-format mp3` for audio-only.",
+    "Use pandoc to convert documents between formats (like MD, DOC, DOCX, PDF) -- and don't forget that you can create PDFs by writing markdown and then converting.",
+    "In Python, the following modules are available:",
+    "Use numpy, pandas, scipy and sympy for working with data.",
+    "Use matplotlib to plot charts.",
+    "Use pillow (import PIL) to create and manipulate images.",
+    "Use moviepy for complex manipulations with videos.",
+    "Use PyPDF2, pikepdf, or fitz to manipulate PDFs.",
+    "Use pdf2image to convert PDF to images.",
+    "Other Python libraries are also available -- however, prefer using the ones above.",
+    "Prefer using qualified imports -- `import library; library.thing()` instead of `import thing from library`.",
+    "Do not attempt to install libraries manually -- it will not work.",
+    "Each invocation of Python runs in a completely fresh VM -- it will not remember anything from before.",
+    "Do not use this tool multiple times in a row, always write the full code you want to run in a single invocation."
+  ].join(" ");
+  storage;
+  files = [];
+  emitter = Emitter.root.child({
+    namespace: [
+      "tool",
+      "python"
+    ],
+    creator: this
+  });
+  async inputSchema() {
+    this.files = await this.storage.list();
+    const fileNames = unique(map(this.files, ({ filename }) => filename));
+    return z.object({
+      language: z.enum([
+        "python",
+        "shell"
+      ]).describe("Use shell for ffmpeg, pandoc, yt-dlp"),
+      code: z.string().describe("full source code file that will be executed"),
+      ...hasMinLength(fileNames, 1) ? {
+        inputFiles: z.array(z.enum(fileNames)).describe("To access an existing file, you must specify it; otherwise, the file will not be accessible. IMPORTANT: If the file is not provided in the input, it will not be accessible.")
+      } : {}
+    });
+  }
+  preprocess;
+  constructor(options) {
+    super(options);
+    if (!options.codeInterpreter.url) {
+      throw new ValidationError([
+        {
+          message: "Property must be a valid URL!",
+          data: options,
+          propertyName: "codeInterpreter.url"
+        }
+      ]);
+    }
+    this.preprocess = options.preprocess;
+    this.storage = options.storage;
+  }
+  static {
+    this.register();
+  }
+  async _run(input, _options, run) {
+    const inputFiles = await pipe(this.files ?? await this.storage.list(), uniqueBy((f) => f.filename), filter((file) => isIncludedIn(file.filename, input.inputFiles ?? [])), (files) => this.storage.upload(files));
+    const getSourceCode = /* @__PURE__ */ __name(async () => {
+      if (this.preprocess) {
+        const { llm, promptTemplate } = this.preprocess;
+        const response = await llm.create({
+          messages: [
+            new UserMessage(promptTemplate.render({
+              input: input.code
+            }))
+          ],
+          abortSignal: run.signal
+        });
+        return response.getTextContent().trim();
+      }
+      return input.code;
+    }, "getSourceCode");
+    const prefix = "/workspace/";
+    const result = await callCodeInterpreter({
+      url: `${this.options.codeInterpreter.url}/v1/execute`,
+      body: {
+        source_code: await getSourceCode(),
+        files: Object.fromEntries(inputFiles.map((file) => [
+          `${prefix}${file.filename}`,
+          file.pythonId
+        ]))
+      },
+      signal: run.signal
+    });
+    const filesOutput = await this.storage.download(Object.entries(result.files).filter(([path, _]) => path.startsWith(prefix)).map(([path, pythonId]) => ({
+      path,
+      pythonId: String(pythonId)
+    })).map((file) => ({
+      filename: file.path.slice(prefix.length),
+      pythonId: file.pythonId
+    })).filter((file) => inputFiles.every((input2) => input2.filename !== file.filename || input2.pythonId !== file.pythonId)));
+    return new PythonToolOutput(result.stdout, result.stderr, result.exit_code, filesOutput);
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      files: shallowCopy(this.files),
+      storage: this.storage,
+      preprocess: this.preprocess
+    };
+  }
+  loadSnapshot(snapshot) {
+    super.loadSnapshot(snapshot);
+  }
+}
+async function callCodeInterpreter({ url, body, signal }) {
+  const response = await fetch(url, {
+    method: "POST",
+    headers: {
+      "Accept": "application/json",
+      "Content-Type": "application/json"
+    },
+    body: JSON.stringify(body),
+    signal
+  }).catch((error) => {
+    if (error.cause.name == "HTTPParserError") {
+      throw new ToolError("Request to bee-code-interpreter has failed -- ensure that CODE_INTERPRETER_URL points to the new HTTP endpoint (default port: 50081).", [
+        error
+      ]);
+    } else {
+      throw new ToolError("Request to bee-code-interpreter has failed.", [
+        error
+      ]);
+    }
+  });
+  if (!response?.ok && response.status > 400) {
+    throw new ToolError(`Request to bee-code-interpreter has failed with HTTP status code ${response.status}.`, [
+      new Error(await response.text())
+    ]);
+  }
+  return await response.json();
+}
+__name(callCodeInterpreter, "callCodeInterpreter");
+
+export { PythonTool, callCodeInterpreter };
+//# sourceMappingURL=python.js.map
+//# sourceMappingURL=python.js.map

package/dist/tools/python/python.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/python/python.ts"],"names":["PythonTool","Tool","name","description","join","storage","files","emitter","Emitter","root","child","namespace","creator","inputSchema","list","fileNames","unique","map","filename","z","object","language","enum","describe","code","string","hasMinLength","inputFiles","array","preprocess","options","codeInterpreter","url","ValidationError","message","data","propertyName","register","_run","input","_options","run","pipe","uniqueBy","f","filter","file","isIncludedIn","upload","getSourceCode","llm","promptTemplate","response","create","messages","UserMessage","render","abortSignal","signal","getTextContent","trim","prefix","result","callCodeInterpreter","body","source_code","Object","fromEntries","pythonId","filesOutput","download","entries","path","_","startsWith","String","slice","length","every","PythonToolOutput","stdout","stderr","exit_code","createSnapshot","shallowCopy","loadSnapshot","snapshot","fetch","method","headers","JSON","stringify","catch","error","cause","ToolError","ok","status","Error","text","json"],"mappings":";;;;;;;;;;;;AAoDO,MAAMA,mBAAmBC,IAAAA,CAAAA;EApDhC;;;EAqDEC,IAAO,GAAA,QAAA;EACPC,WAAc,GAAA;AACZ,IAAA,0OAAA;AACA,IAAA,yKAAA;AACA,IAAA,6CAAA;AACA,IAAA,+BAAA;AACA,IAAA,yJAAA;AACA,IAAA,iKAAA;AACA,IAAA,iDAAA;AACA,IAAA,2DAAA;AACA,IAAA,gCAAA;AACA,IAAA,0DAAA;AACA,IAAA,oDAAA;AACA,IAAA,kDAAA;AACA,IAAA,yCAAA;AACA,IAAA,oFAAA;AACA,IAAA,6GAAA;AACA,IAAA,mEAAA;AACA,IAAA,uGAAA;AACA,IAAA;AACAC,GAAAA,CAAAA,IAAAA,CAAK,GAAA,CAAA;AAESC,EAAAA,OAAAA;AACNC,EAAAA,KAAAA,GAAsB,EAAA;EAEhBC,OAA0DC,GAAAA,OAAAA,CAAQC,KAAKC,KAAM,CAAA;IAC3FC,SAAW,EAAA;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAS,EAAA;GACX,CAAA;AAEA,EAAA,MAAMC,WAAc,GAAA;AAClB,IAAA,IAAA,CAAKP,KAAQ,GAAA,MAAM,IAAKD,CAAAA,OAAAA,CAAQS,IAAI,EAAA;AACpC,IAAMC,MAAAA,SAAAA,GAAYC,MAAOC,CAAAA,GAAAA,CAAI,IAAKX,CAAAA,KAAAA,EAAO,CAAC,EAAEY,QAAAA,EAAeA,KAAAA,QAAAA,CAAAA,CAAAA;AAC3D,IAAA,OAAOC,EAAEC,MAAO,CAAA;AACdC,MAAAA,QAAAA,EAAUF,EAAEG,IAAK,CAAA;AAAC,QAAA,QAAA;AAAU,QAAA;AAAQ,OAAA,CAAA,CAAEC,SAAS,sCAAA,CAAA;AAC/CC,MAAAA,IAAAA,EAAML,CAAEM,CAAAA,MAAAA,EAASF,CAAAA,QAAAA,CAAS,6CAAA,CAAA;MAC1B,GAAIG,YAAAA,CAAaX,SAAW,EAAA,CAAA,CACxB,GAAA;QACEY,UAAYR,EAAAA,CAAAA,CACTS,MAAMT,CAAEG,CAAAA,IAAAA,CAAKP,SAAAA,CAAAA,CAAAA,CACbQ,SACC,8KAAA;AAEN,OAAA,GACA;KACN,CAAA;AACF;AAEmBM,EAAAA,UAAAA;AAEnB,EAAA,WAAA,CAAmBC,OAA4B,EAAA;AAC7C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAI,IAAA,CAACA,OAAQC,CAAAA,eAAAA,CAAgBC,GAAK,EAAA;AAChC,MAAA,MAAM,IAAIC,eAAgB,CAAA;AACxB,QAAA;UACEC,OAAS,EAAA,+BAAA;UACTC,IAAML,EAAAA,OAAAA;UACNM,YAAc,EAAA;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAQD,CAAAA,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAQzB,CAAAA,OAAAA;AACzB;EAEA;AACE,IAAA,IAAA,CAAKgC,QAAQ,EAAA;AACf;EAEA,MAAgBC,IAAAA,CACdC,KACAC,EAAAA,QAAAA,EACAC,GACA,EAAA;AACA,IAAA,MAAMd,UAAa,GAAA,MAAMe,IACvB,CAAA,IAAA,CAAKpC,SAAU,MAAM,IAAA,CAAKD,OAAQS,CAAAA,IAAAA,IAClC6B,QAAS,CAAA,CAACC,CAAMA,KAAAA,CAAAA,CAAE1B,QAAQ,CAC1B2B,EAAAA,MAAAA,CAAO,CAACC,IAAAA,KAASC,YAAaD,CAAAA,IAAAA,CAAK5B,QAAWqB,EAAAA,KAAAA,CAAMZ,cAAc,EAAE,CAAA,CAAA,EACpE,CAACrB,KAAU,KAAA,IAAA,CAAKD,OAAQ2C,CAAAA,MAAAA,CAAO1C,KAAAA,CAAAA,CAAAA;AAGjC,IAAA,MAAM2C,gCAAgB,MAAA,CAAA,YAAA;AACpB,MAAA,IAAI,KAAKpB,UAAY,EAAA;AACnB,QAAA,MAAM,EAAEqB,GAAAA,EAAKC,cAAc,EAAA,GAAK,IAAKtB,CAAAA,UAAAA;AACrC,QAAMuB,MAAAA,QAAAA,GAAW,MAAMF,GAAAA,CAAIG,MAAO,CAAA;UAChCC,QAAU,EAAA;YAAC,IAAIC,WAAAA,CAAYJ,eAAeK,MAAO,CAAA;AAAEjB,cAAAA,KAAAA,EAAOA,KAAMf,CAAAA;AAAK,aAAA,CAAA;;AACrEiC,UAAAA,WAAAA,EAAahB,GAAIiB,CAAAA;SACnB,CAAA;AACA,QAAON,OAAAA,QAAAA,CAASO,cAAc,EAAA,CAAGC,IAAI,EAAA;AACvC;AACA,MAAA,OAAOrB,KAAMf,CAAAA,IAAAA;KATO,EAAA,eAAA,CAAA;AAYtB,IAAA,MAAMqC,MAAS,GAAA,aAAA;AAEf,IAAMC,MAAAA,MAAAA,GAAS,MAAMC,mBAAoB,CAAA;AACvC/B,MAAAA,GAAAA,EAAK,CAAG,EAAA,IAAA,CAAKF,OAAQC,CAAAA,eAAAA,CAAgBC,GAAG,CAAA,WAAA,CAAA;MACxCgC,IAAM,EAAA;AACJC,QAAAA,WAAAA,EAAa,MAAMhB,aAAAA,EAAAA;AACnB3C,QAAAA,KAAAA,EAAO4D,MAAOC,CAAAA,WAAAA,CACZxC,UAAWV,CAAAA,GAAAA,CAAI,CAAC6B,IAAS,KAAA;UAAC,CAAGe,EAAAA,MAAAA,CAASf,EAAAA,IAAAA,CAAK5B,QAAQ,CAAA,CAAA;UAAI4B,IAAKsB,CAAAA;SAAS,CAAA;AAEzE,OAAA;AACAV,MAAAA,MAAAA,EAAQjB,GAAIiB,CAAAA;KACd,CAAA;AAEA,IAAMW,MAAAA,WAAAA,GAAc,MAAM,IAAA,CAAKhE,OAAQiE,CAAAA,QAAAA,CACrCJ,MAAOK,CAAAA,OAAAA,CAAQT,MAAOxD,CAAAA,KAAK,CACxBuC,CAAAA,MAAAA,CAAO,CAAC,CAAC2B,MAAMC,CAAAA,CAAAA,KAAOD,IAAKE,CAAAA,UAAAA,CAAWb,MAAAA,CAAAA,CACtC5C,CAAAA,GAAAA,CAAI,CAAC,CAACuD,IAAMJ,EAAAA,QAAAA,CAAe,MAAA;AAAEI,MAAAA,IAAAA;AAAYJ,MAAAA,QAAAA,EAAUO,OAAOP,QAAAA;KAAU,CAAA,CAAA,CACpEnD,GAAI,CAAA,CAAC6B,IAAU,MAAA;AAAE5B,MAAAA,QAAAA,EAAU4B,IAAK0B,CAAAA,IAAAA,CAAKI,KAAMf,CAAAA,MAAAA,CAAOgB,MAAM,CAAA;AAAGT,MAAAA,QAAAA,EAAUtB,IAAKsB,CAAAA;AAAS,KAAA,CAAA,EACnFvB,MAAO,CAAA,CAACC,IACPnB,KAAAA,UAAAA,CAAWmD,MACT,CAACvC,MAAAA,KAAUA,MAAMrB,CAAAA,QAAAA,KAAa4B,KAAK5B,QAAYqB,IAAAA,MAAAA,CAAM6B,aAAatB,IAAKsB,CAAAA,QAAQ,CAAA,CAAA,CAAA;AAKvF,IAAO,OAAA,IAAIW,iBAAiBjB,MAAOkB,CAAAA,MAAAA,EAAQlB,OAAOmB,MAAQnB,EAAAA,MAAAA,CAAOoB,WAAWb,WAAAA,CAAAA;AAC9E;EAEAc,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7E,KAAO8E,EAAAA,WAAAA,CAAY,KAAK9E,KAAK,CAAA;AAC7BD,MAAAA,OAAAA,EAAS,IAAKA,CAAAA,OAAAA;AACdwB,MAAAA,UAAAA,EAAY,IAAKA,CAAAA;AACnB,KAAA;AACF;AAEAwD,EAAAA,YAAAA,CAAaC,QAAwD,EAAA;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB;AACF;AAEA,eAAsBvB,mBAAoB,CAAA,EACxC/B,GACAgC,EAAAA,IAAAA,EACAN,QAKD,EAAA;AACC,EAAMN,MAAAA,QAAAA,GAAW,MAAMmC,KAAAA,CAAMvD,GAAK,EAAA;IAChCwD,MAAQ,EAAA,MAAA;IACRC,OAAS,EAAA;MACP,QAAU,EAAA,kBAAA;MACV,cAAgB,EAAA;AAClB,KAAA;IACAzB,IAAM0B,EAAAA,IAAAA,CAAKC,UAAU3B,IAAAA,CAAAA;AACrBN,IAAAA;GACF,CAAA,CAAGkC,KAAM,CAAA,CAACC,KAAAA,KAAAA;AACR,IAAIA,IAAAA,KAAAA,CAAMC,KAAM5F,CAAAA,IAAAA,IAAQ,iBAAmB,EAAA;AACzC,MAAM,MAAA,IAAI6F,UACR,uIACA,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEJ,MAAA;AACL,MAAM,MAAA,IAAIE,UAAU,6CAA+C,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;AAC5E;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAM5C,IAAAA,QAAAA,CAAS6C,SAAS,GAAK,EAAA;AAC1C,IAAA,MAAM,IAAIF,SAAAA,CACR,CAAoE3C,iEAAAA,EAAAA,QAAAA,CAAS6C,MAAM,CACnF,CAAA,CAAA,EAAA;AAAC,MAAA,IAAIC,KAAM,CAAA,MAAM9C,QAAS+C,CAAAA,IAAAA,EAAI;AAAI,KAAA,CAAA;AAEtC;AAEA,EAAO,OAAA,MAAM/C,SAASgD,IAAI,EAAA;AAC5B;AApCsBrC,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"python.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  ToolEmitter,\n  Tool,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { filter, isIncludedIn, map, pipe, unique, uniqueBy } from \"remeda\";\nimport { PythonFile, PythonStorage } from \"@/tools/python/storage.js\";\nimport { PythonToolOutput } from \"@/tools/python/output.js\";\nimport { ValidationError } from \"ajv\";\nimport { ConnectionOptions } from \"node:tls\";\nimport { RunContext } from \"@/context.js\";\nimport { hasMinLength } from \"@/internals/helpers/array.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { UserMessage } from \"@/backend/message.js\";\n\nexport interface CodeInterpreterOptions {\n  url: string;\n  connectionOptions?: ConnectionOptions;\n}\n\nexport interface PythonToolOptions extends BaseToolOptions {\n  codeInterpreter: CodeInterpreterOptions;\n  preprocess?: {\n    llm: ChatModel;\n    promptTemplate: PromptTemplate.infer<{ input: string }>;\n  };\n  storage: PythonStorage;\n}\n\nexport class PythonTool extends Tool<PythonToolOutput, PythonToolOptions> {\n  name = \"Python\";\n  description = [\n    \"Run Python and/or shell code and return the console output. Use for isolated calculations, computations, data or file manipulation but still prefer assistant's capabilities (IMPORTANT: Do not use for text analysis or summarization).\",\n    \"Files provided by the user, or created in a previous run, will be accessible if and only if they are specified in the input. It is necessary to always print() results.\",\n    \"The following shell commands are available:\",\n    \"Use ffmpeg to convert videos.\",\n    \"Use yt-dlp to download videos, and unless specified otherwise use `-S vcodec:h264,res,acodec:m4a` for video and `-x --audio-format mp3` for audio-only.\",\n    \"Use pandoc to convert documents between formats (like MD, DOC, DOCX, PDF) -- and don't forget that you can create PDFs by writing markdown and then converting.\",\n    \"In Python, the following modules are available:\",\n    \"Use numpy, pandas, scipy and sympy for working with data.\",\n    \"Use matplotlib to plot charts.\",\n    \"Use pillow (import PIL) to create and manipulate images.\",\n    \"Use moviepy for complex manipulations with videos.\",\n    \"Use PyPDF2, pikepdf, or fitz to manipulate PDFs.\",\n    \"Use pdf2image to convert PDF to images.\",\n    \"Other Python libraries are also available -- however, prefer using the ones above.\",\n    \"Prefer using qualified imports -- `import library; library.thing()` instead of `import thing from library`.\",\n    \"Do not attempt to install libraries manually -- it will not work.\",\n    \"Each invocation of Python runs in a completely fresh VM -- it will not remember anything from before.\",\n    \"Do not use this tool multiple times in a row, always write the full code you want to run in a single invocation.\",\n  ].join(\" \");\n\n  public readonly storage: PythonStorage;\n  protected files: PythonFile[] = [];\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, PythonToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"python\"],\n    creator: this,\n  });\n\n  async inputSchema() {\n    this.files = await this.storage.list();\n    const fileNames = unique(map(this.files, ({ filename }) => filename));\n    return z.object({\n      language: z.enum([\"python\", \"shell\"]).describe(\"Use shell for ffmpeg, pandoc, yt-dlp\"),\n      code: z.string().describe(\"full source code file that will be executed\"),\n      ...(hasMinLength(fileNames, 1)\n        ? {\n            inputFiles: z\n              .array(z.enum(fileNames))\n              .describe(\n                \"To access an existing file, you must specify it; otherwise, the file will not be accessible. IMPORTANT: If the file is not provided in the input, it will not be accessible.\",\n              ),\n          }\n        : {}),\n    });\n  }\n\n  protected readonly preprocess;\n\n  public constructor(options: PythonToolOptions) {\n    super(options);\n    if (!options.codeInterpreter.url) {\n      throw new ValidationError([\n        {\n          message: \"Property must be a valid URL!\",\n          data: options,\n          propertyName: \"codeInterpreter.url\",\n        },\n      ]);\n    }\n    this.preprocess = options.preprocess;\n    this.storage = options.storage;\n  }\n\n  static {\n    this.register();\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: RunContext<this>,\n  ) {\n    const inputFiles = await pipe(\n      this.files ?? (await this.storage.list()),\n      uniqueBy((f) => f.filename),\n      filter((file) => isIncludedIn(file.filename, (input.inputFiles ?? []) as string[])),\n      (files) => this.storage.upload(files),\n    );\n\n    const getSourceCode = async () => {\n      if (this.preprocess) {\n        const { llm, promptTemplate } = this.preprocess;\n        const response = await llm.create({\n          messages: [new UserMessage(promptTemplate.render({ input: input.code }))],\n          abortSignal: run.signal,\n        });\n        return response.getTextContent().trim();\n      }\n      return input.code;\n    };\n\n    const prefix = \"/workspace/\";\n\n    const result = await callCodeInterpreter({\n      url: `${this.options.codeInterpreter.url}/v1/execute`,\n      body: {\n        source_code: await getSourceCode(),\n        files: Object.fromEntries(\n          inputFiles.map((file) => [`${prefix}${file.filename}`, file.pythonId]),\n        ),\n      },\n      signal: run.signal,\n    });\n\n    const filesOutput = await this.storage.download(\n      Object.entries(result.files)\n        .filter(([path, _]) => path.startsWith(prefix))\n        .map(([path, pythonId]) => ({ path: path, pythonId: String(pythonId) }))\n        .map((file) => ({ filename: file.path.slice(prefix.length), pythonId: file.pythonId }))\n        .filter((file) =>\n          inputFiles.every(\n            (input) => input.filename !== file.filename || input.pythonId !== file.pythonId,\n          ),\n        ),\n    );\n\n    return new PythonToolOutput(result.stdout, result.stderr, result.exit_code, filesOutput);\n  }\n\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      files: shallowCopy(this.files),\n      storage: this.storage,\n      preprocess: this.preprocess,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void {\n    super.loadSnapshot(snapshot);\n  }\n}\n\nexport async function callCodeInterpreter({\n  url,\n  body,\n  signal,\n}: {\n  url: string;\n  body: unknown;\n  signal?: AbortSignal;\n}): Promise<Record<string, any>> {\n  const response = await fetch(url, {\n    method: \"POST\",\n    headers: {\n      \"Accept\": \"application/json\",\n      \"Content-Type\": \"application/json\",\n    },\n    body: JSON.stringify(body),\n    signal,\n  }).catch((error) => {\n    if (error.cause.name == \"HTTPParserError\") {\n      throw new ToolError(\n        \"Request to bee-code-interpreter has failed -- ensure that CODE_INTERPRETER_URL points to the new HTTP endpoint (default port: 50081).\",\n        [error],\n      );\n    } else {\n      throw new ToolError(\"Request to bee-code-interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to bee-code-interpreter has failed with HTTP status code ${response.status}.`,\n      [new Error(await response.text())],\n    );\n  }\n\n  return await response.json();\n}\n"]}

package/dist/tools/python/storage.cjs

@@ -0,0 +1,170 @@
+'use strict';
+
+var crypto = require('node:crypto');
+var promises$1 = require('node:stream/promises');
+var fs = require('node:fs');
+var promises = require('node:fs/promises');
+var path = require('node:path');
+var decoratorCache_cjs = require('../../cache/decoratorCache.cjs');
+var serializable_cjs = require('../../internals/serializable.cjs');
+var utils_cjs = require('../../serializer/utils.cjs');
+
+function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
+
+function _interopNamespace(e) {
+  if (e && e.__esModule) return e;
+  var n = Object.create(null);
+  if (e) {
+    Object.keys(e).forEach(function (k) {
+      if (k !== 'default') {
+        var d = Object.getOwnPropertyDescriptor(e, k);
+        Object.defineProperty(n, k, d.get ? d : {
+          enumerable: true,
+          get: function () { return e[k]; }
+        });
+      }
+    });
+  }
+  n.default = e;
+  return Object.freeze(n);
+}
+
+var crypto__namespace = /*#__PURE__*/_interopNamespace(crypto);
+var fs__default = /*#__PURE__*/_interopDefault(fs);
+var path__default = /*#__PURE__*/_interopDefault(path);
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate, "_ts_decorate");
+function _ts_metadata(k, v) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+}
+__name(_ts_metadata, "_ts_metadata");
+class PythonStorage extends serializable_cjs.Serializable {
+  static {
+    __name(this, "PythonStorage");
+  }
+}
+class TemporaryStorage extends PythonStorage {
+  static {
+    __name(this, "TemporaryStorage");
+  }
+  files = [];
+  async list() {
+    return this.files.slice();
+  }
+  async upload(files) {
+    return files.map((file) => ({
+      id: file.id,
+      pythonId: file.id,
+      filename: file.filename
+    }));
+  }
+  async download(files) {
+    this.files = [
+      ...this.files.filter((file) => files.every((f) => f.filename !== file.filename)),
+      ...files.map((file) => ({
+        id: file.pythonId,
+        ...file
+      }))
+    ];
+    return this.files.slice();
+  }
+  createSnapshot() {
+    return {
+      files: this.files.slice()
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+class LocalPythonStorage extends PythonStorage {
+  static {
+    __name(this, "LocalPythonStorage");
+  }
+  input;
+  constructor(input) {
+    super();
+    this.input = {
+      ...input,
+      ignoredFiles: input?.ignoredFiles ?? /* @__PURE__ */ new Set([
+        ".gitkeep"
+      ])
+    };
+  }
+  async init() {
+    await fs__default.default.promises.mkdir(this.input.localWorkingDir, {
+      recursive: true
+    });
+    await fs__default.default.promises.mkdir(this.input.interpreterWorkingDir, {
+      recursive: true
+    });
+  }
+  async list() {
+    await this.init();
+    const files = await fs__default.default.promises.readdir(this.input.localWorkingDir, {
+      withFileTypes: true,
+      recursive: false
+    });
+    return Promise.all(files.filter((file) => file.isFile() && !this.input.ignoredFiles.has(file.name)).map(async (file) => {
+      const pythonId = await this.computeHash(path__default.default.join(this.input.localWorkingDir.toString(), file.name));
+      return {
+        id: pythonId,
+        filename: file.name,
+        pythonId
+      };
+    }));
+  }
+  async upload(files) {
+    await this.init();
+    const fileList = await this.list();
+    await Promise.all(files.map((file) => {
+      const filesystemFile = fileList.find((filesystemFile2) => filesystemFile2.id === file.id);
+      return promises.copyFile(path__default.default.join(this.input.localWorkingDir.toString(), filesystemFile.filename), path__default.default.join(this.input.interpreterWorkingDir.toString(), filesystemFile.id));
+    }));
+    return files.map((file) => ({
+      ...file,
+      pythonId: file.id
+    }));
+  }
+  async download(files) {
+    await this.init();
+    await Promise.all(files.map((file) => promises.copyFile(path__default.default.join(this.input.interpreterWorkingDir.toString(), file.pythonId), path__default.default.join(this.input.localWorkingDir.toString(), file.filename))));
+    return files.map((file) => ({
+      ...file,
+      id: file.pythonId
+    }));
+  }
+  async computeHash(file) {
+    const hash = crypto__namespace.createHash("sha256");
+    await promises$1.pipeline(fs.createReadStream(file), hash);
+    return hash.digest("hex");
+  }
+  createSnapshot() {
+    return {
+      input: utils_cjs.shallowCopy(this.input)
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+_ts_decorate([
+  decoratorCache_cjs.Cache(),
+  _ts_metadata("design:type", Function),
+  _ts_metadata("design:paramtypes", []),
+  _ts_metadata("design:returntype", Promise)
+], LocalPythonStorage.prototype, "init", null);
+
+exports.LocalPythonStorage = LocalPythonStorage;
+exports.PythonStorage = PythonStorage;
+exports.TemporaryStorage = TemporaryStorage;
+//# sourceMappingURL=storage.cjs.map
+//# sourceMappingURL=storage.cjs.map

package/dist/tools/python/storage.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/python/storage.ts"],"names":["PythonStorage","Serializable","TemporaryStorage","files","list","slice","upload","map","file","id","pythonId","filename","download","filter","every","f","createSnapshot","loadSnapshot","snapshot","Object","assign","LocalPythonStorage","input","constructor","ignoredFiles","Set","init","fs","promises","mkdir","localWorkingDir","recursive","interpreterWorkingDir","readdir","withFileTypes","Promise","all","isFile","has","name","computeHash","path","join","toString","fileList","filesystemFile","find","copyFile","hash","crypto","createHash","pipeline","createReadStream","digest","shallowCopy"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CO,MAAeA,sBAAsBC,6BAAAA,CAAAA;AAAAA,EAAAA;;;AAe5C;AAEO,MAAMC,yBAAyBF,aAAAA,CAAAA;AAAAA,EAAAA;;;AAC1BG,EAAAA,KAAAA,GAAsB,EAAA;AAEhC,EAAA,MAAMC,IAAO,GAAA;AACX,IAAO,OAAA,IAAA,CAAKD,MAAME,KAAK,EAAA;AACzB;AAEA,EAAA,MAAMC,OAAOH,KAAkD,EAAA;AAC7D,IAAOA,OAAAA,KAAAA,CAAMI,GAAI,CAAA,CAACC,IAAU,MAAA;AAC1BC,MAAAA,EAAAA,EAAID,IAAKC,CAAAA,EAAAA;AACTC,MAAAA,QAAAA,EAAUF,IAAKC,CAAAA,EAAAA;AACfE,MAAAA,QAAAA,EAAUH,IAAKG,CAAAA;KACjB,CAAA,CAAA;AACF;AAEA,EAAA,MAAMC,SAAST,KAA6B,EAAA;AAC1C,IAAA,IAAA,CAAKA,KAAQ,GAAA;AACR,MAAA,GAAA,IAAA,CAAKA,KAAMU,CAAAA,MAAAA,CAAO,CAACL,IAAAA,KAASL,KAAMW,CAAAA,KAAAA,CAAM,CAACC,CAAAA,KAAMA,CAAEJ,CAAAA,QAAAA,KAAaH,IAAKG,CAAAA,QAAQ,CAAA,CAAA;SAC3ER,KAAMI,CAAAA,GAAAA,CAAI,CAACC,IAAU,MAAA;AAAEC,QAAAA,EAAAA,EAAID,IAAKE,CAAAA,QAAAA;QAAU,GAAGF;OAAK,CAAA;;AAEvD,IAAO,OAAA,IAAA,CAAKL,MAAME,KAAK,EAAA;AACzB;EAEAW,cAAiB,GAAA;AACf,IAAO,OAAA;MAAEb,KAAO,EAAA,IAAA,CAAKA,MAAME,KAAK;AAAG,KAAA;AACrC;AAEAY,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF;AAQO,MAAMG,2BAA2BrB,aAAAA,CAAAA;AAAAA,EAAAA;;;AACnBsB,EAAAA,KAAAA;AAEnBC,EAAAA,WAAAA,CAAYD,KAAc,EAAA;AACxB,IAAK,KAAA,EAAA;AACL,IAAA,IAAA,CAAKA,KAAQ,GAAA;MACX,GAAGA,KAAAA;MACHE,YAAcF,EAAAA,KAAAA,EAAOE,YAAgB,oBAAA,IAAIC,GAAI,CAAA;AAAC,QAAA;AAAW,OAAA;AAC3D,KAAA;AACF;AAEA,EAAA,MACgBC,IAAO,GAAA;AACrB,IAAA,MAAMC,mBAAGC,CAAAA,QAAAA,CAASC,KAAM,CAAA,IAAA,CAAKP,MAAMQ,eAAiB,EAAA;MAAEC,SAAW,EAAA;KAAK,CAAA;AACtE,IAAA,MAAMJ,mBAAGC,CAAAA,QAAAA,CAASC,KAAM,CAAA,IAAA,CAAKP,MAAMU,qBAAuB,EAAA;MAAED,SAAW,EAAA;KAAK,CAAA;AAC9E;AAEA,EAAA,MAAM3B,IAA8B,GAAA;AAClC,IAAA,MAAM,KAAKsB,IAAI,EAAA;AAEf,IAAA,MAAMvB,QAAQ,MAAMwB,mBAAAA,CAAGC,SAASK,OAAQ,CAAA,IAAA,CAAKX,MAAMQ,eAAiB,EAAA;MAClEI,aAAe,EAAA,IAAA;MACfH,SAAW,EAAA;KACb,CAAA;AACA,IAAOI,OAAAA,OAAAA,CAAQC,IACbjC,KACGU,CAAAA,MAAAA,CAAO,CAACL,IAASA,KAAAA,IAAAA,CAAK6B,QAAY,IAAA,CAAC,KAAKf,KAAME,CAAAA,YAAAA,CAAac,IAAI9B,IAAK+B,CAAAA,IAAI,CAAA,CACxEhC,CAAAA,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACV,MAAA,MAAME,QAAW,GAAA,MAAM,IAAK8B,CAAAA,WAAAA,CAC1BC,qBAAKC,CAAAA,IAAAA,CAAK,IAAKpB,CAAAA,KAAAA,CAAMQ,eAAgBa,CAAAA,QAAAA,EAAYnC,EAAAA,IAAAA,CAAK+B,IAAI,CAAA,CAAA;AAG5D,MAAO,OAAA;QACL9B,EAAIC,EAAAA,QAAAA;AACJC,QAAAA,QAAAA,EAAUH,IAAK+B,CAAAA,IAAAA;AACf7B,QAAAA;AACF,OAAA;AACF,KAAA,CAAA,CAAA;AAEN;AAEA,EAAA,MAAMJ,OAAOH,KAAkD,EAAA;AAC7D,IAAA,MAAM,KAAKuB,IAAI,EAAA;AAEf,IAAMkB,MAAAA,QAAAA,GAAW,MAAM,IAAA,CAAKxC,IAAI,EAAA;AAEhC,IAAA,MAAM+B,OAAQC,CAAAA,GAAAA,CACZjC,KAAMI,CAAAA,GAAAA,CAAI,CAACC,IAAAA,KAAAA;AACT,MAAMqC,MAAAA,cAAAA,GAAiBD,SAASE,IAAK,CAAA,CAACD,oBAAmBA,eAAepC,CAAAA,EAAAA,KAAOD,KAAKC,EAAE,CAAA;AACtF,MAAOsC,OAAAA,iBAAAA,CACLN,sBAAKC,IAAK,CAAA,IAAA,CAAKpB,MAAMQ,eAAgBa,CAAAA,QAAAA,IAAYE,cAAelC,CAAAA,QAAQ,GACxE8B,qBAAKC,CAAAA,IAAAA,CAAK,KAAKpB,KAAMU,CAAAA,qBAAAA,CAAsBW,UAAYE,EAAAA,cAAAA,CAAepC,EAAE,CAAA,CAAA;AAE5E,KAAA,CAAA,CAAA;AAEF,IAAON,OAAAA,KAAAA,CAAMI,GAAI,CAAA,CAACC,IAAU,MAAA;MAAE,GAAGA,IAAAA;AAAME,MAAAA,QAAAA,EAAUF,IAAKC,CAAAA;KAAG,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAMG,SAAST,KAA6B,EAAA;AAC1C,IAAA,MAAM,KAAKuB,IAAI,EAAA;AAEf,IAAA,MAAMS,OAAQC,CAAAA,GAAAA,CACZjC,KAAMI,CAAAA,GAAAA,CAAI,CAACC,IAAAA,KACTuC,iBACEN,CAAAA,qBAAAA,CAAKC,IAAK,CAAA,IAAA,CAAKpB,KAAMU,CAAAA,qBAAAA,CAAsBW,QAAQ,EAAA,EAAInC,IAAKE,CAAAA,QAAQ,CACpE+B,EAAAA,qBAAAA,CAAKC,IAAK,CAAA,IAAA,CAAKpB,KAAMQ,CAAAA,eAAAA,CAAgBa,QAAQ,EAAA,EAAInC,IAAKG,CAAAA,QAAQ,CAAA,CAAA,CAAA,CAAA;AAIpE,IAAOR,OAAAA,KAAAA,CAAMI,GAAI,CAAA,CAACC,IAAU,MAAA;MAAE,GAAGA,IAAAA;AAAMC,MAAAA,EAAAA,EAAID,IAAKE,CAAAA;KAAS,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAgB8B,YAAYhC,IAAgB,EAAA;AAC1C,IAAMwC,MAAAA,IAAAA,GAAOC,iBAAOC,CAAAA,UAAAA,CAAW,QAAA,CAAA;AAC/B,IAAA,MAAMC,mBAASC,CAAAA,mBAAAA,CAAiB5C,IAAAA,CAAAA,EAAOwC,IAAAA,CAAAA;AACvC,IAAOA,OAAAA,IAAAA,CAAKK,OAAO,KAAA,CAAA;AACrB;EAEArC,cAAiB,GAAA;AACf,IAAO,OAAA;MAAEM,KAAOgC,EAAAA,qBAAAA,CAAY,KAAKhC,KAAK;AAAE,KAAA;AAC1C;AAEAL,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"storage.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { PathLike } from \"fs\";\nimport * as crypto from \"node:crypto\";\nimport { pipeline } from \"node:stream/promises\";\nimport fs, { createReadStream } from \"node:fs\";\nimport { copyFile } from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\n\nexport interface PythonFile {\n  id: string;\n  pythonId: string;\n  filename: string;\n}\n\nexport interface PythonUploadFile {\n  id: string;\n  filename: string;\n}\n\nexport interface PythonDownloadFile {\n  id?: string;\n  filename: string;\n  pythonId: string;\n}\n\nexport abstract class PythonStorage extends Serializable {\n  /**\n   * List all files that code interpreter can use.\n   */\n  abstract list(): Promise<PythonFile[]>;\n\n  /**\n   * Prepare subset of available files to code interpreter.\n   */\n  abstract upload(files: PythonUploadFile[]): Promise<PythonFile[]>;\n\n  /**\n   * Process updated/modified/deleted files from code interpreter response.\n   */\n  abstract download(files: PythonDownloadFile[]): Promise<PythonFile[]>;\n}\n\nexport class TemporaryStorage extends PythonStorage {\n  protected files: PythonFile[] = [];\n\n  async list() {\n    return this.files.slice();\n  }\n\n  async upload(files: PythonUploadFile[]): Promise<PythonFile[]> {\n    return files.map((file) => ({\n      id: file.id,\n      pythonId: file.id,\n      filename: file.filename,\n    }));\n  }\n\n  async download(files: PythonDownloadFile[]) {\n    this.files = [\n      ...this.files.filter((file) => files.every((f) => f.filename !== file.filename)),\n      ...files.map((file) => ({ id: file.pythonId, ...file })),\n    ];\n    return this.files.slice();\n  }\n\n  createSnapshot() {\n    return { files: this.files.slice() };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n\ninterface Input {\n  localWorkingDir: PathLike;\n  interpreterWorkingDir: PathLike;\n  ignoredFiles?: Set<string>;\n}\n\nexport class LocalPythonStorage extends PythonStorage {\n  protected readonly input: Required<Input>;\n\n  constructor(input: Input) {\n    super();\n    this.input = {\n      ...input,\n      ignoredFiles: input?.ignoredFiles ?? new Set([\".gitkeep\"]),\n    };\n  }\n\n  @Cache()\n  protected async init() {\n    await fs.promises.mkdir(this.input.localWorkingDir, { recursive: true });\n    await fs.promises.mkdir(this.input.interpreterWorkingDir, { recursive: true });\n  }\n\n  async list(): Promise<PythonFile[]> {\n    await this.init();\n\n    const files = await fs.promises.readdir(this.input.localWorkingDir, {\n      withFileTypes: true,\n      recursive: false,\n    });\n    return Promise.all(\n      files\n        .filter((file) => file.isFile() && !this.input.ignoredFiles.has(file.name))\n        .map(async (file) => {\n          const pythonId = await this.computeHash(\n            path.join(this.input.localWorkingDir.toString(), file.name),\n          );\n\n          return {\n            id: pythonId,\n            filename: file.name,\n            pythonId,\n          };\n        }),\n    );\n  }\n\n  async upload(files: PythonUploadFile[]): Promise<PythonFile[]> {\n    await this.init();\n\n    const fileList = await this.list();\n\n    await Promise.all(\n      files.map((file) => {\n        const filesystemFile = fileList.find((filesystemFile) => filesystemFile.id === file.id)!;\n        return copyFile(\n          path.join(this.input.localWorkingDir.toString(), filesystemFile.filename),\n          path.join(this.input.interpreterWorkingDir.toString(), filesystemFile.id),\n        );\n      }),\n    );\n    return files.map((file) => ({ ...file, pythonId: file.id }));\n  }\n\n  async download(files: PythonDownloadFile[]) {\n    await this.init();\n\n    await Promise.all(\n      files.map((file) =>\n        copyFile(\n          path.join(this.input.interpreterWorkingDir.toString(), file.pythonId),\n          path.join(this.input.localWorkingDir.toString(), file.filename),\n        ),\n      ),\n    );\n    return files.map((file) => ({ ...file, id: file.pythonId }));\n  }\n\n  protected async computeHash(file: PathLike) {\n    const hash = crypto.createHash(\"sha256\");\n    await pipeline(createReadStream(file), hash);\n    return hash.digest(\"hex\");\n  }\n\n  createSnapshot() {\n    return { input: shallowCopy(this.input) };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}