@huggingface/inference 3.13.2 → 3.14.0

package/dist/esm/tasks/nlp/featureExtraction.d.ts

@@ -0,0 +1,17 @@
+import type { FeatureExtractionInput } from "@huggingface/tasks";
+import type { BaseArgs, Options } from "../../types.js";
+interface FeatureExtractionOAICompatInput {
+    encoding_format?: "float" | "base64";
+    dimensions?: number | null;
+}
+export type FeatureExtractionArgs = BaseArgs & FeatureExtractionInput & FeatureExtractionOAICompatInput;
+/**
+ * Returned values are a multidimensional array of floats (dimension depending on if you sent a string or a list of string, and if the automatic reduction, usually mean_pooling for instance was applied for you or not. This should be explained on the model's README).
+ */
+export type FeatureExtractionOutput = (number | number[] | number[][])[];
+/**
+ * This task reads some text and outputs raw float values, that are usually consumed as part of a semantic database/semantic search.
+ */
+export declare function featureExtraction(args: FeatureExtractionArgs, options?: Options): Promise<FeatureExtractionOutput>;
+export {};
+//# sourceMappingURL=featureExtraction.d.ts.map

package/dist/esm/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/esm/tasks/nlp/featureExtraction.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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.
+ */
+export async function featureExtraction(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "feature-extraction");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "feature-extraction",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/fillMask.d.ts

@@ -0,0 +1,8 @@
+import type { FillMaskInput, FillMaskOutput } from "@huggingface/tasks";
+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.
+ */
+export declare function fillMask(args: FillMaskArgs, options?: Options): Promise<FillMaskOutput>;
+//# sourceMappingURL=fillMask.d.ts.map

package/dist/esm/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/esm/tasks/nlp/fillMask.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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.
+ */
+export async function fillMask(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "fill-mask");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "fill-mask",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/questionAnswering.d.ts

@@ -0,0 +1,8 @@
+import type { QuestionAnsweringInput, QuestionAnsweringOutput } from "@huggingface/tasks";
+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
+ */
+export declare function questionAnswering(args: QuestionAnsweringArgs, options?: Options): Promise<QuestionAnsweringOutput[number]>;
+//# sourceMappingURL=questionAnswering.d.ts.map

package/dist/esm/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/esm/tasks/nlp/questionAnswering.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * Want to have a nice know-it-all bot that can answer any question?. Recommended model: deepset/roberta-base-squad2
+ */
+export async function questionAnswering(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "question-answering");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/sentenceSimilarity.d.ts

@@ -0,0 +1,8 @@
+import type { SentenceSimilarityInput, SentenceSimilarityOutput } from "@huggingface/tasks";
+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.
+ */
+export declare function sentenceSimilarity(args: SentenceSimilarityArgs, options?: Options): Promise<SentenceSimilarityOutput>;
+//# sourceMappingURL=sentenceSimilarity.d.ts.map

package/dist/esm/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/esm/tasks/nlp/sentenceSimilarity.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * Calculate the semantic similarity between one text and a list of other sentences by comparing their embeddings.
+ */
+export async function sentenceSimilarity(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "sentence-similarity");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "sentence-similarity",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/summarization.d.ts

@@ -0,0 +1,8 @@
+import type { SummarizationInput, SummarizationOutput } from "@huggingface/tasks";
+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.
+ */
+export declare function summarization(args: SummarizationArgs, options?: Options): Promise<SummarizationOutput>;
+//# sourceMappingURL=summarization.d.ts.map

package/dist/esm/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/esm/tasks/nlp/summarization.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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.
+ */
+export async function summarization(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "summarization");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "summarization",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/tableQuestionAnswering.d.ts

@@ -0,0 +1,8 @@
+import type { TableQuestionAnsweringInput, TableQuestionAnsweringOutput } from "@huggingface/tasks";
+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.
+ */
+export declare function tableQuestionAnswering(args: TableQuestionAnsweringArgs, options?: Options): Promise<TableQuestionAnsweringOutput[number]>;
+//# sourceMappingURL=tableQuestionAnswering.d.ts.map

package/dist/esm/tasks/nlp/tableQuestionAnswering.d.ts.map

@@ -0,0 +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,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/esm/tasks/nlp/tableQuestionAnswering.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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.
+ */
+export async function tableQuestionAnswering(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "table-question-answering");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "table-question-answering",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/textClassification.d.ts

@@ -0,0 +1,8 @@
+import type { TextClassificationInput, TextClassificationOutput } from "@huggingface/tasks";
+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
+ */
+export declare function textClassification(args: TextClassificationArgs, options?: Options): Promise<TextClassificationOutput>;
+//# sourceMappingURL=textClassification.d.ts.map

package/dist/esm/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/esm/tasks/nlp/textClassification.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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
+ */
+export async function textClassification(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "text-classification");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "text-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/textGeneration.d.ts

@@ -0,0 +1,8 @@
+import type { TextGenerationInput, TextGenerationOutput } from "@huggingface/tasks";
+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).
+ */
+export declare function textGeneration(args: BaseArgs & TextGenerationInput, options?: Options): Promise<TextGenerationOutput>;
+//# sourceMappingURL=textGeneration.d.ts.map

package/dist/esm/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/esm/tasks/nlp/textGeneration.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../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).
+ */
+export async function textGeneration(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "text-generation");
+    const { data: response } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "text-generation",
+    });
+    return providerHelper.getResponse(response);
+}

package/dist/esm/tasks/nlp/textGenerationStream.d.ts

@@ -0,0 +1,81 @@
+import type { TextGenerationInput } from "@huggingface/tasks";
+import type { BaseArgs, Options } from "../../types.js";
+export interface TextGenerationStreamToken {
+    /** Token ID from the model tokenizer */
+    id: number;
+    /** Token text */
+    text: string;
+    /** Logprob */
+    logprob: number;
+    /**
+     * Is the token a special token
+     * Can be used to ignore tokens when concatenating
+     */
+    special: boolean;
+}
+export interface TextGenerationStreamPrefillToken {
+    /** Token ID from the model tokenizer */
+    id: number;
+    /** Token text */
+    text: string;
+    /**
+     * Logprob
+     * Optional since the logprob of the first token cannot be computed
+     */
+    logprob?: number;
+}
+export interface TextGenerationStreamBestOfSequence {
+    /** Generated text */
+    generated_text: string;
+    /** Generation finish reason */
+    finish_reason: TextGenerationStreamFinishReason;
+    /** Number of generated tokens */
+    generated_tokens: number;
+    /** Sampling seed if sampling was activated */
+    seed?: number;
+    /** Prompt tokens */
+    prefill: TextGenerationStreamPrefillToken[];
+    /** Generated tokens */
+    tokens: TextGenerationStreamToken[];
+}
+export type TextGenerationStreamFinishReason = 
+/** number of generated tokens == `max_new_tokens` */
+"length"
+/** the model generated its end of sequence token */
+ | "eos_token"
+/** the model generated a text included in `stop_sequences` */
+ | "stop_sequence";
+export interface TextGenerationStreamDetails {
+    /** Generation finish reason */
+    finish_reason: TextGenerationStreamFinishReason;
+    /** Number of generated tokens */
+    generated_tokens: number;
+    /** Sampling seed if sampling was activated */
+    seed?: number;
+    /** Prompt tokens */
+    prefill: TextGenerationStreamPrefillToken[];
+    /** */
+    tokens: TextGenerationStreamToken[];
+    /** Additional sequences when using the `best_of` parameter */
+    best_of_sequences?: TextGenerationStreamBestOfSequence[];
+}
+export interface TextGenerationStreamOutput {
+    index?: number;
+    /** Generated token, one at a time */
+    token: TextGenerationStreamToken;
+    /**
+     * Complete generated text
+     * Only available when the generation is finished
+     */
+    generated_text: string | null;
+    /**
+     * Generation details
+     * Only available when the generation is finished
+     */
+    details: TextGenerationStreamDetails | null;
+}
+/**
+ * Use to continue text from a prompt. Same as `textGeneration` but returns generator that can be read one token at a time
+ */
+export declare function textGenerationStream(args: BaseArgs & TextGenerationInput, options?: Options): AsyncGenerator<TextGenerationStreamOutput>;
+//# sourceMappingURL=textGenerationStream.d.ts.map

package/dist/esm/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/esm/tasks/nlp/textGenerationStream.js

@@ -0,0 +1,14 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerStreamingRequest } from "../../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
+ */
+export async function* textGenerationStream(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "text-generation");
+    yield* innerStreamingRequest(args, providerHelper, {
+        ...options,
+        task: "text-generation",
+    });
+}

package/dist/esm/tasks/nlp/tokenClassification.d.ts

@@ -0,0 +1,8 @@
+import type { TokenClassificationInput, TokenClassificationOutput } from "@huggingface/tasks";
+import type { BaseArgs, Options } from "../../types.js";
+export type TokenClassificationArgs = BaseArgs & TokenClassificationInput;
+/**
+ * Usually used for sentence parsing, either grammatical, or Named Entity Recognition (NER) to understand keywords contained within text. Recommended model: dbmdz/bert-large-cased-finetuned-conll03-english
+ */
+export declare function tokenClassification(args: TokenClassificationArgs, options?: Options): Promise<TokenClassificationOutput>;
+//# sourceMappingURL=tokenClassification.d.ts.map

package/dist/esm/tasks/nlp/tokenClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/tokenClassification.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;AAGxD,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG,wBAAwB,CAAC;AAE1E;;GAEG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CAYpC"}

package/dist/esm/tasks/nlp/tokenClassification.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * Usually used for sentence parsing, either grammatical, or Named Entity Recognition (NER) to understand keywords contained within text. Recommended model: dbmdz/bert-large-cased-finetuned-conll03-english
+ */
+export async function tokenClassification(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "token-classification");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "token-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/translation.d.ts

@@ -0,0 +1,8 @@
+import type { TranslationInput, TranslationOutput } from "@huggingface/tasks";
+import type { BaseArgs, Options } from "../../types.js";
+export type TranslationArgs = BaseArgs & TranslationInput;
+/**
+ * This task is well known to translate text from one language to another. Recommended model: Helsinki-NLP/opus-mt-ru-en.
+ */
+export declare function translation(args: TranslationArgs, options?: Options): Promise<TranslationOutput>;
+//# sourceMappingURL=translation.d.ts.map

package/dist/esm/tasks/nlp/translation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"translation.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/translation.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;AAGxD,MAAM,MAAM,eAAe,GAAG,QAAQ,GAAG,gBAAgB,CAAC;AAC1D;;GAEG;AACH,wBAAsB,WAAW,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAQtG"}

package/dist/esm/tasks/nlp/translation.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * This task is well known to translate text from one language to another. Recommended model: Helsinki-NLP/opus-mt-ru-en.
+ */
+export async function translation(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "translation");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "translation",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/nlp/zeroShotClassification.d.ts

@@ -0,0 +1,8 @@
+import type { ZeroShotClassificationInput, ZeroShotClassificationOutput } from "@huggingface/tasks";
+import type { BaseArgs, Options } from "../../types.js";
+export type ZeroShotClassificationArgs = BaseArgs & ZeroShotClassificationInput;
+/**
+ * This task is super useful to try out classification with zero code, you simply pass a sentence/paragraph and the possible labels for that sentence, and you get a result. Recommended model: facebook/bart-large-mnli.
+ */
+export declare function zeroShotClassification(args: ZeroShotClassificationArgs, options?: Options): Promise<ZeroShotClassificationOutput>;
+//# sourceMappingURL=zeroShotClassification.d.ts.map

package/dist/esm/tasks/nlp/zeroShotClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"zeroShotClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/nlp/zeroShotClassification.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,CAYvC"}

package/dist/esm/tasks/nlp/zeroShotClassification.js

@@ -0,0 +1,15 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * This task is super useful to try out classification with zero code, you simply pass a sentence/paragraph and the possible labels for that sentence, and you get a result. Recommended model: facebook/bart-large-mnli.
+ */
+export async function zeroShotClassification(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "zero-shot-classification");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "zero-shot-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/tabular/tabularClassification.d.ts

@@ -0,0 +1,20 @@
+import type { BaseArgs, Options } from "../../types.js";
+export type TabularClassificationArgs = BaseArgs & {
+    inputs: {
+        /**
+         * A table of data represented as a dict of list where entries are headers and the lists are all the values, all lists must have the same size.
+         */
+        data: Record<string, string[]>;
+    };
+};
+/**
+ * A list of predicted labels for each row
+ */
+export type TabularClassificationOutput = number[];
+/**
+ * Predicts target label for a given set of features in tabular form.
+ * Typically, you will want to train a classification model on your training data and use it with your new data of the same format.
+ * Example model: vvmnnnkv/wine-quality
+ */
+export declare function tabularClassification(args: TabularClassificationArgs, options?: Options): Promise<TabularClassificationOutput>;
+//# sourceMappingURL=tabularClassification.d.ts.map

package/dist/esm/tasks/tabular/tabularClassification.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tabularClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/tabular/tabularClassification.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,yBAAyB,GAAG,QAAQ,GAAG;IAClD,MAAM,EAAE;QACP;;WAEG;QACH,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;KAC/B,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,2BAA2B,GAAG,MAAM,EAAE,CAAC;AAEnD;;;;GAIG;AACH,wBAAsB,qBAAqB,CAC1C,IAAI,EAAE,yBAAyB,EAC/B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,2BAA2B,CAAC,CAQtC"}

package/dist/esm/tasks/tabular/tabularClassification.js

@@ -0,0 +1,17 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * Predicts target label for a given set of features in tabular form.
+ * Typically, you will want to train a classification model on your training data and use it with your new data of the same format.
+ * Example model: vvmnnnkv/wine-quality
+ */
+export async function tabularClassification(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "tabular-classification");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "tabular-classification",
+    });
+    return providerHelper.getResponse(res);
+}

package/dist/esm/tasks/tabular/tabularRegression.d.ts

@@ -0,0 +1,20 @@
+import type { BaseArgs, Options } from "../../types.js";
+export type TabularRegressionArgs = BaseArgs & {
+    inputs: {
+        /**
+         * A table of data represented as a dict of list where entries are headers and the lists are all the values, all lists must have the same size.
+         */
+        data: Record<string, string[]>;
+    };
+};
+/**
+ * a list of predicted values for each row
+ */
+export type TabularRegressionOutput = number[];
+/**
+ * Predicts target value for a given set of features in tabular form.
+ * Typically, you will want to train a regression model on your training data and use it with your new data of the same format.
+ * Example model: scikit-learn/Fish-Weight
+ */
+export declare function tabularRegression(args: TabularRegressionArgs, options?: Options): Promise<TabularRegressionOutput>;
+//# sourceMappingURL=tabularRegression.d.ts.map

package/dist/esm/tasks/tabular/tabularRegression.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tabularRegression.d.ts","sourceRoot":"","sources":["../../../../src/tasks/tabular/tabularRegression.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAGxD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG;IAC9C,MAAM,EAAE;QACP;;WAEG;QACH,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;KAC/B,CAAC;CACF,CAAC;AAEF;;GAEG;AACH,MAAM,MAAM,uBAAuB,GAAG,MAAM,EAAE,CAAC;AAE/C;;;;GAIG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CAQlC"}

package/dist/esm/tasks/tabular/tabularRegression.js

@@ -0,0 +1,17 @@
+import { resolveProvider } from "../../lib/getInferenceProviderMapping.js";
+import { getProviderHelper } from "../../lib/getProviderHelper.js";
+import { innerRequest } from "../../utils/request.js";
+/**
+ * Predicts target value for a given set of features in tabular form.
+ * Typically, you will want to train a regression model on your training data and use it with your new data of the same format.
+ * Example model: scikit-learn/Fish-Weight
+ */
+export async function tabularRegression(args, options) {
+    const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
+    const providerHelper = getProviderHelper(provider, "tabular-regression");
+    const { data: res } = await innerRequest(args, providerHelper, {
+        ...options,
+        task: "tabular-regression",
+    });
+    return providerHelper.getResponse(res);
+}