throw error instead

Author: hanouticelina

packages/inference/src/providers/hf-inference.ts

@@ -36,7 +36,7 @@ import type {
 	ZeroShotImageClassificationOutput,
 } from "@huggingface/tasks";
 import { HF_ROUTER_URL } from "../config.js";
-import { InferenceClientProviderOutputError } from "../errors.js";
+import { InferenceClientInputError, InferenceClientProviderOutputError } from "../errors.js";
 import type { TabularClassificationOutput } from "../tasks/tabular/tabularClassification.js";
 import type { BodyParams, RequestArgs, UrlParams } from "../types.js";
 import { toArray } from "../utils/toArray.js";
@@ -141,10 +141,9 @@ export class HFInferenceTextToImageTask extends HFInferenceTask implements TextT
 			if ("data" in response && Array.isArray(response.data) && response.data[0].b64_json) {
 				const base64Data = response.data[0].b64_json;
 				if (outputType === "url") {
-					console.warn(
-						"hf-inference provider does not support URL output for this model, returning base64 data URL instead"
+					throw new InferenceClientInputError(
+						"hf-inference provider does not support URL output for this model. Use outputType 'blob' or 'json' instead."
 					);
-					return `data:image/jpeg;base64,${base64Data}`;
 				}
 				const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);
 				return await base64Response.blob();
@@ -159,12 +158,14 @@ export class HFInferenceTextToImageTask extends HFInferenceTask implements TextT
 			}
 		}
 		if (response instanceof Blob) {
-			if (outputType === "url" || outputType === "json") {
-				console.warn(
-					"hf-inference provider does not support URL output for this model, returning base64 data URL instead"
+			if (outputType === "url") {
+				throw new InferenceClientInputError(
+					"hf-inference provider does not support URL output for this model. Use outputType 'blob' or 'json' instead."
 				);
+			}
+			if (outputType === "json") {
 				const b64 = await response.arrayBuffer().then((buf) => Buffer.from(buf).toString("base64"));
-				return outputType === "url" ? `data:image/jpeg;base64,${b64}` : { output: `data:image/jpeg;base64,${b64}` };
+				return { output: `data:image/jpeg;base64,${b64}` };
 			}
 			return response;
 		}

packages/inference/src/providers/hyperbolic.ts

@@ -23,7 +23,7 @@ import {
 	TaskProviderHelper,
 	type TextToImageTaskHelper,
 } from "./providerHelper.js";
-import { InferenceClientProviderOutputError } from "../errors.js";
+import { InferenceClientInputError, InferenceClientProviderOutputError } from "../errors.js";
 const HYPERBOLIC_API_BASE_URL = "https://api.hyperbolic.xyz";
 
 export interface HyperbolicTextCompletionOutput extends Omit<ChatCompletionOutput, "choices"> {
@@ -118,8 +118,9 @@ export class HyperbolicTextToImageTask extends TaskProviderHelper implements Tex
 				return { ...response };
 			}
 			if (outputType === "url") {
-				console.warn("hyperbolic provider does not support URL output, returning base64 data URL instead");
-				return `data:image/jpeg;base64,${response.images[0].image}`;
+				throw new InferenceClientInputError(
+					"hyperbolic provider does not support URL output. Use outputType 'blob' or 'json' instead."
+				);
 			}
 			return fetch(`data:image/jpeg;base64,${response.images[0].image}`).then((res) => res.blob());
 		}

packages/inference/src/providers/nscale.ts

@@ -18,7 +18,7 @@ import type { TextToImageInput } from "@huggingface/tasks";
 import type { BodyParams } from "../types.js";
 import { omit } from "../utils/omit.js";
 import { BaseConversationalTask, TaskProviderHelper, type TextToImageTaskHelper } from "./providerHelper.js";
-import { InferenceClientProviderOutputError } from "../errors.js";
+import { InferenceClientInputError, InferenceClientProviderOutputError } from "../errors.js";
 
 const NSCALE_API_BASE_URL = "https://inference.api.nscale.com";
 
@@ -72,8 +72,9 @@ export class NscaleTextToImageTask extends TaskProviderHelper implements TextToI
 			}
 			const base64Data = response.data[0].b64_json;
 			if (outputType === "url") {
-				console.warn("nscale provider does not support URL output, returning base64 data URL instead");
-				return `data:image/jpeg;base64,${base64Data}`;
+				throw new InferenceClientInputError(
+					"nscale provider does not support URL output. Use outputType 'blob' or 'json' instead."
+				);
 			}
 			return fetch(`data:image/jpeg;base64,${base64Data}`).then((res) => res.blob());
 		}