fix texttoImage

Author: connorch

packages/inference/src/lib/makeRequestOptions.ts

@@ -240,6 +240,13 @@ function makeUrl(params: {
 			const baseUrl = shouldProxy
 				? HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", params.provider)
 				: HYPERBOLIC_API_BASE_URL;
+
+			if (params.taskHint === "text-generation") {
+				return `${baseUrl}/v1/chat/completions`;
+			}
+			if (params.taskHint === "text-to-image") {
+				return `${baseUrl}/v1/images/generations`;
+			}
 			return `${baseUrl}/v1/chat/completions`;
 		}
 		default: {

packages/inference/src/tasks/cv/textToImage.ts

@@ -24,7 +24,8 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
 		args.provider === "together" ||
 		args.provider === "fal-ai" ||
 		args.provider === "replicate" ||
-		args.provider === "nebius"
+		args.provider === "nebius" ||
+		args.provider === "hyperbolic"
 			? {
 					...omit(args, ["inputs", "parameters"]),
 					...args.parameters,
@@ -42,6 +43,11 @@ export async function textToImage(args: TextToImageArgs, options?: Options): Pro
 			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].image) {
+			const base64Response = await fetch(`data:image/jpeg;base64,${res.images[0].image}`);
+			const blob = await base64Response.blob();
+			return blob;
+		}
 		if ("data" in res && Array.isArray(res.data) && res.data[0].b64_json) {
 			const base64Data = res.data[0].b64_json;
 			const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);

packages/inference/test/HfInference.spec.ts

@@ -2,6 +2,7 @@ import { assert, describe, expect, it } from "vitest";
 
 import type { ChatCompletionStreamOutput } from "@huggingface/tasks";
 
+import type { TextToImageArgs } from "../src";
 import { chatCompletion, HfInference } from "../src";
 import { textToVideo } from "../src/tasks/cv/textToVideo";
 import { readTestFile } from "./test-files";
@@ -1227,8 +1228,13 @@ describe.concurrent("HfInference", () => {
 				const res = await client.textToImage({
 					model: "stabilityai/stable-diffusion-2",
 					provider: "hyperbolic",
-					messages: [{ role: "user", content: "award winning high resolution photo of a giant tortoise" }],
-				});
+					inputs: "award winning high resolution photo of a giant tortoise",
+					parameters: {
+						model_name: "SD2",
+						height: 1024,
+						width: 1024,
+					},
+				} satisfies TextToImageArgs);
 				expect(res).toBeInstanceOf(Blob);
 			});