@huggingface/inference 2.0.0-rc1 → 2.0.0

package/dist/index.js

@@ -96,7 +96,7 @@ function makeRequestOptions(args, options) {
       headers["X-Load-Model"] = "0";
     }
   }
-  const url = /^http(s?):/.test(model) ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
+  const url = /^http(s?):/.test(model) || model.startsWith("/") ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
   const info = {
     headers,
     method: "POST",

package/dist/index.mjs

@@ -53,7 +53,7 @@ function makeRequestOptions(args, options) {
       headers["X-Load-Model"] = "0";
     }
   }
-  const url = /^http(s?):/.test(model) ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
+  const url = /^http(s?):/.test(model) || model.startsWith("/") ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
   const info = {
     headers,
     method: "POST",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@huggingface/inference",
-  "version": "2.0.0rc1",
+  "version": "2.0.0",
   "license": "MIT",
   "author": "Tim Mikeladze <tim.mikeladze@gmail.com>",
   "description": "Typescript wrapper for the Hugging Face Inference API",

package/src/HfInference.ts

@@ -1,18 +1,19 @@
 import * as tasks from "./tasks";
 import type { Options, RequestArgs } from "./types";
+import type { DistributiveOmit } from "./utils/distributive-omit";
 
 type Task = typeof tasks;
 
 type TaskWithNoAccessToken = {
 	[key in keyof Task]: (
-		args: Omit<Parameters<Task[key]>[0], "accessToken">,
+		args: DistributiveOmit<Parameters<Task[key]>[0], "accessToken">,
 		options?: Parameters<Task[key]>[1]
 	) => ReturnType<Task[key]>;
 };
 
 type TaskWithNoAccessTokenNoModel = {
 	[key in keyof Task]: (
-		args: Omit<Parameters<Task[key]>[0], "accessToken" | "model">,
+		args: DistributiveOmit<Parameters<Task[key]>[0], "accessToken" | "model">,
 		options?: Parameters<Task[key]>[1]
 	) => ReturnType<Task[key]>;
 };
@@ -51,7 +52,7 @@ export class HfInferenceEndpoint {
 		for (const [name, fn] of Object.entries(tasks)) {
 			Object.defineProperty(this, name, {
 				enumerable: false,
-				value: (params: Omit<RequestArgs, "model">, options: Options) =>
+				value: (params: RequestArgs, options: Options) =>
 					// eslint-disable-next-line @typescript-eslint/no-explicit-any
 					fn({ ...params, accessToken, model: endpointUrl } as any, { ...defaultOptions, ...options }),
 			});

package/src/lib/makeRequestOptions.ts

@@ -38,7 +38,7 @@ export function makeRequestOptions(
 		}
 	}
 
-	const url = /^http(s?):/.test(model) ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
+	const url = /^http(s?):/.test(model) || model.startsWith("/") ? model : `${HF_INFERENCE_API_BASE_URL}${model}`;
 	const info: RequestInit = {
 		headers,
 		method: "POST",

package/src/types.ts

@@ -36,7 +36,7 @@ export interface BaseArgs {
 }
 
 export type RequestArgs = BaseArgs &
-	({ data?: Blob | ArrayBuffer } | { inputs: unknown }) & {
+	({ data: Blob | ArrayBuffer } | { inputs: unknown }) & {
 		parameters?: Record<string, unknown>;
 		accessToken?: string;
 	};

package/src/utils/distributive-omit.d.ts

@@ -0,0 +1,15 @@
+// https://dev.to/safareli/pick-omit-and-union-types-in-typescript-4nd9
+// https://github.com/microsoft/TypeScript/issues/28339#issuecomment-467393437
+/**
+ * This allows omitting keys from objects inside unions, without merging the individual components of the union.
+ */
+
+type Keys<T> = keyof T;
+type DistributiveKeys<T> = T extends unknown ? Keys<T> : never;
+type Omit_<T, K> = Omit<T, Extract<keyof T, K>>;
+
+export type DistributiveOmit<T, K> = T extends unknown
+	? keyof Omit_<T, K> extends never
+		? never
+		: { [P in keyof Omit_<T, K>]: Omit_<T, K>[P] }
+	: never;