beeai-framework 0.1.15 → 0.1.17

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

@@ -1 +1 @@
-{"version":3,"sources":["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,mIACA,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEJ,MAAA;AACL,MAAM,MAAA,IAAIE,mBAAU,yCAA2C,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAM5C,IAAAA,QAAAA,CAAS6C,SAAS,GAAK,EAAA;AAC1C,IAAA,MAAM,IAAIF,kBAAAA,CACR,CAAgE3C,6DAAAA,EAAAA,QAAAA,CAAS6C,MAAM,CAC/E,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 © BeeAI a Series of LF Projects, LLC\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 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 code interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to 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"]}
+{"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":";;;;;;;;;;;;;;AAyCO,MAAMA,mBAAmBC,aAAAA,CAAAA;EAzChC;;;EA0CEC,IAAAA,GAAO,QAAA;EACPC,WAAAA,GAAc;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,OAAAA,GAA0DC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,MAAMC,WAAAA,GAAc;AAClB,IAAA,IAAA,CAAKP,KAAAA,GAAQ,MAAM,IAAA,CAAKD,OAAAA,CAAQS,IAAAA,EAAI;AACpC,IAAA,MAAMC,SAAAA,GAAYC,aAAAA,CAAOC,UAAAA,CAAI,IAAA,CAAKX,KAAAA,EAAO,CAAC,EAAEY,QAAAA,EAAQ,KAAOA,QAAAA,CAAAA,CAAAA;AAC3D,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,QAAAA,EAAUF,MAAEG,IAAAA,CAAK;AAAC,QAAA,QAAA;AAAU,QAAA;AAAQ,OAAA,CAAA,CAAEC,SAAS,sCAAA,CAAA;AAC/CC,MAAAA,IAAAA,EAAML,KAAAA,CAAEM,MAAAA,EAAM,CAAGF,QAAAA,CAAS,6CAAA,CAAA;MAC1B,GAAIG,sBAAAA,CAAaX,SAAAA,EAAW,CAAA,CAAA,GACxB;QACEY,UAAAA,EAAYR,KAAAA,CACTS,MAAMT,KAAAA,CAAEG,IAAAA,CAAKP,SAAAA,CAAAA,CAAAA,CACbQ,SACC,8KAAA;AAEN,OAAA,GACA;KACN,CAAA;AACF;AAEmBM,EAAAA,UAAAA;AAEnB,EAAA,WAAA,CAAmBC,OAAAA,EAA4B;AAC7C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,eAAAA,CAAgBC,GAAAA,EAAK;AAChC,MAAA,MAAM,IAAIC,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,+BAAA;UACTC,IAAAA,EAAML,OAAAA;UACNM,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAAA,CAAQD,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAAA,CAAQzB,OAAAA;AACzB;EAEA;AACE,IAAA,IAAA,CAAKgC,QAAAA,EAAQ;AACf;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMd,UAAAA,GAAa,MAAMe,WAAAA,CACvB,IAAA,CAAKpC,SAAU,MAAM,IAAA,CAAKD,OAAAA,CAAQS,IAAAA,IAClC6B,eAAAA,CAAS,CAACC,CAAAA,KAAMA,CAAAA,CAAE1B,QAAQ,CAAA,EAC1B2B,aAAAA,CAAO,CAACC,IAAAA,KAASC,mBAAAA,CAAaD,IAAAA,CAAK5B,QAAAA,EAAWqB,KAAAA,CAAMZ,cAAc,EAAE,CAAA,CAAA,EACpE,CAACrB,KAAAA,KAAU,IAAA,CAAKD,OAAAA,CAAQ2C,MAAAA,CAAO1C,KAAAA,CAAAA,CAAAA;AAGjC,IAAA,MAAM2C,gCAAgB,MAAA,CAAA,YAAA;AACpB,MAAA,IAAI,KAAKpB,UAAAA,EAAY;AACnB,QAAA,MAAM,EAAEqB,GAAAA,EAAKC,cAAAA,EAAc,GAAK,IAAA,CAAKtB,UAAAA;AACrC,QAAA,MAAMuB,QAAAA,GAAW,MAAMF,GAAAA,CAAIG,MAAAA,CAAO;UAChCC,QAAAA,EAAU;YAAC,IAAIC,uBAAAA,CAAYJ,eAAeK,MAAAA,CAAO;AAAEjB,cAAAA,KAAAA,EAAOA,KAAAA,CAAMf;AAAK,aAAA,CAAA;;AACrEiC,UAAAA,WAAAA,EAAahB,GAAAA,CAAIiB;SACnB,CAAA;AACA,QAAA,OAAON,QAAAA,CAASO,cAAAA,EAAc,CAAGC,IAAAA,EAAI;AACvC;AACA,MAAA,OAAOrB,KAAAA,CAAMf,IAAAA;KACf,EAVsB,eAAA,CAAA;AAYtB,IAAA,MAAMqC,MAAAA,GAAS,aAAA;AAEf,IAAA,MAAMC,MAAAA,GAAS,MAAMC,mBAAAA,CAAoB;AACvC/B,MAAAA,GAAAA,EAAK,CAAA,EAAG,IAAA,CAAKF,OAAAA,CAAQC,eAAAA,CAAgBC,GAAG,CAAA,WAAA,CAAA;MACxCgC,IAAAA,EAAM;AACJC,QAAAA,WAAAA,EAAa,MAAMhB,aAAAA,EAAAA;AACnB3C,QAAAA,KAAAA,EAAO4D,MAAAA,CAAOC,WAAAA,CACZxC,UAAAA,CAAWV,GAAAA,CAAI,CAAC6B,IAAAA,KAAS;UAAC,CAAA,EAAGe,MAAAA,CAAAA,EAASf,IAAAA,CAAK5B,QAAQ,CAAA,CAAA;UAAI4B,IAAAA,CAAKsB;SAAS,CAAA;AAEzE,OAAA;AACAV,MAAAA,MAAAA,EAAQjB,GAAAA,CAAIiB;KACd,CAAA;AAEA,IAAA,MAAMW,WAAAA,GAAc,MAAM,IAAA,CAAKhE,OAAAA,CAAQiE,QAAAA,CACrCJ,MAAAA,CAAOK,OAAAA,CAAQT,MAAAA,CAAOxD,KAAK,CAAA,CACxBuC,MAAAA,CAAO,CAAC,CAAC2B,MAAMC,CAAAA,CAAAA,KAAOD,IAAAA,CAAKE,UAAAA,CAAWb,MAAAA,CAAAA,CAAAA,CACtC5C,GAAAA,CAAI,CAAC,CAACuD,IAAAA,EAAMJ,QAAAA,CAAAA,MAAe;AAAEI,MAAAA,IAAAA;AAAYJ,MAAAA,QAAAA,EAAUO,OAAOP,QAAAA;KAAU,CAAA,CAAA,CACpEnD,GAAAA,CAAI,CAAC6B,IAAAA,MAAU;AAAE5B,MAAAA,QAAAA,EAAU4B,IAAAA,CAAK0B,IAAAA,CAAKI,KAAAA,CAAMf,MAAAA,CAAOgB,MAAM,CAAA;AAAGT,MAAAA,QAAAA,EAAUtB,IAAAA,CAAKsB;AAAS,KAAA,CAAA,EACnFvB,MAAAA,CAAO,CAACC,IAAAA,KACPnB,UAAAA,CAAWmD,MACT,CAACvC,MAAAA,KAAUA,MAAAA,CAAMrB,QAAAA,KAAa4B,KAAK5B,QAAAA,IAAYqB,MAAAA,CAAM6B,aAAatB,IAAAA,CAAKsB,QAAQ,CAAA,CAAA,CAAA;AAKvF,IAAA,OAAO,IAAIW,4BAAiBjB,MAAAA,CAAOkB,MAAAA,EAAQlB,OAAOmB,MAAAA,EAAQnB,MAAAA,CAAOoB,WAAWb,WAAAA,CAAAA;AAC9E;EAEAc,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7E,KAAAA,EAAO8E,qBAAAA,CAAY,KAAK9E,KAAK,CAAA;AAC7BD,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdwB,MAAAA,UAAAA,EAAY,IAAA,CAAKA;AACnB,KAAA;AACF;AAEAwD,EAAAA,YAAAA,CAAaC,QAAAA,EAAwD;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB;AACF;AAEA,eAAsBvB,mBAAAA,CAAoB,EACxC/B,GAAAA,EACAgC,IAAAA,EACAN,QAAM,EAKP;AACC,EAAA,MAAMN,QAAAA,GAAW,MAAMmC,KAAAA,CAAMvD,GAAAA,EAAK;IAChCwD,MAAAA,EAAQ,MAAA;IACRC,OAAAA,EAAS;MACP,QAAA,EAAU,kBAAA;MACV,cAAA,EAAgB;AAClB,KAAA;IACAzB,IAAAA,EAAM0B,IAAAA,CAAKC,UAAU3B,IAAAA,CAAAA;AACrBN,IAAAA;GACF,CAAA,CAAGkC,KAAAA,CAAM,CAACC,KAAAA,KAAAA;AACR,IAAA,IAAIA,KAAAA,CAAMC,KAAAA,CAAM5F,IAAAA,IAAQ,iBAAA,EAAmB;AACzC,MAAA,MAAM,IAAI6F,mBACR,mIAAA,EACA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEX,MAAO;AACL,MAAA,MAAM,IAAIE,mBAAU,yCAAA,EAA2C;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAAA,IAAM5C,QAAAA,CAAS6C,SAAS,GAAA,EAAK;AAC1C,IAAA,MAAM,IAAIF,kBAAAA,CACR,CAAA,6DAAA,EAAgE3C,QAAAA,CAAS6C,MAAM,CAAA,CAAA,CAAA,EAC/E;AAAC,MAAA,IAAIC,KAAAA,CAAM,MAAM9C,QAAAA,CAAS+C,IAAAA,EAAI;AAAI,KAAA,CAAA;AAEtC;AAEA,EAAA,OAAO,MAAM/C,SAASgD,IAAAA,EAAI;AAC5B;AApCsBrC,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"python.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\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 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 code interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to 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

@@ -7,8 +7,8 @@ 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-D5Mu0EEH.cjs';
-import { m as ChatModel } from '../../chat-DvMzAHj6.cjs';
+import { E as Emitter } from '../../emitter-DWEsd-ud.cjs';
+import { m as ChatModel } from '../../chat-CR4yfOR9.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';

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

@@ -7,8 +7,8 @@ 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-C3dO-s2P.js';
-import { m as ChatModel } from '../../chat-BdN8hGWn.js';
+import { E as Emitter } from '../../emitter-C9nwGLt6.js';
+import { m as ChatModel } from '../../chat-W9j8T6UU.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';

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

@@ -1 +1 @@
-{"version":3,"sources":["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,mIACA,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEJ,MAAA;AACL,MAAM,MAAA,IAAIE,UAAU,yCAA2C,EAAA;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAM5C,IAAAA,QAAAA,CAAS6C,SAAS,GAAK,EAAA;AAC1C,IAAA,MAAM,IAAIF,SAAAA,CACR,CAAgE3C,6DAAAA,EAAAA,QAAAA,CAAS6C,MAAM,CAC/E,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 © BeeAI a Series of LF Projects, LLC\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 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 code interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to 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"]}
+{"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":";;;;;;;;;;;;AAyCO,MAAMA,mBAAmBC,IAAAA,CAAAA;EAzChC;;;EA0CEC,IAAAA,GAAO,QAAA;EACPC,WAAAA,GAAc;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,OAAAA,GAA0DC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,MAAMC,WAAAA,GAAc;AAClB,IAAA,IAAA,CAAKP,KAAAA,GAAQ,MAAM,IAAA,CAAKD,OAAAA,CAAQS,IAAAA,EAAI;AACpC,IAAA,MAAMC,SAAAA,GAAYC,MAAAA,CAAOC,GAAAA,CAAI,IAAA,CAAKX,KAAAA,EAAO,CAAC,EAAEY,QAAAA,EAAQ,KAAOA,QAAAA,CAAAA,CAAAA;AAC3D,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,QAAAA,EAAUF,EAAEG,IAAAA,CAAK;AAAC,QAAA,QAAA;AAAU,QAAA;AAAQ,OAAA,CAAA,CAAEC,SAAS,sCAAA,CAAA;AAC/CC,MAAAA,IAAAA,EAAML,CAAAA,CAAEM,MAAAA,EAAM,CAAGF,QAAAA,CAAS,6CAAA,CAAA;MAC1B,GAAIG,YAAAA,CAAaX,SAAAA,EAAW,CAAA,CAAA,GACxB;QACEY,UAAAA,EAAYR,CAAAA,CACTS,MAAMT,CAAAA,CAAEG,IAAAA,CAAKP,SAAAA,CAAAA,CAAAA,CACbQ,SACC,8KAAA;AAEN,OAAA,GACA;KACN,CAAA;AACF;AAEmBM,EAAAA,UAAAA;AAEnB,EAAA,WAAA,CAAmBC,OAAAA,EAA4B;AAC7C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,eAAAA,CAAgBC,GAAAA,EAAK;AAChC,MAAA,MAAM,IAAIC,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,+BAAA;UACTC,IAAAA,EAAML,OAAAA;UACNM,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAAA,CAAQD,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAAA,CAAQzB,OAAAA;AACzB;EAEA;AACE,IAAA,IAAA,CAAKgC,QAAAA,EAAQ;AACf;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMd,UAAAA,GAAa,MAAMe,IAAAA,CACvB,IAAA,CAAKpC,SAAU,MAAM,IAAA,CAAKD,OAAAA,CAAQS,IAAAA,IAClC6B,QAAAA,CAAS,CAACC,CAAAA,KAAMA,CAAAA,CAAE1B,QAAQ,CAAA,EAC1B2B,MAAAA,CAAO,CAACC,IAAAA,KAASC,YAAAA,CAAaD,IAAAA,CAAK5B,QAAAA,EAAWqB,KAAAA,CAAMZ,cAAc,EAAE,CAAA,CAAA,EACpE,CAACrB,KAAAA,KAAU,IAAA,CAAKD,OAAAA,CAAQ2C,MAAAA,CAAO1C,KAAAA,CAAAA,CAAAA;AAGjC,IAAA,MAAM2C,gCAAgB,MAAA,CAAA,YAAA;AACpB,MAAA,IAAI,KAAKpB,UAAAA,EAAY;AACnB,QAAA,MAAM,EAAEqB,GAAAA,EAAKC,cAAAA,EAAc,GAAK,IAAA,CAAKtB,UAAAA;AACrC,QAAA,MAAMuB,QAAAA,GAAW,MAAMF,GAAAA,CAAIG,MAAAA,CAAO;UAChCC,QAAAA,EAAU;YAAC,IAAIC,WAAAA,CAAYJ,eAAeK,MAAAA,CAAO;AAAEjB,cAAAA,KAAAA,EAAOA,KAAAA,CAAMf;AAAK,aAAA,CAAA;;AACrEiC,UAAAA,WAAAA,EAAahB,GAAAA,CAAIiB;SACnB,CAAA;AACA,QAAA,OAAON,QAAAA,CAASO,cAAAA,EAAc,CAAGC,IAAAA,EAAI;AACvC;AACA,MAAA,OAAOrB,KAAAA,CAAMf,IAAAA;KACf,EAVsB,eAAA,CAAA;AAYtB,IAAA,MAAMqC,MAAAA,GAAS,aAAA;AAEf,IAAA,MAAMC,MAAAA,GAAS,MAAMC,mBAAAA,CAAoB;AACvC/B,MAAAA,GAAAA,EAAK,CAAA,EAAG,IAAA,CAAKF,OAAAA,CAAQC,eAAAA,CAAgBC,GAAG,CAAA,WAAA,CAAA;MACxCgC,IAAAA,EAAM;AACJC,QAAAA,WAAAA,EAAa,MAAMhB,aAAAA,EAAAA;AACnB3C,QAAAA,KAAAA,EAAO4D,MAAAA,CAAOC,WAAAA,CACZxC,UAAAA,CAAWV,GAAAA,CAAI,CAAC6B,IAAAA,KAAS;UAAC,CAAA,EAAGe,MAAAA,CAAAA,EAASf,IAAAA,CAAK5B,QAAQ,CAAA,CAAA;UAAI4B,IAAAA,CAAKsB;SAAS,CAAA;AAEzE,OAAA;AACAV,MAAAA,MAAAA,EAAQjB,GAAAA,CAAIiB;KACd,CAAA;AAEA,IAAA,MAAMW,WAAAA,GAAc,MAAM,IAAA,CAAKhE,OAAAA,CAAQiE,QAAAA,CACrCJ,MAAAA,CAAOK,OAAAA,CAAQT,MAAAA,CAAOxD,KAAK,CAAA,CACxBuC,MAAAA,CAAO,CAAC,CAAC2B,MAAMC,CAAAA,CAAAA,KAAOD,IAAAA,CAAKE,UAAAA,CAAWb,MAAAA,CAAAA,CAAAA,CACtC5C,GAAAA,CAAI,CAAC,CAACuD,IAAAA,EAAMJ,QAAAA,CAAAA,MAAe;AAAEI,MAAAA,IAAAA;AAAYJ,MAAAA,QAAAA,EAAUO,OAAOP,QAAAA;KAAU,CAAA,CAAA,CACpEnD,GAAAA,CAAI,CAAC6B,IAAAA,MAAU;AAAE5B,MAAAA,QAAAA,EAAU4B,IAAAA,CAAK0B,IAAAA,CAAKI,KAAAA,CAAMf,MAAAA,CAAOgB,MAAM,CAAA;AAAGT,MAAAA,QAAAA,EAAUtB,IAAAA,CAAKsB;AAAS,KAAA,CAAA,EACnFvB,MAAAA,CAAO,CAACC,IAAAA,KACPnB,UAAAA,CAAWmD,MACT,CAACvC,MAAAA,KAAUA,MAAAA,CAAMrB,QAAAA,KAAa4B,KAAK5B,QAAAA,IAAYqB,MAAAA,CAAM6B,aAAatB,IAAAA,CAAKsB,QAAQ,CAAA,CAAA,CAAA;AAKvF,IAAA,OAAO,IAAIW,iBAAiBjB,MAAAA,CAAOkB,MAAAA,EAAQlB,OAAOmB,MAAAA,EAAQnB,MAAAA,CAAOoB,WAAWb,WAAAA,CAAAA;AAC9E;EAEAc,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACT7E,KAAAA,EAAO8E,WAAAA,CAAY,KAAK9E,KAAK,CAAA;AAC7BD,MAAAA,OAAAA,EAAS,IAAA,CAAKA,OAAAA;AACdwB,MAAAA,UAAAA,EAAY,IAAA,CAAKA;AACnB,KAAA;AACF;AAEAwD,EAAAA,YAAAA,CAAaC,QAAAA,EAAwD;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB;AACF;AAEA,eAAsBvB,mBAAAA,CAAoB,EACxC/B,GAAAA,EACAgC,IAAAA,EACAN,QAAM,EAKP;AACC,EAAA,MAAMN,QAAAA,GAAW,MAAMmC,KAAAA,CAAMvD,GAAAA,EAAK;IAChCwD,MAAAA,EAAQ,MAAA;IACRC,OAAAA,EAAS;MACP,QAAA,EAAU,kBAAA;MACV,cAAA,EAAgB;AAClB,KAAA;IACAzB,IAAAA,EAAM0B,IAAAA,CAAKC,UAAU3B,IAAAA,CAAAA;AACrBN,IAAAA;GACF,CAAA,CAAGkC,KAAAA,CAAM,CAACC,KAAAA,KAAAA;AACR,IAAA,IAAIA,KAAAA,CAAMC,KAAAA,CAAM5F,IAAAA,IAAQ,iBAAA,EAAmB;AACzC,MAAA,MAAM,IAAI6F,UACR,mIAAA,EACA;AAACF,QAAAA;AAAM,OAAA,CAAA;KAEX,MAAO;AACL,MAAA,MAAM,IAAIE,UAAU,yCAAA,EAA2C;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE;GACF,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU4C,EAAAA,IAAM5C,QAAAA,CAAS6C,SAAS,GAAA,EAAK;AAC1C,IAAA,MAAM,IAAIF,SAAAA,CACR,CAAA,6DAAA,EAAgE3C,QAAAA,CAAS6C,MAAM,CAAA,CAAA,CAAA,EAC/E;AAAC,MAAA,IAAIC,KAAAA,CAAM,MAAM9C,QAAAA,CAAS+C,IAAAA,EAAI;AAAI,KAAA,CAAA;AAEtC;AAEA,EAAA,OAAO,MAAM/C,SAASgD,IAAAA,EAAI;AAC5B;AApCsBrC,MAAAA,CAAAA,mBAAAA,EAAAA,qBAAAA,CAAAA","file":"python.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\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 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 code interpreter has failed.\", [error]);\n    }\n  });\n\n  if (!response?.ok && response.status > 400) {\n    throw new ToolError(\n      `Request to 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.map

@@ -1 +1 @@
-{"version":3,"sources":["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 © BeeAI a Series of LF Projects, LLC\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"]}
+{"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","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":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgCO,MAAeA,sBAAsBC,6BAAAA,CAAAA;AAAAA,EAAAA;;;AAe5C;AAEO,MAAMC,yBAAyBF,aAAAA,CAAAA;AAAAA,EAAAA;;;AAC1BG,EAAAA,KAAAA,GAAsB,EAAA;AAEhC,EAAA,MAAMC,IAAAA,GAAO;AACX,IAAA,OAAO,IAAA,CAAKD,MAAME,KAAAA,EAAK;AACzB;AAEA,EAAA,MAAMC,OAAOH,KAAAA,EAAkD;AAC7D,IAAA,OAAOA,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;AAC1BC,MAAAA,EAAAA,EAAID,IAAAA,CAAKC,EAAAA;AACTC,MAAAA,QAAAA,EAAUF,IAAAA,CAAKC,EAAAA;AACfE,MAAAA,QAAAA,EAAUH,IAAAA,CAAKG;KACjB,CAAA,CAAA;AACF;AAEA,EAAA,MAAMC,SAAST,KAAAA,EAA6B;AAC1C,IAAA,IAAA,CAAKA,KAAAA,GAAQ;AACR,MAAA,GAAA,IAAA,CAAKA,KAAAA,CAAMU,MAAAA,CAAO,CAACL,IAAAA,KAASL,KAAAA,CAAMW,KAAAA,CAAM,CAACC,CAAAA,KAAMA,CAAAA,CAAEJ,QAAAA,KAAaH,IAAAA,CAAKG,QAAQ,CAAA,CAAA;SAC3ER,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;AAAEC,QAAAA,EAAAA,EAAID,IAAAA,CAAKE,QAAAA;QAAU,GAAGF;OAAK,CAAA;;AAEvD,IAAA,OAAO,IAAA,CAAKL,MAAME,KAAAA,EAAK;AACzB;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEb,KAAAA,EAAO,IAAA,CAAKA,MAAME,KAAAA;AAAQ,KAAA;AACrC;AAEAY,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF;AAQO,MAAMG,2BAA2BrB,aAAAA,CAAAA;AAAAA,EAAAA;;;AACnBsB,EAAAA,KAAAA;AAEnB,EAAA,WAAA,CAAYA,KAAAA,EAAc;AACxB,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHC,YAAAA,EAAcD,KAAAA,EAAOC,YAAAA,oBAAgB,IAAIC,GAAAA,CAAI;AAAC,QAAA;AAAW,OAAA;AAC3D,KAAA;AACF;AAEA,EAAA,MACgBC,IAAAA,GAAO;AACrB,IAAA,MAAMC,mBAAAA,CAAGC,QAAAA,CAASC,KAAAA,CAAM,IAAA,CAAKN,MAAMO,eAAAA,EAAiB;MAAEC,SAAAA,EAAW;KAAK,CAAA;AACtE,IAAA,MAAMJ,mBAAAA,CAAGC,QAAAA,CAASC,KAAAA,CAAM,IAAA,CAAKN,MAAMS,qBAAAA,EAAuB;MAAED,SAAAA,EAAW;KAAK,CAAA;AAC9E;AAEA,EAAA,MAAM1B,IAAAA,GAA8B;AAClC,IAAA,MAAM,KAAKqB,IAAAA,EAAI;AAEf,IAAA,MAAMtB,QAAQ,MAAMuB,mBAAAA,CAAGC,SAASK,OAAAA,CAAQ,IAAA,CAAKV,MAAMO,eAAAA,EAAiB;MAClEI,aAAAA,EAAe,IAAA;MACfH,SAAAA,EAAW;KACb,CAAA;AACA,IAAA,OAAOI,OAAAA,CAAQC,IACbhC,KAAAA,CACGU,MAAAA,CAAO,CAACL,IAAAA,KAASA,IAAAA,CAAK4B,QAAM,IAAM,CAAC,KAAKd,KAAAA,CAAMC,YAAAA,CAAac,IAAI7B,IAAAA,CAAK8B,IAAI,CAAA,CAAA,CACxE/B,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACV,MAAA,MAAME,QAAAA,GAAW,MAAM,IAAA,CAAK6B,WAAAA,CAC1BC,qBAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMO,eAAAA,CAAgBa,QAAAA,EAAQ,EAAIlC,IAAAA,CAAK8B,IAAI,CAAA,CAAA;AAG5D,MAAA,OAAO;QACL7B,EAAAA,EAAIC,QAAAA;AACJC,QAAAA,QAAAA,EAAUH,IAAAA,CAAK8B,IAAAA;AACf5B,QAAAA;AACF,OAAA;AACF,KAAA,CAAA,CAAA;AAEN;AAEA,EAAA,MAAMJ,OAAOH,KAAAA,EAAkD;AAC7D,IAAA,MAAM,KAAKsB,IAAAA,EAAI;AAEf,IAAA,MAAMkB,QAAAA,GAAW,MAAM,IAAA,CAAKvC,IAAAA,EAAI;AAEhC,IAAA,MAAM8B,OAAAA,CAAQC,GAAAA,CACZhC,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,KAAAA;AACT,MAAA,MAAMoC,cAAAA,GAAiBD,SAASE,IAAAA,CAAK,CAACD,oBAAmBA,eAAAA,CAAenC,EAAAA,KAAOD,KAAKC,EAAE,CAAA;AACtF,MAAA,OAAOqC,iBAAAA,CACLN,sBAAKC,IAAAA,CAAK,IAAA,CAAKnB,MAAMO,eAAAA,CAAgBa,QAAAA,IAAYE,cAAAA,CAAejC,QAAQ,GACxE6B,qBAAAA,CAAKC,IAAAA,CAAK,KAAKnB,KAAAA,CAAMS,qBAAAA,CAAsBW,UAAQ,EAAIE,cAAAA,CAAenC,EAAE,CAAA,CAAA;AAE5E,KAAA,CAAA,CAAA;AAEF,IAAA,OAAON,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;MAAE,GAAGA,IAAAA;AAAME,MAAAA,QAAAA,EAAUF,IAAAA,CAAKC;KAAG,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAMG,SAAST,KAAAA,EAA6B;AAC1C,IAAA,MAAM,KAAKsB,IAAAA,EAAI;AAEf,IAAA,MAAMS,OAAAA,CAAQC,GAAAA,CACZhC,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,KACTsC,iBAAAA,CACEN,qBAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMS,qBAAAA,CAAsBW,QAAAA,EAAQ,EAAIlC,IAAAA,CAAKE,QAAQ,CAAA,EACpE8B,qBAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMO,eAAAA,CAAgBa,QAAAA,EAAQ,EAAIlC,IAAAA,CAAKG,QAAQ,CAAA,CAAA,CAAA,CAAA;AAIpE,IAAA,OAAOR,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;MAAE,GAAGA,IAAAA;AAAMC,MAAAA,EAAAA,EAAID,IAAAA,CAAKE;KAAS,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAgB6B,YAAY/B,IAAAA,EAAgB;AAC1C,IAAA,MAAMuC,IAAAA,GAAOC,iBAAAA,CAAOC,UAAAA,CAAW,QAAA,CAAA;AAC/B,IAAA,MAAMC,mBAAAA,CAASC,mBAAAA,CAAiB3C,IAAAA,CAAAA,EAAOuC,IAAAA,CAAAA;AACvC,IAAA,OAAOA,IAAAA,CAAKK,OAAO,KAAA,CAAA;AACrB;EAEApC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEM,KAAAA,EAAO+B,qBAAAA,CAAY,KAAK/B,KAAK;AAAE,KAAA;AAC1C;AAEAL,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"storage.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\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"]}

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

@@ -5,18 +5,7 @@ import '../../internals/helpers/guards.cjs';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface PythonFile {

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

@@ -5,18 +5,7 @@ import '../../internals/helpers/guards.js';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface PythonFile {

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

@@ -1 +1 @@
-{"version":3,"sources":["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,YAAAA,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,EAAGC,CAAAA,QAAAA,CAASC,KAAM,CAAA,IAAA,CAAKP,MAAMQ,eAAiB,EAAA;MAAEC,SAAW,EAAA;KAAK,CAAA;AACtE,IAAA,MAAMJ,EAAGC,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,EAAAA,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,IAAKC,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,QAAAA,CACLN,KAAKC,IAAK,CAAA,IAAA,CAAKpB,MAAMQ,eAAgBa,CAAAA,QAAAA,IAAYE,cAAelC,CAAAA,QAAQ,GACxE8B,IAAKC,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,QACEN,CAAAA,IAAAA,CAAKC,IAAK,CAAA,IAAA,CAAKpB,KAAMU,CAAAA,qBAAAA,CAAsBW,QAAQ,EAAA,EAAInC,IAAKE,CAAAA,QAAQ,CACpE+B,EAAAA,IAAAA,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,MAAOC,CAAAA,UAAAA,CAAW,QAAA,CAAA;AAC/B,IAAA,MAAMC,QAASC,CAAAA,gBAAAA,CAAiB5C,IAAAA,CAAAA,EAAOwC,IAAAA,CAAAA;AACvC,IAAOA,OAAAA,IAAAA,CAAKK,OAAO,KAAA,CAAA;AACrB;EAEArC,cAAiB,GAAA;AACf,IAAO,OAAA;MAAEM,KAAOgC,EAAAA,WAAAA,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.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\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"]}
+{"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","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":";;;;;;;;;;;;;;;;;;;;;;AAgCO,MAAeA,sBAAsBC,YAAAA,CAAAA;AAAAA,EAAAA;;;AAe5C;AAEO,MAAMC,yBAAyBF,aAAAA,CAAAA;AAAAA,EAAAA;;;AAC1BG,EAAAA,KAAAA,GAAsB,EAAA;AAEhC,EAAA,MAAMC,IAAAA,GAAO;AACX,IAAA,OAAO,IAAA,CAAKD,MAAME,KAAAA,EAAK;AACzB;AAEA,EAAA,MAAMC,OAAOH,KAAAA,EAAkD;AAC7D,IAAA,OAAOA,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;AAC1BC,MAAAA,EAAAA,EAAID,IAAAA,CAAKC,EAAAA;AACTC,MAAAA,QAAAA,EAAUF,IAAAA,CAAKC,EAAAA;AACfE,MAAAA,QAAAA,EAAUH,IAAAA,CAAKG;KACjB,CAAA,CAAA;AACF;AAEA,EAAA,MAAMC,SAAST,KAAAA,EAA6B;AAC1C,IAAA,IAAA,CAAKA,KAAAA,GAAQ;AACR,MAAA,GAAA,IAAA,CAAKA,KAAAA,CAAMU,MAAAA,CAAO,CAACL,IAAAA,KAASL,KAAAA,CAAMW,KAAAA,CAAM,CAACC,CAAAA,KAAMA,CAAAA,CAAEJ,QAAAA,KAAaH,IAAAA,CAAKG,QAAQ,CAAA,CAAA;SAC3ER,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;AAAEC,QAAAA,EAAAA,EAAID,IAAAA,CAAKE,QAAAA;QAAU,GAAGF;OAAK,CAAA;;AAEvD,IAAA,OAAO,IAAA,CAAKL,MAAME,KAAAA,EAAK;AACzB;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEb,KAAAA,EAAO,IAAA,CAAKA,MAAME,KAAAA;AAAQ,KAAA;AACrC;AAEAY,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF;AAQO,MAAMG,2BAA2BrB,aAAAA,CAAAA;AAAAA,EAAAA;;;AACnBsB,EAAAA,KAAAA;AAEnB,EAAA,WAAA,CAAYA,KAAAA,EAAc;AACxB,IAAA,KAAA,EAAK;AACL,IAAA,IAAA,CAAKA,KAAAA,GAAQ;MACX,GAAGA,KAAAA;MACHC,YAAAA,EAAcD,KAAAA,EAAOC,YAAAA,oBAAgB,IAAIC,GAAAA,CAAI;AAAC,QAAA;AAAW,OAAA;AAC3D,KAAA;AACF;AAEA,EAAA,MACgBC,IAAAA,GAAO;AACrB,IAAA,MAAMC,EAAAA,CAAGC,QAAAA,CAASC,KAAAA,CAAM,IAAA,CAAKN,MAAMO,eAAAA,EAAiB;MAAEC,SAAAA,EAAW;KAAK,CAAA;AACtE,IAAA,MAAMJ,EAAAA,CAAGC,QAAAA,CAASC,KAAAA,CAAM,IAAA,CAAKN,MAAMS,qBAAAA,EAAuB;MAAED,SAAAA,EAAW;KAAK,CAAA;AAC9E;AAEA,EAAA,MAAM1B,IAAAA,GAA8B;AAClC,IAAA,MAAM,KAAKqB,IAAAA,EAAI;AAEf,IAAA,MAAMtB,QAAQ,MAAMuB,EAAAA,CAAGC,SAASK,OAAAA,CAAQ,IAAA,CAAKV,MAAMO,eAAAA,EAAiB;MAClEI,aAAAA,EAAe,IAAA;MACfH,SAAAA,EAAW;KACb,CAAA;AACA,IAAA,OAAOI,OAAAA,CAAQC,IACbhC,KAAAA,CACGU,MAAAA,CAAO,CAACL,IAAAA,KAASA,IAAAA,CAAK4B,QAAM,IAAM,CAAC,KAAKd,KAAAA,CAAMC,YAAAA,CAAac,IAAI7B,IAAAA,CAAK8B,IAAI,CAAA,CAAA,CACxE/B,GAAAA,CAAI,OAAOC,IAAAA,KAAAA;AACV,MAAA,MAAME,QAAAA,GAAW,MAAM,IAAA,CAAK6B,WAAAA,CAC1BC,IAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMO,eAAAA,CAAgBa,QAAAA,EAAQ,EAAIlC,IAAAA,CAAK8B,IAAI,CAAA,CAAA;AAG5D,MAAA,OAAO;QACL7B,EAAAA,EAAIC,QAAAA;AACJC,QAAAA,QAAAA,EAAUH,IAAAA,CAAK8B,IAAAA;AACf5B,QAAAA;AACF,OAAA;AACF,KAAA,CAAA,CAAA;AAEN;AAEA,EAAA,MAAMJ,OAAOH,KAAAA,EAAkD;AAC7D,IAAA,MAAM,KAAKsB,IAAAA,EAAI;AAEf,IAAA,MAAMkB,QAAAA,GAAW,MAAM,IAAA,CAAKvC,IAAAA,EAAI;AAEhC,IAAA,MAAM8B,OAAAA,CAAQC,GAAAA,CACZhC,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,KAAAA;AACT,MAAA,MAAMoC,cAAAA,GAAiBD,SAASE,IAAAA,CAAK,CAACD,oBAAmBA,eAAAA,CAAenC,EAAAA,KAAOD,KAAKC,EAAE,CAAA;AACtF,MAAA,OAAOqC,QAAAA,CACLN,KAAKC,IAAAA,CAAK,IAAA,CAAKnB,MAAMO,eAAAA,CAAgBa,QAAAA,IAAYE,cAAAA,CAAejC,QAAQ,GACxE6B,IAAAA,CAAKC,IAAAA,CAAK,KAAKnB,KAAAA,CAAMS,qBAAAA,CAAsBW,UAAQ,EAAIE,cAAAA,CAAenC,EAAE,CAAA,CAAA;AAE5E,KAAA,CAAA,CAAA;AAEF,IAAA,OAAON,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;MAAE,GAAGA,IAAAA;AAAME,MAAAA,QAAAA,EAAUF,IAAAA,CAAKC;KAAG,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAMG,SAAST,KAAAA,EAA6B;AAC1C,IAAA,MAAM,KAAKsB,IAAAA,EAAI;AAEf,IAAA,MAAMS,OAAAA,CAAQC,GAAAA,CACZhC,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,KACTsC,QAAAA,CACEN,IAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMS,qBAAAA,CAAsBW,QAAAA,EAAQ,EAAIlC,IAAAA,CAAKE,QAAQ,CAAA,EACpE8B,IAAAA,CAAKC,IAAAA,CAAK,IAAA,CAAKnB,KAAAA,CAAMO,eAAAA,CAAgBa,QAAAA,EAAQ,EAAIlC,IAAAA,CAAKG,QAAQ,CAAA,CAAA,CAAA,CAAA;AAIpE,IAAA,OAAOR,KAAAA,CAAMI,GAAAA,CAAI,CAACC,IAAAA,MAAU;MAAE,GAAGA,IAAAA;AAAMC,MAAAA,EAAAA,EAAID,IAAAA,CAAKE;KAAS,CAAA,CAAA;AAC3D;AAEA,EAAA,MAAgB6B,YAAY/B,IAAAA,EAAgB;AAC1C,IAAA,MAAMuC,IAAAA,GAAOC,MAAAA,CAAOC,UAAAA,CAAW,QAAA,CAAA;AAC/B,IAAA,MAAMC,QAAAA,CAASC,gBAAAA,CAAiB3C,IAAAA,CAAAA,EAAOuC,IAAAA,CAAAA;AACvC,IAAA,OAAOA,IAAAA,CAAKK,OAAO,KAAA,CAAA;AACrB;EAEApC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEM,KAAAA,EAAO+B,WAAAA,CAAY,KAAK/B,KAAK;AAAE,KAAA;AAC1C;AAEAL,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"storage.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\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"]}

package/dist/tools/search/base.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["base.ts"],"names":["SearchToolOutput","ToolOutput","constructor","results","sources","R","unique","map","result","url","isEmpty","length","getTextContent","JSON","stringify","join","cacheKey","WeakRefKeyFn","from","self","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAgBM,MAAeA,yBAEZC,mBAAAA,CAAAA;EAhCV;;;;AAiCEC,EAAAA,WAAAA,CAA4BC,OAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,OAAAA,GAAAA,OAAAA;AAE5B;AAEA,EAAA,IAIIC,OAAU,GAAA;AACZ,IAAOC,OAAAA,YAAAA,CAAEC,OAAO,IAAKH,CAAAA,OAAAA,CAAQI,IAAI,CAACC,MAAAA,KAAWA,MAAOC,CAAAA,GAAG,CAAA,CAAA;AACzD;EAEAC,OAAU,GAAA;AACR,IAAO,OAAA,IAAA,CAAKP,QAAQQ,MAAW,KAAA,CAAA;AACjC;EAKAC,cAAyB,GAAA;AACvB,IAAA,OAAO,IAAKT,CAAAA,OAAAA,CAAQI,GAAI,CAAA,CAACC,MAAWK,KAAAA,IAAAA,CAAKC,SAAUN,CAAAA,MAAAA,EAAQ,IAAM,EAAA,CAAA,CAAA,CAAA,CAAIO,KAAK,MAAA,CAAA;AAC5E;AACF;;;AAjBIC,IAAAA,QAAAA,EAAUC,+BAAaC,CAAAA,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO,CAAA;IACpEiB,UAAY,EAAA;;;;;;;AAWZJ,IAAAA,QAAAA,EAAUC,+BAAaC,CAAAA,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\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 { BaseToolOptions, BaseToolRunOptions, ToolOutput } from \"@/tools/base.js\";\nimport { Cache, WeakRefKeyFn } from \"@/cache/decoratorCache.js\";\nimport * as R from \"remeda\";\n\nexport interface SearchToolOptions extends BaseToolOptions {}\n\nexport interface SearchToolRunOptions extends BaseToolRunOptions {}\n\nexport interface SearchToolResult {\n  title: string;\n  description: string;\n  url: string;\n}\n\nexport abstract class SearchToolOutput<\n  TSearchToolResult extends SearchToolResult = SearchToolResult,\n> extends ToolOutput {\n  constructor(public readonly results: TSearchToolResult[]) {\n    super();\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n    enumerable: false,\n  })\n  get sources() {\n    return R.unique(this.results.map((result) => result.url));\n  }\n\n  isEmpty() {\n    return this.results.length === 0;\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n  })\n  getTextContent(): string {\n    return this.results.map((result) => JSON.stringify(result, null, 2)).join(\"\\n\\n\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/search/base.ts"],"names":["SearchToolOutput","ToolOutput","results","sources","R","unique","map","result","url","isEmpty","length","getTextContent","JSON","stringify","join","cacheKey","WeakRefKeyFn","from","self","enumerable"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAgBM,MAAeA,yBAEZC,mBAAAA,CAAAA;EArBV;;;;AAsBE,EAAA,WAAA,CAA4BC,OAAAA,EAA8B;AACxD,IAAA,KAAA,EAAK,EAAA,KADqBA,OAAAA,GAAAA,OAAAA;AAE5B;AAEA,EAAA,IAIIC,OAAAA,GAAU;AACZ,IAAA,OAAOC,YAAAA,CAAEC,OAAO,IAAA,CAAKH,OAAAA,CAAQI,IAAI,CAACC,MAAAA,KAAWA,MAAAA,CAAOC,GAAG,CAAA,CAAA;AACzD;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,IAAA,CAAKP,QAAQQ,MAAAA,KAAW,CAAA;AACjC;EAKAC,cAAAA,GAAyB;AACvB,IAAA,OAAO,IAAA,CAAKT,OAAAA,CAAQI,GAAAA,CAAI,CAACC,MAAAA,KAAWK,IAAAA,CAAKC,SAAAA,CAAUN,MAAAA,EAAQ,IAAA,EAAM,CAAA,CAAA,CAAA,CAAIO,KAAK,MAAA,CAAA;AAC5E;AACF;;;AAjBIC,IAAAA,QAAAA,EAAUC,+BAAAA,CAAaC,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO,CAAA;IACpEiB,UAAAA,EAAY;;;;;;;AAWZJ,IAAAA,QAAAA,EAAUC,+BAAAA,CAAaC,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO","file":"base.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolOptions, BaseToolRunOptions, ToolOutput } from \"@/tools/base.js\";\nimport { Cache, WeakRefKeyFn } from \"@/cache/decoratorCache.js\";\nimport * as R from \"remeda\";\n\nexport interface SearchToolOptions extends BaseToolOptions {}\n\nexport interface SearchToolRunOptions extends BaseToolRunOptions {}\n\nexport interface SearchToolResult {\n  title: string;\n  description: string;\n  url: string;\n}\n\nexport abstract class SearchToolOutput<\n  TSearchToolResult extends SearchToolResult = SearchToolResult,\n> extends ToolOutput {\n  constructor(public readonly results: TSearchToolResult[]) {\n    super();\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n    enumerable: false,\n  })\n  get sources() {\n    return R.unique(this.results.map((result) => result.url));\n  }\n\n  isEmpty() {\n    return this.results.length === 0;\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n  })\n  getTextContent(): string {\n    return this.results.map((result) => JSON.stringify(result, null, 2)).join(\"\\n\\n\");\n  }\n}\n"]}

package/dist/tools/search/base.d.cts

@@ -1,7 +1,7 @@
 import { BaseToolOptions, BaseToolRunOptions, ToolOutput } from '../base.cjs';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-D5Mu0EEH.cjs';
+import '../../emitter-DWEsd-ud.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';
@@ -15,18 +15,7 @@ import 'zod-to-json-schema';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface SearchToolOptions extends BaseToolOptions {

package/dist/tools/search/base.d.ts

@@ -1,7 +1,7 @@
 import { BaseToolOptions, BaseToolRunOptions, ToolOutput } from '../base.js';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-C3dO-s2P.js';
+import '../../emitter-C9nwGLt6.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';
@@ -15,18 +15,7 @@ import 'zod-to-json-schema';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface SearchToolOptions extends BaseToolOptions {

package/dist/tools/search/base.js.map

@@ -1 +1 @@
-{"version":3,"sources":["base.ts"],"names":["SearchToolOutput","ToolOutput","constructor","results","sources","R","unique","map","result","url","isEmpty","length","getTextContent","JSON","stringify","join","cacheKey","WeakRefKeyFn","from","self","enumerable"],"mappings":";;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAgBM,MAAeA,yBAEZC,UAAAA,CAAAA;EAhCV;;;;AAiCEC,EAAAA,WAAAA,CAA4BC,OAA8B,EAAA;AACxD,IAAK,KAAA,EAAA,EAAA,KADqBA,OAAAA,GAAAA,OAAAA;AAE5B;AAEA,EAAA,IAIIC,OAAU,GAAA;AACZ,IAAOC,OAAAA,CAAAA,CAAEC,OAAO,IAAKH,CAAAA,OAAAA,CAAQI,IAAI,CAACC,MAAAA,KAAWA,MAAOC,CAAAA,GAAG,CAAA,CAAA;AACzD;EAEAC,OAAU,GAAA;AACR,IAAO,OAAA,IAAA,CAAKP,QAAQQ,MAAW,KAAA,CAAA;AACjC;EAKAC,cAAyB,GAAA;AACvB,IAAA,OAAO,IAAKT,CAAAA,OAAAA,CAAQI,GAAI,CAAA,CAACC,MAAWK,KAAAA,IAAAA,CAAKC,SAAUN,CAAAA,MAAAA,EAAQ,IAAM,EAAA,CAAA,CAAA,CAAA,CAAIO,KAAK,MAAA,CAAA;AAC5E;AACF;;;AAjBIC,IAAAA,QAAAA,EAAUC,YAAaC,CAAAA,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO,CAAA;IACpEiB,UAAY,EAAA;;;;;;;AAWZJ,IAAAA,QAAAA,EAAUC,YAAaC,CAAAA,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\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 { BaseToolOptions, BaseToolRunOptions, ToolOutput } from \"@/tools/base.js\";\nimport { Cache, WeakRefKeyFn } from \"@/cache/decoratorCache.js\";\nimport * as R from \"remeda\";\n\nexport interface SearchToolOptions extends BaseToolOptions {}\n\nexport interface SearchToolRunOptions extends BaseToolRunOptions {}\n\nexport interface SearchToolResult {\n  title: string;\n  description: string;\n  url: string;\n}\n\nexport abstract class SearchToolOutput<\n  TSearchToolResult extends SearchToolResult = SearchToolResult,\n> extends ToolOutput {\n  constructor(public readonly results: TSearchToolResult[]) {\n    super();\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n    enumerable: false,\n  })\n  get sources() {\n    return R.unique(this.results.map((result) => result.url));\n  }\n\n  isEmpty() {\n    return this.results.length === 0;\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n  })\n  getTextContent(): string {\n    return this.results.map((result) => JSON.stringify(result, null, 2)).join(\"\\n\\n\");\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/search/base.ts"],"names":["SearchToolOutput","ToolOutput","results","sources","R","unique","map","result","url","isEmpty","length","getTextContent","JSON","stringify","join","cacheKey","WeakRefKeyFn","from","self","enumerable"],"mappings":";;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAgBM,MAAeA,yBAEZC,UAAAA,CAAAA;EArBV;;;;AAsBE,EAAA,WAAA,CAA4BC,OAAAA,EAA8B;AACxD,IAAA,KAAA,EAAK,EAAA,KADqBA,OAAAA,GAAAA,OAAAA;AAE5B;AAEA,EAAA,IAIIC,OAAAA,GAAU;AACZ,IAAA,OAAOC,CAAAA,CAAEC,OAAO,IAAA,CAAKH,OAAAA,CAAQI,IAAI,CAACC,MAAAA,KAAWA,MAAAA,CAAOC,GAAG,CAAA,CAAA;AACzD;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,IAAA,CAAKP,QAAQQ,MAAAA,KAAW,CAAA;AACjC;EAKAC,cAAAA,GAAyB;AACvB,IAAA,OAAO,IAAA,CAAKT,OAAAA,CAAQI,GAAAA,CAAI,CAACC,MAAAA,KAAWK,IAAAA,CAAKC,SAAAA,CAAUN,MAAAA,EAAQ,IAAA,EAAM,CAAA,CAAA,CAAA,CAAIO,KAAK,MAAA,CAAA;AAC5E;AACF;;;AAjBIC,IAAAA,QAAAA,EAAUC,YAAAA,CAAaC,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO,CAAA;IACpEiB,UAAAA,EAAY;;;;;;;AAWZJ,IAAAA,QAAAA,EAAUC,YAAAA,CAAaC,IAAAA,CAAuB,CAACC,IAAAA,KAASA,KAAKhB,OAAO","file":"base.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolOptions, BaseToolRunOptions, ToolOutput } from \"@/tools/base.js\";\nimport { Cache, WeakRefKeyFn } from \"@/cache/decoratorCache.js\";\nimport * as R from \"remeda\";\n\nexport interface SearchToolOptions extends BaseToolOptions {}\n\nexport interface SearchToolRunOptions extends BaseToolRunOptions {}\n\nexport interface SearchToolResult {\n  title: string;\n  description: string;\n  url: string;\n}\n\nexport abstract class SearchToolOutput<\n  TSearchToolResult extends SearchToolResult = SearchToolResult,\n> extends ToolOutput {\n  constructor(public readonly results: TSearchToolResult[]) {\n    super();\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n    enumerable: false,\n  })\n  get sources() {\n    return R.unique(this.results.map((result) => result.url));\n  }\n\n  isEmpty() {\n    return this.results.length === 0;\n  }\n\n  @Cache({\n    cacheKey: WeakRefKeyFn.from<SearchToolOutput>((self) => self.results),\n  })\n  getTextContent(): string {\n    return this.results.map((result) => JSON.stringify(result, null, 2)).join(\"\\n\\n\");\n  }\n}\n"]}

package/dist/tools/search/duckDuckGoSearch.cjs

@@ -8,6 +8,7 @@ var zod = require('zod');
 var decoratorCache_cjs = require('../../cache/decoratorCache.cjs');
 var paginate_cjs = require('../../internals/helpers/paginate.cjs');
 var emitter_cjs = require('../../emitter/emitter.cjs');
+var env_cjs = require('../../internals/env.cjs');
 
 var __defProp = Object.defineProperty;
 var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
@@ -66,7 +67,12 @@ class DuckDuckGoSearchTool extends base_cjs$1.Tool {
   constructor(options = {}) {
     super({
       ...options,
-      maxResults: options?.maxResults ?? 15
+      maxResults: options?.maxResults ?? 15,
+      httpClientOptions: {
+        proxy: env_cjs.getEnv("BEEAI_DDG_TOOL_PROXY"),
+        rejectUnauthorized: env_cjs.parseEnv.asBoolean("BEEAI_DDG_TOOL_PROXY_VERIFY", true),
+        ...options?.httpClientOptions
+      }
     });
   }
   static {

package/dist/tools/search/duckDuckGoSearch.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["duckDuckGoSearch.ts"],"names":["DuckDuckGoSearchToolOutput","SearchToolOutput","constructor","results","register","createSnapshot","loadSnapshot","snapshot","Object","assign","DuckDuckGoSearchTool","Tool","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","query","string","min","max","options","maxResults","_createClient","throttle","rawDDGSearch","default","pThrottle","limit","interval","_run","input","run","headers","HeaderGenerator","getHeaders","client","paginate","size","handler","cursor","data","noResults","done","safeSearch","SafeSearchType","MODERATE","search","offset","user_agent","httpClientOptions","signal","uri_modifier","rawUrl","url","URL","searchParams","delete","toString","nextCursor","undefined","length","stripHtml","map","result","title","enumerable"],"mappings":";;;;;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAmCM,MAAMA,mCAAmCC,yBAAAA,CAAAA;EAjDhD;;;;AAkDEC,EAAAA,WAAAA,CAA4BC,OAAuC,EAAA;AACjE,IAAMA,KAAAA,CAAAA,OAAAA,CAAAA,EAAAA,IAAAA,CADoBA,OAAAA,GAAAA,OAAAA;AAE5B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;EAEAC,cAAiB,GAAA;AACf,IAAO,OAAA;AACLF,MAAAA,OAAAA,EAAS,IAAKA,CAAAA;AAChB,KAAA;AACF;AAEAG,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF;AAEO,MAAMG,6BAA6BC,eAAAA,CAAAA;EArE1C;;;EA0EEC,IAAO,GAAA,YAAA;EACPC,WACE,GAAA,4FAAA;EAEcC,OACdC,GAAAA,mBAAAA,CAAQC,KAAKC,KAAM,CAAA;IACjBC,SAAW,EAAA;AAAC,MAAA,MAAA;AAAQ,MAAA,QAAA;AAAU,MAAA;;IAC9BC,OAAS,EAAA;GACX,CAAA;EAGFC,WAAc,GAAA;AACZ,IAAA,OAAOC,MAAEC,MAAO,CAAA;AACdC,MAAAA,KAAAA,EAAOF,MAAEG,MAAO,CAAA;QAAEX,WAAa,EAAA,CAAA,YAAA;AAAe,OAAA,CAAGY,CAAAA,GAAAA,CAAI,CAAA,CAAA,CAAGC,IAAI,GAAA;KAC9D,CAAA;AACF;EAEA,WAAmBC,CAAAA,OAAAA,GAAgD,EAAI,EAAA;AACrE,IAAM,KAAA,CAAA;MAAE,GAAGA,OAAAA;AAASC,MAAAA,UAAAA,EAAYD,SAASC,UAAc,IAAA;KAAG,CAAA;AAC5D;EAEA;AACE,IAAA,IAAA,CAAKxB,QAAQ,EAAA;AACf;AAEA,EAAA,MACgByB,aAAgB,GAAA;AAC9B,IAAM,MAAA,EAAEC,QAAQ,EAAA,GAAK,IAAKH,CAAAA,OAAAA;AAC1B,IAAA,IAAIG,aAAa,KAAO,EAAA;AACtB,MAAOC,OAAAA,qBAAAA;AACT;AAEA,IAAA,MAAM,EAAEC,OAASC,EAAAA,SAAAA,EAAc,GAAA,MAAM,OAAO,YAAA,CAAA;AAC5C,IAAA,OAAOA,SAAU,CAAA;MACf,GAAGH,QAAAA;AACHI,MAAAA,KAAAA,EAAOJ,UAAUI,KAAS,IAAA,CAAA;AAC1BC,MAAAA,QAAAA,EAAUL,UAAUK,QAAY,IAAA;AAClC,KAAA,EAAGJ,qBAAAA,CAAAA;AACL;AAEA,EAAA,MAAgBK,KACd,EAAEb,KAAAA,EAAOc,KAAK,EAAA,EACdV,SACAW,GACA,EAAA;AACA,IAAA,MAAMC,OAAU,GAAA,IAAIC,+BAAAA,EAAAA,CAAkBC,UAAU,EAAA;AAChD,IAAMC,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKb,aAAa,EAAA;AAEvC,IAAM1B,MAAAA,OAAAA,GAAU,MAAMwC,qBAAS,CAAA;AAC7BC,MAAAA,IAAAA,EAAM,KAAKjB,OAAQC,CAAAA,UAAAA;AACnBiB,MAAAA,OAAAA,kBAAgB,MAAA,CAAA,OAAA,EAAEC,MAAS,GAAA,CAAA,EAAG,KAAA;AAC5B,QAAM,MAAA,EAAE3C,SAAS4C,IAAMC,EAAAA,SAAAA,EAAWC,MAAS,GAAA,MAAMP,OAC/CL,KACA,EAAA;AACEa,UAAAA,UAAAA,EAAYC,6BAAeC,CAAAA,QAAAA;AAC3B,UAAA,GAAG,KAAKzB,OAAQ0B,CAAAA,MAAAA;AAChB,UAAA,GAAG1B,OAAS0B,EAAAA,MAAAA;UACZC,MAAQR,EAAAA;SAEV,EAAA;AACEP,UAAAA,OAAAA;AACAgB,UAAAA,UAAAA,EAAYhB,QAAQ,YAAA,CAAA;AACpB,UAAA,GAAG,KAAKZ,OAAS6B,EAAAA,iBAAAA;AACjB,UAAA,GAAG7B,OAAS6B,EAAAA,iBAAAA;AACZC,UAAAA,MAAAA,EAAQnB,GAAImB,CAAAA,MAAAA;AACZC,UAAAA,YAAAA,0BAAeC,MAAAA,KAAAA;AACb,YAAMC,MAAAA,GAAAA,GAAM,IAAIC,GAAAA,CAAIF,MAAAA,CAAAA;AACpBC,YAAIE,GAAAA,CAAAA,YAAAA,CAAaC,OAAO,QAAA,CAAA;AACxB,YAAA,OAAOH,IAAII,QAAQ,EAAA;WAHP,EAAA,cAAA;SAKhB,CAAA;AAGF,QAAO,OAAA;AACLjB,UAAAA,IAAAA;UACAkB,UAAYhB,EAAAA,IAAAA,GAAOiB,MAAYpB,GAAAA,MAAAA,GAASC,IAAKoB,CAAAA;AAC/C,SAAA;OA1BO,EAAA,SAAA;KA4BX,CAAA;AAEA,IAAA,MAAM,EAAEC,SAAAA,EAAc,GAAA,MAAM,OAAO,mBAAA,CAAA;AAEnC,IAAA,OAAO,IAAIpE,0BAAAA,CACTG,OAAQkE,CAAAA,GAAAA,CAAI,CAACC,MAAY,MAAA;MACvBC,KAAOH,EAAAA,SAAAA,CAAUE,MAAOC,CAAAA,KAAK,CAAED,CAAAA,MAAAA;MAC/BzD,WAAauD,EAAAA,SAAAA,CAAUE,MAAOzD,CAAAA,WAAW,CAAEyD,CAAAA,MAAAA;AAC3CV,MAAAA,GAAAA,EAAKU,MAAOV,CAAAA;AACd,KAAA,CAAA,CAAA,CAAA;AAEJ;AACF;;;;;;;;;IAjEWY,UAAY,EAAA","file":"duckDuckGoSearch.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\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 { SearchOptions, search as rawDDGSearch, SafeSearchType } from \"duck-duck-scrape\";\nimport { Options as ThrottleOptions } from \"p-throttle\";\nimport {\n  SearchToolOptions,\n  SearchToolOutput,\n  SearchToolResult,\n  SearchToolRunOptions,\n} from \"./base.js\";\nimport { ToolEmitter, Tool, ToolInput } from \"@/tools/base.js\";\nimport { HeaderGenerator } from \"header-generator\";\nimport type { NeedleOptions } from \"needle\";\nimport { z } from \"zod\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { RunContext } from \"@/context.js\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport { SafeSearchType as DuckDuckGoSearchToolSearchType };\n\nexport interface DuckDuckGoSearchToolOptions extends SearchToolOptions {\n  search?: SearchOptions;\n  throttle?: ThrottleOptions | false;\n  httpClientOptions?: NeedleOptions;\n  maxResults: number;\n}\n\nexport interface DuckDuckGoSearchToolRunOptions extends SearchToolRunOptions {\n  search?: SearchOptions;\n  httpClientOptions?: NeedleOptions;\n}\n\nexport interface DuckDuckGoSearchToolResult extends SearchToolResult {}\n\nexport class DuckDuckGoSearchToolOutput extends SearchToolOutput<DuckDuckGoSearchToolResult> {\n  constructor(public readonly results: DuckDuckGoSearchToolResult[]) {\n    super(results);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      results: this.results,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n\nexport class DuckDuckGoSearchTool extends Tool<\n  DuckDuckGoSearchToolOutput,\n  DuckDuckGoSearchToolOptions,\n  DuckDuckGoSearchToolRunOptions\n> {\n  name = \"DuckDuckGo\";\n  description =\n    \"Search for online trends, news, current events, real-time information, or research topics.\";\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, DuckDuckGoSearchToolOutput> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"search\", \"ddg\"],\n      creator: this,\n    });\n\n  @Cache()\n  inputSchema() {\n    return z.object({\n      query: z.string({ description: `Search query` }).min(1).max(128),\n    });\n  }\n\n  public constructor(options: Partial<DuckDuckGoSearchToolOptions> = {}) {\n    super({ ...options, maxResults: options?.maxResults ?? 15 });\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache({ enumerable: false })\n  protected async _createClient() {\n    const { throttle } = this.options;\n    if (throttle === false) {\n      return rawDDGSearch;\n    }\n\n    const { default: pThrottle } = await import(\"p-throttle\");\n    return pThrottle({\n      ...throttle,\n      limit: throttle?.limit ?? 1,\n      interval: throttle?.interval ?? 3000,\n    })(rawDDGSearch);\n  }\n\n  protected async _run(\n    { query: input }: ToolInput<this>,\n    options: Partial<DuckDuckGoSearchToolRunOptions>,\n    run: RunContext<this>,\n  ) {\n    const headers = new HeaderGenerator().getHeaders();\n    const client = await this._createClient();\n\n    const results = await paginate({\n      size: this.options.maxResults,\n      handler: async ({ cursor = 0 }) => {\n        const { results: data, noResults: done } = await client(\n          input,\n          {\n            safeSearch: SafeSearchType.MODERATE,\n            ...this.options.search,\n            ...options?.search,\n            offset: cursor,\n          },\n          {\n            headers,\n            user_agent: headers[\"user-agent\"],\n            ...this.options?.httpClientOptions,\n            ...options?.httpClientOptions,\n            signal: run.signal,\n            uri_modifier: (rawUrl) => {\n              const url = new URL(rawUrl);\n              url.searchParams.delete(\"ss_mkt\");\n              return url.toString();\n            },\n          },\n        );\n\n        return {\n          data,\n          nextCursor: done ? undefined : cursor + data.length,\n        };\n      },\n    });\n\n    const { stripHtml } = await import(\"string-strip-html\");\n\n    return new DuckDuckGoSearchToolOutput(\n      results.map((result) => ({\n        title: stripHtml(result.title).result,\n        description: stripHtml(result.description).result,\n        url: result.url,\n      })),\n    );\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/search/duckDuckGoSearch.ts"],"names":["DuckDuckGoSearchToolOutput","SearchToolOutput","results","register","createSnapshot","loadSnapshot","snapshot","Object","assign","DuckDuckGoSearchTool","Tool","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","query","string","min","max","options","maxResults","httpClientOptions","proxy","getEnv","rejectUnauthorized","parseEnv","asBoolean","_createClient","throttle","rawDDGSearch","default","pThrottle","limit","interval","_run","input","run","headers","HeaderGenerator","getHeaders","client","paginate","size","handler","cursor","data","noResults","done","safeSearch","SafeSearchType","MODERATE","search","offset","user_agent","signal","uri_modifier","rawUrl","url","URL","searchParams","delete","toString","nextCursor","undefined","length","stripHtml","map","result","title","enumerable"],"mappings":";;;;;;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAoCM,MAAMA,mCAAmCC,yBAAAA,CAAAA;EAvChD;;;;AAwCE,EAAA,WAAA,CAA4BC,OAAAA,EAAuC;AACjE,IAAA,KAAA,CAAMA,OAAAA,CAAAA,EAAAA,IAAAA,CADoBA,OAAAA,GAAAA,OAAAA;AAE5B;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;EAEAC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLF,MAAAA,OAAAA,EAAS,IAAA,CAAKA;AAChB,KAAA;AACF;AAEAG,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF;AAEO,MAAMG,6BAA6BC,eAAAA,CAAAA;EA3D1C;;;EAgEEC,IAAAA,GAAO,YAAA;EACPC,WAAAA,GACE,4FAAA;EAEcC,OAAAA,GACdC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACjBC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,QAAA;AAAU,MAAA;;IAC9BC,OAAAA,EAAS;GACX,CAAA;EAGFC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,KAAAA,EAAOF,MAAEG,MAAAA,CAAO;QAAEX,WAAAA,EAAa,CAAA,YAAA;AAAe,OAAA,CAAA,CAAGY,GAAAA,CAAI,CAAA,CAAA,CAAGC,IAAI,GAAA;KAC9D,CAAA;AACF;EAEA,WAAA,CAAmBC,OAAAA,GAAgD,EAAC,EAAG;AACrE,IAAA,KAAA,CAAM;MACJ,GAAGA,OAAAA;AACHC,MAAAA,UAAAA,EAAYD,SAASC,UAAAA,IAAc,EAAA;MACnCC,iBAAAA,EAAmB;AACjBC,QAAAA,KAAAA,EAAOC,eAAO,sBAAA,CAAA;QACdC,kBAAAA,EAAoBC,gBAAAA,CAASC,SAAAA,CAAU,6BAAA,EAA+B,IAAA,CAAA;AACtE,QAAA,GAAGP,OAAAA,EAASE;AACd;KACF,CAAA;AACF;EAEA;AACE,IAAA,IAAA,CAAKzB,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgB+B,aAAAA,GAAgB;AAC9B,IAAA,MAAM,EAAEC,QAAAA,EAAQ,GAAK,IAAA,CAAKT,OAAAA;AAC1B,IAAA,IAAIS,aAAa,KAAA,EAAO;AACtB,MAAA,OAAOC,qBAAAA;AACT;AAEA,IAAA,MAAM,EAAEC,OAAAA,EAASC,SAAAA,EAAS,GAAK,MAAM,OAAO,YAAA,CAAA;AAC5C,IAAA,OAAOA,SAAAA,CAAU;MACf,GAAGH,QAAAA;AACHI,MAAAA,KAAAA,EAAOJ,UAAUI,KAAAA,IAAS,CAAA;AAC1BC,MAAAA,QAAAA,EAAUL,UAAUK,QAAAA,IAAY;AAClC,KAAA,EAAGJ,qBAAAA,CAAAA;AACL;AAEA,EAAA,MAAgBK,KACd,EAAEnB,KAAAA,EAAOoB,KAAAA,EAAK,EACdhB,SACAiB,GAAAA,EACA;AACA,IAAA,MAAMC,OAAAA,GAAU,IAAIC,+BAAAA,EAAAA,CAAkBC,UAAAA,EAAU;AAChD,IAAA,MAAMC,MAAAA,GAAS,MAAM,IAAA,CAAKb,aAAAA,EAAa;AAEvC,IAAA,MAAMhC,OAAAA,GAAU,MAAM8C,qBAAAA,CAAS;AAC7BC,MAAAA,IAAAA,EAAM,KAAKvB,OAAAA,CAAQC,UAAAA;AACnBuB,MAAAA,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,MAAAA,GAAS,CAAA,EAAC,KAAE;AAC5B,QAAA,MAAM,EAAEjD,SAASkD,IAAAA,EAAMC,SAAAA,EAAWC,MAAI,GAAK,MAAMP,OAC/CL,KAAAA,EACA;AACEa,UAAAA,UAAAA,EAAYC,6BAAAA,CAAeC,QAAAA;AAC3B,UAAA,GAAG,KAAK/B,OAAAA,CAAQgC,MAAAA;AAChB,UAAA,GAAGhC,OAAAA,EAASgC,MAAAA;UACZC,MAAAA,EAAQR;SACV,EACA;AACEP,UAAAA,OAAAA;AACAgB,UAAAA,UAAAA,EAAYhB,QAAQ,YAAA,CAAA;AACpB,UAAA,GAAG,KAAKlB,OAAAA,EAASE,iBAAAA;AACjB,UAAA,GAAGF,OAAAA,EAASE,iBAAAA;AACZiC,UAAAA,MAAAA,EAAQlB,GAAAA,CAAIkB,MAAAA;AACZC,UAAAA,YAAAA,0BAAeC,MAAAA,KAAAA;AACb,YAAA,MAAMC,GAAAA,GAAM,IAAIC,GAAAA,CAAIF,MAAAA,CAAAA;AACpBC,YAAAA,GAAAA,CAAIE,YAAAA,CAAaC,OAAO,QAAA,CAAA;AACxB,YAAA,OAAOH,IAAII,QAAAA,EAAQ;WACrB,EAJc,cAAA;SAKhB,CAAA;AAGF,QAAA,OAAO;AACLhB,UAAAA,IAAAA;UACAiB,UAAAA,EAAYf,IAAAA,GAAOgB,MAAAA,GAAYnB,MAAAA,GAASC,IAAAA,CAAKmB;AAC/C,SAAA;OACF,EA3BS,SAAA;KA4BX,CAAA;AAEA,IAAA,MAAM,EAAEC,SAAAA,EAAS,GAAK,MAAM,OAAO,mBAAA,CAAA;AAEnC,IAAA,OAAO,IAAIxE,0BAAAA,CACTE,OAAAA,CAAQuE,GAAAA,CAAI,CAACC,MAAAA,MAAY;MACvBC,KAAAA,EAAOH,SAAAA,CAAUE,MAAAA,CAAOC,KAAK,CAAA,CAAED,MAAAA;MAC/B9D,WAAAA,EAAa4D,SAAAA,CAAUE,MAAAA,CAAO9D,WAAW,CAAA,CAAE8D,MAAAA;AAC3CV,MAAAA,GAAAA,EAAKU,MAAAA,CAAOV;AACd,KAAA,CAAA,CAAA,CAAA;AAEJ;AACF;;;;;;;;;IAjEWY,UAAAA,EAAY","file":"duckDuckGoSearch.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { SearchOptions, search as rawDDGSearch, SafeSearchType } from \"duck-duck-scrape\";\nimport { Options as ThrottleOptions } from \"p-throttle\";\nimport {\n  SearchToolOptions,\n  SearchToolOutput,\n  SearchToolResult,\n  SearchToolRunOptions,\n} from \"./base.js\";\nimport { ToolEmitter, Tool, ToolInput } from \"@/tools/base.js\";\nimport { HeaderGenerator } from \"header-generator\";\nimport type { NeedleOptions } from \"needle\";\nimport { z } from \"zod\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { RunContext } from \"@/context.js\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { getEnv, parseEnv } from \"@/internals/env.js\";\n\nexport { SafeSearchType as DuckDuckGoSearchToolSearchType };\n\nexport interface DuckDuckGoSearchToolOptions extends SearchToolOptions {\n  search?: SearchOptions;\n  throttle?: ThrottleOptions | false;\n  httpClientOptions?: NeedleOptions;\n  maxResults: number;\n}\n\nexport interface DuckDuckGoSearchToolRunOptions extends SearchToolRunOptions {\n  search?: SearchOptions;\n  httpClientOptions?: NeedleOptions;\n}\n\nexport interface DuckDuckGoSearchToolResult extends SearchToolResult {}\n\nexport class DuckDuckGoSearchToolOutput extends SearchToolOutput<DuckDuckGoSearchToolResult> {\n  constructor(public readonly results: DuckDuckGoSearchToolResult[]) {\n    super(results);\n  }\n\n  static {\n    this.register();\n  }\n\n  createSnapshot() {\n    return {\n      results: this.results,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n\nexport class DuckDuckGoSearchTool extends Tool<\n  DuckDuckGoSearchToolOutput,\n  DuckDuckGoSearchToolOptions,\n  DuckDuckGoSearchToolRunOptions\n> {\n  name = \"DuckDuckGo\";\n  description =\n    \"Search for online trends, news, current events, real-time information, or research topics.\";\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, DuckDuckGoSearchToolOutput> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"search\", \"ddg\"],\n      creator: this,\n    });\n\n  @Cache()\n  inputSchema() {\n    return z.object({\n      query: z.string({ description: `Search query` }).min(1).max(128),\n    });\n  }\n\n  public constructor(options: Partial<DuckDuckGoSearchToolOptions> = {}) {\n    super({\n      ...options,\n      maxResults: options?.maxResults ?? 15,\n      httpClientOptions: {\n        proxy: getEnv(\"BEEAI_DDG_TOOL_PROXY\"),\n        rejectUnauthorized: parseEnv.asBoolean(\"BEEAI_DDG_TOOL_PROXY_VERIFY\", true),\n        ...options?.httpClientOptions,\n      },\n    });\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache({ enumerable: false })\n  protected async _createClient() {\n    const { throttle } = this.options;\n    if (throttle === false) {\n      return rawDDGSearch;\n    }\n\n    const { default: pThrottle } = await import(\"p-throttle\");\n    return pThrottle({\n      ...throttle,\n      limit: throttle?.limit ?? 1,\n      interval: throttle?.interval ?? 3000,\n    })(rawDDGSearch);\n  }\n\n  protected async _run(\n    { query: input }: ToolInput<this>,\n    options: Partial<DuckDuckGoSearchToolRunOptions>,\n    run: RunContext<this>,\n  ) {\n    const headers = new HeaderGenerator().getHeaders();\n    const client = await this._createClient();\n\n    const results = await paginate({\n      size: this.options.maxResults,\n      handler: async ({ cursor = 0 }) => {\n        const { results: data, noResults: done } = await client(\n          input,\n          {\n            safeSearch: SafeSearchType.MODERATE,\n            ...this.options.search,\n            ...options?.search,\n            offset: cursor,\n          },\n          {\n            headers,\n            user_agent: headers[\"user-agent\"],\n            ...this.options?.httpClientOptions,\n            ...options?.httpClientOptions,\n            signal: run.signal,\n            uri_modifier: (rawUrl) => {\n              const url = new URL(rawUrl);\n              url.searchParams.delete(\"ss_mkt\");\n              return url.toString();\n            },\n          },\n        );\n\n        return {\n          data,\n          nextCursor: done ? undefined : cursor + data.length,\n        };\n      },\n    });\n\n    const { stripHtml } = await import(\"string-strip-html\");\n\n    return new DuckDuckGoSearchToolOutput(\n      results.map((result) => ({\n        title: stripHtml(result.title).result,\n        description: stripHtml(result.description).result,\n        url: result.url,\n      })),\n    );\n  }\n}\n"]}

package/dist/tools/search/duckDuckGoSearch.d.cts

@@ -7,7 +7,7 @@ import { NeedleOptions } from 'needle';
 import { z } from 'zod';
 import { RunContext } from '../../context.cjs';
 import 'ajv';
-import '../../emitter-D5Mu0EEH.cjs';
+import '../../emitter-DWEsd-ud.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';
@@ -20,18 +20,7 @@ import 'zod-to-json-schema';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface DuckDuckGoSearchToolOptions extends SearchToolOptions {

package/dist/tools/search/duckDuckGoSearch.d.ts

@@ -7,7 +7,7 @@ import { NeedleOptions } from 'needle';
 import { z } from 'zod';
 import { RunContext } from '../../context.js';
 import 'ajv';
-import '../../emitter-C3dO-s2P.js';
+import '../../emitter-C9nwGLt6.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';
@@ -20,18 +20,7 @@ import 'zod-to-json-schema';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
+ * SPDX-License-Identifier: Apache-2.0
  */
 
 interface DuckDuckGoSearchToolOptions extends SearchToolOptions {