@huggingface/inference 3.14.0 → 4.0.0

package/src/providers/fal-ai.ts

@@ -17,7 +17,6 @@
 import { base64FromBytes } from "../utils/base64FromBytes.js";
 
 import type { AutomaticSpeechRecognitionOutput } from "@huggingface/tasks";
-import { InferenceOutputError } from "../lib/InferenceOutputError.js";
 import { isUrl } from "../lib/isUrl.js";
 import type { BodyParams, HeaderParams, ModelId, RequestArgs, UrlParams } from "../types.js";
 import { delay } from "../utils/delay.js";
@@ -30,6 +29,11 @@ import {
 } from "./providerHelper.js";
 import { HF_HUB_URL } from "../config.js";
 import type { AutomaticSpeechRecognitionArgs } from "../tasks/audio/automaticSpeechRecognition.js";
+import {
+	InferenceClientInputError,
+	InferenceClientProviderApiError,
+	InferenceClientProviderOutputError,
+} from "../errors.js";
 
 export interface FalAiQueueOutput {
 	request_id: string;
@@ -122,7 +126,7 @@ export class FalAITextToImageTask extends FalAITask implements TextToImageTaskHe
 			return await urlResponse.blob();
 		}
 
-		throw new InferenceOutputError("Expected Fal.ai text-to-image response format");
+		throw new InferenceClientProviderOutputError("Received malformed response from Fal.ai text-to-image API");
 	}
 }
 
@@ -150,11 +154,13 @@ export class FalAITextToVideoTask extends FalAITask implements TextToVideoTaskHe
 		headers?: Record<string, string>
 	): Promise<Blob> {
 		if (!url || !headers) {
-			throw new InferenceOutputError("URL and headers are required for text-to-video task");
+			throw new InferenceClientInputError("URL and headers are required for text-to-video task");
 		}
 		const requestId = response.request_id;
 		if (!requestId) {
-			throw new InferenceOutputError("No request ID found in the response");
+			throw new InferenceClientProviderOutputError(
+				"Received malformed response from Fal.ai text-to-video API: no request ID found in the response"
+			);
 		}
 		let status = response.status;
 
@@ -176,12 +182,22 @@ export class FalAITextToVideoTask extends FalAITask implements TextToVideoTaskHe
 			const statusResponse = await fetch(statusUrl, { headers });
 
 			if (!statusResponse.ok) {
-				throw new InferenceOutputError("Failed to fetch response status from fal-ai API");
+				throw new InferenceClientProviderApiError(
+					"Failed to fetch response status from fal-ai API",
+					{ url: statusUrl, method: "GET" },
+					{
+						requestId: statusResponse.headers.get("x-request-id") ?? "",
+						status: statusResponse.status,
+						body: await statusResponse.text(),
+					}
+				);
 			}
 			try {
 				status = (await statusResponse.json()).status;
 			} catch (error) {
-				throw new InferenceOutputError("Failed to parse status response from fal-ai API");
+				throw new InferenceClientProviderOutputError(
+					"Failed to parse status response from fal-ai API: received malformed response"
+				);
 			}
 		}
 
@@ -190,7 +206,9 @@ export class FalAITextToVideoTask extends FalAITask implements TextToVideoTaskHe
 		try {
 			result = await resultResponse.json();
 		} catch (error) {
-			throw new InferenceOutputError("Failed to parse result response from fal-ai API");
+			throw new InferenceClientProviderOutputError(
+				"Failed to parse result response from fal-ai API: received malformed response"
+			);
 		}
 		if (
 			typeof result === "object" &&
@@ -205,8 +223,10 @@ export class FalAITextToVideoTask extends FalAITask implements TextToVideoTaskHe
 			const urlResponse = await fetch(result.video.url);
 			return await urlResponse.blob();
 		} else {
-			throw new InferenceOutputError(
-				"Expected { video: { url: string } } result format, got instead: " + JSON.stringify(result)
+			throw new InferenceClientProviderOutputError(
+				`Received malformed response from Fal.ai text-to-video API: expected { video: { url: string } } result format, got instead: ${JSON.stringify(
+					result
+				)}`
 			);
 		}
 	}
@@ -221,8 +241,10 @@ export class FalAIAutomaticSpeechRecognitionTask extends FalAITask implements Au
 	override async getResponse(response: unknown): Promise<AutomaticSpeechRecognitionOutput> {
 		const res = response as FalAIAutomaticSpeechRecognitionOutput;
 		if (typeof res?.text !== "string") {
-			throw new InferenceOutputError(
-				`Expected { text: string } format from Fal.ai Automatic Speech Recognition, got: ${JSON.stringify(response)}`
+			throw new InferenceClientProviderOutputError(
+				`Received malformed response from Fal.ai Automatic Speech Recognition API: expected { text: string } format, got instead: ${JSON.stringify(
+					response
+				)}`
 			);
 		}
 		return { text: res.text };
@@ -232,12 +254,12 @@ export class FalAIAutomaticSpeechRecognitionTask extends FalAITask implements Au
 		const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : undefined;
 		const contentType = blob?.type;
 		if (!contentType) {
-			throw new Error(
+			throw new InferenceClientInputError(
 				`Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
 			);
 		}
 		if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
-			throw new Error(
+			throw new InferenceClientInputError(
 				`Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
 					", "
 				)}`
@@ -263,21 +285,35 @@ export class FalAITextToSpeechTask extends FalAITask {
 	override async getResponse(response: unknown): Promise<Blob> {
 		const res = response as FalAITextToSpeechOutput;
 		if (typeof res?.audio?.url !== "string") {
-			throw new InferenceOutputError(
-				`Expected { audio: { url: string } } format from Fal.ai Text-to-Speech, got: ${JSON.stringify(response)}`
+			throw new InferenceClientProviderOutputError(
+				`Received malformed response from Fal.ai Text-to-Speech API: expected { audio: { url: string } } format, got instead: ${JSON.stringify(
+					response
+				)}`
+			);
+		}
+		const urlResponse = await fetch(res.audio.url);
+		if (!urlResponse.ok) {
+			throw new InferenceClientProviderApiError(
+				`Failed to fetch audio from ${res.audio.url}: ${urlResponse.statusText}`,
+				{ url: res.audio.url, method: "GET", headers: { "Content-Type": "application/json" } },
+				{
+					requestId: urlResponse.headers.get("x-request-id") ?? "",
+					status: urlResponse.status,
+					body: await urlResponse.text(),
+				}
 			);
 		}
 		try {
-			const urlResponse = await fetch(res.audio.url);
-			if (!urlResponse.ok) {
-				throw new Error(`Failed to fetch audio from ${res.audio.url}: ${urlResponse.statusText}`);
-			}
 			return await urlResponse.blob();
 		} catch (error) {
-			throw new InferenceOutputError(
-				`Error fetching or processing audio from Fal.ai Text-to-Speech URL: ${res.audio.url}. ${
-					error instanceof Error ? error.message : String(error)
-				}`
+			throw new InferenceClientProviderApiError(
+				`Failed to fetch audio from ${res.audio.url}: ${error instanceof Error ? error.message : String(error)}`,
+				{ url: res.audio.url, method: "GET", headers: { "Content-Type": "application/json" } },
+				{
+					requestId: urlResponse.headers.get("x-request-id") ?? "",
+					status: urlResponse.status,
+					body: await urlResponse.text(),
+				}
 			);
 		}
 	}

package/src/providers/featherless-ai.ts

@@ -4,9 +4,10 @@ import type {
 	TextGenerationOutput,
 	TextGenerationOutputFinishReason,
 } from "@huggingface/tasks";
-import { InferenceOutputError } from "../lib/InferenceOutputError.js";
 import type { BodyParams } from "../types.js";
 import { BaseConversationalTask, BaseTextGenerationTask } from "./providerHelper.js";
+import { omit } from "../utils/omit.js";
+import { InferenceClientProviderOutputError } from "../errors.js";
 
 interface FeatherlessAITextCompletionOutput extends Omit<ChatCompletionOutput, "choices"> {
 	choices: Array<{
@@ -33,9 +34,14 @@ export class FeatherlessAITextGenerationTask extends BaseTextGenerationTask {
 
 	override preparePayload(params: BodyParams<TextGenerationInput>): Record<string, unknown> {
 		return {
-			...params.args,
-			...params.args.parameters,
 			model: params.model,
+			...omit(params.args, ["inputs", "parameters"]),
+			...(params.args.parameters
+				? {
+						max_tokens: params.args.parameters.max_new_tokens,
+						...omit(params.args.parameters, "max_new_tokens"),
+				  }
+				: undefined),
 			prompt: params.args.inputs,
 		};
 	}
@@ -52,6 +58,6 @@ export class FeatherlessAITextGenerationTask extends BaseTextGenerationTask {
 				generated_text: completion.text,
 			};
 		}
-		throw new InferenceOutputError("Expected Featherless AI text generation response format");
+		throw new InferenceClientProviderOutputError("Received malformed response from Featherless AI text generation API");
 	}
 }

package/src/providers/hf-inference.ts

@@ -34,7 +34,7 @@ import type {
 	ZeroShotImageClassificationOutput,
 } from "@huggingface/tasks";
 import { HF_ROUTER_URL } from "../config.js";
-import { InferenceOutputError } from "../lib/InferenceOutputError.js";
+import { 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";
@@ -127,7 +127,9 @@ export class HFInferenceTextToImageTask extends HFInferenceTask implements TextT
 		outputType?: "url" | "blob"
 	): Promise<string | Blob> {
 		if (!response) {
-			throw new InferenceOutputError("response is undefined");
+			throw new InferenceClientProviderOutputError(
+				"Received malformed response from HF-Inference text-to-image API: response is undefined"
+			);
 		}
 		if (typeof response == "object") {
 			if ("data" in response && Array.isArray(response.data) && response.data[0].b64_json) {
@@ -154,7 +156,9 @@ export class HFInferenceTextToImageTask extends HFInferenceTask implements TextT
 			}
 			return response;
 		}
-		throw new InferenceOutputError("Expected a Blob ");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference text-to-image API: expected a Blob"
+		);
 	}
 }
 
@@ -195,13 +199,14 @@ export class HFInferenceTextGenerationTask extends HFInferenceTask implements Te
 		if (Array.isArray(res) && res.every((x) => "generated_text" in x && typeof x?.generated_text === "string")) {
 			return (res as TextGenerationOutput[])?.[0];
 		}
-		throw new InferenceOutputError("Expected Array<{generated_text: string}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference text generation API: expected Array<{generated_text: string}>"
+		);
 	}
 }
 
 export class HFInferenceAudioClassificationTask extends HFInferenceTask implements AudioClassificationTaskHelper {
 	override async getResponse(response: unknown): Promise<AudioClassificationOutput> {
-		// Add type checking/validation for the 'unknown' input
 		if (
 			Array.isArray(response) &&
 			response.every(
@@ -209,11 +214,11 @@ export class HFInferenceAudioClassificationTask extends HFInferenceTask implemen
 					typeof x === "object" && x !== null && typeof x.label === "string" && typeof x.score === "number"
 			)
 		) {
-			// If validation passes, it's safe to return as AudioClassificationOutput
 			return response;
 		}
-		// If validation fails, throw an error
-		throw new InferenceOutputError("Expected Array<{label: string, score: number}> but received different format");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference audio-classification API: expected Array<{label: string, score: number}> but received different format"
+		);
 	}
 }
 
@@ -238,7 +243,9 @@ export class HFInferenceAutomaticSpeechRecognitionTask
 export class HFInferenceAudioToAudioTask extends HFInferenceTask implements AudioToAudioTaskHelper {
 	override async getResponse(response: AudioToAudioOutput[]): Promise<AudioToAudioOutput[]> {
 		if (!Array.isArray(response)) {
-			throw new InferenceOutputError("Expected Array");
+			throw new InferenceClientProviderOutputError(
+				"Received malformed response from HF-Inference audio-to-audio API: expected Array"
+			);
 		}
 		if (
 			!response.every((elem): elem is AudioToAudioOutput => {
@@ -254,7 +261,9 @@ export class HFInferenceAudioToAudioTask extends HFInferenceTask implements Audi
 				);
 			})
 		) {
-			throw new InferenceOutputError("Expected Array<{label: string, audio: Blob}>");
+			throw new InferenceClientProviderOutputError(
+				"Received malformed response from HF-Inference audio-to-audio API: expected Array<{label: string, audio: Blob}>"
+			);
 		}
 		return response;
 	}
@@ -281,7 +290,9 @@ export class HFInferenceDocumentQuestionAnsweringTask
 		) {
 			return response[0];
 		}
-		throw new InferenceOutputError("Expected Array<{answer: string, end: number, score: number, start: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference document-question-answering API: expected Array<{answer: string, end: number, score: number, start: number}>"
+		);
 	}
 }
 
@@ -298,7 +309,9 @@ export class HFInferenceFeatureExtractionTask extends HFInferenceTask implements
 		if (Array.isArray(response) && isNumArrayRec(response, 3, 0)) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<number[][][] | number[][] | number[] | number>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference feature-extraction API: expected Array<number[][][] | number[][] | number[] | number>"
+		);
 	}
 }
 
@@ -307,7 +320,9 @@ export class HFInferenceImageClassificationTask extends HFInferenceTask implemen
 		if (Array.isArray(response) && response.every((x) => typeof x.label === "string" && typeof x.score === "number")) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<{label: string, score: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference image-classification API: expected Array<{label: string, score: number}>"
+		);
 	}
 }
 
@@ -324,14 +339,18 @@ export class HFInferenceImageSegmentationTask extends HFInferenceTask implements
 		) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<{label: string, mask: string, score: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference image-segmentation API: expected Array<{label: string, mask: string, score: number}>"
+		);
 	}
 }
 
 export class HFInferenceImageToTextTask extends HFInferenceTask implements ImageToTextTaskHelper {
 	override async getResponse(response: ImageToTextOutput): Promise<ImageToTextOutput> {
 		if (typeof response?.generated_text !== "string") {
-			throw new InferenceOutputError("Expected {generated_text: string}");
+			throw new InferenceClientProviderOutputError(
+				"Received malformed response from HF-Inference image-to-text API: expected {generated_text: string}"
+			);
 		}
 		return response;
 	}
@@ -359,7 +378,9 @@ export class HFInferenceImageToImageTask extends HFInferenceTask implements Imag
 		if (response instanceof Blob) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Blob");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference image-to-image API: expected Blob"
+		);
 	}
 }
 
@@ -379,8 +400,8 @@ export class HFInferenceObjectDetectionTask extends HFInferenceTask implements O
 		) {
 			return response;
 		}
-		throw new InferenceOutputError(
-			"Expected Array<{label: string, score: number, box: {xmin: number, ymin: number, xmax: number, ymax: number}}>"
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference object-detection API: expected Array<{label: string, score: number, box: {xmin: number, ymin: number, xmax: number, ymax: number}}>"
 		);
 	}
 }
@@ -393,7 +414,9 @@ export class HFInferenceZeroShotImageClassificationTask
 		if (Array.isArray(response) && response.every((x) => typeof x.label === "string" && typeof x.score === "number")) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<{label: string, score: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference zero-shot-image-classification API: expected Array<{label: string, score: number}>"
+		);
 	}
 }
 
@@ -403,7 +426,9 @@ export class HFInferenceTextClassificationTask extends HFInferenceTask implement
 		if (Array.isArray(output) && output.every((x) => typeof x?.label === "string" && typeof x.score === "number")) {
 			return output;
 		}
-		throw new InferenceOutputError("Expected Array<{label: string, score: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference text-classification API: expected Array<{label: string, score: number}>"
+		);
 	}
 }
 
@@ -431,7 +456,9 @@ export class HFInferenceQuestionAnsweringTask extends HFInferenceTask implements
 		) {
 			return Array.isArray(response) ? response[0] : response;
 		}
-		throw new InferenceOutputError("Expected Array<{answer: string, end: number, score: number, start: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference question-answering API: expected Array<{answer: string, end: number, score: number, start: number}>"
+		);
 	}
 }
 
@@ -449,8 +476,8 @@ export class HFInferenceFillMaskTask extends HFInferenceTask implements FillMask
 		) {
 			return response;
 		}
-		throw new InferenceOutputError(
-			"Expected Array<{score: number, sequence: string, token: number, token_str: string}>"
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference fill-mask API: expected Array<{score: number, sequence: string, token: number, token_str: string}>"
 		);
 	}
 }
@@ -470,7 +497,9 @@ export class HFInferenceZeroShotClassificationTask extends HFInferenceTask imple
 		) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<{labels: string[], scores: number[], sequence: string}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference zero-shot-classification API: expected Array<{labels: string[], scores: number[], sequence: string}>"
+		);
 	}
 }
 
@@ -479,7 +508,9 @@ export class HFInferenceSentenceSimilarityTask extends HFInferenceTask implement
 		if (Array.isArray(response) && response.every((x) => typeof x === "number")) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<number>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference sentence-similarity API: expected Array<number>"
+		);
 	}
 }
 
@@ -510,8 +541,8 @@ export class HFInferenceTableQuestionAnsweringTask extends HFInferenceTask imple
 		) {
 			return Array.isArray(response) ? response[0] : response;
 		}
-		throw new InferenceOutputError(
-			"Expected {aggregator: string, answer: string, cells: string[], coordinates: number[][]}"
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference table-question-answering API: expected {aggregator: string, answer: string, cells: string[], coordinates: number[][]}"
 		);
 	}
 }
@@ -531,8 +562,8 @@ export class HFInferenceTokenClassificationTask extends HFInferenceTask implemen
 		) {
 			return response;
 		}
-		throw new InferenceOutputError(
-			"Expected Array<{end: number, entity_group: string, score: number, start: number, word: string}>"
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference token-classification API: expected Array<{end: number, entity_group: string, score: number, start: number, word: string}>"
 		);
 	}
 }
@@ -542,7 +573,9 @@ export class HFInferenceTranslationTask extends HFInferenceTask implements Trans
 		if (Array.isArray(response) && response.every((x) => typeof x?.translation_text === "string")) {
 			return response?.length === 1 ? response?.[0] : response;
 		}
-		throw new InferenceOutputError("Expected Array<{translation_text: string}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference translation API: expected Array<{translation_text: string}>"
+		);
 	}
 }
 
@@ -551,7 +584,9 @@ export class HFInferenceSummarizationTask extends HFInferenceTask implements Sum
 		if (Array.isArray(response) && response.every((x) => typeof x?.summary_text === "string")) {
 			return response?.[0];
 		}
-		throw new InferenceOutputError("Expected Array<{summary_text: string}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference summarization API: expected Array<{summary_text: string}>"
+		);
 	}
 }
 
@@ -566,7 +601,9 @@ export class HFInferenceTabularClassificationTask extends HFInferenceTask implem
 		if (Array.isArray(response) && response.every((x) => typeof x === "number")) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<number>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference tabular-classification API: expected Array<number>"
+		);
 	}
 }
 
@@ -584,7 +621,9 @@ export class HFInferenceVisualQuestionAnsweringTask
 		) {
 			return response[0];
 		}
-		throw new InferenceOutputError("Expected Array<{answer: string, score: number}>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference visual-question-answering API: expected Array<{answer: string, score: number}>"
+		);
 	}
 }
 
@@ -593,7 +632,9 @@ export class HFInferenceTabularRegressionTask extends HFInferenceTask implements
 		if (Array.isArray(response) && response.every((x) => typeof x === "number")) {
 			return response;
 		}
-		throw new InferenceOutputError("Expected Array<number>");
+		throw new InferenceClientProviderOutputError(
+			"Received malformed response from HF-Inference tabular-regression API: expected Array<number>"
+		);
 	}
 }
 

package/src/providers/hyperbolic.ts

@@ -15,7 +15,6 @@
  * Thanks!
  */
 import type { ChatCompletionOutput, TextGenerationOutput } from "@huggingface/tasks";
-import { InferenceOutputError } from "../lib/InferenceOutputError.js";
 import type { BodyParams, UrlParams } from "../types.js";
 import { omit } from "../utils/omit.js";
 import {
@@ -24,7 +23,7 @@ import {
 	TaskProviderHelper,
 	type TextToImageTaskHelper,
 } from "./providerHelper.js";
-
+import { InferenceClientProviderOutputError } from "../errors.js";
 const HYPERBOLIC_API_BASE_URL = "https://api.hyperbolic.xyz";
 
 export interface HyperbolicTextCompletionOutput extends Omit<ChatCompletionOutput, "choices"> {
@@ -79,7 +78,7 @@ export class HyperbolicTextGenerationTask extends BaseTextGenerationTask {
 			};
 		}
 
-		throw new InferenceOutputError("Expected Hyperbolic text generation response format");
+		throw new InferenceClientProviderOutputError("Received malformed response from Hyperbolic text generation API");
 	}
 }
 
@@ -121,6 +120,6 @@ export class HyperbolicTextToImageTask extends TaskProviderHelper implements Tex
 			return fetch(`data:image/jpeg;base64,${response.images[0].image}`).then((res) => res.blob());
 		}
 
-		throw new InferenceOutputError("Expected Hyperbolic text-to-image response format");
+		throw new InferenceClientProviderOutputError("Received malformed response from Hyperbolic text-to-image API");
 	}
 }

package/src/providers/nebius.ts

@@ -15,7 +15,6 @@
  * Thanks!
  */
 import type { FeatureExtractionOutput } from "@huggingface/tasks";
-import { InferenceOutputError } from "../lib/InferenceOutputError.js";
 import type { BodyParams } from "../types.js";
 import { omit } from "../utils/omit.js";
 import {
@@ -25,6 +24,7 @@ import {
 	type FeatureExtractionTaskHelper,
 	type TextToImageTaskHelper,
 } from "./providerHelper.js";
+import { InferenceClientProviderOutputError } from "../errors.js";
 
 const NEBIUS_API_BASE_URL = "https://api.studio.nebius.ai";
 
@@ -92,7 +92,7 @@ export class NebiusTextToImageTask extends TaskProviderHelper implements TextToI
 			return fetch(`data:image/jpeg;base64,${base64Data}`).then((res) => res.blob());
 		}
 
-		throw new InferenceOutputError("Expected Nebius text-to-image response format");
+		throw new InferenceClientProviderOutputError("Received malformed response from Nebius text-to-image API");
 	}
 }