@huggingface/inference 3.13.2 → 3.14.0

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",
+    });
+}

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

@@ -1,5 +1,5 @@
 import type { FeatureExtractionInput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 interface FeatureExtractionOAICompatInput {
     encoding_format?: "float" | "base64";
     dimensions?: number | null;

package/dist/commonjs/tasks/nlp/featureExtraction.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"featureExtraction.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/featureExtraction.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAGjE,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,UAAU,+BAA+B;IACxC,eAAe,CAAC,EAAE,OAAO,GAAG,QAAQ,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC3B;AAED,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,sBAAsB,GAAG,+BAA+B,CAAC;AAExG;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,CAAC,MAAM,GAAG,MAAM,EAAE,GAAG,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;AAEzE;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CAQlC"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.featureExtraction = featureExtraction;
+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 and outputs raw float values, that are usually consumed as part of a semantic database/semantic search.
+ */
+async function featureExtraction(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "feature-extraction");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "feature-extraction",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { FillMaskInput, FillMaskOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type FillMaskArgs = BaseArgs & FillMaskInput;
 /**
  * Tries to fill in a hole with a missing word (token to be precise). That’s the base task for BERT models.

package/dist/commonjs/tasks/nlp/fillMask.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fillMask.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/fillMask.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAGxE,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,YAAY,GAAG,QAAQ,GAAG,aAAa,CAAC;AAEpD;;GAEG;AACH,wBAAsB,QAAQ,CAAC,IAAI,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAQ7F"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fillMask = fillMask;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Tries to fill in a hole with a missing word (token to be precise). That’s the base task for BERT models.
+ */
+async function fillMask(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "fill-mask");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "fill-mask",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { QuestionAnsweringInput, QuestionAnsweringOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type QuestionAnsweringArgs = BaseArgs & QuestionAnsweringInput;
 /**
  * Want to have a nice know-it-all bot that can answer any question?. Recommended model: deepset/roberta-base-squad2

package/dist/commonjs/tasks/nlp/questionAnswering.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"questionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/questionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAI1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,sBAAsB,CAAC;AAEtE;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,MAAM,CAAC,CAAC,CAY1C"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.questionAnswering = questionAnswering;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Want to have a nice know-it-all bot that can answer any question?. Recommended model: deepset/roberta-base-squad2
+ */
+async function questionAnswering(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "question-answering");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { SentenceSimilarityInput, SentenceSimilarityOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type SentenceSimilarityArgs = BaseArgs & SentenceSimilarityInput;
 /**
  * Calculate the semantic similarity between one text and a list of other sentences by comparing their embeddings.

package/dist/commonjs/tasks/nlp/sentenceSimilarity.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sentenceSimilarity.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/sentenceSimilarity.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAG5F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG,uBAAuB,CAAC;AAExE;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAQnC"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sentenceSimilarity = sentenceSimilarity;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Calculate the semantic similarity between one text and a list of other sentences by comparing their embeddings.
+ */
+async function sentenceSimilarity(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "sentence-similarity");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "sentence-similarity",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { SummarizationInput, SummarizationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type SummarizationArgs = BaseArgs & SummarizationInput;
 /**
  * This task is well known to summarize longer text into shorter text. Be careful, some models have a maximum length of input. That means that the summary cannot handle full books for instance. Be careful when choosing your model.

package/dist/commonjs/tasks/nlp/summarization.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"summarization.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/summarization.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAGlF,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,iBAAiB,GAAG,QAAQ,GAAG,kBAAkB,CAAC;AAE9D;;GAEG;AACH,wBAAsB,aAAa,CAAC,IAAI,EAAE,iBAAiB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAQ5G"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.summarization = summarization;
+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 is well known to summarize longer text into shorter text. Be careful, some models have a maximum length of input. That means that the summary cannot handle full books for instance. Be careful when choosing your model.
+ */
+async function summarization(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "summarization");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "summarization",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { TableQuestionAnsweringInput, TableQuestionAnsweringOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type TableQuestionAnsweringArgs = BaseArgs & TableQuestionAnsweringInput;
 /**
  * Don’t know SQL? Don’t want to dive into a large spreadsheet? Ask questions in plain english! Recommended model: google/tapas-base-finetuned-wtq.

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

@@ -1 +1 @@
-{"version":3,"file":"tableQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/tableQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AAGpG,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,0BAA0B,GAAG,QAAQ,GAAG,2BAA2B,CAAC;AAEhF;;GAEG;AACH,wBAAsB,sBAAsB,CAC3C,IAAI,EAAE,0BAA0B,EAChC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAY/C"}
+{"version":3,"file":"tableQuestionAnswering.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/tableQuestionAnswering.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AAGpG,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,0BAA0B,GAAG,QAAQ,GAAG,2BAA2B,CAAC;AAEhF;;GAEG;AACH,wBAAsB,sBAAsB,CAC3C,IAAI,EAAE,0BAA0B,EAChC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC,CAY/C"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tableQuestionAnswering = tableQuestionAnswering;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Don’t know SQL? Don’t want to dive into a large spreadsheet? Ask questions in plain english! Recommended model: google/tapas-base-finetuned-wtq.
+ */
+async function tableQuestionAnswering(args, options) {
+    const provider = await (0, getInferenceProviderMapping_js_1.resolveProvider)(args.provider, args.model, args.endpointUrl);
+    const providerHelper = (0, getProviderHelper_js_1.getProviderHelper)(provider, "table-question-answering");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "table-question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { TextClassificationInput, TextClassificationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type TextClassificationArgs = BaseArgs & TextClassificationInput;
 /**
  * Usually used for sentiment-analysis this will output the likelihood of classes of an input. Recommended model: distilbert-base-uncased-finetuned-sst-2-english

package/dist/commonjs/tasks/nlp/textClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"textClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/textClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAG5F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,sBAAsB,GAAG,QAAQ,GAAG,uBAAuB,CAAC;AAExE;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,sBAAsB,EAC5B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,wBAAwB,CAAC,CAQnC"}

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textClassification = textClassification;
+const getInferenceProviderMapping_js_1 = require("../../lib/getInferenceProviderMapping.js");
+const getProviderHelper_js_1 = require("../../lib/getProviderHelper.js");
+const request_js_1 = require("../../utils/request.js");
+/**
+ * Usually used for sentiment-analysis this will output the likelihood of classes of an input. Recommended model: distilbert-base-uncased-finetuned-sst-2-english
+ */
+async function textClassification(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-classification");
+    const { data: res } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "text-classification",
+    });
+    return providerHelper.getResponse(res);
+}

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

@@ -1,5 +1,5 @@
 import type { TextGenerationInput, TextGenerationOutput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export type { TextGenerationInput, TextGenerationOutput };
 /**
  * Use to continue text from a prompt. This is a very generic task. Recommended model: gpt2 (it’s a simple model, but fun to play with).

package/dist/commonjs/tasks/nlp/textGeneration.d.ts.map

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

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

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textGeneration = textGeneration;
+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. This is a very generic task. Recommended model: gpt2 (it’s a simple model, but fun to play with).
+ */
+async function textGeneration(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-generation");
+    const { data: response } = await (0, request_js_1.innerRequest)(args, providerHelper, {
+        ...options,
+        task: "text-generation",
+    });
+    return providerHelper.getResponse(response);
+}

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

@@ -1,5 +1,5 @@
 import type { TextGenerationInput } from "@huggingface/tasks";
-import type { BaseArgs, Options } from "../../types";
+import type { BaseArgs, Options } from "../../types.js";
 export interface TextGenerationStreamToken {
     /** Token ID from the model tokenizer */
     id: number;

package/dist/commonjs/tasks/nlp/textGenerationStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"textGenerationStream.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/textGenerationStream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAG9D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,WAAW,yBAAyB;IACzC,wCAAwC;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,cAAc;IACd,OAAO,EAAE,MAAM,CAAC;IAChB;;;OAGG;IACH,OAAO,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,gCAAgC;IAChD,wCAAwC;IACxC,EAAE,EAAE,MAAM,CAAC;IACX,iBAAiB;IACjB,IAAI,EAAE,MAAM,CAAC;IACb;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,kCAAkC;IAClD,qBAAqB;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,+BAA+B;IAC/B,aAAa,EAAE,gCAAgC,CAAC;IAChD,iCAAiC;IACjC,gBAAgB,EAAE,MAAM,CAAC;IACzB,8CAA8C;IAC9C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,oBAAoB;IACpB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,uBAAuB;IACvB,MAAM,EAAE,yBAAyB,EAAE,CAAC;CACpC;AAED,MAAM,MAAM,gCAAgC;AAC3C,qDAAqD;AACnD,QAAQ;AACV,oDAAoD;GAClD,WAAW;AACb,8DAA8D;GAC5D,eAAe,CAAC;AAEnB,MAAM,WAAW,2BAA2B;IAC3C,+BAA+B;IAC/B,aAAa,EAAE,gCAAgC,CAAC;IAChD,iCAAiC;IACjC,gBAAgB,EAAE,MAAM,CAAC;IACzB,8CAA8C;IAC9C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,oBAAoB;IACpB,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC5C,MAAM;IACN,MAAM,EAAE,yBAAyB,EAAE,CAAC;IACpC,8DAA8D;IAC9D,iBAAiB,CAAC,EAAE,kCAAkC,EAAE,CAAC;CACzD;AAED,MAAM,WAAW,0BAA0B;IAC1C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qCAAqC;IACrC,KAAK,EAAE,yBAAyB,CAAC;IACjC;;;OAGG;IACH,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B;;;OAGG;IACH,OAAO,EAAE,2BAA2B,GAAG,IAAI,CAAC;CAC5C;AAED;;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/textGenerationStream.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.textGenerationStream = textGenerationStream;
+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* textGenerationStream(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-generation");
+    yield* (0, request_js_1.innerStreamingRequest)(args, providerHelper, {
+        ...options,
+        task: "text-generation",
+    });
+}