@huggingface/inference 4.13.2 → 4.13.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (11) hide show
  1. package/dist/commonjs/package.d.ts +1 -1
  2. package/dist/commonjs/package.js +1 -1
  3. package/dist/commonjs/providers/wavespeed.d.ts.map +1 -1
  4. package/dist/commonjs/providers/wavespeed.js +14 -11
  5. package/dist/esm/package.d.ts +1 -1
  6. package/dist/esm/package.js +1 -1
  7. package/dist/esm/providers/wavespeed.d.ts.map +1 -1
  8. package/dist/esm/providers/wavespeed.js +14 -11
  9. package/package.json +3 -3
  10. package/src/package.ts +1 -1
  11. package/src/providers/wavespeed.ts +23 -15
package/dist/commonjs/package.d.ts CHANGED
@@ -1,3 +1,3 @@
1
- export declare const PACKAGE_VERSION = "4.13.2";
1
+ export declare const PACKAGE_VERSION = "4.13.3";
2
2
  export declare const PACKAGE_NAME = "@huggingface/inference";
3
3
  //# sourceMappingURL=package.d.ts.map
package/dist/commonjs/package.js CHANGED
@@ -2,5 +2,5 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.PACKAGE_NAME = exports.PACKAGE_VERSION = void 0;
4
4
  // Generated file from package.json. Issues importing JSON directly when publishing on commonjs/ESM - see https://github.com/microsoft/TypeScript/issues/51783
5
- exports.PACKAGE_VERSION = "4.13.2";
5
+ exports.PACKAGE_VERSION = "4.13.3";
6
6
  exports.PACKAGE_NAME = "@huggingface/inference";
package/dist/commonjs/providers/wavespeed.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"wavespeed.d.ts","sourceRoot":"","sources":["../../../src/providers/wavespeed.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAItE,OAAO,KAAK,EACX,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AA6BzD;;GAEG;AACH,UAAU,yBAAyB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACL,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC;CACF;AAWD;;GAEG;AACH,UAAU,6BAA6B;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,yBAAyB,CAAC;CAChC;AAED,uBAAe,eAAgB,SAAQ,kBAAkB;gBAC5C,GAAG,CAAC,EAAE,MAAM;IAIxB,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAIpC,cAAc,CACb,MAAM,EAAE,UAAU,CAAC,gBAAgB,GAAG,eAAe,GAAG,eAAe,GAAG,gBAAgB,CAAC,GACzF,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAkBX,WAAW,CACzB,QAAQ,EAAE,6BAA6B,EACvC,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAkEhB;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CASvE;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CASvE"}
1
+ {"version":3,"file":"wavespeed.d.ts","sourceRoot":"","sources":["../../../src/providers/wavespeed.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAItE,OAAO,KAAK,EACX,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AA6BzD;;GAEG;AACH,UAAU,yBAAyB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACL,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC;CACF;AAWD;;GAEG;AACH,UAAU,6BAA6B;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,yBAAyB,CAAC;CAChC;AAgBD,uBAAe,eAAgB,SAAQ,kBAAkB;gBAC5C,GAAG,CAAC,EAAE,MAAM;IAIxB,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAIpC,cAAc,CACb,MAAM,EAAE,UAAU,CAAC,gBAAgB,GAAG,eAAe,GAAG,eAAe,GAAG,gBAAgB,CAAC,GACzF,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAkBX,WAAW,CACzB,QAAQ,EAAE,6BAA6B,EACvC,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAkEhB;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CAMvE;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CAMvE"}
package/dist/commonjs/providers/wavespeed.js CHANGED
@@ -7,6 +7,13 @@ const base64FromBytes_js_1 = require("../utils/base64FromBytes.js");
7
7
  const providerHelper_js_1 = require("./providerHelper.js");
8
8
  const errors_js_1 = require("../errors.js");
9
9
  const WAVESPEEDAI_API_BASE_URL = "https://api.wavespeed.ai";
10
+ async function buildImagesField(inputs, hasImages) {
11
+ const base = (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(inputs instanceof ArrayBuffer ? inputs : await inputs.arrayBuffer()));
12
+ const images = Array.isArray(hasImages) && hasImages.every((value) => typeof value === "string")
13
+ ? hasImages
14
+ : [base];
15
+ return { base, images };
16
+ }
10
17
  class WavespeedAITask extends providerHelper_js_1.TaskProviderHelper {
11
18
  constructor(url) {
12
19
  super("wavespeed", url || WAVESPEEDAI_API_BASE_URL);
@@ -17,7 +24,7 @@ class WavespeedAITask extends providerHelper_js_1.TaskProviderHelper {
17
24
  preparePayload(params) {
18
25
  const payload = {
19
26
  ...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
20
- ...params.args.parameters,
27
+ ...(params.args.parameters ? (0, omit_js_1.omit)(params.args.parameters, ["images"]) : undefined),
21
28
  prompt: params.args.inputs,
22
29
  };
23
30
  // Add LoRA support if adapter is specified in the mapping
@@ -97,11 +104,9 @@ class WavespeedAIImageToImageTask extends WavespeedAITask {
97
104
  super(WAVESPEEDAI_API_BASE_URL);
98
105
  }
99
106
  async preparePayloadAsync(args) {
100
- return {
101
- ...args,
102
- inputs: args.parameters?.prompt,
103
- image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())),
104
- };
107
+ const hasImages = args.images ?? args.parameters?.images;
108
+ const { base, images } = await buildImagesField(args.inputs, hasImages);
109
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
105
110
  }
106
111
  }
107
112
  exports.WavespeedAIImageToImageTask = WavespeedAIImageToImageTask;
@@ -110,11 +115,9 @@ class WavespeedAIImageToVideoTask extends WavespeedAITask {
110
115
  super(WAVESPEEDAI_API_BASE_URL);
111
116
  }
112
117
  async preparePayloadAsync(args) {
113
- return {
114
- ...args,
115
- inputs: args.parameters?.prompt,
116
- image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())),
117
- };
118
+ const hasImages = args.images ?? args.parameters?.images;
119
+ const { base, images } = await buildImagesField(args.inputs, hasImages);
120
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
118
121
  }
119
122
  }
120
123
  exports.WavespeedAIImageToVideoTask = WavespeedAIImageToVideoTask;
package/dist/esm/package.d.ts CHANGED
@@ -1,3 +1,3 @@
1
- export declare const PACKAGE_VERSION = "4.13.2";
1
+ export declare const PACKAGE_VERSION = "4.13.3";
2
2
  export declare const PACKAGE_NAME = "@huggingface/inference";
3
3
  //# sourceMappingURL=package.d.ts.map
package/dist/esm/package.js CHANGED
@@ -1,3 +1,3 @@
1
1
  // Generated file from package.json. Issues importing JSON directly when publishing on commonjs/ESM - see https://github.com/microsoft/TypeScript/issues/51783
2
- export const PACKAGE_VERSION = "4.13.2";
2
+ export const PACKAGE_VERSION = "4.13.3";
3
3
  export const PACKAGE_NAME = "@huggingface/inference";
package/dist/esm/providers/wavespeed.d.ts.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"wavespeed.d.ts","sourceRoot":"","sources":["../../../src/providers/wavespeed.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAItE,OAAO,KAAK,EACX,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AA6BzD;;GAEG;AACH,UAAU,yBAAyB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACL,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC;CACF;AAWD;;GAEG;AACH,UAAU,6BAA6B;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,yBAAyB,CAAC;CAChC;AAED,uBAAe,eAAgB,SAAQ,kBAAkB;gBAC5C,GAAG,CAAC,EAAE,MAAM;IAIxB,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAIpC,cAAc,CACb,MAAM,EAAE,UAAU,CAAC,gBAAgB,GAAG,eAAe,GAAG,eAAe,GAAG,gBAAgB,CAAC,GACzF,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAkBX,WAAW,CACzB,QAAQ,EAAE,6BAA6B,EACvC,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAkEhB;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CASvE;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CASvE"}
1
+ {"version":3,"file":"wavespeed.d.ts","sourceRoot":"","sources":["../../../src/providers/wavespeed.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAItE,OAAO,KAAK,EACX,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,sBAAsB,EACtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AA6BzD;;GAEG;AACH,UAAU,yBAAyB;IAClC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE;QACL,GAAG,EAAE,MAAM,CAAC;KACZ,CAAC;CACF;AAWD;;GAEG;AACH,UAAU,6BAA6B;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,yBAAyB,CAAC;CAChC;AAgBD,uBAAe,eAAgB,SAAQ,kBAAkB;gBAC5C,GAAG,CAAC,EAAE,MAAM;IAIxB,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAIpC,cAAc,CACb,MAAM,EAAE,UAAU,CAAC,gBAAgB,GAAG,eAAe,GAAG,eAAe,GAAG,gBAAgB,CAAC,GACzF,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAkBX,WAAW,CACzB,QAAQ,EAAE,6BAA6B,EACvC,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAkEhB;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;;CAI/F;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CAMvE;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;;IAK3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;CAMvE"}
package/dist/esm/providers/wavespeed.js CHANGED
@@ -4,6 +4,13 @@ import { base64FromBytes } from "../utils/base64FromBytes.js";
4
4
  import { TaskProviderHelper } from "./providerHelper.js";
5
5
  import { InferenceClientInputError, InferenceClientProviderApiError, InferenceClientProviderOutputError, } from "../errors.js";
6
6
  const WAVESPEEDAI_API_BASE_URL = "https://api.wavespeed.ai";
7
+ async function buildImagesField(inputs, hasImages) {
8
+ const base = base64FromBytes(new Uint8Array(inputs instanceof ArrayBuffer ? inputs : await inputs.arrayBuffer()));
9
+ const images = Array.isArray(hasImages) && hasImages.every((value) => typeof value === "string")
10
+ ? hasImages
11
+ : [base];
12
+ return { base, images };
13
+ }
7
14
  class WavespeedAITask extends TaskProviderHelper {
8
15
  constructor(url) {
9
16
  super("wavespeed", url || WAVESPEEDAI_API_BASE_URL);
@@ -14,7 +21,7 @@ class WavespeedAITask extends TaskProviderHelper {
14
21
  preparePayload(params) {
15
22
  const payload = {
16
23
  ...omit(params.args, ["inputs", "parameters"]),
17
- ...params.args.parameters,
24
+ ...(params.args.parameters ? omit(params.args.parameters, ["images"]) : undefined),
18
25
  prompt: params.args.inputs,
19
26
  };
20
27
  // Add LoRA support if adapter is specified in the mapping
@@ -92,11 +99,9 @@ export class WavespeedAIImageToImageTask extends WavespeedAITask {
92
99
  super(WAVESPEEDAI_API_BASE_URL);
93
100
  }
94
101
  async preparePayloadAsync(args) {
95
- return {
96
- ...args,
97
- inputs: args.parameters?.prompt,
98
- image: base64FromBytes(new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())),
99
- };
102
+ const hasImages = args.images ?? args.parameters?.images;
103
+ const { base, images } = await buildImagesField(args.inputs, hasImages);
104
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
100
105
  }
101
106
  }
102
107
  export class WavespeedAIImageToVideoTask extends WavespeedAITask {
@@ -104,10 +109,8 @@ export class WavespeedAIImageToVideoTask extends WavespeedAITask {
104
109
  super(WAVESPEEDAI_API_BASE_URL);
105
110
  }
106
111
  async preparePayloadAsync(args) {
107
- return {
108
- ...args,
109
- inputs: args.parameters?.prompt,
110
- image: base64FromBytes(new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())),
111
- };
112
+ const hasImages = args.images ?? args.parameters?.images;
113
+ const { base, images } = await buildImagesField(args.inputs, hasImages);
114
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
112
115
  }
113
116
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@huggingface/inference",
3
- "version": "4.13.2",
3
+ "version": "4.13.3",
4
4
  "license": "MIT",
5
5
  "author": "Hugging Face and Tim Mikeladze <tim.mikeladze@gmail.com>",
6
6
  "description": "Typescript client for the Hugging Face Inference Providers and Inference Endpoints",
@@ -40,8 +40,8 @@
40
40
  },
41
41
  "type": "module",
42
42
  "dependencies": {
43
- "@huggingface/jinja": "^0.5.1",
44
- "@huggingface/tasks": "^0.19.63"
43
+ "@huggingface/tasks": "^0.19.63",
44
+ "@huggingface/jinja": "^0.5.1"
45
45
  },
46
46
  "devDependencies": {
47
47
  "@types/node": "18.13.0"
package/src/package.ts CHANGED
@@ -1,3 +1,3 @@
1
1
  // Generated file from package.json. Issues importing JSON directly when publishing on commonjs/ESM - see https://github.com/microsoft/TypeScript/issues/51783
2
- export const PACKAGE_VERSION = "4.13.2";
2
+ export const PACKAGE_VERSION = "4.13.3";
3
3
  export const PACKAGE_NAME = "@huggingface/inference";
package/src/providers/wavespeed.ts CHANGED
@@ -69,6 +69,20 @@ interface WaveSpeedAISubmitTaskResponse {
69
69
  data: WaveSpeedAISubmitResponse;
70
70
  }
71
71
 
72
+ async function buildImagesField(
73
+ inputs: Blob | ArrayBuffer,
74
+ hasImages: unknown
75
+ ): Promise<{ base: string; images: string[] }> {
76
+ const base = base64FromBytes(
77
+ new Uint8Array(inputs instanceof ArrayBuffer ? inputs : await (inputs as Blob).arrayBuffer())
78
+ );
79
+ const images =
80
+ Array.isArray(hasImages) && hasImages.every((value): value is string => typeof value === "string")
81
+ ? hasImages
82
+ : [base];
83
+ return { base, images };
84
+ }
85
+
72
86
  abstract class WavespeedAITask extends TaskProviderHelper {
73
87
  constructor(url?: string) {
74
88
  super("wavespeed", url || WAVESPEEDAI_API_BASE_URL);
@@ -83,7 +97,7 @@ abstract class WavespeedAITask extends TaskProviderHelper {
83
97
  ): Record<string, unknown> {
84
98
  const payload: Record<string, unknown> = {
85
99
  ...omit(params.args, ["inputs", "parameters"]),
86
- ...params.args.parameters,
100
+ ...(params.args.parameters ? omit(params.args.parameters as Record<string, unknown>, ["images"]) : undefined),
87
101
  prompt: params.args.inputs,
88
102
  };
89
103
  // Add LoRA support if adapter is specified in the mapping
@@ -188,13 +202,10 @@ export class WavespeedAIImageToImageTask extends WavespeedAITask implements Imag
188
202
  }
189
203
 
190
204
  async preparePayloadAsync(args: ImageToImageArgs): Promise<RequestArgs> {
191
- return {
192
- ...args,
193
- inputs: args.parameters?.prompt,
194
- image: base64FromBytes(
195
- new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await (args.inputs as Blob).arrayBuffer())
196
- ),
197
- };
205
+ const hasImages =
206
+ (args as { images?: unknown }).images ?? (args.parameters as Record<string, unknown> | undefined)?.images;
207
+ const { base, images } = await buildImagesField(args.inputs as Blob | ArrayBuffer, hasImages);
208
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
198
209
  }
199
210
  }
200
211
 
@@ -204,12 +215,9 @@ export class WavespeedAIImageToVideoTask extends WavespeedAITask implements Imag
204
215
  }
205
216
 
206
217
  async preparePayloadAsync(args: ImageToVideoArgs): Promise<RequestArgs> {
207
- return {
208
- ...args,
209
- inputs: args.parameters?.prompt,
210
- image: base64FromBytes(
211
- new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await (args.inputs as Blob).arrayBuffer())
212
- ),
213
- };
218
+ const hasImages =
219
+ (args as { images?: unknown }).images ?? (args.parameters as Record<string, unknown> | undefined)?.images;
220
+ const { base, images } = await buildImagesField(args.inputs as Blob | ArrayBuffer, hasImages);
221
+ return { ...args, inputs: args.parameters?.prompt, image: base, images };
214
222
  }
215
223
  }