@huggingface/inference 3.13.2 → 3.15.0

package/dist/commonjs/tasks/custom/request.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.request = request;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Primitive to make custom calls to the inference provider
+ * @deprecated Use specific task functions instead. This function will be removed in a future version.
+ */
+async function request(args, options) {
+    console.warn("The request method is deprecated and will be removed in a future version of huggingface.js. Use specific task functions instead.");
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, options?.task);
+    const result = await (0, request_js_1.innerRequest)(args, providerHelper, options);
+    return result.data;
+}

package/dist/{src → commonjs}/tasks/custom/streamingRequest.d.ts

@@ -1,4 +1,4 @@
-import type { InferenceTask, Options, RequestArgs } from "../../types";
+import type { InferenceTask, Options, RequestArgs } from "../../types.js";
 /**
  * Primitive to make custom inference calls that expect server-sent events, and returns the response through a generator
  * @deprecated Use specific task functions instead. This function will be removed in a future version.

package/dist/commonjs/tasks/custom/streamingRequest.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamingRequest.d.ts","sourceRoot":"","sources":["../../../../src/tasks/custom/streamingRequest.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG1E;;;GAGG;AACH,wBAAuB,gBAAgB,CAAC,CAAC,EACxC,IAAI,EAAE,WAAW,EACjB,OAAO,CAAC,EAAE,OAAO,GAAG;IACnB,oEAAoE;IACpE,IAAI,CAAC,EAAE,aAAa,CAAC;CACrB,GACC,cAAc,CAAC,CAAC,CAAC,CAOnB"}

package/dist/commonjs/tasks/custom/streamingRequest.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.streamingRequest = streamingRequest;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Primitive to make custom inference calls that expect server-sent events, and returns the response through a generator
+ * @deprecated Use specific task functions instead. This function will be removed in a future version.
+ */
+async function* streamingRequest(args, options) {
+    console.warn("The streamingRequest method is deprecated and will be removed in a future version of huggingface.js. Use specific task functions instead.");
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, options?.task);
+    yield* (0, request_js_1.innerStreamingRequest)(args, providerHelper, options);
+}

package/dist/{src → commonjs}/tasks/cv/imageClassification.d.ts

@@ -1,6 +1,6 @@
 import type { ImageClassificationInput, ImageClassificationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
-import { type LegacyImageInput } from "./utils";
+import type { BaseArgs, Options } from "../../types.js";
+import { type LegacyImageInput } from "./utils.js";
 export type ImageClassificationArgs = BaseArgs & (ImageClassificationInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes.

package/dist/commonjs/tasks/cv/imageClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAG9F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAExD,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnE,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG,CAAC,wBAAwB,GAAG,gBAAgB,CAAC,CAAC;AAE/F;;;GAGG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CASpC"}

package/dist/commonjs/tasks/cv/imageClassification.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.imageClassification = imageClassification;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+const utils_js_1 = require("./utils.js");
+/**
+ * This task reads some image input and outputs the likelihood of classes.
+ * Recommended model: google/vit-base-patch16-224
+ */
+async function imageClassification(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "image-classification");
+    const payload = (0, utils_js_1.preparePayload)(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "image-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/cv/imageSegmentation.d.ts

@@ -1,6 +1,6 @@
 import type { ImageSegmentationInput, ImageSegmentationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
-import { type LegacyImageInput } from "./utils";
+import type { BaseArgs, Options } from "../../types.js";
+import { type LegacyImageInput } from "./utils.js";
 export type ImageSegmentationArgs = BaseArgs & (ImageSegmentationInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.

package/dist/commonjs/tasks/cv/imageSegmentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageSegmentation.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageSegmentation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAG1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAExD,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnE,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,CAAC,sBAAsB,GAAG,gBAAgB,CAAC,CAAC;AAE3F;;;GAGG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CASlC"}

package/dist/commonjs/tasks/cv/imageSegmentation.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.imageSegmentation = imageSegmentation;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+const utils_js_1 = require("./utils.js");
+/**
+ * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.
+ * Recommended model: facebook/detr-resnet-50-panoptic
+ */
+async function imageSegmentation(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "image-segmentation");
+    const payload = (0, utils_js_1.preparePayload)(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "image-segmentation",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/cv/imageToImage.d.ts

@@ -1,5 +1,5 @@
 import type { ImageToImageInput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type ImageToImageArgs = BaseArgs & ImageToImageInput;
 /**
  * This task reads some text input and outputs an image.

package/dist/commonjs/tasks/cv/imageToImage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAG5D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,iBAAiB,CAAC;AAE5D;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAS3F"}

package/dist/commonjs/tasks/cv/imageToImage.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.imageToImage = imageToImage;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * This task reads some text input and outputs an image.
+ * Recommended model: lllyasviel/sd-controlnet-depth
+ */
+async function imageToImage(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "image-to-image");
+    const payload = await providerHelper.preparePayloadAsync(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "image-to-image",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/cv/imageToText.d.ts

@@ -1,6 +1,6 @@
 import type { ImageToTextInput, ImageToTextOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
-import type { LegacyImageInput } from "./utils";
+import type { BaseArgs, Options } from "../../types.js";
+import type { LegacyImageInput } from "./utils.js";
 export type ImageToTextArgs = BaseArgs & (ImageToTextInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the text caption.

package/dist/commonjs/tasks/cv/imageToText.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageToText.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageToText.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAG9E,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAExD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAGnD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,CAAC,gBAAgB,GAAG,gBAAgB,CAAC,CAAC;AAC/E;;GAEG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAUtG"}

package/dist/commonjs/tasks/cv/imageToText.js

@@ -0,0 +1,20 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.imageToText = imageToText;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+const utils_js_1 = require("./utils.js");
+/**
+ * This task reads some image input and outputs the text caption.
+ */
+async function imageToText(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "image-to-text");
+    const payload = (0, utils_js_1.preparePayload)(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "image-to-text",
+    });
+    return providerHelper.getResponse(res[0]);
+}

package/dist/{src → commonjs}/tasks/cv/objectDetection.d.ts

@@ -1,6 +1,6 @@
 import type { ObjectDetectionInput, ObjectDetectionOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
-import { type LegacyImageInput } from "./utils";
+import type { BaseArgs, Options } from "../../types.js";
+import { type LegacyImageInput } from "./utils.js";
 export type ObjectDetectionArgs = BaseArgs & (ObjectDetectionInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.

package/dist/commonjs/tasks/cv/objectDetection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"objectDetection.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/objectDetection.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAGtF,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAExD,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAEnE,MAAM,MAAM,mBAAmB,GAAG,QAAQ,GAAG,CAAC,oBAAoB,GAAG,gBAAgB,CAAC,CAAC;AAEvF;;;GAGG;AACH,wBAAsB,eAAe,CAAC,IAAI,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,qBAAqB,CAAC,CASlH"}

package/dist/commonjs/tasks/cv/objectDetection.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.objectDetection = objectDetection;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+const utils_js_1 = require("./utils.js");
+/**
+ * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.
+ * Recommended model: facebook/detr-resnet-50
+ */
+async function objectDetection(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "object-detection");
+    const payload = (0, utils_js_1.preparePayload)(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "object-detection",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/cv/textToImage.d.ts

@@ -1,5 +1,5 @@
 import type { TextToImageInput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type TextToImageArgs = BaseArgs & TextToImageInput;
 interface TextToImageOptions extends Options {
     outputType?: "url" | "blob";

package/dist/commonjs/tasks/cv/textToImage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"textToImage.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToImage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAI3D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAE1D,UAAU,kBAAmB,SAAQ,OAAO;IAC3C,UAAU,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC;CAC5B;AAED;;;GAGG;AACH,wBAAsB,WAAW,CAChC,IAAI,EAAE,eAAe,EACrB,OAAO,CAAC,EAAE,kBAAkB,GAAG;IAAE,UAAU,EAAE,KAAK,CAAA;CAAE,GAClD,OAAO,CAAC,MAAM,CAAC,CAAC;AACnB,wBAAsB,WAAW,CAChC,IAAI,EAAE,eAAe,EACrB,OAAO,CAAC,EAAE,kBAAkB,GAAG;IAAE,UAAU,CAAC,EAAE,SAAS,GAAG,MAAM,CAAA;CAAE,GAChE,OAAO,CAAC,IAAI,CAAC,CAAC"}

package/dist/commonjs/tasks/cv/textToImage.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textToImage = textToImage;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const makeRequestOptions_js_1 = require("../../lib/makeRequestOptions.js");
+const request_js_1 = require("../../utils/request.js");
+async function textToImage(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "text-to-image");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "text-to-image",
+    });
+    const { url, info } = await (0, makeRequestOptions_js_1.makeRequestOptions)(args, providerHelper, { ...options, task: "text-to-image" });
+    return providerHelper.getResponse(res, url, info.headers, options?.outputType);
+}

package/dist/{src → commonjs}/tasks/cv/textToVideo.d.ts

@@ -1,5 +1,5 @@
 import type { TextToVideoInput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type TextToVideoArgs = BaseArgs & TextToVideoInput;
 export type TextToVideoOutput = Blob;
 export declare function textToVideo(args: TextToVideoArgs, options?: Options): Promise<TextToVideoOutput>;

package/dist/commonjs/tasks/cv/textToVideo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"textToVideo.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/textToVideo.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAO3D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAE1D,MAAM,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAErC,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAatG"}

package/dist/commonjs/tasks/cv/textToVideo.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textToVideo = textToVideo;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const makeRequestOptions_js_1 = require("../../lib/makeRequestOptions.js");
+const request_js_1 = require("../../utils/request.js");
+async function textToVideo(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "text-to-video");
+    const { data: response } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "text-to-video",
+    });
+    const { url, info } = await (0, makeRequestOptions_js_1.makeRequestOptions)(args, providerHelper, { ...options, task: "text-to-video" });
+    return providerHelper.getResponse(response, url, info.headers);
+}

package/dist/{src → commonjs}/tasks/cv/utils.d.ts

@@ -1,4 +1,4 @@
-import type { BaseArgs, RequestArgs } from "../../types";
+import type { BaseArgs, RequestArgs } from "../../types.js";
 /**
  * @deprecated
  */

package/dist/commonjs/tasks/cv/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG5D;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,QAAQ,GAAG,CAAC;IAAE,MAAM,EAAE,IAAI,CAAA;CAAE,GAAG,gBAAgB,CAAC,GAAG,WAAW,CAElG"}

package/dist/commonjs/tasks/cv/utils.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.preparePayload = preparePayload;
+const omit_js_1 = require("../../utils/omit.js");
+function preparePayload(args) {
+    return "data" in args ? args : { ...(0, omit_js_1.omit)(args, "inputs"), data: args.inputs };
+}

package/dist/{src → commonjs}/tasks/cv/zeroShotImageClassification.d.ts

@@ -1,5 +1,5 @@
 import type { ZeroShotImageClassificationInput, ZeroShotImageClassificationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 /**
  * @deprecated
  */

package/dist/commonjs/tasks/cv/zeroShotImageClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"zeroShotImageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/zeroShotImageClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gCAAgC,EAAE,iCAAiC,EAAE,MAAM,oBAAoB,CAAC;AAG9G,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,gBAAgB,CAAC;AAIrE;;GAEG;AACH,UAAU,sCAAsC;IAC/C,MAAM,EAAE;QAAE,KAAK,EAAE,IAAI,GAAG,WAAW,CAAA;KAAE,CAAC;CACtC;AAED,MAAM,MAAM,+BAA+B,GAAG,QAAQ,GACrD,CAAC,gCAAgC,GAAG,sCAAsC,CAAC,CAAC;AAwB7E;;;GAGG;AACH,wBAAsB,2BAA2B,CAChD,IAAI,EAAE,+BAA+B,EACrC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,iCAAiC,CAAC,CAS5C"}

package/dist/commonjs/tasks/cv/zeroShotImageClassification.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.zeroShotImageClassification = zeroShotImageClassification;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const base64FromBytes_js_1 = require("../../utils/base64FromBytes.js");
+const request_js_1 = require("../../utils/request.js");
+async function preparePayload(args) {
+    if (args.inputs instanceof Blob) {
+        return {
+            ...args,
+            inputs: {
+                image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(await args.inputs.arrayBuffer())),
+            },
+        };
+    }
+    else {
+        return {
+            ...args,
+            inputs: {
+                image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(args.inputs.image instanceof ArrayBuffer ? args.inputs.image : await args.inputs.image.arrayBuffer())),
+            },
+        };
+    }
+}
+/**
+ * Classify an image to specified classes.
+ * Recommended model: openai/clip-vit-large-patch14-336
+ */
+async function zeroShotImageClassification(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "zero-shot-image-classification");
+    const payload = await preparePayload(args);
+    const { data: res } = await (0, request_js_1.innerRequest)(payload, providerHelper, {
+        ...options,
+        task: "zero-shot-image-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/commonjs/tasks/index.d.ts

@@ -0,0 +1,33 @@
+export * from "./custom/request.js";
+export * from "./custom/streamingRequest.js";
+export * from "./audio/audioClassification.js";
+export * from "./audio/audioToAudio.js";
+export * from "./audio/automaticSpeechRecognition.js";
+export * from "./audio/textToSpeech.js";
+export * from "./cv/imageClassification.js";
+export * from "./cv/imageSegmentation.js";
+export * from "./cv/imageToImage.js";
+export * from "./cv/imageToText.js";
+export * from "./cv/objectDetection.js";
+export * from "./cv/textToImage.js";
+export * from "./cv/textToVideo.js";
+export * from "./cv/zeroShotImageClassification.js";
+export * from "./nlp/chatCompletion.js";
+export * from "./nlp/chatCompletionStream.js";
+export * from "./nlp/featureExtraction.js";
+export * from "./nlp/fillMask.js";
+export * from "./nlp/questionAnswering.js";
+export * from "./nlp/sentenceSimilarity.js";
+export * from "./nlp/summarization.js";
+export * from "./nlp/tableQuestionAnswering.js";
+export * from "./nlp/textClassification.js";
+export * from "./nlp/textGeneration.js";
+export * from "./nlp/textGenerationStream.js";
+export * from "./nlp/tokenClassification.js";
+export * from "./nlp/translation.js";
+export * from "./nlp/zeroShotClassification.js";
+export * from "./multimodal/documentQuestionAnswering.js";
+export * from "./multimodal/visualQuestionAnswering.js";
+export * from "./tabular/tabularClassification.js";
+export * from "./tabular/tabularRegression.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/commonjs/tasks/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/tasks/index.ts"],"names":[],"mappings":"AACA,cAAc,qBAAqB,CAAC;AACpC,cAAc,8BAA8B,CAAC;AAG7C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,yBAAyB,CAAC;AACxC,cAAc,uCAAuC,CAAC;AACtD,cAAc,yBAAyB,CAAC;AAGxC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,yBAAyB,CAAC;AACxC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,qCAAqC,CAAC;AAGpD,cAAc,yBAAyB,CAAC;AACxC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,wBAAwB,CAAC;AACvC,cAAc,iCAAiC,CAAC;AAChD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,8BAA8B,CAAC;AAC7C,cAAc,sBAAsB,CAAC;AACrC,cAAc,iCAAiC,CAAC;AAGhD,cAAc,2CAA2C,CAAC;AAC1D,cAAc,yCAAyC,CAAC;AAGxD,cAAc,oCAAoC,CAAC;AACnD,cAAc,gCAAgC,CAAC"}

package/dist/commonjs/tasks/index.js

@@ -0,0 +1,54 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+// Custom tasks with arbitrary inputs and outputs
+__exportStar(require("./custom/request.js"), exports);
+__exportStar(require("./custom/streamingRequest.js"), exports);
+// Audio tasks
+__exportStar(require("./audio/audioClassification.js"), exports);
+__exportStar(require("./audio/audioToAudio.js"), exports);
+__exportStar(require("./audio/automaticSpeechRecognition.js"), exports);
+__exportStar(require("./audio/textToSpeech.js"), exports);
+// Computer Vision tasks
+__exportStar(require("./cv/imageClassification.js"), exports);
+__exportStar(require("./cv/imageSegmentation.js"), exports);
+__exportStar(require("./cv/imageToImage.js"), exports);
+__exportStar(require("./cv/imageToText.js"), exports);
+__exportStar(require("./cv/objectDetection.js"), exports);
+__exportStar(require("./cv/textToImage.js"), exports);
+__exportStar(require("./cv/textToVideo.js"), exports);
+__exportStar(require("./cv/zeroShotImageClassification.js"), exports);
+// Natural Language Processing tasks
+__exportStar(require("./nlp/chatCompletion.js"), exports);
+__exportStar(require("./nlp/chatCompletionStream.js"), exports);
+__exportStar(require("./nlp/featureExtraction.js"), exports);
+__exportStar(require("./nlp/fillMask.js"), exports);
+__exportStar(require("./nlp/questionAnswering.js"), exports);
+__exportStar(require("./nlp/sentenceSimilarity.js"), exports);
+__exportStar(require("./nlp/summarization.js"), exports);
+__exportStar(require("./nlp/tableQuestionAnswering.js"), exports);
+__exportStar(require("./nlp/textClassification.js"), exports);
+__exportStar(require("./nlp/textGeneration.js"), exports);
+__exportStar(require("./nlp/textGenerationStream.js"), exports);
+__exportStar(require("./nlp/tokenClassification.js"), exports);
+__exportStar(require("./nlp/translation.js"), exports);
+__exportStar(require("./nlp/zeroShotClassification.js"), exports);
+// Multimodal tasks
+__exportStar(require("./multimodal/documentQuestionAnswering.js"), exports);
+__exportStar(require("./multimodal/visualQuestionAnswering.js"), exports);
+// Tabular tasks
+__exportStar(require("./tabular/tabularClassification.js"), exports);
+__exportStar(require("./tabular/tabularRegression.js"), exports);

package/dist/{src → commonjs}/tasks/multimodal/documentQuestionAnswering.d.ts

@@ -1,5 +1,5 @@
 import type { DocumentQuestionAnsweringInput, DocumentQuestionAnsweringInputData, DocumentQuestionAnsweringOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type DocumentQuestionAnsweringArgs = BaseArgs & DocumentQuestionAnsweringInput & {
     inputs: DocumentQuestionAnsweringInputData & {
         image: Blob;

package/dist/commonjs/tasks/multimodal/documentQuestionAnswering.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"documentQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/documentQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,8BAA8B,EAC9B,kCAAkC,EAClC,+BAA+B,EAC/B,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,gBAAgB,CAAC;AAKrE,MAAM,MAAM,6BAA6B,GAAG,QAAQ,GACnD,8BAA8B,GAAG;IAAE,MAAM,EAAE,kCAAkC,GAAG;QAAE,KAAK,EAAE,IAAI,CAAA;KAAE,CAAA;CAAE,CAAC;AAEnG;;GAEG;AACH,wBAAsB,yBAAyB,CAC9C,IAAI,EAAE,6BAA6B,EACnC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,+BAA+B,CAAC,MAAM,CAAC,CAAC,CAoBlD"}

package/dist/commonjs/tasks/multimodal/documentQuestionAnswering.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.documentQuestionAnswering = documentQuestionAnswering;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const base64FromBytes_js_1 = require("../../utils/base64FromBytes.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Answers a question on a document image. Recommended model: impira/layoutlm-document-qa.
+ */
+async function documentQuestionAnswering(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "document-question-answering");
+    const reqArgs = {
+        ...args,
+        inputs: {
+            question: args.inputs.question,
+            // convert Blob or ArrayBuffer to base64
+            image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(await args.inputs.image.arrayBuffer())),
+        },
+    };
+    const { data: res } = await (0, request_js_1.innerRequest)(reqArgs, providerHelper, {
+        ...options,
+        task: "document-question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/multimodal/visualQuestionAnswering.d.ts

@@ -1,5 +1,5 @@
 import type { VisualQuestionAnsweringInput, VisualQuestionAnsweringInputData, VisualQuestionAnsweringOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type VisualQuestionAnsweringArgs = BaseArgs & VisualQuestionAnsweringInput & {
     inputs: VisualQuestionAnsweringInputData & {
         image: Blob;

package/dist/commonjs/tasks/multimodal/visualQuestionAnswering.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"visualQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/multimodal/visualQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,4BAA4B,EAC5B,gCAAgC,EAChC,6BAA6B,EAC7B,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,gBAAgB,CAAC;AAKrE,MAAM,MAAM,2BAA2B,GAAG,QAAQ,GACjD,4BAA4B,GAAG;IAAE,MAAM,EAAE,gCAAgC,GAAG;QAAE,KAAK,EAAE,IAAI,CAAA;KAAE,CAAA;CAAE,CAAC;AAE/F;;GAEG;AACH,wBAAsB,uBAAuB,CAC5C,IAAI,EAAE,2BAA2B,EACjC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,6BAA6B,CAAC,MAAM,CAAC,CAAC,CAiBhD"}

package/dist/commonjs/tasks/multimodal/visualQuestionAnswering.js

@@ -0,0 +1,27 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.visualQuestionAnswering = visualQuestionAnswering;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const base64FromBytes_js_1 = require("../../utils/base64FromBytes.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Answers a question on an image. Recommended model: dandelin/vilt-b32-finetuned-vqa.
+ */
+async function visualQuestionAnswering(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "visual-question-answering");
+    const reqArgs = {
+        ...args,
+        inputs: {
+            question: args.inputs.question,
+            // convert Blob or ArrayBuffer to base64
+            image: (0, base64FromBytes_js_1.base64FromBytes)(new Uint8Array(await args.inputs.image.arrayBuffer())),
+        },
+    };
+    const { data: res } = await (0, request_js_1.innerRequest)(reqArgs, providerHelper, {
+        ...options,
+        task: "visual-question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/{src → commonjs}/tasks/nlp/chatCompletion.d.ts

@@ -1,5 +1,5 @@
 import type { ChatCompletionInput, ChatCompletionOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 /**
  * Use the chat completion endpoint to generate a response to a prompt, using OpenAI message completion API no stream
  */

package/dist/{src → commonjs}/tasks/nlp/chatCompletion.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatCompletion.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/chatCompletion.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAGpF,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD;;GAEG;AACH,wBAAsB,cAAc,CACnC,IAAI,EAAE,QAAQ,GAAG,mBAAmB,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,oBAAoB,CAAC,CAQ/B"}
+{"version":3,"file":"chatCompletion.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/chatCompletion.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAGpF,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD;;GAEG;AACH,wBAAsB,cAAc,CACnC,IAAI,EAAE,QAAQ,GAAG,mBAAmB,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,oBAAoB,CAAC,CAQ/B"}

package/dist/commonjs/tasks/nlp/chatCompletion.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.chatCompletion = chatCompletion;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Use the chat completion endpoint to generate a response to a prompt, using OpenAI message completion API no stream
+ */
+async function chatCompletion(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "conversational");
+    const { data: response } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "conversational",
+    });
+    return providerHelper.getResponse(response);
+}

package/dist/{src → commonjs}/tasks/nlp/chatCompletionStream.d.ts

@@ -1,5 +1,5 @@
 import type { ChatCompletionInput, ChatCompletionStreamOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 /**
  * Use to continue text from a prompt. Same as `textGeneration` but returns generator that can be read one token at a time
  */

package/dist/{src → commonjs}/tasks/nlp/chatCompletionStream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatCompletionStream.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/chatCompletionStream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAG1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD;;GAEG;AACH,wBAAuB,oBAAoB,CAC1C,IAAI,EAAE,QAAQ,GAAG,mBAAmB,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,cAAc,CAAC,0BAA0B,CAAC,CAO5C"}
+{"version":3,"file":"chatCompletionStream.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/chatCompletionStream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAG1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD;;GAEG;AACH,wBAAuB,oBAAoB,CAC1C,IAAI,EAAE,QAAQ,GAAG,mBAAmB,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,cAAc,CAAC,0BAA0B,CAAC,CAO5C"}

package/dist/commonjs/tasks/nlp/chatCompletionStream.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.chatCompletionStream = chatCompletionStream;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Use to continue text from a prompt. Same as `textGeneration` but returns generator that can be read one token at a time
+ */
+async function* chatCompletionStream(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "conversational");
+    yield* (0, request_js_1.innerStreamingRequest)(args, providerHelper, {
+        ...options,
+        task: "conversational",
+    });
+}