lint & format

Author: SBrandeis

packages/inference/src/lib/makeRequestOptions.ts

@@ -68,21 +68,21 @@ export async function makeRequestOptions(
 			? "hf-token"
 			: "provider-key"
 		: includeCredentials === "include"
-			? "credentials-include"
-			: "none";
+		  ? "credentials-include"
+		  : "none";
 
 	const url = endpointUrl
 		? chatCompletion
 			? endpointUrl + `/v1/chat/completions`
 			: endpointUrl
 		: makeUrl({
-			authMethod,
-			chatCompletion: chatCompletion ?? false,
-			forceTask,
-			model,
-			provider: provider ?? "hf-inference",
-			taskHint,
-		});
+				authMethod,
+				chatCompletion: chatCompletion ?? false,
+				forceTask,
+				model,
+				provider: provider ?? "hf-inference",
+				taskHint,
+		  });
 
 	const headers: Record<string, string> = {};
 	if (accessToken) {
@@ -148,9 +148,9 @@ export async function makeRequestOptions(
 		body: binary
 			? args.data
 			: JSON.stringify({
-				...otherArgs,
-				...(chatCompletion || provider === "together" ? { model } : undefined),
-			}),
+					...otherArgs,
+					...(chatCompletion || provider === "together" ? { model } : undefined),
+			  }),
 		...(credentials ? { credentials } : undefined),
 		signal: options?.signal,
 	};
@@ -176,7 +176,7 @@ function makeUrl(params: {
 			const baseUrl = shouldProxy
 				? HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", params.provider)
 				: BLACKFORESTLABS_AI_API_BASE_URL;
-			return `${baseUrl}/${params.model}`
+			return `${baseUrl}/${params.model}`;
 		}
 		case "fal-ai": {
 			const baseUrl = shouldProxy

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

@@ -27,15 +27,20 @@ interface BlackForestLabsResponse {
  */
 export async function textToImage(args: TextToImageArgs, options?: Options): Promise<Blob> {
 	const payload =
-		args.provider === "together" || args.provider === "fal-ai" || args.provider === "replicate" || args.provider === "black-forest-labs"
+		args.provider === "together" ||
+		args.provider === "fal-ai" ||
+		args.provider === "replicate" ||
+		args.provider === "black-forest-labs"
 			? {
-				...omit(args, ["inputs", "parameters"]),
-				...args.parameters,
-				...(args.provider !== "replicate" ? { response_format: "base64" } : undefined),
-				prompt: args.inputs,
-			}
+					...omit(args, ["inputs", "parameters"]),
+					...args.parameters,
+					...(args.provider !== "replicate" ? { response_format: "base64" } : undefined),
+					prompt: args.inputs,
+			  }
 			: args;
-	const res = await request<TextToImageOutput | Base64ImageGeneration | OutputUrlImageGeneration | BlackForestLabsResponse>(payload, {
+	const res = await request<
+		TextToImageOutput | Base64ImageGeneration | OutputUrlImageGeneration | BlackForestLabsResponse
+	>(payload, {
 		...options,
 		taskHint: "text-to-image",
 	});
@@ -77,11 +82,21 @@ async function pollBflResponse(url: string): Promise<Blob> {
 			throw new InferenceOutputError("Failed to fetch result from black forest labs API");
 		}
 		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 (
+			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"
+		) {
 			const image = await fetch(payload.result.sample);
 			return await image.blob();
 		}
 	}
 	throw new InferenceOutputError("Failed to fetch result from black forest labs API");
 }
-

packages/inference/src/utils/delay.ts

@@ -1,5 +1,5 @@
 export function delay(ms: number): Promise<void> {
-	return new Promise(resolve => {
+	return new Promise((resolve) => {
 		setTimeout(() => resolve(), ms);
 	});
 }

packages/inference/test/HfInference.spec.ts

@@ -1129,7 +1129,6 @@ describe.concurrent("HfInference", () => {
 	describe.concurrent(
 		"Black Forest Labs",
 		() => {
-
 			HARDCODED_MODEL_ID_MAPPING["black-forest-labs"] = {
 				"black-forest-labs/FLUX.1-dev": "flux-dev",
 				// "black-forest-labs/FLUX.1-schnell": "flux-pro",
@@ -1146,11 +1145,10 @@ describe.concurrent("HfInference", () => {
 						width: 256,
 						num_inference_steps: 4,
 						seed: 8817,
-					}
+					},
 				});
 				expect(res).toBeInstanceOf(Blob);
 			});
-
 		},
 		TIMEOUT
 	);