@huggingface/inference 3.3.4 → 3.3.6

package/dist/index.cjs

@@ -134,7 +134,7 @@ function isUrl(modelOrUrl) {
 
 // package.json
 var name = "@huggingface/inference";
-var version = "3.3.4";
+var version = "3.3.6";
 
 // src/providers/consts.ts
 var HARDCODED_MODEL_ID_MAPPING = {
@@ -840,24 +840,35 @@ async function textToImage(args, options) {
   });
   if (res && typeof res === "object") {
     if (args.provider === "black-forest-labs" && "polling_url" in res && typeof res.polling_url === "string") {
-      return await pollBflResponse(res.polling_url);
+      return await pollBflResponse(res.polling_url, options?.outputType);
     }
     if (args.provider === "fal-ai" && "images" in res && Array.isArray(res.images) && res.images[0].url) {
-      const image = await fetch(res.images[0].url);
-      return await image.blob();
+      if (options?.outputType === "url") {
+        return res.images[0].url;
+      } else {
+        const image = await fetch(res.images[0].url);
+        return await image.blob();
+      }
     }
     if (args.provider === "hyperbolic" && "images" in res && Array.isArray(res.images) && res.images[0] && typeof res.images[0].image === "string") {
+      if (options?.outputType === "url") {
+        return `data:image/jpeg;base64,${res.images[0].image}`;
+      }
       const base64Response = await fetch(`data:image/jpeg;base64,${res.images[0].image}`);
-      const blob = await base64Response.blob();
-      return blob;
+      return await base64Response.blob();
     }
     if ("data" in res && Array.isArray(res.data) && res.data[0].b64_json) {
       const base64Data = res.data[0].b64_json;
+      if (options?.outputType === "url") {
+        return `data:image/jpeg;base64,${base64Data}`;
+      }
       const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);
-      const blob = await base64Response.blob();
-      return blob;
+      return await base64Response.blob();
     }
     if ("output" in res && Array.isArray(res.output)) {
+      if (options?.outputType === "url") {
+        return res.output[0];
+      }
       const urlResponse = await fetch(res.output[0]);
       const blob = await urlResponse.blob();
       return blob;
@@ -867,9 +878,13 @@ async function textToImage(args, options) {
   if (!isValidOutput) {
     throw new InferenceOutputError("Expected Blob");
   }
+  if (options?.outputType === "url") {
+    const b64 = await res.arrayBuffer().then((buf) => Buffer.from(buf).toString("base64"));
+    return `data:image/jpeg;base64,${b64}`;
+  }
   return res;
 }
-async function pollBflResponse(url) {
+async function pollBflResponse(url, outputType) {
   const urlObj = new URL(url);
   for (let step = 0; step < 5; step++) {
     await delay(1e3);
@@ -881,6 +896,9 @@ async function pollBflResponse(url) {
     }
     const payload = await resp.json();
     if (typeof payload === "object" && payload && "status" in payload && typeof payload.status === "string" && payload.status === "Ready" && "result" in payload && typeof payload.result === "object" && payload.result && "sample" in payload.result && typeof payload.result.sample === "string") {
+      if (outputType === "url") {
+        return payload.result.sample;
+      }
       const image = await fetch(payload.result.sample);
       return await image.blob();
     }

package/dist/index.js

@@ -79,7 +79,7 @@ function isUrl(modelOrUrl) {
 
 // package.json
 var name = "@huggingface/inference";
-var version = "3.3.4";
+var version = "3.3.6";
 
 // src/providers/consts.ts
 var HARDCODED_MODEL_ID_MAPPING = {
@@ -785,24 +785,35 @@ async function textToImage(args, options) {
   });
   if (res && typeof res === "object") {
     if (args.provider === "black-forest-labs" && "polling_url" in res && typeof res.polling_url === "string") {
-      return await pollBflResponse(res.polling_url);
+      return await pollBflResponse(res.polling_url, options?.outputType);
     }
     if (args.provider === "fal-ai" && "images" in res && Array.isArray(res.images) && res.images[0].url) {
-      const image = await fetch(res.images[0].url);
-      return await image.blob();
+      if (options?.outputType === "url") {
+        return res.images[0].url;
+      } else {
+        const image = await fetch(res.images[0].url);
+        return await image.blob();
+      }
     }
     if (args.provider === "hyperbolic" && "images" in res && Array.isArray(res.images) && res.images[0] && typeof res.images[0].image === "string") {
+      if (options?.outputType === "url") {
+        return `data:image/jpeg;base64,${res.images[0].image}`;
+      }
       const base64Response = await fetch(`data:image/jpeg;base64,${res.images[0].image}`);
-      const blob = await base64Response.blob();
-      return blob;
+      return await base64Response.blob();
     }
     if ("data" in res && Array.isArray(res.data) && res.data[0].b64_json) {
       const base64Data = res.data[0].b64_json;
+      if (options?.outputType === "url") {
+        return `data:image/jpeg;base64,${base64Data}`;
+      }
       const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);
-      const blob = await base64Response.blob();
-      return blob;
+      return await base64Response.blob();
     }
     if ("output" in res && Array.isArray(res.output)) {
+      if (options?.outputType === "url") {
+        return res.output[0];
+      }
       const urlResponse = await fetch(res.output[0]);
       const blob = await urlResponse.blob();
       return blob;
@@ -812,9 +823,13 @@ async function textToImage(args, options) {
   if (!isValidOutput) {
     throw new InferenceOutputError("Expected Blob");
   }
+  if (options?.outputType === "url") {
+    const b64 = await res.arrayBuffer().then((buf) => Buffer.from(buf).toString("base64"));
+    return `data:image/jpeg;base64,${b64}`;
+  }
   return res;
 }
-async function pollBflResponse(url) {
+async function pollBflResponse(url, outputType) {
   const urlObj = new URL(url);
   for (let step = 0; step < 5; step++) {
     await delay(1e3);
@@ -826,6 +841,9 @@ async function pollBflResponse(url) {
     }
     const payload = await resp.json();
     if (typeof payload === "object" && payload && "status" in payload && typeof payload.status === "string" && payload.status === "Ready" && "result" in payload && typeof payload.result === "object" && payload.result && "sample" in payload.result && typeof payload.result.sample === "string") {
+      if (outputType === "url") {
+        return payload.result.sample;
+      }
       const image = await fetch(payload.result.sample);
       return await image.blob();
     }

package/dist/src/tasks/cv/textToImage.d.ts

@@ -1,9 +1,18 @@
 import type { TextToImageInput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
 export type TextToImageArgs = BaseArgs & TextToImageInput;
+interface TextToImageOptions extends Options {
+    outputType?: "url" | "blob";
+}
 /**
  * This task reads some text input and outputs an image.
  * Recommended model: stabilityai/stable-diffusion-2
  */
-export declare function textToImage(args: TextToImageArgs, options?: Options): Promise<Blob>;
+export declare function textToImage(args: TextToImageArgs, options?: TextToImageOptions & {
+    outputType: "url";
+}): Promise<string>;
+export declare function textToImage(args: TextToImageArgs, options?: TextToImageOptions & {
+    outputType?: undefined | "blob";
+}): Promise<Blob>;
+export {};
 //# sourceMappingURL=textToImage.d.ts.map

package/dist/src/tasks/cv/textToImage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"textToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAqB,MAAM,oBAAoB,CAAC;AAE9E,OAAO,KAAK,EAAE,QAAQ,EAAqB,OAAO,EAAE,MAAM,aAAa,CAAC;AAKxE,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAkC1D;;;GAGG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAyDzF"}
+{"version":3,"file":"textToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAqB,MAAM,oBAAoB,CAAC;AAE9E,OAAO,KAAK,EAAE,QAAQ,EAAqB,OAAO,EAAE,MAAM,aAAa,CAAC;AAKxE,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAmB1D,UAAU,kBAAmB,SAAQ,OAAO;IAC3C,UAAU,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;CAC5B;AAiBD;;;GAGG;AACH,wBAAsB,WAAW,CAChC,IAAI,EAAE,eAAe,EACrB,OAAO,CAAC,EAAE,kBAAkB,GAAG;IAAE,UAAU,EAAE,KAAK,CAAA;CAAE,GAClD,OAAO,CAAC,MAAM,CAAC,CAAC;AACnB,wBAAsB,WAAW,CAChC,IAAI,EAAE,eAAe,EACrB,OAAO,CAAC,EAAE,kBAAkB,GAAG;IAAE,UAAU,CAAC,EAAE,SAAS,GAAG,MAAM,CAAA;CAAE,GAChE,OAAO,CAAC,IAAI,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@huggingface/inference",
-  "version": "3.3.4",
+  "version": "3.3.6",
   "packageManager": "pnpm@8.10.5",
   "license": "MIT",
   "author": "Tim Mikeladze <tim.mikeladze@gmail.com>",
@@ -39,7 +39,7 @@
   },
   "type": "module",
   "dependencies": {
-    "@huggingface/tasks": "^0.15.9"
+    "@huggingface/tasks": "^0.16.0"
   },
   "devDependencies": {
     "@types/node": "18.13.0"

package/src/tasks/cv/textToImage.ts

@@ -24,6 +24,10 @@ interface BlackForestLabsResponse {
 	polling_url: string;
 }
 
+interface TextToImageOptions extends Options {
+	outputType?: "url" | "blob";
+}
+
 function getResponseFormatArg(provider: InferenceProvider) {
 	switch (provider) {
 		case "fal-ai":
@@ -43,7 +47,15 @@ function getResponseFormatArg(provider: InferenceProvider) {
  * This task reads some text input and outputs an image.
  * Recommended model: stabilityai/stable-diffusion-2
  */
-export async function textToImage(args: TextToImageArgs, options?: Options): Promise<Blob> {
+export async function textToImage(
+	args: TextToImageArgs,
+	options?: TextToImageOptions & { outputType: "url" }
+): Promise<string>;
+export async function textToImage(
+	args: TextToImageArgs,
+	options?: TextToImageOptions & { outputType?: undefined | "blob" }
+): Promise<Blob>;
+export async function textToImage(args: TextToImageArgs, options?: TextToImageOptions): Promise<Blob | string> {
 	const payload =
 		!args.provider || args.provider === "hf-inference" || args.provider === "sambanova"
 			? args
@@ -66,11 +78,15 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
 
 	if (res && typeof res === "object") {
 		if (args.provider === "black-forest-labs" && "polling_url" in res && typeof res.polling_url === "string") {
-			return await pollBflResponse(res.polling_url);
+			return await pollBflResponse(res.polling_url, options?.outputType);
 		}
 		if (args.provider === "fal-ai" && "images" in res && Array.isArray(res.images) && res.images[0].url) {
-			const image = await fetch(res.images[0].url);
-			return await image.blob();
+			if (options?.outputType === "url") {
+				return res.images[0].url;
+			} else {
+				const image = await fetch(res.images[0].url);
+				return await image.blob();
+			}
 		}
 		if (
 			args.provider === "hyperbolic" &&
@@ -79,17 +95,24 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
 			res.images[0] &&
 			typeof res.images[0].image === "string"
 		) {
+			if (options?.outputType === "url") {
+				return `data:image/jpeg;base64,${res.images[0].image}`;
+			}
 			const base64Response = await fetch(`data:image/jpeg;base64,${res.images[0].image}`);
-			const blob = await base64Response.blob();
-			return blob;
+			return await base64Response.blob();
 		}
 		if ("data" in res && Array.isArray(res.data) && res.data[0].b64_json) {
 			const base64Data = res.data[0].b64_json;
+			if (options?.outputType === "url") {
+				return `data:image/jpeg;base64,${base64Data}`;
+			}
 			const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);
-			const blob = await base64Response.blob();
-			return blob;
+			return await base64Response.blob();
 		}
 		if ("output" in res && Array.isArray(res.output)) {
+			if (options?.outputType === "url") {
+				return res.output[0];
+			}
 			const urlResponse = await fetch(res.output[0]);
 			const blob = await urlResponse.blob();
 			return blob;
@@ -99,10 +122,14 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
 	if (!isValidOutput) {
 		throw new InferenceOutputError("Expected Blob");
 	}
+	if (options?.outputType === "url") {
+		const b64 = await res.arrayBuffer().then((buf) => Buffer.from(buf).toString("base64"));
+		return `data:image/jpeg;base64,${b64}`;
+	}
 	return res;
 }
 
-async function pollBflResponse(url: string): Promise<Blob> {
+async function pollBflResponse(url: string, outputType?: "url" | "blob"): Promise<Blob> {
 	const urlObj = new URL(url);
 	for (let step = 0; step < 5; step++) {
 		await delay(1000);
@@ -125,6 +152,9 @@ async function pollBflResponse(url: string): Promise<Blob> {
 			"sample" in payload.result &&
 			typeof payload.result.sample === "string"
 		) {
+			if (outputType === "url") {
+				return payload.result.sample;
+			}
 			const image = await fetch(payload.result.sample);
 			return await image.blob();
 		}