Merge branch 'main' of github.com:huggingface/huggingface.js into update-request-func

Author: hanouticelina

packages/inference/src/config.ts

@@ -1,2 +1,3 @@
 export const HF_HUB_URL = "https://huggingface.co";
 export const HF_ROUTER_URL = "https://router.huggingface.co";
+export const HF_HEADER_X_BILL_TO = "X-HF-Bill-To";

packages/inference/src/lib/makeRequestOptions.ts

@@ -1,4 +1,4 @@
-import { HF_HUB_URL, HF_ROUTER_URL } from "../config";
+import { HF_HUB_URL, HF_ROUTER_URL, HF_HEADER_X_BILL_TO } from "../config";
 import { BLACK_FOREST_LABS_CONFIG } from "../providers/black-forest-labs";
 import { CEREBRAS_CONFIG } from "../providers/cerebras";
 import { COHERE_CONFIG } from "../providers/cohere";
@@ -117,7 +117,7 @@ export function makeRequestOptionsFromResolvedModel(
 	const provider = maybeProvider ?? "hf-inference";
 	const providerConfig = providerConfigs[provider];
 
-	const { includeCredentials, task, chatCompletion, signal } = options ?? {};
+	const { includeCredentials, task, chatCompletion, signal, billTo } = options ?? {};
 
 	const authMethod = (() => {
 		if (providerConfig.clientSideRoutingOnly) {
@@ -159,6 +159,9 @@ export function makeRequestOptionsFromResolvedModel(
 		accessToken,
 		authMethod,
 	});
+	if (billTo) {
+		headers[HF_HEADER_X_BILL_TO] = billTo;
+	}
 
 	// Add content-type to headers
 	if (!binary) {

packages/inference/src/snippets/templates/js/fetch/textToImage.jinja

@@ -14,6 +14,7 @@ async function query(data) {
 	return result;
 }
 
-query({ inputs: {{ providerInputs.asObj.inputs }} }).then((response) => {
+
+query({ {{ providerInputs.asTsString }} }).then((response) => {
     // Use image
 });

packages/inference/src/tasks/nlp/sentenceSimilarity.ts

@@ -1,7 +1,6 @@
 import type { SentenceSimilarityInput, SentenceSimilarityOutput } from "@huggingface/tasks";
 import { InferenceOutputError } from "../../lib/InferenceOutputError";
 import type { BaseArgs, Options } from "../../types";
-import { omit } from "../../utils/omit";
 import { innerRequest } from "../../utils/request";
 
 export type SentenceSimilarityArgs = BaseArgs & SentenceSimilarityInput;
@@ -14,7 +13,7 @@ export async function sentenceSimilarity(
 	options?: Options
 ): Promise<SentenceSimilarityOutput> {
 	const res = (
-		await innerRequest<SentenceSimilarityOutput>(prepareInput(args), {
+		await innerRequest<SentenceSimilarityOutput>(args, {
 			...options,
 			task: "sentence-similarity",
 		})
@@ -26,11 +25,3 @@ export async function sentenceSimilarity(
 	}
 	return res;
 }
-
-function prepareInput(args: SentenceSimilarityArgs) {
-	return {
-		...omit(args, ["inputs", "parameters"]),
-		inputs: { ...omit(args.inputs, "sourceSentence") },
-		parameters: { source_sentence: args.inputs.sourceSentence, ...args.parameters },
-	};
-}

packages/inference/src/types.ts

@@ -24,6 +24,14 @@ export interface Options {
 	 * (Default: "same-origin"). String | Boolean. Credentials to use for the request. If this is a string, it will be passed straight on. If it's a boolean, true will be "include" and false will not send credentials at all.
 	 */
 	includeCredentials?: string | boolean;
+
+	/**
+	 * The billing account to use for the requests.
+	 *
+	 * By default the requests are billed on the user's account.
+	 * Requests can only be billed to an organization the user is a member of, and which has subscribed to Enterprise Hub.
+	 */
+	billTo?: string;
 }
 
 export type InferenceTask = Exclude<PipelineType, "other">;