beeai-framework 0.1.28 → 0.1.30

package/dist/tools/mcp.d.cts

@@ -1,4 +1,4 @@
-import { JSONToolOutput, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.cjs';
+import { Tool, JSONToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.cjs';
 import { GetRunContext } from '../context.cjs';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { ListToolsResult } from '@modelcontextprotocol/sdk/types.js';
@@ -12,7 +12,7 @@ import '../cache/base.cjs';
 import '../internals/helpers/schema.cjs';
 import 'zod';
 import 'zod-to-json-schema';
-import '../emitter-UkIkNrek.cjs';
+import '../emitter-D4OcelZ9.cjs';
 import '../internals/helpers/promise.cjs';
 
 /**

package/dist/tools/mcp.d.ts

@@ -1,4 +1,4 @@
-import { JSONToolOutput, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.js';
+import { Tool, JSONToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.js';
 import { GetRunContext } from '../context.js';
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 import { ListToolsResult } from '@modelcontextprotocol/sdk/types.js';
@@ -12,7 +12,7 @@ import '../cache/base.js';
 import '../internals/helpers/schema.js';
 import 'zod';
 import 'zod-to-json-schema';
-import '../emitter-CL56qab4.js';
+import '../emitter-kHxkUxco.js';
 import '../internals/helpers/promise.js';
 
 /**

package/dist/tools/mcp.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/mcp.ts"],"names":["MCPToolOutput","JSONToolOutput","MCPTool","Tool","name","description","emitter","client","tool","options","Emitter","root","child","namespace","toCamelCase","creator","inputSchema","_run","input","_options","run","result","callTool","arguments","undefined","signal","fromClient","tools","paginate","size","Infinity","handler","cursor","nextCursor","listTools","data","map"],"mappings":";;;;;;;AAmBO,MAAMA,sBAAsBC,cAAAA,CAAAA;EAnBnC;;;AAmBwD;AAEjD,MAAMC,gBAAgBC,IAAAA,CAAAA;EArB7B;;;AAsBkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,OAAAA;AAEAC,EAAAA,MAAAA;AACCC,EAAAA,IAAAA;AAEjB,EAAA,WAAA,CAAmB,EAAED,MAAAA,EAAQC,IAAAA,EAAM,GAAGC,SAAAA,EAAyB;AAC7D,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,IAAAA,GAAOA,IAAAA;AACZ,IAAA,IAAA,CAAKJ,OAAOI,IAAAA,CAAKJ,IAAAA;AACjB,IAAA,IAAA,CAAKC,WAAAA,GACHG,KAAKH,WAAAA,IAAe,sEAAA;AACtB,IAAA,IAAA,CAAKC,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAOC,QAAAA,WAAAA,CAAY,KAAKV,IAAI;;MAChDW,OAAAA,EAAS;KACX,CAAA;AACF;EAEOC,WAAAA,GAAc;AACnB,IAAA,OAAO,KAAKR,IAAAA,CAAKQ,WAAAA;AACnB;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAAS,MAAM,IAAA,CAAKd,MAAAA,CAAOe,QAAAA,CAAS;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMmB,SAAAA,EAAWL;AAAM,KAAA,EAAGM,MAAAA,EAAW;AAC1FC,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,OAAO,IAAIzB,cAAcqB,MAAAA,CAAAA;AAC3B;AAEA,EAAA,aAAoBK,WAAWnB,MAAAA,EAAuC;AACpE,IAAA,MAAMoB,KAAAA,GAAQ,MAAMC,QAAAA,CAAS;MAC3BC,IAAAA,EAAMC,QAAAA;MACNC,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,MAAAA,EAAM,KAAuB;AAC7C,QAAA,MAAM,EAAEL,KAAAA,EAAAA,MAAAA,EAAOM,YAAU,GAAK,MAAM1B,OAAO2B,SAAAA,CAAU;AAAEF,UAAAA;SAAO,CAAA;AAC9D,QAAA,OAAO;UAAEG,IAAAA,EAAMR,MAAAA;AAAOM,UAAAA;AAAW,SAAA;OACnC,EAHS,SAAA;KAIX,CAAA;AACA,IAAA,OAAON,KAAAA,CAAMS,GAAAA,CAAI,CAAC5B,IAAAA,KAAS,IAAIN,OAAAA,CAAQ;AAAEK,MAAAA,MAAAA;AAAQC,MAAAA;AAAK,KAAA,CAAA,CAAA;AACxD;AACF","file":"mcp.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolRunOptions, ToolEmitter, ToolInput, JSONToolOutput, Tool } from \"@/tools/base.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { ListToolsResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport { SchemaObject } from \"ajv\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { toCamelCase } from \"remeda\";\n\nexport interface MCPToolInput {\n  client: MCPClient;\n  tool: ListToolsResult[\"tools\"][number];\n}\n\nexport class MCPToolOutput extends JSONToolOutput<any> {}\n\nexport class MCPTool extends Tool<MCPToolOutput> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly emitter: ToolEmitter<ToolInput<this>, MCPToolOutput>;\n\n  public readonly client: MCPClient;\n  private readonly tool: ListToolsResult[\"tools\"][number];\n\n  public constructor({ client, tool, ...options }: MCPToolInput) {\n    super(options);\n    this.client = client;\n    this.tool = tool;\n    this.name = tool.name;\n    this.description =\n      tool.description ?? \"No available description, use the tool based on its name and schema.\";\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"mcp\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  public inputSchema() {\n    return this.tool.inputSchema as SchemaObject;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: BaseToolRunOptions,\n    run: GetRunContext<typeof this>,\n  ) {\n    const result = await this.client.callTool({ name: this.name, arguments: input }, undefined, {\n      signal: run.signal,\n    });\n    return new MCPToolOutput(result);\n  }\n\n  public static async fromClient(client: MCPClient): Promise<MCPTool[]> {\n    const tools = await paginate({\n      size: Infinity,\n      handler: async ({ cursor }: { cursor?: string }) => {\n        const { tools, nextCursor } = await client.listTools({ cursor });\n        return { data: tools, nextCursor } as const;\n      },\n    });\n    return tools.map((tool) => new MCPTool({ client, tool }));\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/mcp.ts"],"names":["MCPToolOutput","JSONToolOutput","MCPTool","Tool","name","description","emitter","client","tool","options","Emitter","root","child","namespace","toCamelCase","creator","inputSchema","_run","input","_options","run","result","callTool","arguments","undefined","signal","fromClient","tools","paginate","size","Infinity","handler","cursor","nextCursor","listTools","data","map"],"mappings":";;;;;;;AAmBO,MAAMA,sBAAsBC,cAAAA,CAAAA;EAnBnC;;;AAmBwD;AAEjD,MAAMC,gBAAgBC,IAAAA,CAAAA;EArB7B;;;AAsBkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,OAAAA;AAEAC,EAAAA,MAAAA;AACCC,EAAAA,IAAAA;AAEjB,EAAA,WAAA,CAAmB,EAAED,MAAAA,EAAQC,IAAAA,EAAM,GAAGC,SAAAA,EAAyB;AAC7D,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,IAAAA,GAAOA,IAAAA;AACZ,IAAA,IAAA,CAAKJ,OAAOI,IAAAA,CAAKJ,IAAAA;AACjB,IAAA,IAAA,CAAKC,WAAAA,GACHG,KAAKH,WAAAA,IAAe,sEAAA;AACtB,IAAA,IAAA,CAAKC,OAAAA,GAAUI,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAOC,QAAAA,WAAAA,CAAY,KAAKV,IAAI;;MAChDW,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEOC,WAAAA,GAAc;AACnB,IAAA,OAAO,KAAKR,IAAAA,CAAKQ,WAAAA;AACnB,EAAA;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAAS,MAAM,IAAA,CAAKd,MAAAA,CAAOe,QAAAA,CAAS;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMmB,SAAAA,EAAWL;AAAM,KAAA,EAAGM,MAAAA,EAAW;AAC1FC,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,OAAO,IAAIzB,cAAcqB,MAAAA,CAAAA;AAC3B,EAAA;AAEA,EAAA,aAAoBK,WAAWnB,MAAAA,EAAuC;AACpE,IAAA,MAAMoB,KAAAA,GAAQ,MAAMC,QAAAA,CAAS;MAC3BC,IAAAA,EAAMC,QAAAA;MACNC,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,MAAAA,EAAM,KAAuB;AAC7C,QAAA,MAAM,EAAEL,KAAAA,EAAAA,MAAAA,EAAOM,YAAU,GAAK,MAAM1B,OAAO2B,SAAAA,CAAU;AAAEF,UAAAA;SAAO,CAAA;AAC9D,QAAA,OAAO;UAAEG,IAAAA,EAAMR,MAAAA;AAAOM,UAAAA;AAAW,SAAA;MACnC,CAAA,EAHS,SAAA;KAIX,CAAA;AACA,IAAA,OAAON,KAAAA,CAAMS,GAAAA,CAAI,CAAC5B,IAAAA,KAAS,IAAIN,OAAAA,CAAQ;AAAEK,MAAAA,MAAAA;AAAQC,MAAAA;AAAK,KAAA,CAAA,CAAA;AACxD,EAAA;AACF","file":"mcp.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolRunOptions, ToolEmitter, ToolInput, JSONToolOutput, Tool } from \"@/tools/base.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { ListToolsResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport { SchemaObject } from \"ajv\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { toCamelCase } from \"remeda\";\n\nexport interface MCPToolInput {\n  client: MCPClient;\n  tool: ListToolsResult[\"tools\"][number];\n}\n\nexport class MCPToolOutput extends JSONToolOutput<any> {}\n\nexport class MCPTool extends Tool<MCPToolOutput> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly emitter: ToolEmitter<ToolInput<this>, MCPToolOutput>;\n\n  public readonly client: MCPClient;\n  private readonly tool: ListToolsResult[\"tools\"][number];\n\n  public constructor({ client, tool, ...options }: MCPToolInput) {\n    super(options);\n    this.client = client;\n    this.tool = tool;\n    this.name = tool.name;\n    this.description =\n      tool.description ?? \"No available description, use the tool based on its name and schema.\";\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"mcp\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  public inputSchema() {\n    return this.tool.inputSchema as SchemaObject;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: BaseToolRunOptions,\n    run: GetRunContext<typeof this>,\n  ) {\n    const result = await this.client.callTool({ name: this.name, arguments: input }, undefined, {\n      signal: run.signal,\n    });\n    return new MCPToolOutput(result);\n  }\n\n  public static async fromClient(client: MCPClient): Promise<MCPTool[]> {\n    const tools = await paginate({\n      size: Infinity,\n      handler: async ({ cursor }: { cursor?: string }) => {\n        const { tools, nextCursor } = await client.listTools({ cursor });\n        return { data: tools, nextCursor } as const;\n      },\n    });\n    return tools.map((tool) => new MCPTool({ client, tool }));\n  }\n}\n"]}

package/dist/tools/openapi.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/openapi.ts"],"names":["OpenAPIToolOutput","StringToolOutput","status","statusText","result","OpenAPITool","Tool","name","description","url","openApiSchema","inputSchema","getReferencedObject","json","refPath","pathSegments","split","currentObject","segment","type","required","oneOf","Object","entries","paths","flatMap","path","pathSpec","map","method","methodSpec","additionalProperties","properties","const","summary","requestBody","content","schema","body","parameters","filter","p","reduce","acc","$ref","emitter","register","options","parse","servers","find","s","ValueError","info","title","trim","Emitter","root","child","namespace","toCamelCase","creator","_run","input","_options","run","URL","keys","forEach","key","value","newPath","replace","searchParams","append","pathname","join","fetchOptions","signal","AbortSignal","any","isTruthy","isEmpty","undefined","toLowerCase","headers","Accept","emit","response","fetch","toString","text","output","data","err","ToolError","createSnapshot","clone"],"mappings":";;;;;;;;;;;AAmCO,MAAMA,0BAA0BC,yBAAAA,CAAAA;EAnCvC;;;;;;EAoCE,WAAA,CACkBC,MAAAA,EACAC,UAAAA,EACAC,MAAAA,GAAS,EAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAJWF,MAAAA,GAAAA,MAAAA,EAAAA,KACAC,UAAAA,GAAAA,UAAAA,EAAAA,KACAC,MAAAA,GAAAA,MAAAA;AAGhB,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,UAAAA,GAAaA,UAAAA;AAClB,IAAA,IAAA,CAAKC,SAASA,MAAAA,IAAU,EAAA;AAC1B;AACF;AAEO,MAAMC,oBAAoBC,aAAAA,CAAAA;EAhDjC;;;AAiDkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,GAAAA;AAEAC,EAAAA,aAAAA;EAEhBC,WAAAA,GAA4B;AAC1B,IAAA,MAAMC,mBAAAA,mBAAsB,MAAA,CAAA,CAACC,IAAAA,EAAWC,OAAAA,KAAAA;AACtC,MAAA,MAAMC,YAAAA,GAAeD,OAAAA,CAAQE,KAAAA,CAAM,GAAA,CAAA;AACnC,MAAA,IAAIC,aAAAA,GAAgBJ,IAAAA;AACpB,MAAA,KAAA,MAAWK,WAAWH,YAAAA,EAAc;AAClC,QAAA,IAAIG,YAAY,GAAA,EAAK;AACnB,UAAA;AACF;AACAD,QAAAA,aAAAA,GAAgBA,cAAcC,OAAAA,CAAAA;AAChC;AACA,MAAA,OAAOD,aAAAA;KACT,EAV4B,qBAAA,CAAA;AAY5B,IAAA,OAAO;MACLE,IAAAA,EAAM,QAAA;MACNC,QAAAA,EAAU;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACnBC,KAAAA,EAAOC,MAAAA,CAAOC,QAAQ,IAAA,CAAKb,aAAAA,CAAcc,KAAK,CAAA,CAAEC,OAAAA,CAAQ,CAAC,CAACC,IAAAA,EAAMC,QAAAA,CAAAA,KAC9DL,MAAAA,CAAOC,QAAQI,QAAAA,CAAAA,CAAUC,IAAI,CAAC,CAACC,MAAAA,EAAQC,UAAAA,CAAAA,MAAgC;QACrEC,oBAAAA,EAAsB,KAAA;QACtBC,UAAAA,EAAY;UACVN,IAAAA,EAAM;YACJO,KAAAA,EAAOP,IAAAA;YACPlB,WAAAA,EACE;AACJ,WAAA;UACAqB,MAAAA,EAAQ;YAAEI,KAAAA,EAAOJ,MAAAA;YAAQrB,WAAAA,EAAasB,UAAAA,CAAWI,WAAWJ,UAAAA,CAAWtB;AAAY,WAAA;AACnF,UAAA,GAAIsB,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,GAAqBC,MAAAA,GACvD;AACEC,YAAAA,IAAAA,EAAMR,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,CAAA,EAAqBC;AAC/D,WAAA,GACA,EAAC;AACL,UAAA,GAAIP,WAAWS,UAAAA,GACX;YACEA,UAAAA,EAAY;cACVpB,IAAAA,EAAM,QAAA;cACNY,oBAAAA,EAAsB,KAAA;AACtBX,cAAAA,QAAAA,EAAUU,UAAAA,CAAWS,UAAAA,CAClBC,MAAAA,CAAO,CAACC,CAAAA,KAAWA,CAAAA,CAAErB,QAAAA,KAAa,IAAA,CAAA,CAClCQ,GAAAA,CAAI,CAACa,CAAAA,KAAWA,EAAElC,IAAI,CAAA;cACzByB,UAAAA,EAAYF,UAAAA,CAAWS,WAAWG,MAAAA,CAChC,CAACC,KAAUF,CAAAA,KACT,CAACA,EAAEG,IAAAA,GACC;gBAAE,GAAGD,GAAAA;gBAAK,CAACF,CAAAA,CAAElC,IAAI,GAAG;AAAE,kBAAA,GAAGkC,CAAAA,CAAEJ,MAAAA;AAAQ7B,kBAAAA,WAAAA,EAAaiC,CAAAA,CAAElC;AAAK;eAAE,GACzD;gBACE,GAAGoC,GAAAA;AACH,gBAAA,CAAC/B,oBAAoB,IAAA,CAAKF,aAAAA,EAAe+B,EAAEG,IAAI,CAAA,EAAGrC,IAAAA,GAAO;AACvD,kBAAA,GAAGK,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGP,MAAAA;AACpD7B,kBAAAA,WAAAA,EAAaI,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGrC;AAChE;AACF,eAAA,EACN,EAAC;AAEL;AACF,WAAA,GACA;AACN;AACF,OAAA,CAAA,CAAA;AAEJ,KAAA;AACF;AAEgBsC,EAAAA,OAAAA;EAEhB;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAA6B;AAC9C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AAEN,IAAA,IAAA,CAAKrC,aAAAA,GAAgBsC,UAAAA,CAAMD,OAAAA,CAAQrC,aAAa,CAAA;AAEhD,IAAA,IAAA,CAAKD,GAAAA,GAAM,IAAA,CAAKsC,OAAAA,CAAQtC,GAAAA,IAAO,IAAA,CAAKC,aAAAA,CAAcuC,OAAAA,EAASC,IAAAA,CAAK,CAACC,CAAAA,KAAWA,CAAAA,EAAG1C,GAAAA,CAAAA,EAAMA,GAAAA;AACrF,IAAA,IAAI,CAAC,KAAKA,GAAAA,EAAK;AACb,MAAA,MAAM,IAAI2C,sBACR,wEAAA,CAAA;AAEJ;AAEA,IAAA,IAAA,CAAK7C,OAAOwC,OAAAA,CAAQxC,IAAAA,IAAQ,KAAKG,aAAAA,EAAe2C,IAAAA,EAAMC,OAAOC,IAAAA,EAAAA;AAC7D,IAAA,IAAI,CAAC,KAAKhD,IAAAA,EAAM;AACd,MAAA,MAAM,IAAI6C,sBAAW,2DAAA,CAAA;AACvB;AACA,IAAA,IAAA,CAAKP,OAAAA,GAAUW,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAO,QAAA,SAAA;AAAWC,QAAAA,kBAAAA,CAAY,KAAKrD,IAAI;;MAC3DsD,OAAAA,EAAS;KACX,CAAA;AAEA,IAAA,IAAA,CAAKrD,cACHuC,OAAAA,CAAQvC,WAAAA,IACR,IAAA,CAAKE,aAAAA,EAAe2C,MAAM7C,WAAAA,IAC1B,4MAAA;AAGF,IAAA,IAAI,CAAC,KAAKA,WAAAA,EAAa;AACrB,MAAA,MAAM,IAAI4C,sBAAW,kEAAA,CAAA;AACvB;AACF;EAEA,MAAgBU,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,IAAIvC,MAAAA,GAAeqC,MAAMrC,IAAAA,IAAQ,EAAA;AACjC,IAAA,MAAMjB,GAAAA,GAAM,IAAIyD,GAAAA,CAAI,IAAA,CAAKzD,GAAG,CAAA;AAC5Ba,IAAAA,MAAAA,CAAO6C,IAAAA,CAAKJ,MAAMxB,UAAAA,IAAc,EAAC,CAAA,CAAG6B,OAAAA,CAAQ,CAACC,GAAAA,KAAAA;AAC3C,MAAA,MAAMC,KAAAA,GAAQP,KAAAA,CAAMxB,UAAAA,CAAW8B,GAAAA,CAAAA;AAC/B,MAAA,MAAME,UAAU7C,MAAAA,CAAK8C,OAAAA,CAAQ,CAAA,CAAA,EAAIH,GAAAA,KAAQC,KAAAA,CAAAA;AACzC,MAAA,IAAIC,WAAW7C,MAAAA,EAAM;AACnBjB,QAAAA,GAAAA,CAAIgE,YAAAA,CAAaC,MAAAA,CAAOL,GAAAA,EAAKC,KAAAA,CAAAA;OAC/B,MAAO;AACL5C,QAAAA,MAAAA,GAAO6C,OAAAA;AACT;KACF,CAAA;AACA9D,IAAAA,GAAAA,CAAIkE,QAAAA,GAAWC,SAAAA,CAAKnE,GAAAA,CAAIkE,QAAAA,EAAUjD,MAAAA,CAAAA;AAClC,IAAA,MAAMqB,OAAAA,GAAuB;AAC3B,MAAA,GAAG,KAAKA,OAAAA,CAAQ8B,YAAAA;AAChBC,MAAAA,MAAAA,EAAQC,YAAYC,GAAAA,CAAI;QAACf,GAAAA,CAAIa,MAAAA;AAAQ,QAAA,IAAA,CAAK/B,QAAQ8B,YAAAA,EAAcC;AAAQtC,OAAAA,CAAAA,MAAAA,CAAOyC,eAAAA,CAAAA,CAAAA;AAC/E3C,MAAAA,IAAAA,EAAM,CAAC4C,cAAAA,CAAQnB,KAAAA,CAAMzB,IAAI,CAAA,GAAIyB,MAAMzB,IAAAA,GAAO6C,MAAAA;MAC1CtD,MAAAA,EAAQkC,KAAAA,CAAMlC,OAAOuD,WAAAA,EAAW;MAChCC,OAAAA,EAAS;QAAEC,MAAAA,EAAQ,kBAAA;QAAoB,GAAG,IAAA,CAAKvC,QAAQ8B,YAAAA,EAAcQ;AAAQ;AAC/E,KAAA;AACA,IAAA,MAAMpB,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,aAAA,EAAe;AAAExC,MAAAA,OAAAA;AAAkBtC,MAAAA;KAAS,CAAA;AACnE,IAAA,IAAI;AACF,MAAA,MAAM+E,WAAW,MAAMC,KAAAA,CAAMhF,GAAAA,CAAIiF,QAAAA,IAAY3C,OAAAA,CAAAA;AAC7C,MAAA,MAAM4C,IAAAA,GAAO,MAAMH,QAAAA,CAASG,IAAAA,EAAI;AAChC,MAAA,MAAMC,SAAS,IAAI5F,iBAAAA,CAAkBwF,SAAStF,MAAAA,EAAQsF,QAAAA,CAASrF,YAAYwF,IAAAA,CAAAA;AAC3E,MAAA,MAAM1B,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,YAAA,EAAc;QAAEM,IAAAA,EAAMD,MAAAA;AAAQnF,QAAAA;OAAI,CAAA;AACzD,MAAA,OAAOmF,MAAAA;AACT,KAAA,CAAA,OAASE,GAAAA,EAAK;AACZ,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,WAAA,EAActF,GAAAA,CAAAA,YAAAA,CAAAA,EAAmB;AAACqF,QAAAA;AAAI,OAAA,CAAA;AAC5D;AACF;EACAE,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTtF,aAAAA,EAAeuF,YAAAA,CAAM,KAAKvF,aAAa,CAAA;AACvCH,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;AACXC,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;AAClBC,MAAAA,GAAAA,EAAK,IAAA,CAAKA;AACZ,KAAA;AACF;AACF","file":"openapi.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { join } from \"path\";\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ValueError } from \"@/errors.js\";\nimport { SchemaObject } from \"ajv\";\nimport { parse } from \"yaml\";\nimport { isEmpty, isTruthy, clone, toCamelCase } from \"remeda\";\n\nexport interface OpenAPIToolOptions extends BaseToolOptions {\n  name?: string;\n  description?: string;\n  openApiSchema: any;\n  url?: string;\n  fetchOptions?: RequestInit;\n}\n\nexport interface OpenAPIEvents {\n  beforeFetch: Callback<{ options: RequestInit; url: URL }>;\n  afterFetch: Callback<{ data: OpenAPIToolOutput; url: URL }>;\n}\n\nexport class OpenAPIToolOutput extends StringToolOutput {\n  constructor(\n    public readonly status: number,\n    public readonly statusText: string,\n    public readonly result = \"\",\n  ) {\n    super();\n    this.status = status;\n    this.statusText = statusText;\n    this.result = result ?? \"\";\n  }\n}\n\nexport class OpenAPITool extends Tool<OpenAPIToolOutput, OpenAPIToolOptions> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly url: string;\n\n  public readonly openApiSchema: any;\n\n  inputSchema(): SchemaObject {\n    const getReferencedObject = (json: any, refPath: string) => {\n      const pathSegments = refPath.split(\"/\");\n      let currentObject = json;\n      for (const segment of pathSegments) {\n        if (segment === \"#\") {\n          continue;\n        }\n        currentObject = currentObject[segment];\n      }\n      return currentObject;\n    };\n\n    return {\n      type: \"object\",\n      required: [\"path\", \"method\"],\n      oneOf: Object.entries(this.openApiSchema.paths).flatMap(([path, pathSpec]: [string, any]) =>\n        Object.entries(pathSpec).map(([method, methodSpec]: [string, any]) => ({\n          additionalProperties: false,\n          properties: {\n            path: {\n              const: path,\n              description:\n                \"Do not replace variables in path, instead of, put them to the parameters object.\",\n            },\n            method: { const: method, description: methodSpec.summary || methodSpec.description },\n            ...(methodSpec.requestBody?.content?.[\"application/json\"]?.schema\n              ? {\n                  body: methodSpec.requestBody?.content?.[\"application/json\"]?.schema,\n                }\n              : {}),\n            ...(methodSpec.parameters\n              ? {\n                  parameters: {\n                    type: \"object\",\n                    additionalProperties: false,\n                    required: methodSpec.parameters\n                      .filter((p: any) => p.required === true)\n                      .map((p: any) => p.name),\n                    properties: methodSpec.parameters.reduce(\n                      (acc: any, p: any) =>\n                        !p.$ref\n                          ? { ...acc, [p.name]: { ...p.schema, description: p.name } }\n                          : {\n                              ...acc,\n                              [getReferencedObject(this.openApiSchema, p.$ref)?.name]: {\n                                ...getReferencedObject(this.openApiSchema, p.$ref)?.schema,\n                                description: getReferencedObject(this.openApiSchema, p.$ref)?.name,\n                              },\n                            },\n                      {},\n                    ),\n                  },\n                }\n              : {}),\n          },\n        })),\n      ),\n    } as const satisfies SchemaObject;\n  }\n\n  public readonly emitter: ToolEmitter<Record<string, any>, OpenAPIToolOutput, OpenAPIEvents>;\n\n  static {\n    this.register();\n  }\n\n  public constructor(options: OpenAPIToolOptions) {\n    super(options);\n\n    this.openApiSchema = parse(options.openApiSchema);\n\n    this.url = this.options.url || this.openApiSchema.servers?.find((s: any) => s?.url)?.url;\n    if (!this.url) {\n      throw new ValueError(\n        \"OpenAPI schema hasn't any server with url specified. Pass it manually.\",\n      );\n    }\n\n    this.name = options.name ?? this.openApiSchema?.info?.title?.trim();\n    if (!this.name) {\n      throw new ValueError(\"OpenAPI schema hasn't 'name' specified. Pass it manually.\");\n    }\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"web\", \"openAPI\", toCamelCase(this.name)],\n      creator: this,\n    });\n\n    this.description =\n      options.description ||\n      this.openApiSchema?.info?.description ||\n      \"Performs REST API requests to the servers and retrieves the response. The server API interfaces are defined in OpenAPI schema. \\n\" +\n        \"Only use the OpenAPI tool if you need to communicate to external servers.\";\n\n    if (!this.description) {\n      throw new ValueError(\"OpenAPI schema hasn't 'description' specified. Pass it manually.\");\n    }\n  }\n\n  protected async _run(\n    input: Record<string, any>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ) {\n    let path: string = input.path || \"\";\n    const url = new URL(this.url);\n    Object.keys(input.parameters ?? {}).forEach((key) => {\n      const value = input.parameters[key];\n      const newPath = path.replace(`{${key}}`, value);\n      if (newPath == path) {\n        url.searchParams.append(key, value);\n      } else {\n        path = newPath;\n      }\n    });\n    url.pathname = join(url.pathname, path);\n    const options: RequestInit = {\n      ...this.options.fetchOptions,\n      signal: AbortSignal.any([run.signal, this.options.fetchOptions?.signal].filter(isTruthy)),\n      body: !isEmpty(input.body) ? input.body : undefined,\n      method: input.method.toLowerCase(),\n      headers: { Accept: \"application/json\", ...this.options.fetchOptions?.headers },\n    };\n    await run.emitter.emit(\"beforeFetch\", { options: options, url: url });\n    try {\n      const response = await fetch(url.toString(), options);\n      const text = await response.text();\n      const output = new OpenAPIToolOutput(response.status, response.statusText, text);\n      await run.emitter.emit(\"afterFetch\", { data: output, url });\n      return output;\n    } catch (err) {\n      throw new ToolError(`Request to ${url} has failed.`, [err]);\n    }\n  }\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      openApiSchema: clone(this.openApiSchema),\n      name: this.name,\n      description: this.description,\n      url: this.url,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/openapi.ts"],"names":["OpenAPIToolOutput","StringToolOutput","status","statusText","result","OpenAPITool","Tool","name","description","url","openApiSchema","inputSchema","getReferencedObject","json","refPath","pathSegments","split","currentObject","segment","type","required","oneOf","Object","entries","paths","flatMap","path","pathSpec","map","method","methodSpec","additionalProperties","properties","const","summary","requestBody","content","schema","body","parameters","filter","p","reduce","acc","$ref","emitter","register","options","parse","servers","find","s","ValueError","info","title","trim","Emitter","root","child","namespace","toCamelCase","creator","_run","input","_options","run","URL","keys","forEach","key","value","newPath","replace","searchParams","append","pathname","join","fetchOptions","signal","AbortSignal","any","isTruthy","isEmpty","undefined","toLowerCase","headers","Accept","emit","response","fetch","toString","text","output","data","err","ToolError","createSnapshot","clone"],"mappings":";;;;;;;;;;;AAmCO,MAAMA,0BAA0BC,yBAAAA,CAAAA;EAnCvC;;;;;;EAoCE,WAAA,CACkBC,MAAAA,EACAC,UAAAA,EACAC,MAAAA,GAAS,EAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAJWF,MAAAA,GAAAA,MAAAA,EAAAA,KACAC,UAAAA,GAAAA,UAAAA,EAAAA,KACAC,MAAAA,GAAAA,MAAAA;AAGhB,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,UAAAA,GAAaA,UAAAA;AAClB,IAAA,IAAA,CAAKC,SAASA,MAAAA,IAAU,EAAA;AAC1B,EAAA;AACF;AAEO,MAAMC,oBAAoBC,aAAAA,CAAAA;EAhDjC;;;AAiDkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,GAAAA;AAEAC,EAAAA,aAAAA;EAEhBC,WAAAA,GAA4B;AAC1B,IAAA,MAAMC,mBAAAA,mBAAsB,MAAA,CAAA,CAACC,IAAAA,EAAWC,OAAAA,KAAAA;AACtC,MAAA,MAAMC,YAAAA,GAAeD,OAAAA,CAAQE,KAAAA,CAAM,GAAA,CAAA;AACnC,MAAA,IAAIC,aAAAA,GAAgBJ,IAAAA;AACpB,MAAA,KAAA,MAAWK,WAAWH,YAAAA,EAAc;AAClC,QAAA,IAAIG,YAAY,GAAA,EAAK;AACnB,UAAA;AACF,QAAA;AACAD,QAAAA,aAAAA,GAAgBA,cAAcC,OAAAA,CAAAA;AAChC,MAAA;AACA,MAAA,OAAOD,aAAAA;IACT,CAAA,EAV4B,qBAAA,CAAA;AAY5B,IAAA,OAAO;MACLE,IAAAA,EAAM,QAAA;MACNC,QAAAA,EAAU;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACnBC,KAAAA,EAAOC,MAAAA,CAAOC,QAAQ,IAAA,CAAKb,aAAAA,CAAcc,KAAK,CAAA,CAAEC,OAAAA,CAAQ,CAAC,CAACC,IAAAA,EAAMC,QAAAA,CAAAA,KAC9DL,MAAAA,CAAOC,QAAQI,QAAAA,CAAAA,CAAUC,IAAI,CAAC,CAACC,MAAAA,EAAQC,UAAAA,CAAAA,MAAgC;QACrEC,oBAAAA,EAAsB,KAAA;QACtBC,UAAAA,EAAY;UACVN,IAAAA,EAAM;YACJO,KAAAA,EAAOP,IAAAA;YACPlB,WAAAA,EACE;AACJ,WAAA;UACAqB,MAAAA,EAAQ;YAAEI,KAAAA,EAAOJ,MAAAA;YAAQrB,WAAAA,EAAasB,UAAAA,CAAWI,WAAWJ,UAAAA,CAAWtB;AAAY,WAAA;AACnF,UAAA,GAAIsB,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,GAAqBC,MAAAA,GACvD;AACEC,YAAAA,IAAAA,EAAMR,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,CAAA,EAAqBC;AAC/D,WAAA,GACA,EAAC;AACL,UAAA,GAAIP,WAAWS,UAAAA,GACX;YACEA,UAAAA,EAAY;cACVpB,IAAAA,EAAM,QAAA;cACNY,oBAAAA,EAAsB,KAAA;AACtBX,cAAAA,QAAAA,EAAUU,UAAAA,CAAWS,UAAAA,CAClBC,MAAAA,CAAO,CAACC,CAAAA,KAAWA,CAAAA,CAAErB,QAAAA,KAAa,IAAA,CAAA,CAClCQ,GAAAA,CAAI,CAACa,CAAAA,KAAWA,EAAElC,IAAI,CAAA;cACzByB,UAAAA,EAAYF,UAAAA,CAAWS,WAAWG,MAAAA,CAChC,CAACC,KAAUF,CAAAA,KACT,CAACA,EAAEG,IAAAA,GACC;gBAAE,GAAGD,GAAAA;gBAAK,CAACF,CAAAA,CAAElC,IAAI,GAAG;AAAE,kBAAA,GAAGkC,CAAAA,CAAEJ,MAAAA;AAAQ7B,kBAAAA,WAAAA,EAAaiC,CAAAA,CAAElC;AAAK;eAAE,GACzD;gBACE,GAAGoC,GAAAA;AACH,gBAAA,CAAC/B,oBAAoB,IAAA,CAAKF,aAAAA,EAAe+B,EAAEG,IAAI,CAAA,EAAGrC,IAAAA,GAAO;AACvD,kBAAA,GAAGK,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGP,MAAAA;AACpD7B,kBAAAA,WAAAA,EAAaI,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGrC;AAChE;AACF,eAAA,EACN,EAAC;AAEL;AACF,WAAA,GACA;AACN;AACF,OAAA,CAAA,CAAA;AAEJ,KAAA;AACF,EAAA;AAEgBsC,EAAAA,OAAAA;EAEhB;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAA6B;AAC9C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AAEN,IAAA,IAAA,CAAKrC,aAAAA,GAAgBsC,UAAAA,CAAMD,OAAAA,CAAQrC,aAAa,CAAA;AAEhD,IAAA,IAAA,CAAKD,GAAAA,GAAM,IAAA,CAAKsC,OAAAA,CAAQtC,GAAAA,IAAO,IAAA,CAAKC,aAAAA,CAAcuC,OAAAA,EAASC,IAAAA,CAAK,CAACC,CAAAA,KAAWA,CAAAA,EAAG1C,GAAAA,CAAAA,EAAMA,GAAAA;AACrF,IAAA,IAAI,CAAC,KAAKA,GAAAA,EAAK;AACb,MAAA,MAAM,IAAI2C,sBACR,wEAAA,CAAA;AAEJ,IAAA;AAEA,IAAA,IAAA,CAAK7C,OAAOwC,OAAAA,CAAQxC,IAAAA,IAAQ,KAAKG,aAAAA,EAAe2C,IAAAA,EAAMC,OAAOC,IAAAA,EAAAA;AAC7D,IAAA,IAAI,CAAC,KAAKhD,IAAAA,EAAM;AACd,MAAA,MAAM,IAAI6C,sBAAW,2DAAA,CAAA;AACvB,IAAA;AACA,IAAA,IAAA,CAAKP,OAAAA,GAAUW,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAO,QAAA,SAAA;AAAWC,QAAAA,kBAAAA,CAAY,KAAKrD,IAAI;;MAC3DsD,OAAAA,EAAS;KACX,CAAA;AAEA,IAAA,IAAA,CAAKrD,cACHuC,OAAAA,CAAQvC,WAAAA,IACR,IAAA,CAAKE,aAAAA,EAAe2C,MAAM7C,WAAAA,IAC1B,4MAAA;AAGF,IAAA,IAAI,CAAC,KAAKA,WAAAA,EAAa;AACrB,MAAA,MAAM,IAAI4C,sBAAW,kEAAA,CAAA;AACvB,IAAA;AACF,EAAA;EAEA,MAAgBU,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,IAAIvC,MAAAA,GAAeqC,MAAMrC,IAAAA,IAAQ,EAAA;AACjC,IAAA,MAAMjB,GAAAA,GAAM,IAAIyD,GAAAA,CAAI,IAAA,CAAKzD,GAAG,CAAA;AAC5Ba,IAAAA,MAAAA,CAAO6C,IAAAA,CAAKJ,MAAMxB,UAAAA,IAAc,EAAC,CAAA,CAAG6B,OAAAA,CAAQ,CAACC,GAAAA,KAAAA;AAC3C,MAAA,MAAMC,KAAAA,GAAQP,KAAAA,CAAMxB,UAAAA,CAAW8B,GAAAA,CAAAA;AAC/B,MAAA,MAAME,UAAU7C,MAAAA,CAAK8C,OAAAA,CAAQ,CAAA,CAAA,EAAIH,GAAAA,KAAQC,KAAAA,CAAAA;AACzC,MAAA,IAAIC,WAAW7C,MAAAA,EAAM;AACnBjB,QAAAA,GAAAA,CAAIgE,YAAAA,CAAaC,MAAAA,CAAOL,GAAAA,EAAKC,KAAAA,CAAAA;MAC/B,CAAA,MAAO;AACL5C,QAAAA,MAAAA,GAAO6C,OAAAA;AACT,MAAA;IACF,CAAA,CAAA;AACA9D,IAAAA,GAAAA,CAAIkE,QAAAA,GAAWC,SAAAA,CAAKnE,GAAAA,CAAIkE,QAAAA,EAAUjD,MAAAA,CAAAA;AAClC,IAAA,MAAMqB,OAAAA,GAAuB;AAC3B,MAAA,GAAG,KAAKA,OAAAA,CAAQ8B,YAAAA;AAChBC,MAAAA,MAAAA,EAAQC,YAAYC,GAAAA,CAAI;QAACf,GAAAA,CAAIa,MAAAA;AAAQ,QAAA,IAAA,CAAK/B,QAAQ8B,YAAAA,EAAcC;AAAQtC,OAAAA,CAAAA,MAAAA,CAAOyC,eAAAA,CAAAA,CAAAA;AAC/E3C,MAAAA,IAAAA,EAAM,CAAC4C,cAAAA,CAAQnB,KAAAA,CAAMzB,IAAI,CAAA,GAAIyB,MAAMzB,IAAAA,GAAO6C,MAAAA;MAC1CtD,MAAAA,EAAQkC,KAAAA,CAAMlC,OAAOuD,WAAAA,EAAW;MAChCC,OAAAA,EAAS;QAAEC,MAAAA,EAAQ,kBAAA;QAAoB,GAAG,IAAA,CAAKvC,QAAQ8B,YAAAA,EAAcQ;AAAQ;AAC/E,KAAA;AACA,IAAA,MAAMpB,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,aAAA,EAAe;AAAExC,MAAAA,OAAAA;AAAkBtC,MAAAA;KAAS,CAAA;AACnE,IAAA,IAAI;AACF,MAAA,MAAM+E,WAAW,MAAMC,KAAAA,CAAMhF,GAAAA,CAAIiF,QAAAA,IAAY3C,OAAAA,CAAAA;AAC7C,MAAA,MAAM4C,IAAAA,GAAO,MAAMH,QAAAA,CAASG,IAAAA,EAAI;AAChC,MAAA,MAAMC,SAAS,IAAI5F,iBAAAA,CAAkBwF,SAAStF,MAAAA,EAAQsF,QAAAA,CAASrF,YAAYwF,IAAAA,CAAAA;AAC3E,MAAA,MAAM1B,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,YAAA,EAAc;QAAEM,IAAAA,EAAMD,MAAAA;AAAQnF,QAAAA;OAAI,CAAA;AACzD,MAAA,OAAOmF,MAAAA;AACT,IAAA,CAAA,CAAA,OAASE,GAAAA,EAAK;AACZ,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,WAAA,EAActF,GAAAA,CAAAA,YAAAA,CAAAA,EAAmB;AAACqF,QAAAA;AAAI,OAAA,CAAA;AAC5D,IAAA;AACF,EAAA;EACAE,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTtF,aAAAA,EAAeuF,YAAAA,CAAM,KAAKvF,aAAa,CAAA;AACvCH,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;AACXC,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;AAClBC,MAAAA,GAAAA,EAAK,IAAA,CAAKA;AACZ,KAAA;AACF,EAAA;AACF","file":"openapi.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { join } from \"path\";\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ValueError } from \"@/errors.js\";\nimport { SchemaObject } from \"ajv\";\nimport { parse } from \"yaml\";\nimport { isEmpty, isTruthy, clone, toCamelCase } from \"remeda\";\n\nexport interface OpenAPIToolOptions extends BaseToolOptions {\n  name?: string;\n  description?: string;\n  openApiSchema: any;\n  url?: string;\n  fetchOptions?: RequestInit;\n}\n\nexport interface OpenAPIEvents {\n  beforeFetch: Callback<{ options: RequestInit; url: URL }>;\n  afterFetch: Callback<{ data: OpenAPIToolOutput; url: URL }>;\n}\n\nexport class OpenAPIToolOutput extends StringToolOutput {\n  constructor(\n    public readonly status: number,\n    public readonly statusText: string,\n    public readonly result = \"\",\n  ) {\n    super();\n    this.status = status;\n    this.statusText = statusText;\n    this.result = result ?? \"\";\n  }\n}\n\nexport class OpenAPITool extends Tool<OpenAPIToolOutput, OpenAPIToolOptions> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly url: string;\n\n  public readonly openApiSchema: any;\n\n  inputSchema(): SchemaObject {\n    const getReferencedObject = (json: any, refPath: string) => {\n      const pathSegments = refPath.split(\"/\");\n      let currentObject = json;\n      for (const segment of pathSegments) {\n        if (segment === \"#\") {\n          continue;\n        }\n        currentObject = currentObject[segment];\n      }\n      return currentObject;\n    };\n\n    return {\n      type: \"object\",\n      required: [\"path\", \"method\"],\n      oneOf: Object.entries(this.openApiSchema.paths).flatMap(([path, pathSpec]: [string, any]) =>\n        Object.entries(pathSpec).map(([method, methodSpec]: [string, any]) => ({\n          additionalProperties: false,\n          properties: {\n            path: {\n              const: path,\n              description:\n                \"Do not replace variables in path, instead of, put them to the parameters object.\",\n            },\n            method: { const: method, description: methodSpec.summary || methodSpec.description },\n            ...(methodSpec.requestBody?.content?.[\"application/json\"]?.schema\n              ? {\n                  body: methodSpec.requestBody?.content?.[\"application/json\"]?.schema,\n                }\n              : {}),\n            ...(methodSpec.parameters\n              ? {\n                  parameters: {\n                    type: \"object\",\n                    additionalProperties: false,\n                    required: methodSpec.parameters\n                      .filter((p: any) => p.required === true)\n                      .map((p: any) => p.name),\n                    properties: methodSpec.parameters.reduce(\n                      (acc: any, p: any) =>\n                        !p.$ref\n                          ? { ...acc, [p.name]: { ...p.schema, description: p.name } }\n                          : {\n                              ...acc,\n                              [getReferencedObject(this.openApiSchema, p.$ref)?.name]: {\n                                ...getReferencedObject(this.openApiSchema, p.$ref)?.schema,\n                                description: getReferencedObject(this.openApiSchema, p.$ref)?.name,\n                              },\n                            },\n                      {},\n                    ),\n                  },\n                }\n              : {}),\n          },\n        })),\n      ),\n    } as const satisfies SchemaObject;\n  }\n\n  public readonly emitter: ToolEmitter<Record<string, any>, OpenAPIToolOutput, OpenAPIEvents>;\n\n  static {\n    this.register();\n  }\n\n  public constructor(options: OpenAPIToolOptions) {\n    super(options);\n\n    this.openApiSchema = parse(options.openApiSchema);\n\n    this.url = this.options.url || this.openApiSchema.servers?.find((s: any) => s?.url)?.url;\n    if (!this.url) {\n      throw new ValueError(\n        \"OpenAPI schema hasn't any server with url specified. Pass it manually.\",\n      );\n    }\n\n    this.name = options.name ?? this.openApiSchema?.info?.title?.trim();\n    if (!this.name) {\n      throw new ValueError(\"OpenAPI schema hasn't 'name' specified. Pass it manually.\");\n    }\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"web\", \"openAPI\", toCamelCase(this.name)],\n      creator: this,\n    });\n\n    this.description =\n      options.description ||\n      this.openApiSchema?.info?.description ||\n      \"Performs REST API requests to the servers and retrieves the response. The server API interfaces are defined in OpenAPI schema. \\n\" +\n        \"Only use the OpenAPI tool if you need to communicate to external servers.\";\n\n    if (!this.description) {\n      throw new ValueError(\"OpenAPI schema hasn't 'description' specified. Pass it manually.\");\n    }\n  }\n\n  protected async _run(\n    input: Record<string, any>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ) {\n    let path: string = input.path || \"\";\n    const url = new URL(this.url);\n    Object.keys(input.parameters ?? {}).forEach((key) => {\n      const value = input.parameters[key];\n      const newPath = path.replace(`{${key}}`, value);\n      if (newPath == path) {\n        url.searchParams.append(key, value);\n      } else {\n        path = newPath;\n      }\n    });\n    url.pathname = join(url.pathname, path);\n    const options: RequestInit = {\n      ...this.options.fetchOptions,\n      signal: AbortSignal.any([run.signal, this.options.fetchOptions?.signal].filter(isTruthy)),\n      body: !isEmpty(input.body) ? input.body : undefined,\n      method: input.method.toLowerCase(),\n      headers: { Accept: \"application/json\", ...this.options.fetchOptions?.headers },\n    };\n    await run.emitter.emit(\"beforeFetch\", { options: options, url: url });\n    try {\n      const response = await fetch(url.toString(), options);\n      const text = await response.text();\n      const output = new OpenAPIToolOutput(response.status, response.statusText, text);\n      await run.emitter.emit(\"afterFetch\", { data: output, url });\n      return output;\n    } catch (err) {\n      throw new ToolError(`Request to ${url} has failed.`, [err]);\n    }\n  }\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      openApiSchema: clone(this.openApiSchema),\n      name: this.name,\n      description: this.description,\n      url: this.url,\n    };\n  }\n}\n"]}

package/dist/tools/openapi.d.cts

@@ -1,7 +1,7 @@
 import { BaseCache } from '../cache/base.cjs';
 import * as promise_based_task from 'promise-based-task';
-import { BaseToolOptions, StringToolOutput, Tool, ToolEmitter, BaseToolRunOptions } from './base.cjs';
-import { C as Callback, E as Emitter } from '../emitter-UkIkNrek.cjs';
+import { StringToolOutput, Tool, BaseToolOptions, ToolEmitter, BaseToolRunOptions } from './base.cjs';
+import { C as Callback, E as Emitter } from '../emitter-D4OcelZ9.cjs';
 import { GetRunContext } from '../context.cjs';
 import { SchemaObject } from 'ajv';
 import '../internals/serializable.cjs';

package/dist/tools/openapi.d.ts

@@ -1,7 +1,7 @@
 import { BaseCache } from '../cache/base.js';
 import * as promise_based_task from 'promise-based-task';
-import { BaseToolOptions, StringToolOutput, Tool, ToolEmitter, BaseToolRunOptions } from './base.js';
-import { C as Callback, E as Emitter } from '../emitter-CL56qab4.js';
+import { StringToolOutput, Tool, BaseToolOptions, ToolEmitter, BaseToolRunOptions } from './base.js';
+import { C as Callback, E as Emitter } from '../emitter-kHxkUxco.js';
 import { GetRunContext } from '../context.js';
 import { SchemaObject } from 'ajv';
 import '../internals/serializable.js';

package/dist/tools/openapi.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/openapi.ts"],"names":["OpenAPIToolOutput","StringToolOutput","status","statusText","result","OpenAPITool","Tool","name","description","url","openApiSchema","inputSchema","getReferencedObject","json","refPath","pathSegments","split","currentObject","segment","type","required","oneOf","Object","entries","paths","flatMap","path","pathSpec","map","method","methodSpec","additionalProperties","properties","const","summary","requestBody","content","schema","body","parameters","filter","p","reduce","acc","$ref","emitter","register","options","parse","servers","find","s","ValueError","info","title","trim","Emitter","root","child","namespace","toCamelCase","creator","_run","input","_options","run","URL","keys","forEach","key","value","newPath","replace","searchParams","append","pathname","join","fetchOptions","signal","AbortSignal","any","isTruthy","isEmpty","undefined","toLowerCase","headers","Accept","emit","response","fetch","toString","text","output","data","err","ToolError","createSnapshot","clone"],"mappings":";;;;;;;;;AAmCO,MAAMA,0BAA0BC,gBAAAA,CAAAA;EAnCvC;;;;;;EAoCE,WAAA,CACkBC,MAAAA,EACAC,UAAAA,EACAC,MAAAA,GAAS,EAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAJWF,MAAAA,GAAAA,MAAAA,EAAAA,KACAC,UAAAA,GAAAA,UAAAA,EAAAA,KACAC,MAAAA,GAAAA,MAAAA;AAGhB,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,UAAAA,GAAaA,UAAAA;AAClB,IAAA,IAAA,CAAKC,SAASA,MAAAA,IAAU,EAAA;AAC1B;AACF;AAEO,MAAMC,oBAAoBC,IAAAA,CAAAA;EAhDjC;;;AAiDkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,GAAAA;AAEAC,EAAAA,aAAAA;EAEhBC,WAAAA,GAA4B;AAC1B,IAAA,MAAMC,mBAAAA,mBAAsB,MAAA,CAAA,CAACC,IAAAA,EAAWC,OAAAA,KAAAA;AACtC,MAAA,MAAMC,YAAAA,GAAeD,OAAAA,CAAQE,KAAAA,CAAM,GAAA,CAAA;AACnC,MAAA,IAAIC,aAAAA,GAAgBJ,IAAAA;AACpB,MAAA,KAAA,MAAWK,WAAWH,YAAAA,EAAc;AAClC,QAAA,IAAIG,YAAY,GAAA,EAAK;AACnB,UAAA;AACF;AACAD,QAAAA,aAAAA,GAAgBA,cAAcC,OAAAA,CAAAA;AAChC;AACA,MAAA,OAAOD,aAAAA;KACT,EAV4B,qBAAA,CAAA;AAY5B,IAAA,OAAO;MACLE,IAAAA,EAAM,QAAA;MACNC,QAAAA,EAAU;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACnBC,KAAAA,EAAOC,MAAAA,CAAOC,QAAQ,IAAA,CAAKb,aAAAA,CAAcc,KAAK,CAAA,CAAEC,OAAAA,CAAQ,CAAC,CAACC,IAAAA,EAAMC,QAAAA,CAAAA,KAC9DL,MAAAA,CAAOC,QAAQI,QAAAA,CAAAA,CAAUC,IAAI,CAAC,CAACC,MAAAA,EAAQC,UAAAA,CAAAA,MAAgC;QACrEC,oBAAAA,EAAsB,KAAA;QACtBC,UAAAA,EAAY;UACVN,IAAAA,EAAM;YACJO,KAAAA,EAAOP,IAAAA;YACPlB,WAAAA,EACE;AACJ,WAAA;UACAqB,MAAAA,EAAQ;YAAEI,KAAAA,EAAOJ,MAAAA;YAAQrB,WAAAA,EAAasB,UAAAA,CAAWI,WAAWJ,UAAAA,CAAWtB;AAAY,WAAA;AACnF,UAAA,GAAIsB,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,GAAqBC,MAAAA,GACvD;AACEC,YAAAA,IAAAA,EAAMR,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,CAAA,EAAqBC;AAC/D,WAAA,GACA,EAAC;AACL,UAAA,GAAIP,WAAWS,UAAAA,GACX;YACEA,UAAAA,EAAY;cACVpB,IAAAA,EAAM,QAAA;cACNY,oBAAAA,EAAsB,KAAA;AACtBX,cAAAA,QAAAA,EAAUU,UAAAA,CAAWS,UAAAA,CAClBC,MAAAA,CAAO,CAACC,CAAAA,KAAWA,CAAAA,CAAErB,QAAAA,KAAa,IAAA,CAAA,CAClCQ,GAAAA,CAAI,CAACa,CAAAA,KAAWA,EAAElC,IAAI,CAAA;cACzByB,UAAAA,EAAYF,UAAAA,CAAWS,WAAWG,MAAAA,CAChC,CAACC,KAAUF,CAAAA,KACT,CAACA,EAAEG,IAAAA,GACC;gBAAE,GAAGD,GAAAA;gBAAK,CAACF,CAAAA,CAAElC,IAAI,GAAG;AAAE,kBAAA,GAAGkC,CAAAA,CAAEJ,MAAAA;AAAQ7B,kBAAAA,WAAAA,EAAaiC,CAAAA,CAAElC;AAAK;eAAE,GACzD;gBACE,GAAGoC,GAAAA;AACH,gBAAA,CAAC/B,oBAAoB,IAAA,CAAKF,aAAAA,EAAe+B,EAAEG,IAAI,CAAA,EAAGrC,IAAAA,GAAO;AACvD,kBAAA,GAAGK,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGP,MAAAA;AACpD7B,kBAAAA,WAAAA,EAAaI,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGrC;AAChE;AACF,eAAA,EACN,EAAC;AAEL;AACF,WAAA,GACA;AACN;AACF,OAAA,CAAA,CAAA;AAEJ,KAAA;AACF;AAEgBsC,EAAAA,OAAAA;EAEhB;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAA6B;AAC9C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AAEN,IAAA,IAAA,CAAKrC,aAAAA,GAAgBsC,KAAAA,CAAMD,OAAAA,CAAQrC,aAAa,CAAA;AAEhD,IAAA,IAAA,CAAKD,GAAAA,GAAM,IAAA,CAAKsC,OAAAA,CAAQtC,GAAAA,IAAO,IAAA,CAAKC,aAAAA,CAAcuC,OAAAA,EAASC,IAAAA,CAAK,CAACC,CAAAA,KAAWA,CAAAA,EAAG1C,GAAAA,CAAAA,EAAMA,GAAAA;AACrF,IAAA,IAAI,CAAC,KAAKA,GAAAA,EAAK;AACb,MAAA,MAAM,IAAI2C,WACR,wEAAA,CAAA;AAEJ;AAEA,IAAA,IAAA,CAAK7C,OAAOwC,OAAAA,CAAQxC,IAAAA,IAAQ,KAAKG,aAAAA,EAAe2C,IAAAA,EAAMC,OAAOC,IAAAA,EAAAA;AAC7D,IAAA,IAAI,CAAC,KAAKhD,IAAAA,EAAM;AACd,MAAA,MAAM,IAAI6C,WAAW,2DAAA,CAAA;AACvB;AACA,IAAA,IAAA,CAAKP,OAAAA,GAAUW,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAO,QAAA,SAAA;AAAWC,QAAAA,WAAAA,CAAY,KAAKrD,IAAI;;MAC3DsD,OAAAA,EAAS;KACX,CAAA;AAEA,IAAA,IAAA,CAAKrD,cACHuC,OAAAA,CAAQvC,WAAAA,IACR,IAAA,CAAKE,aAAAA,EAAe2C,MAAM7C,WAAAA,IAC1B,4MAAA;AAGF,IAAA,IAAI,CAAC,KAAKA,WAAAA,EAAa;AACrB,MAAA,MAAM,IAAI4C,WAAW,kEAAA,CAAA;AACvB;AACF;EAEA,MAAgBU,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,IAAIvC,IAAAA,GAAeqC,MAAMrC,IAAAA,IAAQ,EAAA;AACjC,IAAA,MAAMjB,GAAAA,GAAM,IAAIyD,GAAAA,CAAI,IAAA,CAAKzD,GAAG,CAAA;AAC5Ba,IAAAA,MAAAA,CAAO6C,IAAAA,CAAKJ,MAAMxB,UAAAA,IAAc,EAAC,CAAA,CAAG6B,OAAAA,CAAQ,CAACC,GAAAA,KAAAA;AAC3C,MAAA,MAAMC,KAAAA,GAAQP,KAAAA,CAAMxB,UAAAA,CAAW8B,GAAAA,CAAAA;AAC/B,MAAA,MAAME,UAAU7C,IAAAA,CAAK8C,OAAAA,CAAQ,CAAA,CAAA,EAAIH,GAAAA,KAAQC,KAAAA,CAAAA;AACzC,MAAA,IAAIC,WAAW7C,IAAAA,EAAM;AACnBjB,QAAAA,GAAAA,CAAIgE,YAAAA,CAAaC,MAAAA,CAAOL,GAAAA,EAAKC,KAAAA,CAAAA;OAC/B,MAAO;AACL5C,QAAAA,IAAAA,GAAO6C,OAAAA;AACT;KACF,CAAA;AACA9D,IAAAA,GAAAA,CAAIkE,QAAAA,GAAWC,IAAAA,CAAKnE,GAAAA,CAAIkE,QAAAA,EAAUjD,IAAAA,CAAAA;AAClC,IAAA,MAAMqB,OAAAA,GAAuB;AAC3B,MAAA,GAAG,KAAKA,OAAAA,CAAQ8B,YAAAA;AAChBC,MAAAA,MAAAA,EAAQC,YAAYC,GAAAA,CAAI;QAACf,GAAAA,CAAIa,MAAAA;AAAQ,QAAA,IAAA,CAAK/B,QAAQ8B,YAAAA,EAAcC;AAAQtC,OAAAA,CAAAA,MAAAA,CAAOyC,QAAAA,CAAAA,CAAAA;AAC/E3C,MAAAA,IAAAA,EAAM,CAAC4C,OAAAA,CAAQnB,KAAAA,CAAMzB,IAAI,CAAA,GAAIyB,MAAMzB,IAAAA,GAAO6C,MAAAA;MAC1CtD,MAAAA,EAAQkC,KAAAA,CAAMlC,OAAOuD,WAAAA,EAAW;MAChCC,OAAAA,EAAS;QAAEC,MAAAA,EAAQ,kBAAA;QAAoB,GAAG,IAAA,CAAKvC,QAAQ8B,YAAAA,EAAcQ;AAAQ;AAC/E,KAAA;AACA,IAAA,MAAMpB,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,aAAA,EAAe;AAAExC,MAAAA,OAAAA;AAAkBtC,MAAAA;KAAS,CAAA;AACnE,IAAA,IAAI;AACF,MAAA,MAAM+E,WAAW,MAAMC,KAAAA,CAAMhF,GAAAA,CAAIiF,QAAAA,IAAY3C,OAAAA,CAAAA;AAC7C,MAAA,MAAM4C,IAAAA,GAAO,MAAMH,QAAAA,CAASG,IAAAA,EAAI;AAChC,MAAA,MAAMC,SAAS,IAAI5F,iBAAAA,CAAkBwF,SAAStF,MAAAA,EAAQsF,QAAAA,CAASrF,YAAYwF,IAAAA,CAAAA;AAC3E,MAAA,MAAM1B,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,YAAA,EAAc;QAAEM,IAAAA,EAAMD,MAAAA;AAAQnF,QAAAA;OAAI,CAAA;AACzD,MAAA,OAAOmF,MAAAA;AACT,KAAA,CAAA,OAASE,GAAAA,EAAK;AACZ,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,WAAA,EAActF,GAAAA,CAAAA,YAAAA,CAAAA,EAAmB;AAACqF,QAAAA;AAAI,OAAA,CAAA;AAC5D;AACF;EACAE,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTtF,aAAAA,EAAeuF,KAAAA,CAAM,KAAKvF,aAAa,CAAA;AACvCH,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;AACXC,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;AAClBC,MAAAA,GAAAA,EAAK,IAAA,CAAKA;AACZ,KAAA;AACF;AACF","file":"openapi.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { join } from \"path\";\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ValueError } from \"@/errors.js\";\nimport { SchemaObject } from \"ajv\";\nimport { parse } from \"yaml\";\nimport { isEmpty, isTruthy, clone, toCamelCase } from \"remeda\";\n\nexport interface OpenAPIToolOptions extends BaseToolOptions {\n  name?: string;\n  description?: string;\n  openApiSchema: any;\n  url?: string;\n  fetchOptions?: RequestInit;\n}\n\nexport interface OpenAPIEvents {\n  beforeFetch: Callback<{ options: RequestInit; url: URL }>;\n  afterFetch: Callback<{ data: OpenAPIToolOutput; url: URL }>;\n}\n\nexport class OpenAPIToolOutput extends StringToolOutput {\n  constructor(\n    public readonly status: number,\n    public readonly statusText: string,\n    public readonly result = \"\",\n  ) {\n    super();\n    this.status = status;\n    this.statusText = statusText;\n    this.result = result ?? \"\";\n  }\n}\n\nexport class OpenAPITool extends Tool<OpenAPIToolOutput, OpenAPIToolOptions> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly url: string;\n\n  public readonly openApiSchema: any;\n\n  inputSchema(): SchemaObject {\n    const getReferencedObject = (json: any, refPath: string) => {\n      const pathSegments = refPath.split(\"/\");\n      let currentObject = json;\n      for (const segment of pathSegments) {\n        if (segment === \"#\") {\n          continue;\n        }\n        currentObject = currentObject[segment];\n      }\n      return currentObject;\n    };\n\n    return {\n      type: \"object\",\n      required: [\"path\", \"method\"],\n      oneOf: Object.entries(this.openApiSchema.paths).flatMap(([path, pathSpec]: [string, any]) =>\n        Object.entries(pathSpec).map(([method, methodSpec]: [string, any]) => ({\n          additionalProperties: false,\n          properties: {\n            path: {\n              const: path,\n              description:\n                \"Do not replace variables in path, instead of, put them to the parameters object.\",\n            },\n            method: { const: method, description: methodSpec.summary || methodSpec.description },\n            ...(methodSpec.requestBody?.content?.[\"application/json\"]?.schema\n              ? {\n                  body: methodSpec.requestBody?.content?.[\"application/json\"]?.schema,\n                }\n              : {}),\n            ...(methodSpec.parameters\n              ? {\n                  parameters: {\n                    type: \"object\",\n                    additionalProperties: false,\n                    required: methodSpec.parameters\n                      .filter((p: any) => p.required === true)\n                      .map((p: any) => p.name),\n                    properties: methodSpec.parameters.reduce(\n                      (acc: any, p: any) =>\n                        !p.$ref\n                          ? { ...acc, [p.name]: { ...p.schema, description: p.name } }\n                          : {\n                              ...acc,\n                              [getReferencedObject(this.openApiSchema, p.$ref)?.name]: {\n                                ...getReferencedObject(this.openApiSchema, p.$ref)?.schema,\n                                description: getReferencedObject(this.openApiSchema, p.$ref)?.name,\n                              },\n                            },\n                      {},\n                    ),\n                  },\n                }\n              : {}),\n          },\n        })),\n      ),\n    } as const satisfies SchemaObject;\n  }\n\n  public readonly emitter: ToolEmitter<Record<string, any>, OpenAPIToolOutput, OpenAPIEvents>;\n\n  static {\n    this.register();\n  }\n\n  public constructor(options: OpenAPIToolOptions) {\n    super(options);\n\n    this.openApiSchema = parse(options.openApiSchema);\n\n    this.url = this.options.url || this.openApiSchema.servers?.find((s: any) => s?.url)?.url;\n    if (!this.url) {\n      throw new ValueError(\n        \"OpenAPI schema hasn't any server with url specified. Pass it manually.\",\n      );\n    }\n\n    this.name = options.name ?? this.openApiSchema?.info?.title?.trim();\n    if (!this.name) {\n      throw new ValueError(\"OpenAPI schema hasn't 'name' specified. Pass it manually.\");\n    }\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"web\", \"openAPI\", toCamelCase(this.name)],\n      creator: this,\n    });\n\n    this.description =\n      options.description ||\n      this.openApiSchema?.info?.description ||\n      \"Performs REST API requests to the servers and retrieves the response. The server API interfaces are defined in OpenAPI schema. \\n\" +\n        \"Only use the OpenAPI tool if you need to communicate to external servers.\";\n\n    if (!this.description) {\n      throw new ValueError(\"OpenAPI schema hasn't 'description' specified. Pass it manually.\");\n    }\n  }\n\n  protected async _run(\n    input: Record<string, any>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ) {\n    let path: string = input.path || \"\";\n    const url = new URL(this.url);\n    Object.keys(input.parameters ?? {}).forEach((key) => {\n      const value = input.parameters[key];\n      const newPath = path.replace(`{${key}}`, value);\n      if (newPath == path) {\n        url.searchParams.append(key, value);\n      } else {\n        path = newPath;\n      }\n    });\n    url.pathname = join(url.pathname, path);\n    const options: RequestInit = {\n      ...this.options.fetchOptions,\n      signal: AbortSignal.any([run.signal, this.options.fetchOptions?.signal].filter(isTruthy)),\n      body: !isEmpty(input.body) ? input.body : undefined,\n      method: input.method.toLowerCase(),\n      headers: { Accept: \"application/json\", ...this.options.fetchOptions?.headers },\n    };\n    await run.emitter.emit(\"beforeFetch\", { options: options, url: url });\n    try {\n      const response = await fetch(url.toString(), options);\n      const text = await response.text();\n      const output = new OpenAPIToolOutput(response.status, response.statusText, text);\n      await run.emitter.emit(\"afterFetch\", { data: output, url });\n      return output;\n    } catch (err) {\n      throw new ToolError(`Request to ${url} has failed.`, [err]);\n    }\n  }\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      openApiSchema: clone(this.openApiSchema),\n      name: this.name,\n      description: this.description,\n      url: this.url,\n    };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/openapi.ts"],"names":["OpenAPIToolOutput","StringToolOutput","status","statusText","result","OpenAPITool","Tool","name","description","url","openApiSchema","inputSchema","getReferencedObject","json","refPath","pathSegments","split","currentObject","segment","type","required","oneOf","Object","entries","paths","flatMap","path","pathSpec","map","method","methodSpec","additionalProperties","properties","const","summary","requestBody","content","schema","body","parameters","filter","p","reduce","acc","$ref","emitter","register","options","parse","servers","find","s","ValueError","info","title","trim","Emitter","root","child","namespace","toCamelCase","creator","_run","input","_options","run","URL","keys","forEach","key","value","newPath","replace","searchParams","append","pathname","join","fetchOptions","signal","AbortSignal","any","isTruthy","isEmpty","undefined","toLowerCase","headers","Accept","emit","response","fetch","toString","text","output","data","err","ToolError","createSnapshot","clone"],"mappings":";;;;;;;;;AAmCO,MAAMA,0BAA0BC,gBAAAA,CAAAA;EAnCvC;;;;;;EAoCE,WAAA,CACkBC,MAAAA,EACAC,UAAAA,EACAC,MAAAA,GAAS,EAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAJWF,MAAAA,GAAAA,MAAAA,EAAAA,KACAC,UAAAA,GAAAA,UAAAA,EAAAA,KACAC,MAAAA,GAAAA,MAAAA;AAGhB,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,UAAAA,GAAaA,UAAAA;AAClB,IAAA,IAAA,CAAKC,SAASA,MAAAA,IAAU,EAAA;AAC1B,EAAA;AACF;AAEO,MAAMC,oBAAoBC,IAAAA,CAAAA;EAhDjC;;;AAiDkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,GAAAA;AAEAC,EAAAA,aAAAA;EAEhBC,WAAAA,GAA4B;AAC1B,IAAA,MAAMC,mBAAAA,mBAAsB,MAAA,CAAA,CAACC,IAAAA,EAAWC,OAAAA,KAAAA;AACtC,MAAA,MAAMC,YAAAA,GAAeD,OAAAA,CAAQE,KAAAA,CAAM,GAAA,CAAA;AACnC,MAAA,IAAIC,aAAAA,GAAgBJ,IAAAA;AACpB,MAAA,KAAA,MAAWK,WAAWH,YAAAA,EAAc;AAClC,QAAA,IAAIG,YAAY,GAAA,EAAK;AACnB,UAAA;AACF,QAAA;AACAD,QAAAA,aAAAA,GAAgBA,cAAcC,OAAAA,CAAAA;AAChC,MAAA;AACA,MAAA,OAAOD,aAAAA;IACT,CAAA,EAV4B,qBAAA,CAAA;AAY5B,IAAA,OAAO;MACLE,IAAAA,EAAM,QAAA;MACNC,QAAAA,EAAU;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACnBC,KAAAA,EAAOC,MAAAA,CAAOC,QAAQ,IAAA,CAAKb,aAAAA,CAAcc,KAAK,CAAA,CAAEC,OAAAA,CAAQ,CAAC,CAACC,IAAAA,EAAMC,QAAAA,CAAAA,KAC9DL,MAAAA,CAAOC,QAAQI,QAAAA,CAAAA,CAAUC,IAAI,CAAC,CAACC,MAAAA,EAAQC,UAAAA,CAAAA,MAAgC;QACrEC,oBAAAA,EAAsB,KAAA;QACtBC,UAAAA,EAAY;UACVN,IAAAA,EAAM;YACJO,KAAAA,EAAOP,IAAAA;YACPlB,WAAAA,EACE;AACJ,WAAA;UACAqB,MAAAA,EAAQ;YAAEI,KAAAA,EAAOJ,MAAAA;YAAQrB,WAAAA,EAAasB,UAAAA,CAAWI,WAAWJ,UAAAA,CAAWtB;AAAY,WAAA;AACnF,UAAA,GAAIsB,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,GAAqBC,MAAAA,GACvD;AACEC,YAAAA,IAAAA,EAAMR,UAAAA,CAAWK,WAAAA,EAAaC,OAAAA,GAAU,kBAAA,CAAA,EAAqBC;AAC/D,WAAA,GACA,EAAC;AACL,UAAA,GAAIP,WAAWS,UAAAA,GACX;YACEA,UAAAA,EAAY;cACVpB,IAAAA,EAAM,QAAA;cACNY,oBAAAA,EAAsB,KAAA;AACtBX,cAAAA,QAAAA,EAAUU,UAAAA,CAAWS,UAAAA,CAClBC,MAAAA,CAAO,CAACC,CAAAA,KAAWA,CAAAA,CAAErB,QAAAA,KAAa,IAAA,CAAA,CAClCQ,GAAAA,CAAI,CAACa,CAAAA,KAAWA,EAAElC,IAAI,CAAA;cACzByB,UAAAA,EAAYF,UAAAA,CAAWS,WAAWG,MAAAA,CAChC,CAACC,KAAUF,CAAAA,KACT,CAACA,EAAEG,IAAAA,GACC;gBAAE,GAAGD,GAAAA;gBAAK,CAACF,CAAAA,CAAElC,IAAI,GAAG;AAAE,kBAAA,GAAGkC,CAAAA,CAAEJ,MAAAA;AAAQ7B,kBAAAA,WAAAA,EAAaiC,CAAAA,CAAElC;AAAK;eAAE,GACzD;gBACE,GAAGoC,GAAAA;AACH,gBAAA,CAAC/B,oBAAoB,IAAA,CAAKF,aAAAA,EAAe+B,EAAEG,IAAI,CAAA,EAAGrC,IAAAA,GAAO;AACvD,kBAAA,GAAGK,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGP,MAAAA;AACpD7B,kBAAAA,WAAAA,EAAaI,mBAAAA,CAAoB,IAAA,CAAKF,aAAAA,EAAe+B,CAAAA,CAAEG,IAAI,CAAA,EAAGrC;AAChE;AACF,eAAA,EACN,EAAC;AAEL;AACF,WAAA,GACA;AACN;AACF,OAAA,CAAA,CAAA;AAEJ,KAAA;AACF,EAAA;AAEgBsC,EAAAA,OAAAA;EAEhB;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAA6B;AAC9C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AAEN,IAAA,IAAA,CAAKrC,aAAAA,GAAgBsC,KAAAA,CAAMD,OAAAA,CAAQrC,aAAa,CAAA;AAEhD,IAAA,IAAA,CAAKD,GAAAA,GAAM,IAAA,CAAKsC,OAAAA,CAAQtC,GAAAA,IAAO,IAAA,CAAKC,aAAAA,CAAcuC,OAAAA,EAASC,IAAAA,CAAK,CAACC,CAAAA,KAAWA,CAAAA,EAAG1C,GAAAA,CAAAA,EAAMA,GAAAA;AACrF,IAAA,IAAI,CAAC,KAAKA,GAAAA,EAAK;AACb,MAAA,MAAM,IAAI2C,WACR,wEAAA,CAAA;AAEJ,IAAA;AAEA,IAAA,IAAA,CAAK7C,OAAOwC,OAAAA,CAAQxC,IAAAA,IAAQ,KAAKG,aAAAA,EAAe2C,IAAAA,EAAMC,OAAOC,IAAAA,EAAAA;AAC7D,IAAA,IAAI,CAAC,KAAKhD,IAAAA,EAAM;AACd,MAAA,MAAM,IAAI6C,WAAW,2DAAA,CAAA;AACvB,IAAA;AACA,IAAA,IAAA,CAAKP,OAAAA,GAAUW,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAO,QAAA,SAAA;AAAWC,QAAAA,WAAAA,CAAY,KAAKrD,IAAI;;MAC3DsD,OAAAA,EAAS;KACX,CAAA;AAEA,IAAA,IAAA,CAAKrD,cACHuC,OAAAA,CAAQvC,WAAAA,IACR,IAAA,CAAKE,aAAAA,EAAe2C,MAAM7C,WAAAA,IAC1B,4MAAA;AAGF,IAAA,IAAI,CAAC,KAAKA,WAAAA,EAAa;AACrB,MAAA,MAAM,IAAI4C,WAAW,kEAAA,CAAA;AACvB,IAAA;AACF,EAAA;EAEA,MAAgBU,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,IAAIvC,IAAAA,GAAeqC,MAAMrC,IAAAA,IAAQ,EAAA;AACjC,IAAA,MAAMjB,GAAAA,GAAM,IAAIyD,GAAAA,CAAI,IAAA,CAAKzD,GAAG,CAAA;AAC5Ba,IAAAA,MAAAA,CAAO6C,IAAAA,CAAKJ,MAAMxB,UAAAA,IAAc,EAAC,CAAA,CAAG6B,OAAAA,CAAQ,CAACC,GAAAA,KAAAA;AAC3C,MAAA,MAAMC,KAAAA,GAAQP,KAAAA,CAAMxB,UAAAA,CAAW8B,GAAAA,CAAAA;AAC/B,MAAA,MAAME,UAAU7C,IAAAA,CAAK8C,OAAAA,CAAQ,CAAA,CAAA,EAAIH,GAAAA,KAAQC,KAAAA,CAAAA;AACzC,MAAA,IAAIC,WAAW7C,IAAAA,EAAM;AACnBjB,QAAAA,GAAAA,CAAIgE,YAAAA,CAAaC,MAAAA,CAAOL,GAAAA,EAAKC,KAAAA,CAAAA;MAC/B,CAAA,MAAO;AACL5C,QAAAA,IAAAA,GAAO6C,OAAAA;AACT,MAAA;IACF,CAAA,CAAA;AACA9D,IAAAA,GAAAA,CAAIkE,QAAAA,GAAWC,IAAAA,CAAKnE,GAAAA,CAAIkE,QAAAA,EAAUjD,IAAAA,CAAAA;AAClC,IAAA,MAAMqB,OAAAA,GAAuB;AAC3B,MAAA,GAAG,KAAKA,OAAAA,CAAQ8B,YAAAA;AAChBC,MAAAA,MAAAA,EAAQC,YAAYC,GAAAA,CAAI;QAACf,GAAAA,CAAIa,MAAAA;AAAQ,QAAA,IAAA,CAAK/B,QAAQ8B,YAAAA,EAAcC;AAAQtC,OAAAA,CAAAA,MAAAA,CAAOyC,QAAAA,CAAAA,CAAAA;AAC/E3C,MAAAA,IAAAA,EAAM,CAAC4C,OAAAA,CAAQnB,KAAAA,CAAMzB,IAAI,CAAA,GAAIyB,MAAMzB,IAAAA,GAAO6C,MAAAA;MAC1CtD,MAAAA,EAAQkC,KAAAA,CAAMlC,OAAOuD,WAAAA,EAAW;MAChCC,OAAAA,EAAS;QAAEC,MAAAA,EAAQ,kBAAA;QAAoB,GAAG,IAAA,CAAKvC,QAAQ8B,YAAAA,EAAcQ;AAAQ;AAC/E,KAAA;AACA,IAAA,MAAMpB,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,aAAA,EAAe;AAAExC,MAAAA,OAAAA;AAAkBtC,MAAAA;KAAS,CAAA;AACnE,IAAA,IAAI;AACF,MAAA,MAAM+E,WAAW,MAAMC,KAAAA,CAAMhF,GAAAA,CAAIiF,QAAAA,IAAY3C,OAAAA,CAAAA;AAC7C,MAAA,MAAM4C,IAAAA,GAAO,MAAMH,QAAAA,CAASG,IAAAA,EAAI;AAChC,MAAA,MAAMC,SAAS,IAAI5F,iBAAAA,CAAkBwF,SAAStF,MAAAA,EAAQsF,QAAAA,CAASrF,YAAYwF,IAAAA,CAAAA;AAC3E,MAAA,MAAM1B,GAAAA,CAAIpB,OAAAA,CAAQ0C,IAAAA,CAAK,YAAA,EAAc;QAAEM,IAAAA,EAAMD,MAAAA;AAAQnF,QAAAA;OAAI,CAAA;AACzD,MAAA,OAAOmF,MAAAA;AACT,IAAA,CAAA,CAAA,OAASE,GAAAA,EAAK;AACZ,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,WAAA,EAActF,GAAAA,CAAAA,YAAAA,CAAAA,EAAmB;AAACqF,QAAAA;AAAI,OAAA,CAAA;AAC5D,IAAA;AACF,EAAA;EACAE,cAAAA,GAAiB;AACf,IAAA,OAAO;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTtF,aAAAA,EAAeuF,KAAAA,CAAM,KAAKvF,aAAa,CAAA;AACvCH,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;AACXC,MAAAA,WAAAA,EAAa,IAAA,CAAKA,WAAAA;AAClBC,MAAAA,GAAAA,EAAK,IAAA,CAAKA;AACZ,KAAA;AACF,EAAA;AACF","file":"openapi.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { join } from \"path\";\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Callback, Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { ValueError } from \"@/errors.js\";\nimport { SchemaObject } from \"ajv\";\nimport { parse } from \"yaml\";\nimport { isEmpty, isTruthy, clone, toCamelCase } from \"remeda\";\n\nexport interface OpenAPIToolOptions extends BaseToolOptions {\n  name?: string;\n  description?: string;\n  openApiSchema: any;\n  url?: string;\n  fetchOptions?: RequestInit;\n}\n\nexport interface OpenAPIEvents {\n  beforeFetch: Callback<{ options: RequestInit; url: URL }>;\n  afterFetch: Callback<{ data: OpenAPIToolOutput; url: URL }>;\n}\n\nexport class OpenAPIToolOutput extends StringToolOutput {\n  constructor(\n    public readonly status: number,\n    public readonly statusText: string,\n    public readonly result = \"\",\n  ) {\n    super();\n    this.status = status;\n    this.statusText = statusText;\n    this.result = result ?? \"\";\n  }\n}\n\nexport class OpenAPITool extends Tool<OpenAPIToolOutput, OpenAPIToolOptions> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly url: string;\n\n  public readonly openApiSchema: any;\n\n  inputSchema(): SchemaObject {\n    const getReferencedObject = (json: any, refPath: string) => {\n      const pathSegments = refPath.split(\"/\");\n      let currentObject = json;\n      for (const segment of pathSegments) {\n        if (segment === \"#\") {\n          continue;\n        }\n        currentObject = currentObject[segment];\n      }\n      return currentObject;\n    };\n\n    return {\n      type: \"object\",\n      required: [\"path\", \"method\"],\n      oneOf: Object.entries(this.openApiSchema.paths).flatMap(([path, pathSpec]: [string, any]) =>\n        Object.entries(pathSpec).map(([method, methodSpec]: [string, any]) => ({\n          additionalProperties: false,\n          properties: {\n            path: {\n              const: path,\n              description:\n                \"Do not replace variables in path, instead of, put them to the parameters object.\",\n            },\n            method: { const: method, description: methodSpec.summary || methodSpec.description },\n            ...(methodSpec.requestBody?.content?.[\"application/json\"]?.schema\n              ? {\n                  body: methodSpec.requestBody?.content?.[\"application/json\"]?.schema,\n                }\n              : {}),\n            ...(methodSpec.parameters\n              ? {\n                  parameters: {\n                    type: \"object\",\n                    additionalProperties: false,\n                    required: methodSpec.parameters\n                      .filter((p: any) => p.required === true)\n                      .map((p: any) => p.name),\n                    properties: methodSpec.parameters.reduce(\n                      (acc: any, p: any) =>\n                        !p.$ref\n                          ? { ...acc, [p.name]: { ...p.schema, description: p.name } }\n                          : {\n                              ...acc,\n                              [getReferencedObject(this.openApiSchema, p.$ref)?.name]: {\n                                ...getReferencedObject(this.openApiSchema, p.$ref)?.schema,\n                                description: getReferencedObject(this.openApiSchema, p.$ref)?.name,\n                              },\n                            },\n                      {},\n                    ),\n                  },\n                }\n              : {}),\n          },\n        })),\n      ),\n    } as const satisfies SchemaObject;\n  }\n\n  public readonly emitter: ToolEmitter<Record<string, any>, OpenAPIToolOutput, OpenAPIEvents>;\n\n  static {\n    this.register();\n  }\n\n  public constructor(options: OpenAPIToolOptions) {\n    super(options);\n\n    this.openApiSchema = parse(options.openApiSchema);\n\n    this.url = this.options.url || this.openApiSchema.servers?.find((s: any) => s?.url)?.url;\n    if (!this.url) {\n      throw new ValueError(\n        \"OpenAPI schema hasn't any server with url specified. Pass it manually.\",\n      );\n    }\n\n    this.name = options.name ?? this.openApiSchema?.info?.title?.trim();\n    if (!this.name) {\n      throw new ValueError(\"OpenAPI schema hasn't 'name' specified. Pass it manually.\");\n    }\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"web\", \"openAPI\", toCamelCase(this.name)],\n      creator: this,\n    });\n\n    this.description =\n      options.description ||\n      this.openApiSchema?.info?.description ||\n      \"Performs REST API requests to the servers and retrieves the response. The server API interfaces are defined in OpenAPI schema. \\n\" +\n        \"Only use the OpenAPI tool if you need to communicate to external servers.\";\n\n    if (!this.description) {\n      throw new ValueError(\"OpenAPI schema hasn't 'description' specified. Pass it manually.\");\n    }\n  }\n\n  protected async _run(\n    input: Record<string, any>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ) {\n    let path: string = input.path || \"\";\n    const url = new URL(this.url);\n    Object.keys(input.parameters ?? {}).forEach((key) => {\n      const value = input.parameters[key];\n      const newPath = path.replace(`{${key}}`, value);\n      if (newPath == path) {\n        url.searchParams.append(key, value);\n      } else {\n        path = newPath;\n      }\n    });\n    url.pathname = join(url.pathname, path);\n    const options: RequestInit = {\n      ...this.options.fetchOptions,\n      signal: AbortSignal.any([run.signal, this.options.fetchOptions?.signal].filter(isTruthy)),\n      body: !isEmpty(input.body) ? input.body : undefined,\n      method: input.method.toLowerCase(),\n      headers: { Accept: \"application/json\", ...this.options.fetchOptions?.headers },\n    };\n    await run.emitter.emit(\"beforeFetch\", { options: options, url: url });\n    try {\n      const response = await fetch(url.toString(), options);\n      const text = await response.text();\n      const output = new OpenAPIToolOutput(response.status, response.statusText, text);\n      await run.emitter.emit(\"afterFetch\", { data: output, url });\n      return output;\n    } catch (err) {\n      throw new ToolError(`Request to ${url} has failed.`, [err]);\n    }\n  }\n  createSnapshot() {\n    return {\n      ...super.createSnapshot(),\n      openApiSchema: clone(this.openApiSchema),\n      name: this.name,\n      description: this.description,\n      url: this.url,\n    };\n  }\n}\n"]}

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","stdout","stderr","exitCode","outputFiles","register","isEmpty","getTextContent","executionStatus","trim","isImage","filename","some","ext","toLowerCase","endsWith","files","length","map","file","id","join","filter","Boolean","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;AAQO,MAAMA,yBAAyBC,mBAAAA,CAAAA;EARtC;;;;;;;AASE,EAAA,OAAOC,WAAAA,GAAc,cAAA;AAErB,EAAA,WAAA,CACkBC,MAAAA,EACAC,MAAAA,EACAC,QAAAA,EACAC,WAAAA,GAA4B,EAAA,EAC5C;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CALWH,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,KAAA;AACT;EAEAC,cAAAA,GAAiB;AACf,IAAA,MAAMC,kBACJ,IAAA,CAAKL,QAAAA,KAAa,IACd,iCAAA,GACA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOQ,IAAAA,EAAI,GAAK,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOO,IAAAA,EAAI,GAAK,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAA,MAAMQ,OAAAA,2BAAWC,QAAAA,KACf;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAAA,CAAK,CAACC,QAAQF,QAAAA,CAASG,WAAAA,GAAcC,QAAAA,CAASF,GAAAA,CAAAA,CAAAA,EAD1E,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAAA,GAAQ,IAAA,CAAKZ,WAAAA,CAAYa,MAAAA,GAC3B,oJAAA,GACA,IAAA,CAAKb,WAAAA,CACFc,GAAAA,CACC,CAACC,IAAAA,KACC,CAAA,EAAGT,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAA,GAAM,EAAA,CAAA,CAAA,EAAMQ,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKb,gBAAAA,CAAiBE,WAAW,CAAA,CAAA,EAAImB,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAAA,CAEvGC,IAAAA,CAAK,IAAA,CAAA,GACR,IAAA;AAEJ,IAAA,OAAO;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAAA,CAAOC,OAAAA,CAAAA,CAASF,IAAAA,CAAK,IAAA,CAAA;AACvE;EAEAG,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLvB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAA,CAAKA;AACpB,KAAA;AACF;AAEAqB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"output.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolOutput } from \"@/tools/base.js\";\nimport { PythonFile } from \"@/tools/python/storage.js\";\n\nexport class PythonToolOutput extends ToolOutput {\n  static FILE_PREFIX = \"urn:bee:file\";\n\n  constructor(\n    public readonly stdout: string,\n    public readonly stderr: string,\n    public readonly exitCode?: number,\n    public readonly outputFiles: PythonFile[] = [],\n  ) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  isEmpty() {\n    return false;\n  }\n\n  getTextContent() {\n    const executionStatus =\n      this.exitCode === 0\n        ? \"The code executed successfully.\"\n        : `The code exited with error code ${this.exitCode}.`;\n    const stdout = this.stdout.trim() ? `Standard output: \\n\\`\\`\\`\\n${this.stdout}\\n\\`\\`\\`` : null;\n    const stderr = this.stderr.trim() ? `Error output: \\n\\`\\`\\`\\n${this.stderr}\\n\\`\\`\\`` : null;\n    const isImage = (filename: string) =>\n      [\".png\", \".jpg\", \".jpeg\", \".gif\", \".bmp\"].some((ext) => filename.toLowerCase().endsWith(ext));\n    const files = this.outputFiles.length\n      ? \"The following files were created or modified. The user does not see them yet. To present a file to the user, send them the link below, verbatim:\\n\" +\n        this.outputFiles\n          .map(\n            (file) =>\n              `${isImage(file.filename) ? \"!\" : \"\"}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`,\n          )\n          .join(\"\\n\")\n      : null;\n\n    return [executionStatus, stdout, stderr, files].filter(Boolean).join(\"\\n\");\n  }\n\n  createSnapshot() {\n    return {\n      stdout: this.stdout,\n      stderr: this.stderr,\n      exitCode: this.exitCode,\n      outputFiles: this.outputFiles,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","stdout","stderr","exitCode","outputFiles","register","isEmpty","getTextContent","executionStatus","trim","isImage","filename","some","ext","toLowerCase","endsWith","files","length","map","file","id","join","filter","Boolean","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;;;AAQO,MAAMA,yBAAyBC,mBAAAA,CAAAA;EARtC;;;;;;;AASE,EAAA,OAAOC,WAAAA,GAAc,cAAA;AAErB,EAAA,WAAA,CACkBC,MAAAA,EACAC,MAAAA,EACAC,QAAAA,EACAC,WAAAA,GAA4B,EAAA,EAC5C;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CALWH,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB,EAAA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,KAAA;AACT,EAAA;EAEAC,cAAAA,GAAiB;AACf,IAAA,MAAMC,kBACJ,IAAA,CAAKL,QAAAA,KAAa,IACd,iCAAA,GACA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOQ,IAAAA,EAAI,GAAK,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOO,IAAAA,EAAI,GAAK,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAA,MAAMQ,OAAAA,2BAAWC,QAAAA,KACf;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAAA,CAAK,CAACC,QAAQF,QAAAA,CAASG,WAAAA,GAAcC,QAAAA,CAASF,GAAAA,CAAAA,CAAAA,EAD1E,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAAA,GAAQ,IAAA,CAAKZ,WAAAA,CAAYa,MAAAA,GAC3B,oJAAA,GACA,IAAA,CAAKb,WAAAA,CACFc,GAAAA,CACC,CAACC,IAAAA,KACC,CAAA,EAAGT,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAA,GAAM,EAAA,CAAA,CAAA,EAAMQ,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKb,gBAAAA,CAAiBE,WAAW,CAAA,CAAA,EAAImB,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAAA,CAEvGC,IAAAA,CAAK,IAAA,CAAA,GACR,IAAA;AAEJ,IAAA,OAAO;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAAA,CAAOC,OAAAA,CAAAA,CAASF,IAAAA,CAAK,IAAA,CAAA;AACvE,EAAA;EAEAG,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLvB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAA,CAAKA;AACpB,KAAA;AACF,EAAA;AAEAqB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB,EAAA;AACF","file":"output.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolOutput } from \"@/tools/base.js\";\nimport { PythonFile } from \"@/tools/python/storage.js\";\n\nexport class PythonToolOutput extends ToolOutput {\n  static FILE_PREFIX = \"urn:bee:file\";\n\n  constructor(\n    public readonly stdout: string,\n    public readonly stderr: string,\n    public readonly exitCode?: number,\n    public readonly outputFiles: PythonFile[] = [],\n  ) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  isEmpty() {\n    return false;\n  }\n\n  getTextContent() {\n    const executionStatus =\n      this.exitCode === 0\n        ? \"The code executed successfully.\"\n        : `The code exited with error code ${this.exitCode}.`;\n    const stdout = this.stdout.trim() ? `Standard output: \\n\\`\\`\\`\\n${this.stdout}\\n\\`\\`\\`` : null;\n    const stderr = this.stderr.trim() ? `Error output: \\n\\`\\`\\`\\n${this.stderr}\\n\\`\\`\\`` : null;\n    const isImage = (filename: string) =>\n      [\".png\", \".jpg\", \".jpeg\", \".gif\", \".bmp\"].some((ext) => filename.toLowerCase().endsWith(ext));\n    const files = this.outputFiles.length\n      ? \"The following files were created or modified. The user does not see them yet. To present a file to the user, send them the link below, verbatim:\\n\" +\n        this.outputFiles\n          .map(\n            (file) =>\n              `${isImage(file.filename) ? \"!\" : \"\"}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`,\n          )\n          .join(\"\\n\")\n      : null;\n\n    return [executionStatus, stdout, stderr, files].filter(Boolean).join(\"\\n\");\n  }\n\n  createSnapshot() {\n    return {\n      stdout: this.stdout,\n      stderr: this.stderr,\n      exitCode: this.exitCode,\n      outputFiles: this.outputFiles,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

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

@@ -2,7 +2,7 @@ import { ToolOutput } from '../base.cjs';
 import { PythonFile } from './storage.cjs';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-UkIkNrek.cjs';
+import '../../emitter-D4OcelZ9.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';

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

@@ -2,7 +2,7 @@ import { ToolOutput } from '../base.js';
 import { PythonFile } from './storage.js';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-CL56qab4.js';
+import '../../emitter-kHxkUxco.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","stdout","stderr","exitCode","outputFiles","register","isEmpty","getTextContent","executionStatus","trim","isImage","filename","some","ext","toLowerCase","endsWith","files","length","map","file","id","join","filter","Boolean","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;AAQO,MAAMA,yBAAyBC,UAAAA,CAAAA;EARtC;;;;;;;AASE,EAAA,OAAOC,WAAAA,GAAc,cAAA;AAErB,EAAA,WAAA,CACkBC,MAAAA,EACAC,MAAAA,EACAC,QAAAA,EACAC,WAAAA,GAA4B,EAAA,EAC5C;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CALWH,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,KAAA;AACT;EAEAC,cAAAA,GAAiB;AACf,IAAA,MAAMC,kBACJ,IAAA,CAAKL,QAAAA,KAAa,IACd,iCAAA,GACA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOQ,IAAAA,EAAI,GAAK,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOO,IAAAA,EAAI,GAAK,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAA,MAAMQ,OAAAA,2BAAWC,QAAAA,KACf;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAAA,CAAK,CAACC,QAAQF,QAAAA,CAASG,WAAAA,GAAcC,QAAAA,CAASF,GAAAA,CAAAA,CAAAA,EAD1E,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAAA,GAAQ,IAAA,CAAKZ,WAAAA,CAAYa,MAAAA,GAC3B,oJAAA,GACA,IAAA,CAAKb,WAAAA,CACFc,GAAAA,CACC,CAACC,IAAAA,KACC,CAAA,EAAGT,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAA,GAAM,EAAA,CAAA,CAAA,EAAMQ,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKb,gBAAAA,CAAiBE,WAAW,CAAA,CAAA,EAAImB,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAAA,CAEvGC,IAAAA,CAAK,IAAA,CAAA,GACR,IAAA;AAEJ,IAAA,OAAO;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAAA,CAAOC,OAAAA,CAAAA,CAASF,IAAAA,CAAK,IAAA,CAAA;AACvE;EAEAG,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLvB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAA,CAAKA;AACpB,KAAA;AACF;AAEAqB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"output.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolOutput } from \"@/tools/base.js\";\nimport { PythonFile } from \"@/tools/python/storage.js\";\n\nexport class PythonToolOutput extends ToolOutput {\n  static FILE_PREFIX = \"urn:bee:file\";\n\n  constructor(\n    public readonly stdout: string,\n    public readonly stderr: string,\n    public readonly exitCode?: number,\n    public readonly outputFiles: PythonFile[] = [],\n  ) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  isEmpty() {\n    return false;\n  }\n\n  getTextContent() {\n    const executionStatus =\n      this.exitCode === 0\n        ? \"The code executed successfully.\"\n        : `The code exited with error code ${this.exitCode}.`;\n    const stdout = this.stdout.trim() ? `Standard output: \\n\\`\\`\\`\\n${this.stdout}\\n\\`\\`\\`` : null;\n    const stderr = this.stderr.trim() ? `Error output: \\n\\`\\`\\`\\n${this.stderr}\\n\\`\\`\\`` : null;\n    const isImage = (filename: string) =>\n      [\".png\", \".jpg\", \".jpeg\", \".gif\", \".bmp\"].some((ext) => filename.toLowerCase().endsWith(ext));\n    const files = this.outputFiles.length\n      ? \"The following files were created or modified. The user does not see them yet. To present a file to the user, send them the link below, verbatim:\\n\" +\n        this.outputFiles\n          .map(\n            (file) =>\n              `${isImage(file.filename) ? \"!\" : \"\"}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`,\n          )\n          .join(\"\\n\")\n      : null;\n\n    return [executionStatus, stdout, stderr, files].filter(Boolean).join(\"\\n\");\n  }\n\n  createSnapshot() {\n    return {\n      stdout: this.stdout,\n      stderr: this.stderr,\n      exitCode: this.exitCode,\n      outputFiles: this.outputFiles,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","stdout","stderr","exitCode","outputFiles","register","isEmpty","getTextContent","executionStatus","trim","isImage","filename","some","ext","toLowerCase","endsWith","files","length","map","file","id","join","filter","Boolean","createSnapshot","loadSnapshot","snapshot","Object","assign"],"mappings":";;;;AAQO,MAAMA,yBAAyBC,UAAAA,CAAAA;EARtC;;;;;;;AASE,EAAA,OAAOC,WAAAA,GAAc,cAAA;AAErB,EAAA,WAAA,CACkBC,MAAAA,EACAC,MAAAA,EACAC,QAAAA,EACAC,WAAAA,GAA4B,EAAA,EAC5C;AACA,IAAA,KAAA,EAAK,EAAA,IAAA,CALWH,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB,EAAA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;EAEAC,OAAAA,GAAU;AACR,IAAA,OAAO,KAAA;AACT,EAAA;EAEAC,cAAAA,GAAiB;AACf,IAAA,MAAMC,kBACJ,IAAA,CAAKL,QAAAA,KAAa,IACd,iCAAA,GACA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOQ,IAAAA,EAAI,GAAK,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAAA,GAAS,IAAA,CAAKA,MAAAA,CAAOO,IAAAA,EAAI,GAAK,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAA,MAAMQ,OAAAA,2BAAWC,QAAAA,KACf;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAAA,CAAK,CAACC,QAAQF,QAAAA,CAASG,WAAAA,GAAcC,QAAAA,CAASF,GAAAA,CAAAA,CAAAA,EAD1E,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAAA,GAAQ,IAAA,CAAKZ,WAAAA,CAAYa,MAAAA,GAC3B,oJAAA,GACA,IAAA,CAAKb,WAAAA,CACFc,GAAAA,CACC,CAACC,IAAAA,KACC,CAAA,EAAGT,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAA,GAAM,EAAA,CAAA,CAAA,EAAMQ,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKb,gBAAAA,CAAiBE,WAAW,CAAA,CAAA,EAAImB,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAAA,CAEvGC,IAAAA,CAAK,IAAA,CAAA,GACR,IAAA;AAEJ,IAAA,OAAO;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAAA,CAAOC,OAAAA,CAAAA,CAASF,IAAAA,CAAK,IAAA,CAAA;AACvE,EAAA;EAEAG,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLvB,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAA,CAAKA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAA,CAAKA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAA,CAAKA;AACpB,KAAA;AACF,EAAA;AAEAqB,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB,EAAA;AACF","file":"output.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ToolOutput } from \"@/tools/base.js\";\nimport { PythonFile } from \"@/tools/python/storage.js\";\n\nexport class PythonToolOutput extends ToolOutput {\n  static FILE_PREFIX = \"urn:bee:file\";\n\n  constructor(\n    public readonly stdout: string,\n    public readonly stderr: string,\n    public readonly exitCode?: number,\n    public readonly outputFiles: PythonFile[] = [],\n  ) {\n    super();\n  }\n\n  static {\n    this.register();\n  }\n\n  isEmpty() {\n    return false;\n  }\n\n  getTextContent() {\n    const executionStatus =\n      this.exitCode === 0\n        ? \"The code executed successfully.\"\n        : `The code exited with error code ${this.exitCode}.`;\n    const stdout = this.stdout.trim() ? `Standard output: \\n\\`\\`\\`\\n${this.stdout}\\n\\`\\`\\`` : null;\n    const stderr = this.stderr.trim() ? `Error output: \\n\\`\\`\\`\\n${this.stderr}\\n\\`\\`\\`` : null;\n    const isImage = (filename: string) =>\n      [\".png\", \".jpg\", \".jpeg\", \".gif\", \".bmp\"].some((ext) => filename.toLowerCase().endsWith(ext));\n    const files = this.outputFiles.length\n      ? \"The following files were created or modified. The user does not see them yet. To present a file to the user, send them the link below, verbatim:\\n\" +\n        this.outputFiles\n          .map(\n            (file) =>\n              `${isImage(file.filename) ? \"!\" : \"\"}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`,\n          )\n          .join(\"\\n\")\n      : null;\n\n    return [executionStatus, stdout, stderr, files].filter(Boolean).join(\"\\n\");\n  }\n\n  createSnapshot() {\n    return {\n      stdout: this.stdout,\n      stderr: this.stderr,\n      exitCode: this.exitCode,\n      outputFiles: this.outputFiles,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n}\n"]}

package/dist/tools/python/python.cjs

@@ -138,7 +138,7 @@ async function callCodeInterpreter({ url, body, signal }) {
     body: JSON.stringify(body),
     signal
   }).catch((error) => {
-    if (error.cause.name == "HTTPParserError") {
+    if (error.cause?.name === "HTTPParserError") {
       throw new base_cjs.ToolError("Request to code interpreter has failed -- ensure that CODE_INTERPRETER_URL points to the new HTTP endpoint (default port: 50081).", [
         error
       ]);
@@ -148,7 +148,10 @@ async function callCodeInterpreter({ url, body, signal }) {
       ]);
     }
   });
-  if (!response?.ok && response.status > 400) {
+  if (!response) {
+    throw new base_cjs.ToolError("Request to code interpreter has failed.");
+  }
+  if (!response.ok && response.status >= 400) {
     throw new base_cjs.ToolError(`Request to code interpreter has failed with HTTP status code ${response.status}.`, [
       new Error(await response.text())
     ]);

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/python/python.ts"],"names":["PythonTool","Tool","name","description","join","storage","files","emitter","Emitter","root","child","namespace","creator","inputSchema","list","fileNames","unique","map","filename","z","object","language","enum","describe","code","string","hasMinLength","inputFiles","array","preprocess","options","codeInterpreter","url","ValidationError","message","data","propertyName","register","_run","input","_options","run","pipe","uniqueBy","f","filter","file","isIncludedIn","upload","getSourceCode","llm","promptTemplate","response","create","messages","UserMessage","render","abortSignal","signal","getTextContent","trim","prefix","result","callCodeInterpreter","body","source_code","Object","fromEntries","pythonId","filesOutput","download","entries","path","_","startsWith","String","slice","length","every","PythonToolOutput","stdout","stderr","exit_code","createSnapshot","shallowCopy","loadSnapshot","snapshot","fetch","method","headers","JSON","stringify","catch","error","cause","ToolError","ok","status","Error","text","json"],"mappings":";;;;;;;;;;;;;;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"]}
+{"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,EAAA;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,IAAA;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAAA,CAAQD,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAAA,CAAQzB,OAAAA;AACzB,EAAA;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,MAAA;AACA,MAAA,OAAOrB,KAAAA,CAAMf,IAAAA;IACf,CAAA,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,EAAA;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,EAAA;AAEAwD,EAAAA,YAAAA,CAAaC,QAAAA,EAAwD;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB,EAAA;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,EAAO5F,IAAAA,KAAS,iBAAA,EAAmB;AAC3C,MAAA,MAAM,IAAI6F,mBACR,mIAAA,EACA;AAACF,QAAAA;AAAM,OAAA,CAAA;IAEX,CAAA,MAAO;AACL,MAAA,MAAM,IAAIE,mBAAU,yCAAA,EAA2C;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE,IAAA;EACF,CAAA,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU;AACb,IAAA,MAAM,IAAI2C,mBAAU,yCAAA,CAAA;AACtB,EAAA;AAEA,EAAA,IAAI,CAAC3C,QAAAA,CAAS4C,EAAAA,IAAM5C,QAAAA,CAAS6C,UAAU,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,EAAA;AAEA,EAAA,OAAO,MAAM/C,SAASgD,IAAAA,EAAI;AAC5B;AAxCsBrC,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) {\n    throw new ToolError(\"Request to code interpreter has failed.\");\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

@@ -1,14 +1,14 @@
 import { BaseCache } from '../../cache/base.cjs';
 import * as promise_based_task from 'promise-based-task';
-import { BaseToolOptions, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.cjs';
+import { Tool, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.cjs';
 import { z } from 'zod';
 import { PromptTemplate } from '../../template.cjs';
 import { PythonStorage, PythonFile } from './storage.cjs';
 import { PythonToolOutput } from './output.cjs';
 import { ConnectionOptions } from 'node:tls';
 import { RunContext } from '../../context.cjs';
-import { E as Emitter } from '../../emitter-UkIkNrek.cjs';
-import { m as ChatModel } from '../../chat-DaqV3woF.cjs';
+import { E as Emitter } from '../../emitter-D4OcelZ9.cjs';
+import { a as ChatModel } from '../../chat-C20nYj46.cjs';
 import '../../internals/serializable.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';

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

@@ -1,14 +1,14 @@
 import { BaseCache } from '../../cache/base.js';
 import * as promise_based_task from 'promise-based-task';
-import { BaseToolOptions, Tool, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.js';
+import { Tool, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.js';
 import { z } from 'zod';
 import { PromptTemplate } from '../../template.js';
 import { PythonStorage, PythonFile } from './storage.js';
 import { PythonToolOutput } from './output.js';
 import { ConnectionOptions } from 'node:tls';
 import { RunContext } from '../../context.js';
-import { E as Emitter } from '../../emitter-CL56qab4.js';
-import { m as ChatModel } from '../../chat-DTxmaKP0.js';
+import { E as Emitter } from '../../emitter-kHxkUxco.js';
+import { a as ChatModel } from '../../chat-CfTNfR60.js';
 import '../../internals/serializable.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';

package/dist/tools/python/python.js

@@ -136,7 +136,7 @@ async function callCodeInterpreter({ url, body, signal }) {
     body: JSON.stringify(body),
     signal
   }).catch((error) => {
-    if (error.cause.name == "HTTPParserError") {
+    if (error.cause?.name === "HTTPParserError") {
       throw new ToolError("Request to code interpreter has failed -- ensure that CODE_INTERPRETER_URL points to the new HTTP endpoint (default port: 50081).", [
         error
       ]);
@@ -146,7 +146,10 @@ async function callCodeInterpreter({ url, body, signal }) {
       ]);
     }
   });
-  if (!response?.ok && response.status > 400) {
+  if (!response) {
+    throw new ToolError("Request to code interpreter has failed.");
+  }
+  if (!response.ok && response.status >= 400) {
     throw new ToolError(`Request to code interpreter has failed with HTTP status code ${response.status}.`, [
       new Error(await response.text())
     ]);

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/python/python.ts"],"names":["PythonTool","Tool","name","description","join","storage","files","emitter","Emitter","root","child","namespace","creator","inputSchema","list","fileNames","unique","map","filename","z","object","language","enum","describe","code","string","hasMinLength","inputFiles","array","preprocess","options","codeInterpreter","url","ValidationError","message","data","propertyName","register","_run","input","_options","run","pipe","uniqueBy","f","filter","file","isIncludedIn","upload","getSourceCode","llm","promptTemplate","response","create","messages","UserMessage","render","abortSignal","signal","getTextContent","trim","prefix","result","callCodeInterpreter","body","source_code","Object","fromEntries","pythonId","filesOutput","download","entries","path","_","startsWith","String","slice","length","every","PythonToolOutput","stdout","stderr","exit_code","createSnapshot","shallowCopy","loadSnapshot","snapshot","fetch","method","headers","JSON","stringify","catch","error","cause","ToolError","ok","status","Error","text","json"],"mappings":";;;;;;;;;;;;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"]}
+{"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,EAAA;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,IAAA;AACA,IAAA,IAAA,CAAKP,aAAaC,OAAAA,CAAQD,UAAAA;AAC1B,IAAA,IAAA,CAAKxB,UAAUyB,OAAAA,CAAQzB,OAAAA;AACzB,EAAA;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,MAAA;AACA,MAAA,OAAOrB,KAAAA,CAAMf,IAAAA;IACf,CAAA,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,EAAA;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,EAAA;AAEAwD,EAAAA,YAAAA,CAAaC,QAAAA,EAAwD;AACnE,IAAA,KAAA,CAAMD,aAAaC,QAAAA,CAAAA;AACrB,EAAA;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,EAAO5F,IAAAA,KAAS,iBAAA,EAAmB;AAC3C,MAAA,MAAM,IAAI6F,UACR,mIAAA,EACA;AAACF,QAAAA;AAAM,OAAA,CAAA;IAEX,CAAA,MAAO;AACL,MAAA,MAAM,IAAIE,UAAU,yCAAA,EAA2C;AAACF,QAAAA;AAAM,OAAA,CAAA;AACxE,IAAA;EACF,CAAA,CAAA;AAEA,EAAA,IAAI,CAACzC,QAAAA,EAAU;AACb,IAAA,MAAM,IAAI2C,UAAU,yCAAA,CAAA;AACtB,EAAA;AAEA,EAAA,IAAI,CAAC3C,QAAAA,CAAS4C,EAAAA,IAAM5C,QAAAA,CAAS6C,UAAU,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,EAAA;AAEA,EAAA,OAAO,MAAM/C,SAASgD,IAAAA,EAAI;AAC5B;AAxCsBrC,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) {\n    throw new ToolError(\"Request to code interpreter has failed.\");\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":["../../../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"]}
+{"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,EAAA;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,EAAA;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,EAAA;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEb,KAAAA,EAAO,IAAA,CAAKA,MAAME,KAAAA;AAAQ,KAAA;AACrC,EAAA;AAEAY,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB,EAAA;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,EAAA;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,EAAA;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,IAAA,CAAA,CAAA,CAAA;AAEN,EAAA;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,IAAA,CAAA,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,EAAA;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,EAAA;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,EAAA;EAEApC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAEM,KAAAA,EAAO+B,qBAAAA,CAAY,KAAK/B,KAAK;AAAE,KAAA;AAC1C,EAAA;AAEAL,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB,EAAA;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"]}