beeai-framework 0.1.2

package/dist/tools/openapi.js

@@ -0,0 +1,175 @@
+import { join } from 'path';
+import { StringToolOutput, Tool, ToolError } from './base.js';
+import { Emitter } from '../emitter/emitter.js';
+import { ValueError } from '../errors.js';
+import { parse } from 'yaml';
+import { toCamelCase, isTruthy, isEmpty, clone } from 'remeda';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class OpenAPIToolOutput extends StringToolOutput {
+  static {
+    __name(this, "OpenAPIToolOutput");
+  }
+  status;
+  statusText;
+  result;
+  constructor(status, statusText, result = "") {
+    super(), this.status = status, this.statusText = statusText, this.result = result;
+    this.status = status;
+    this.statusText = statusText;
+    this.result = result ?? "";
+  }
+}
+class OpenAPITool extends Tool {
+  static {
+    __name(this, "OpenAPITool");
+  }
+  name;
+  description;
+  url;
+  openApiSchema;
+  inputSchema() {
+    const getReferencedObject = /* @__PURE__ */ __name((json, refPath) => {
+      const pathSegments = refPath.split("/");
+      let currentObject = json;
+      for (const segment of pathSegments) {
+        if (segment === "#") {
+          continue;
+        }
+        currentObject = currentObject[segment];
+      }
+      return currentObject;
+    }, "getReferencedObject");
+    return {
+      type: "object",
+      required: [
+        "path",
+        "method"
+      ],
+      oneOf: Object.entries(this.openApiSchema.paths).flatMap(([path, pathSpec]) => Object.entries(pathSpec).map(([method, methodSpec]) => ({
+        additionalProperties: false,
+        properties: {
+          path: {
+            const: path,
+            description: "Do not replace variables in path, instead of, put them to the parameters object."
+          },
+          method: {
+            const: method,
+            description: methodSpec.summary || methodSpec.description
+          },
+          ...methodSpec.requestBody?.content?.["application/json"]?.schema ? {
+            body: methodSpec.requestBody?.content?.["application/json"]?.schema
+          } : {},
+          ...methodSpec.parameters ? {
+            parameters: {
+              type: "object",
+              additionalProperties: false,
+              required: methodSpec.parameters.filter((p) => p.required === true).map((p) => p.name),
+              properties: methodSpec.parameters.reduce((acc, p) => !p.$ref ? {
+                ...acc,
+                [p.name]: {
+                  ...p.schema,
+                  description: p.name
+                }
+              } : {
+                ...acc,
+                [getReferencedObject(this.openApiSchema, p.$ref)?.name]: {
+                  ...getReferencedObject(this.openApiSchema, p.$ref)?.schema,
+                  description: getReferencedObject(this.openApiSchema, p.$ref)?.name
+                }
+              }, {})
+            }
+          } : {}
+        }
+      })))
+    };
+  }
+  emitter;
+  static {
+    this.register();
+  }
+  constructor(options) {
+    super(options);
+    this.openApiSchema = parse(options.openApiSchema);
+    this.url = this.options.url || this.openApiSchema.servers?.find((s) => s?.url)?.url;
+    if (!this.url) {
+      throw new ValueError("OpenAPI schema hasn't any server with url specified. Pass it manually.");
+    }
+    this.name = options.name ?? this.openApiSchema?.info?.title?.trim();
+    if (!this.name) {
+      throw new ValueError("OpenAPI schema hasn't 'name' specified. Pass it manually.");
+    }
+    this.emitter = Emitter.root.child({
+      namespace: [
+        "tool",
+        "web",
+        "openAPI",
+        toCamelCase(this.name)
+      ],
+      creator: this
+    });
+    this.description = options.description || this.openApiSchema?.info?.description || "Performs REST API requests to the servers and retrieves the response. The server API interfaces are defined in OpenAPI schema. \nOnly use the OpenAPI tool if you need to communicate to external servers.";
+    if (!this.description) {
+      throw new ValueError("OpenAPI schema hasn't 'description' specified. Pass it manually.");
+    }
+  }
+  async _run(input, _options, run) {
+    let path = input.path || "";
+    const url = new URL(this.url);
+    Object.keys(input.parameters ?? {}).forEach((key) => {
+      const value = input.parameters[key];
+      const newPath = path.replace(`{${key}}`, value);
+      if (newPath == path) {
+        url.searchParams.append(key, value);
+      } else {
+        path = newPath;
+      }
+    });
+    url.pathname = join(url.pathname, path);
+    const options = {
+      ...this.options.fetchOptions,
+      signal: AbortSignal.any([
+        run.signal,
+        this.options.fetchOptions?.signal
+      ].filter(isTruthy)),
+      body: !isEmpty(input.body) ? input.body : void 0,
+      method: input.method.toLowerCase(),
+      headers: {
+        Accept: "application/json",
+        ...this.options.fetchOptions?.headers
+      }
+    };
+    await run.emitter.emit("beforeFetch", {
+      options,
+      url
+    });
+    try {
+      const response = await fetch(url.toString(), options);
+      const text = await response.text();
+      const output = new OpenAPIToolOutput(response.status, response.statusText, text);
+      await run.emitter.emit("afterFetch", {
+        data: output,
+        url
+      });
+      return output;
+    } catch (err) {
+      throw new ToolError(`Request to ${url} has failed.`, [
+        err
+      ]);
+    }
+  }
+  createSnapshot() {
+    return {
+      ...super.createSnapshot(),
+      openApiSchema: clone(this.openApiSchema),
+      name: this.name,
+      description: this.description,
+      url: this.url
+    };
+  }
+}
+
+export { OpenAPITool, OpenAPIToolOutput };
+//# sourceMappingURL=openapi.js.map
+//# sourceMappingURL=openapi.js.map

package/dist/tools/openapi.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tools/openapi.ts"],"names":["OpenAPIToolOutput","StringToolOutput","constructor","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":";;;;;;;;;AA8CO,MAAMA,0BAA0BC,gBAAAA,CAAAA;EA9CvC;;;;;;EA+CEC,WACkBC,CAAAA,MAAAA,EACAC,UACAC,EAAAA,MAAAA,GAAS,EACzB,EAAA;AACA,IAAK,KAAA,EAAA,EAAA,KAJWF,MAAAA,GAAAA,MAAAA,EAAAA,KACAC,UAAAA,GAAAA,UAAAA,EAAAA,KACAC,MAAAA,GAAAA,MAAAA;AAGhB,IAAA,IAAA,CAAKF,MAASA,GAAAA,MAAAA;AACd,IAAA,IAAA,CAAKC,UAAaA,GAAAA,UAAAA;AAClB,IAAA,IAAA,CAAKC,SAASA,MAAU,IAAA,EAAA;AAC1B;AACF;AAEO,MAAMC,oBAAoBC,IAAAA,CAAAA;EA3DjC;;;AA4DkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,GAAAA;AAEAC,EAAAA,aAAAA;EAEhBC,WAA4B,GAAA;AAC1B,IAAMC,MAAAA,mBAAAA,mBAAuBC,MAAAA,CAAAA,CAAAA,IAAAA,EAAWC,OAAAA,KAAAA;AACtC,MAAMC,MAAAA,YAAAA,GAAeD,OAAQE,CAAAA,KAAAA,CAAM,GAAA,CAAA;AACnC,MAAA,IAAIC,aAAgBJ,GAAAA,IAAAA;AACpB,MAAA,KAAA,MAAWK,WAAWH,YAAc,EAAA;AAClC,QAAA,IAAIG,YAAY,GAAK,EAAA;AACnB,UAAA;AACF;AACAD,QAAAA,aAAAA,GAAgBA,cAAcC,OAAAA,CAAAA;AAChC;AACA,MAAOD,OAAAA,aAAAA;KATmB,EAAA,qBAAA,CAAA;AAY5B,IAAO,OAAA;MACLE,IAAM,EAAA,QAAA;MACNC,QAAU,EAAA;AAAC,QAAA,MAAA;AAAQ,QAAA;;MACnBC,KAAOC,EAAAA,MAAAA,CAAOC,QAAQ,IAAKb,CAAAA,aAAAA,CAAcc,KAAK,CAAEC,CAAAA,OAAAA,CAAQ,CAAC,CAACC,IAAAA,EAAMC,QAAAA,CAC9DL,KAAAA,MAAAA,CAAOC,QAAQI,QAAAA,CAAAA,CAAUC,IAAI,CAAC,CAACC,MAAQC,EAAAA,UAAAA,CAAgC,MAAA;QACrEC,oBAAsB,EAAA,KAAA;QACtBC,UAAY,EAAA;UACVN,IAAM,EAAA;YACJO,KAAOP,EAAAA,IAAAA;YACPlB,WACE,EAAA;AACJ,WAAA;UACAqB,MAAQ,EAAA;YAAEI,KAAOJ,EAAAA,MAAAA;YAAQrB,WAAasB,EAAAA,UAAAA,CAAWI,WAAWJ,UAAWtB,CAAAA;AAAY,WAAA;AACnF,UAAA,GAAIsB,UAAWK,CAAAA,WAAAA,EAAaC,OAAU,GAAA,kBAAA,GAAqBC,MACvD,GAAA;AACEC,YAAAA,IAAAA,EAAMR,UAAWK,CAAAA,WAAAA,EAAaC,OAAU,GAAA,kBAAA,CAAqBC,EAAAA;AAC/D,WAAA,GACA,EAAC;AACL,UAAA,GAAIP,WAAWS,UACX,GAAA;YACEA,UAAY,EAAA;cACVpB,IAAM,EAAA,QAAA;cACNY,oBAAsB,EAAA,KAAA;AACtBX,cAAAA,QAAAA,EAAUU,UAAWS,CAAAA,UAAAA,CAClBC,MAAO,CAAA,CAACC,CAAWA,KAAAA,CAAAA,CAAErB,QAAa,KAAA,IAAA,CAClCQ,CAAAA,GAAAA,CAAI,CAACa,CAAAA,KAAWA,EAAElC,IAAI,CAAA;cACzByB,UAAYF,EAAAA,UAAAA,CAAWS,WAAWG,MAChC,CAAA,CAACC,KAAUF,CACT,KAAA,CAACA,EAAEG,IACC,GAAA;gBAAE,GAAGD,GAAAA;gBAAK,CAACF,CAAAA,CAAElC,IAAI,GAAG;AAAE,kBAAA,GAAGkC,CAAEJ,CAAAA,MAAAA;AAAQ7B,kBAAAA,WAAAA,EAAaiC,CAAElC,CAAAA;AAAK;eACvD,GAAA;gBACE,GAAGoC,GAAAA;AACH,gBAAA,CAAC/B,oBAAoB,IAAKF,CAAAA,aAAAA,EAAe+B,EAAEG,IAAI,CAAA,EAAGrC,IAAAA,GAAO;AACvD,kBAAA,GAAGK,mBAAoB,CAAA,IAAA,CAAKF,aAAe+B,EAAAA,CAAAA,CAAEG,IAAI,CAAGP,EAAAA,MAAAA;AACpD7B,kBAAAA,WAAAA,EAAaI,mBAAoB,CAAA,IAAA,CAAKF,aAAe+B,EAAAA,CAAAA,CAAEG,IAAI,CAAGrC,EAAAA;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,QAAQ,EAAA;AACf;AAEA,EAAA,WAAA,CAAmBC,OAA6B,EAAA;AAC9C,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AAEN,IAAKrC,IAAAA,CAAAA,aAAAA,GAAgBsC,KAAMD,CAAAA,OAAAA,CAAQrC,aAAa,CAAA;AAEhD,IAAA,IAAA,CAAKD,GAAM,GAAA,IAAA,CAAKsC,OAAQtC,CAAAA,GAAAA,IAAO,IAAKC,CAAAA,aAAAA,CAAcuC,OAASC,EAAAA,IAAAA,CAAK,CAACC,CAAAA,KAAWA,CAAG1C,EAAAA,GAAAA,CAAMA,EAAAA,GAAAA;AACrF,IAAI,IAAA,CAAC,KAAKA,GAAK,EAAA;AACb,MAAM,MAAA,IAAI2C,WACR,wEAAA,CAAA;AAEJ;AAEA,IAAA,IAAA,CAAK7C,OAAOwC,OAAQxC,CAAAA,IAAAA,IAAQ,KAAKG,aAAe2C,EAAAA,IAAAA,EAAMC,OAAOC,IAAAA,EAAAA;AAC7D,IAAI,IAAA,CAAC,KAAKhD,IAAM,EAAA;AACd,MAAM,MAAA,IAAI6C,WAAW,2DAAA,CAAA;AACvB;AACA,IAAKP,IAAAA,CAAAA,OAAAA,GAAUW,OAAQC,CAAAA,IAAAA,CAAKC,KAAM,CAAA;MAChCC,SAAW,EAAA;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAO,QAAA,SAAA;AAAWC,QAAAA,WAAAA,CAAY,KAAKrD,IAAI;;MAC3DsD,OAAS,EAAA;KACX,CAAA;AAEA,IAAA,IAAA,CAAKrD,cACHuC,OAAQvC,CAAAA,WAAAA,IACR,IAAKE,CAAAA,aAAAA,EAAe2C,MAAM7C,WAC1B,IAAA,4MAAA;AAGF,IAAI,IAAA,CAAC,KAAKA,WAAa,EAAA;AACrB,MAAM,MAAA,IAAI4C,WAAW,kEAAA,CAAA;AACvB;AACF;EAEA,MAAgBU,IAAAA,CACdC,KACAC,EAAAA,QAAAA,EACAC,GACA,EAAA;AACA,IAAIvC,IAAAA,IAAAA,GAAeqC,MAAMrC,IAAQ,IAAA,EAAA;AACjC,IAAA,MAAMjB,GAAM,GAAA,IAAIyD,GAAI,CAAA,IAAA,CAAKzD,GAAG,CAAA;AAC5Ba,IAAO6C,MAAAA,CAAAA,IAAAA,CAAKJ,MAAMxB,UAAc,IAAA,EAAC,CAAG6B,CAAAA,OAAAA,CAAQ,CAACC,GAAAA,KAAAA;AAC3C,MAAMC,MAAAA,KAAAA,GAAQP,KAAMxB,CAAAA,UAAAA,CAAW8B,GAAAA,CAAAA;AAC/B,MAAA,MAAME,UAAU7C,IAAK8C,CAAAA,OAAAA,CAAQ,CAAIH,CAAAA,EAAAA,GAAAA,KAAQC,KAAAA,CAAAA;AACzC,MAAA,IAAIC,WAAW7C,IAAM,EAAA;AACnBjB,QAAIgE,GAAAA,CAAAA,YAAAA,CAAaC,MAAOL,CAAAA,GAAAA,EAAKC,KAAAA,CAAAA;OACxB,MAAA;AACL5C,QAAO6C,IAAAA,GAAAA,OAAAA;AACT;KACF,CAAA;AACA9D,IAAAA,GAAAA,CAAIkE,QAAWC,GAAAA,IAAAA,CAAKnE,GAAIkE,CAAAA,QAAAA,EAAUjD,IAAAA,CAAAA;AAClC,IAAA,MAAMqB,OAAuB,GAAA;AAC3B,MAAA,GAAG,KAAKA,OAAQ8B,CAAAA,YAAAA;AAChBC,MAAAA,MAAAA,EAAQC,YAAYC,GAAI,CAAA;QAACf,GAAIa,CAAAA,MAAAA;AAAQ,QAAA,IAAA,CAAK/B,QAAQ8B,YAAcC,EAAAA;AAAQtC,OAAAA,CAAAA,MAAAA,CAAOyC,QAAAA,CAAAA,CAAAA;AAC/E3C,MAAAA,IAAAA,EAAM,CAAC4C,OAAQnB,CAAAA,KAAAA,CAAMzB,IAAI,CAAA,GAAIyB,MAAMzB,IAAO6C,GAAAA,KAAAA,CAAAA;MAC1CtD,MAAQkC,EAAAA,KAAAA,CAAMlC,OAAOuD,WAAW,EAAA;MAChCC,OAAS,EAAA;QAAEC,MAAQ,EAAA,kBAAA;QAAoB,GAAG,IAAA,CAAKvC,QAAQ8B,YAAcQ,EAAAA;AAAQ;AAC/E,KAAA;AACA,IAAMpB,MAAAA,GAAAA,CAAIpB,OAAQ0C,CAAAA,IAAAA,CAAK,aAAe,EAAA;AAAExC,MAAAA,OAAAA;AAAkBtC,MAAAA;KAAS,CAAA;AACnE,IAAI,IAAA;AACF,MAAA,MAAM+E,WAAW,MAAMC,KAAAA,CAAMhF,GAAIiF,CAAAA,QAAAA,IAAY3C,OAAAA,CAAAA;AAC7C,MAAM4C,MAAAA,IAAAA,GAAO,MAAMH,QAAAA,CAASG,IAAI,EAAA;AAChC,MAAA,MAAMC,SAAS,IAAI7F,iBAAAA,CAAkByF,SAAStF,MAAQsF,EAAAA,QAAAA,CAASrF,YAAYwF,IAAAA,CAAAA;AAC3E,MAAM1B,MAAAA,GAAAA,CAAIpB,OAAQ0C,CAAAA,IAAAA,CAAK,YAAc,EAAA;QAAEM,IAAMD,EAAAA,MAAAA;AAAQnF,QAAAA;OAAI,CAAA;AACzD,MAAOmF,OAAAA,MAAAA;AACT,KAAA,CAAA,OAASE,GAAK,EAAA;AACZ,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAActF,WAAAA,EAAAA,GAAAA,CAAmB,YAAA,CAAA,EAAA;AAACqF,QAAAA;AAAI,OAAA,CAAA;AAC5D;AACF;EACAE,cAAiB,GAAA;AACf,IAAO,OAAA;AACL,MAAA,GAAG,MAAMA,cAAAA,EAAAA;MACTtF,aAAeuF,EAAAA,KAAAA,CAAM,KAAKvF,aAAa,CAAA;AACvCH,MAAAA,IAAAA,EAAM,IAAKA,CAAAA,IAAAA;AACXC,MAAAA,WAAAA,EAAa,IAAKA,CAAAA,WAAAA;AAClBC,MAAAA,GAAAA,EAAK,IAAKA,CAAAA;AACZ,KAAA;AACF;AACF","file":"openapi.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { 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

@@ -0,0 +1,65 @@
+'use strict';
+
+var base_cjs = require('../base.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class PythonToolOutput extends base_cjs.ToolOutput {
+  static {
+    __name(this, "PythonToolOutput");
+  }
+  stdout;
+  stderr;
+  exitCode;
+  outputFiles;
+  static FILE_PREFIX = "urn:bee:file";
+  constructor(stdout, stderr, exitCode, outputFiles = []) {
+    super(), this.stdout = stdout, this.stderr = stderr, this.exitCode = exitCode, this.outputFiles = outputFiles;
+  }
+  static {
+    this.register();
+  }
+  isEmpty() {
+    return false;
+  }
+  getTextContent() {
+    const executionStatus = this.exitCode === 0 ? "The code executed successfully." : `The code exited with error code ${this.exitCode}.`;
+    const stdout = this.stdout.trim() ? `Standard output: 
+\`\`\`
+${this.stdout}
+\`\`\`` : null;
+    const stderr = this.stderr.trim() ? `Error output: 
+\`\`\`
+${this.stderr}
+\`\`\`` : null;
+    const isImage = /* @__PURE__ */ __name((filename) => [
+      ".png",
+      ".jpg",
+      ".jpeg",
+      ".gif",
+      ".bmp"
+    ].some((ext) => filename.toLowerCase().endsWith(ext)), "isImage");
+    const files = this.outputFiles.length ? "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" + this.outputFiles.map((file) => `${isImage(file.filename) ? "!" : ""}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`).join("\n") : null;
+    return [
+      executionStatus,
+      stdout,
+      stderr,
+      files
+    ].filter(Boolean).join("\n");
+  }
+  createSnapshot() {
+    return {
+      stdout: this.stdout,
+      stderr: this.stderr,
+      exitCode: this.exitCode,
+      outputFiles: this.outputFiles
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+
+exports.PythonToolOutput = PythonToolOutput;
+//# sourceMappingURL=output.cjs.map
+//# sourceMappingURL=output.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","constructor","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":";;;;;;AAmBO,MAAMA,yBAAyBC,mBAAAA,CAAAA;EAnBtC;;;;;;;AAoBE,EAAA,OAAOC,WAAc,GAAA,cAAA;AAErBC,EAAAA,WAAAA,CACkBC,MACAC,EAAAA,MAAAA,EACAC,QACAC,EAAAA,WAAAA,GAA4B,EAC5C,EAAA;AACA,IAAK,KAAA,EAAA,EAAA,IALWH,CAAAA,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IACAC,CAAAA,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;EAEAC,OAAU,GAAA;AACR,IAAO,OAAA,KAAA;AACT;EAEAC,cAAiB,GAAA;AACf,IAAA,MAAMC,kBACJ,IAAKL,CAAAA,QAAAA,KAAa,IACd,iCACA,GAAA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAS,GAAA,IAAA,CAAKA,MAAOQ,CAAAA,IAAAA,EAAS,GAAA,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAS,GAAA,IAAA,CAAKA,MAAOO,CAAAA,IAAAA,EAAS,GAAA,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAMQ,MAAAA,OAAAA,2BAAWC,QACf,KAAA;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAK,CAAA,CAACC,QAAQF,QAASG,CAAAA,WAAAA,GAAcC,QAASF,CAAAA,GAAAA,CAAAA,CAD1E,EAAA,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAQ,GAAA,IAAA,CAAKZ,WAAYa,CAAAA,MAAAA,GAC3B,oJACA,GAAA,IAAA,CAAKb,WACFc,CAAAA,GAAAA,CACC,CAACC,IAAAA,KACC,CAAGT,EAAAA,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAM,GAAA,EAAA,CAAMQ,CAAAA,EAAAA,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKd,gBAAiBE,CAAAA,WAAW,CAAIoB,CAAAA,EAAAA,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAEvGC,CAAAA,IAAAA,CAAK,IAAA,CACR,GAAA,IAAA;AAEJ,IAAO,OAAA;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAOC,CAAAA,OAAAA,CAASF,CAAAA,IAAAA,CAAK,IAAA,CAAA;AACvE;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACLvB,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAKA,CAAAA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAKA,CAAAA;AACpB,KAAA;AACF;AAEAqB,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"output.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { 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

@@ -0,0 +1,52 @@
+import { ToolOutput } from '../base.cjs';
+import { PythonFile } from './storage.cjs';
+import 'ajv';
+import '../../context.cjs';
+import '../../emitter-BWtGHYn0.cjs';
+import '../../internals/types.cjs';
+import '../../internals/helpers/guards.cjs';
+import '../../internals/serializable.cjs';
+import '../../internals/helpers/promise.cjs';
+import '../../errors.cjs';
+import 'promise-based-task';
+import '../../cache/base.cjs';
+import '../../internals/helpers/schema.cjs';
+import 'zod';
+import 'zod-to-json-schema';
+import 'fs';
+
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare class PythonToolOutput extends ToolOutput {
+    readonly stdout: string;
+    readonly stderr: string;
+    readonly exitCode?: number | undefined;
+    readonly outputFiles: PythonFile[];
+    static FILE_PREFIX: string;
+    constructor(stdout: string, stderr: string, exitCode?: number | undefined, outputFiles?: PythonFile[]);
+    isEmpty(): boolean;
+    getTextContent(): string;
+    createSnapshot(): {
+        stdout: string;
+        stderr: string;
+        exitCode: number | undefined;
+        outputFiles: PythonFile[];
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+
+export { PythonToolOutput };

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

@@ -0,0 +1,52 @@
+import { ToolOutput } from '../base.js';
+import { PythonFile } from './storage.js';
+import 'ajv';
+import '../../context.js';
+import '../../emitter-l0W9gC1A.js';
+import '../../internals/types.js';
+import '../../internals/helpers/guards.js';
+import '../../internals/serializable.js';
+import '../../internals/helpers/promise.js';
+import '../../errors.js';
+import 'promise-based-task';
+import '../../cache/base.js';
+import '../../internals/helpers/schema.js';
+import 'zod';
+import 'zod-to-json-schema';
+import 'fs';
+
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare class PythonToolOutput extends ToolOutput {
+    readonly stdout: string;
+    readonly stderr: string;
+    readonly exitCode?: number | undefined;
+    readonly outputFiles: PythonFile[];
+    static FILE_PREFIX: string;
+    constructor(stdout: string, stderr: string, exitCode?: number | undefined, outputFiles?: PythonFile[]);
+    isEmpty(): boolean;
+    getTextContent(): string;
+    createSnapshot(): {
+        stdout: string;
+        stderr: string;
+        exitCode: number | undefined;
+        outputFiles: PythonFile[];
+    };
+    loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
+}
+
+export { PythonToolOutput };

package/dist/tools/python/output.js

@@ -0,0 +1,63 @@
+import { ToolOutput } from '../base.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class PythonToolOutput extends ToolOutput {
+  static {
+    __name(this, "PythonToolOutput");
+  }
+  stdout;
+  stderr;
+  exitCode;
+  outputFiles;
+  static FILE_PREFIX = "urn:bee:file";
+  constructor(stdout, stderr, exitCode, outputFiles = []) {
+    super(), this.stdout = stdout, this.stderr = stderr, this.exitCode = exitCode, this.outputFiles = outputFiles;
+  }
+  static {
+    this.register();
+  }
+  isEmpty() {
+    return false;
+  }
+  getTextContent() {
+    const executionStatus = this.exitCode === 0 ? "The code executed successfully." : `The code exited with error code ${this.exitCode}.`;
+    const stdout = this.stdout.trim() ? `Standard output: 
+\`\`\`
+${this.stdout}
+\`\`\`` : null;
+    const stderr = this.stderr.trim() ? `Error output: 
+\`\`\`
+${this.stderr}
+\`\`\`` : null;
+    const isImage = /* @__PURE__ */ __name((filename) => [
+      ".png",
+      ".jpg",
+      ".jpeg",
+      ".gif",
+      ".bmp"
+    ].some((ext) => filename.toLowerCase().endsWith(ext)), "isImage");
+    const files = this.outputFiles.length ? "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" + this.outputFiles.map((file) => `${isImage(file.filename) ? "!" : ""}[${file.filename}](${PythonToolOutput.FILE_PREFIX}:${file.id})`).join("\n") : null;
+    return [
+      executionStatus,
+      stdout,
+      stderr,
+      files
+    ].filter(Boolean).join("\n");
+  }
+  createSnapshot() {
+    return {
+      stdout: this.stdout,
+      stderr: this.stderr,
+      exitCode: this.exitCode,
+      outputFiles: this.outputFiles
+    };
+  }
+  loadSnapshot(snapshot) {
+    Object.assign(this, snapshot);
+  }
+}
+
+export { PythonToolOutput };
+//# sourceMappingURL=output.js.map
+//# sourceMappingURL=output.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/python/output.ts"],"names":["PythonToolOutput","ToolOutput","FILE_PREFIX","constructor","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":";;;;AAmBO,MAAMA,yBAAyBC,UAAAA,CAAAA;EAnBtC;;;;;;;AAoBE,EAAA,OAAOC,WAAc,GAAA,cAAA;AAErBC,EAAAA,WAAAA,CACkBC,MACAC,EAAAA,MAAAA,EACAC,QACAC,EAAAA,WAAAA,GAA4B,EAC5C,EAAA;AACA,IAAK,KAAA,EAAA,EAAA,IALWH,CAAAA,MAAAA,GAAAA,MAAAA,EAAAA,IAAAA,CACAC,MAAAA,GAAAA,MAAAA,EAAAA,IACAC,CAAAA,QAAAA,GAAAA,QAAAA,EAAAA,IAAAA,CACAC,WAAAA,GAAAA,WAAAA;AAGlB;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;EAEAC,OAAU,GAAA;AACR,IAAO,OAAA,KAAA;AACT;EAEAC,cAAiB,GAAA;AACf,IAAA,MAAMC,kBACJ,IAAKL,CAAAA,QAAAA,KAAa,IACd,iCACA,GAAA,CAAA,gCAAA,EAAmC,KAAKA,QAAQ,CAAA,CAAA,CAAA;AACtD,IAAA,MAAMF,MAAS,GAAA,IAAA,CAAKA,MAAOQ,CAAAA,IAAAA,EAAS,GAAA,CAAA;;AAA8B,EAAA,IAAA,CAAKR,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AAC1F,IAAA,MAAMC,MAAS,GAAA,IAAA,CAAKA,MAAOO,CAAAA,IAAAA,EAAS,GAAA,CAAA;;AAA2B,EAAA,IAAA,CAAKP,MAAM;AAAa,MAAA,CAAA,GAAA,IAAA;AACvF,IAAMQ,MAAAA,OAAAA,2BAAWC,QACf,KAAA;AAAC,MAAA,MAAA;AAAQ,MAAA,MAAA;AAAQ,MAAA,OAAA;AAAS,MAAA,MAAA;AAAQ,MAAA;MAAQC,IAAK,CAAA,CAACC,QAAQF,QAASG,CAAAA,WAAAA,GAAcC,QAASF,CAAAA,GAAAA,CAAAA,CAD1E,EAAA,SAAA,CAAA;AAEhB,IAAA,MAAMG,KAAQ,GAAA,IAAA,CAAKZ,WAAYa,CAAAA,MAAAA,GAC3B,oJACA,GAAA,IAAA,CAAKb,WACFc,CAAAA,GAAAA,CACC,CAACC,IAAAA,KACC,CAAGT,EAAAA,OAAAA,CAAQS,KAAKR,QAAQ,CAAA,GAAI,GAAM,GAAA,EAAA,CAAMQ,CAAAA,EAAAA,IAAAA,CAAKR,QAAQ,CAAA,EAAA,EAAKd,gBAAiBE,CAAAA,WAAW,CAAIoB,CAAAA,EAAAA,IAAAA,CAAKC,EAAE,CAAA,CAAA,CAAG,CAEvGC,CAAAA,IAAAA,CAAK,IAAA,CACR,GAAA,IAAA;AAEJ,IAAO,OAAA;AAACb,MAAAA,eAAAA;AAAiBP,MAAAA,MAAAA;AAAQC,MAAAA,MAAAA;AAAQc,MAAAA;MAAOM,MAAOC,CAAAA,OAAAA,CAASF,CAAAA,IAAAA,CAAK,IAAA,CAAA;AACvE;EAEAG,cAAiB,GAAA;AACf,IAAO,OAAA;AACLvB,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA,MAAAA;AACbC,MAAAA,MAAAA,EAAQ,IAAKA,CAAAA,MAAAA;AACbC,MAAAA,QAAAA,EAAU,IAAKA,CAAAA,QAAAA;AACfC,MAAAA,WAAAA,EAAa,IAAKA,CAAAA;AACpB,KAAA;AACF;AAEAqB,EAAAA,YAAAA,CAAaC,QAAkD,EAAA;AAC7DC,IAAOC,MAAAA,CAAAA,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;AACF","file":"output.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { 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

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