@huggingface/tasks 0.13.1 → 0.13.2

package/dist/commonjs/snippets/common.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.stringifyMessages = stringifyMessages;
+exports.stringifyGenerationConfig = stringifyGenerationConfig;
+function stringifyMessages(messages, opts) {
+    let messagesStr = JSON.stringify(messages, null, "\t");
+    if (opts?.indent) {
+        messagesStr = messagesStr.replaceAll("\n", `\n${opts.indent}`);
+    }
+    if (!opts?.attributeKeyQuotes) {
+        messagesStr = messagesStr.replace(/"([^"]+)":/g, "$1:");
+    }
+    if (opts?.customContentEscaper) {
+        messagesStr = opts.customContentEscaper(messagesStr);
+    }
+    return messagesStr;
+}
+function stringifyGenerationConfig(config, opts) {
+    const quote = opts.attributeKeyQuotes ? `"` : "";
+    return Object.entries(config)
+        .map(([key, val]) => `${quote}${key}${quote}${opts.attributeValueConnector}${val}`)
+        .join(`,${opts.indent}`);
+}

package/dist/commonjs/snippets/curl.js

@@ -0,0 +1,100 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.curlSnippets = exports.snippetFile = exports.snippetZeroShotClassification = exports.snippetTextGeneration = exports.snippetBasic = void 0;
+exports.getCurlInferenceSnippet = getCurlInferenceSnippet;
+exports.hasCurlInferenceSnippet = hasCurlInferenceSnippet;
+const common_js_1 = require("./common.js");
+const inputs_js_1 = require("./inputs.js");
+const snippetBasic = (model, accessToken) => ({
+    content: `curl https://api-inference.huggingface.co/models/${model.id} \\
+	-X POST \\
+	-d '{"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model, true)}}' \\
+	-H 'Content-Type: application/json' \\
+	-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`,
+});
+exports.snippetBasic = snippetBasic;
+const snippetTextGeneration = (model, accessToken, opts) => {
+    if (model.tags.includes("conversational")) {
+        // Conversational model detected, so we display a code snippet that features the Messages API
+        const streaming = opts?.streaming ?? true;
+        const exampleMessages = (0, inputs_js_1.getModelInputSnippet)(model);
+        const messages = opts?.messages ?? exampleMessages;
+        const config = {
+            ...(opts?.temperature ? { temperature: opts.temperature } : undefined),
+            max_tokens: opts?.max_tokens ?? 500,
+            ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
+        };
+        return {
+            content: `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
+-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}" \\
+-H 'Content-Type: application/json' \\
+--data '{
+    "model": "${model.id}",
+    "messages": ${(0, common_js_1.stringifyMessages)(messages, {
+                indent: "\t",
+                attributeKeyQuotes: true,
+                customContentEscaper: (str) => str.replace(/'/g, "'\\''"),
+            })},
+    ${(0, common_js_1.stringifyGenerationConfig)(config, {
+                indent: "\n    ",
+                attributeKeyQuotes: true,
+                attributeValueConnector: ": ",
+            })},
+    "stream": ${!!streaming}
+}'`,
+        };
+    }
+    else {
+        return (0, exports.snippetBasic)(model, accessToken);
+    }
+};
+exports.snippetTextGeneration = snippetTextGeneration;
+const snippetZeroShotClassification = (model, accessToken) => ({
+    content: `curl https://api-inference.huggingface.co/models/${model.id} \\
+	-X POST \\
+	-d '{"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
+	-H 'Content-Type: application/json' \\
+	-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`,
+});
+exports.snippetZeroShotClassification = snippetZeroShotClassification;
+const snippetFile = (model, accessToken) => ({
+    content: `curl https://api-inference.huggingface.co/models/${model.id} \\
+	-X POST \\
+	--data-binary '@${(0, inputs_js_1.getModelInputSnippet)(model, true, true)}' \\
+	-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`,
+});
+exports.snippetFile = snippetFile;
+exports.curlSnippets = {
+    // Same order as in js/src/lib/interfaces/Types.ts
+    "text-classification": exports.snippetBasic,
+    "token-classification": exports.snippetBasic,
+    "table-question-answering": exports.snippetBasic,
+    "question-answering": exports.snippetBasic,
+    "zero-shot-classification": exports.snippetZeroShotClassification,
+    translation: exports.snippetBasic,
+    summarization: exports.snippetBasic,
+    "feature-extraction": exports.snippetBasic,
+    "text-generation": exports.snippetTextGeneration,
+    "image-text-to-text": exports.snippetTextGeneration,
+    "text2text-generation": exports.snippetBasic,
+    "fill-mask": exports.snippetBasic,
+    "sentence-similarity": exports.snippetBasic,
+    "automatic-speech-recognition": exports.snippetFile,
+    "text-to-image": exports.snippetBasic,
+    "text-to-speech": exports.snippetBasic,
+    "text-to-audio": exports.snippetBasic,
+    "audio-to-audio": exports.snippetFile,
+    "audio-classification": exports.snippetFile,
+    "image-classification": exports.snippetFile,
+    "image-to-text": exports.snippetFile,
+    "object-detection": exports.snippetFile,
+    "image-segmentation": exports.snippetFile,
+};
+function getCurlInferenceSnippet(model, accessToken, opts) {
+    return model.pipeline_tag && model.pipeline_tag in exports.curlSnippets
+        ? exports.curlSnippets[model.pipeline_tag]?.(model, accessToken, opts) ?? { content: "" }
+        : { content: "" };
+}
+function hasCurlInferenceSnippet(model) {
+    return !!model.pipeline_tag && model.pipeline_tag in exports.curlSnippets;
+}

package/dist/commonjs/snippets/curl.spec.js

@@ -0,0 +1,89 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const curl_js_1 = require("./curl.js");
+(0, vitest_1.describe)("inference API snippets", () => {
+    (0, vitest_1.it)("conversational llm", async () => {
+        const model = {
+            id: "meta-llama/Llama-3.1-8B-Instruct",
+            pipeline_tag: "text-generation",
+            tags: ["conversational"],
+            inference: "",
+        };
+        const snippet = (0, curl_js_1.getCurlInferenceSnippet)(model, "api_token");
+        (0, vitest_1.expect)(snippet.content)
+            .toEqual(`curl 'https://api-inference.huggingface.co/models/meta-llama/Llama-3.1-8B-Instruct/v1/chat/completions' \\
+-H "Authorization: Bearer api_token" \\
+-H 'Content-Type: application/json' \\
+--data '{
+    "model": "meta-llama/Llama-3.1-8B-Instruct",
+    "messages": [
+		{
+			"role": "user",
+			"content": "What is the capital of France?"
+		}
+	],
+    "max_tokens": 500,
+    "stream": true
+}'`);
+    });
+    (0, vitest_1.it)("conversational llm non-streaming", async () => {
+        const model = {
+            id: "meta-llama/Llama-3.1-8B-Instruct",
+            pipeline_tag: "text-generation",
+            tags: ["conversational"],
+            inference: "",
+        };
+        const snippet = (0, curl_js_1.getCurlInferenceSnippet)(model, "api_token", { streaming: false });
+        (0, vitest_1.expect)(snippet.content)
+            .toEqual(`curl 'https://api-inference.huggingface.co/models/meta-llama/Llama-3.1-8B-Instruct/v1/chat/completions' \\
+-H "Authorization: Bearer api_token" \\
+-H 'Content-Type: application/json' \\
+--data '{
+    "model": "meta-llama/Llama-3.1-8B-Instruct",
+    "messages": [
+		{
+			"role": "user",
+			"content": "What is the capital of France?"
+		}
+	],
+    "max_tokens": 500,
+    "stream": false
+}'`);
+    });
+    (0, vitest_1.it)("conversational vlm", async () => {
+        const model = {
+            id: "meta-llama/Llama-3.2-11B-Vision-Instruct",
+            pipeline_tag: "image-text-to-text",
+            tags: ["conversational"],
+            inference: "",
+        };
+        const snippet = (0, curl_js_1.getCurlInferenceSnippet)(model, "api_token");
+        (0, vitest_1.expect)(snippet.content)
+            .toEqual(`curl 'https://api-inference.huggingface.co/models/meta-llama/Llama-3.2-11B-Vision-Instruct/v1/chat/completions' \\
+-H "Authorization: Bearer api_token" \\
+-H 'Content-Type: application/json' \\
+--data '{
+    "model": "meta-llama/Llama-3.2-11B-Vision-Instruct",
+    "messages": [
+		{
+			"role": "user",
+			"content": [
+				{
+					"type": "text",
+					"text": "Describe this image in one sentence."
+				},
+				{
+					"type": "image_url",
+					"image_url": {
+						"url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+					}
+				}
+			]
+		}
+	],
+    "max_tokens": 500,
+    "stream": true
+}'`);
+    });
+});

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

@@ -0,0 +1,7 @@
+import * as inputs from "./inputs.js";
+import * as curl from "./curl.js";
+import * as python from "./python.js";
+import * as js from "./js.js";
+export * from "./types.js";
+export { inputs, curl, python, js };
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/snippets/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,cAAc,YAAY,CAAC;AAE3B,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC"}

package/dist/commonjs/snippets/index.js

@@ -0,0 +1,38 @@
+"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 __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+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 });
+exports.js = exports.python = exports.curl = exports.inputs = void 0;
+const inputs = __importStar(require("./inputs.js"));
+exports.inputs = inputs;
+const curl = __importStar(require("./curl.js"));
+exports.curl = curl;
+const python = __importStar(require("./python.js"));
+exports.python = python;
+const js = __importStar(require("./js.js"));
+exports.js = js;
+__exportStar(require("./types.js"), exports);

package/dist/commonjs/snippets/inputs.d.ts

@@ -0,0 +1,4 @@
+import type { ChatCompletionInputMessage } from "../tasks/index.js";
+import type { ModelDataMinimal } from "./types.js";
+export declare function getModelInputSnippet(model: ModelDataMinimal, noWrap?: boolean, noQuotes?: boolean): string | ChatCompletionInputMessage[];
+//# sourceMappingURL=inputs.d.ts.map

package/dist/commonjs/snippets/inputs.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"inputs.d.ts","sourceRoot":"","sources":["../../../src/snippets/inputs.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,0BAA0B,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AA6InD,wBAAgB,oBAAoB,CACnC,KAAK,EAAE,gBAAgB,EACvB,MAAM,UAAQ,EACd,QAAQ,UAAQ,GACd,MAAM,GAAG,0BAA0B,EAAE,CAmBvC"}

package/dist/commonjs/snippets/inputs.js

@@ -0,0 +1,127 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getModelInputSnippet = getModelInputSnippet;
+const inputsZeroShotClassification = () => `"Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!"`;
+const inputsTranslation = () => `"Меня зовут Вольфганг и я живу в Берлине"`;
+const inputsSummarization = () => `"The tower is 324 metres (1,063 ft) tall, about the same height as an 81-storey building, and the tallest structure in Paris. Its base is square, measuring 125 metres (410 ft) on each side. During its construction, the Eiffel Tower surpassed the Washington Monument to become the tallest man-made structure in the world, a title it held for 41 years until the Chrysler Building in New York City was finished in 1930. It was the first structure to reach a height of 300 metres. Due to the addition of a broadcasting aerial at the top of the tower in 1957, it is now taller than the Chrysler Building by 5.2 metres (17 ft). Excluding transmitters, the Eiffel Tower is the second tallest free-standing structure in France after the Millau Viaduct."`;
+const inputsTableQuestionAnswering = () => `{
+	"query": "How many stars does the transformers repository have?",
+	"table": {
+		"Repository": ["Transformers", "Datasets", "Tokenizers"],
+		"Stars": ["36542", "4512", "3934"],
+		"Contributors": ["651", "77", "34"],
+		"Programming language": [
+			"Python",
+			"Python",
+			"Rust, Python and NodeJS"
+		]
+	}
+}`;
+const inputsVisualQuestionAnswering = () => `{
+	"image": "cat.png",
+	"question": "What is in this image?"
+}`;
+const inputsQuestionAnswering = () => `{
+	"question": "What is my name?",
+	"context": "My name is Clara and I live in Berkeley."
+}`;
+const inputsTextClassification = () => `"I like you. I love you"`;
+const inputsTokenClassification = () => `"My name is Sarah Jessica Parker but you can call me Jessica"`;
+const inputsTextGeneration = (model) => {
+    if (model.tags.includes("conversational")) {
+        return model.pipeline_tag === "text-generation"
+            ? [{ role: "user", content: "What is the capital of France?" }]
+            : [
+                {
+                    role: "user",
+                    content: [
+                        {
+                            type: "text",
+                            text: "Describe this image in one sentence.",
+                        },
+                        {
+                            type: "image_url",
+                            image_url: {
+                                url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg",
+                            },
+                        },
+                    ],
+                },
+            ];
+    }
+    return `"Can you please let us know more details about your "`;
+};
+const inputsText2TextGeneration = () => `"The answer to the universe is"`;
+const inputsFillMask = (model) => `"The answer to the universe is ${model.mask_token}."`;
+const inputsSentenceSimilarity = () => `{
+	"source_sentence": "That is a happy person",
+	"sentences": [
+		"That is a happy dog",
+		"That is a very happy person",
+		"Today is a sunny day"
+	]
+}`;
+const inputsFeatureExtraction = () => `"Today is a sunny day and I will get some ice cream."`;
+const inputsImageClassification = () => `"cats.jpg"`;
+const inputsImageToText = () => `"cats.jpg"`;
+const inputsImageSegmentation = () => `"cats.jpg"`;
+const inputsObjectDetection = () => `"cats.jpg"`;
+const inputsAudioToAudio = () => `"sample1.flac"`;
+const inputsAudioClassification = () => `"sample1.flac"`;
+const inputsTextToImage = () => `"Astronaut riding a horse"`;
+const inputsTextToSpeech = () => `"The answer to the universe is 42"`;
+const inputsTextToAudio = () => `"liquid drum and bass, atmospheric synths, airy sounds"`;
+const inputsAutomaticSpeechRecognition = () => `"sample1.flac"`;
+const inputsTabularPrediction = () => `'{"Height":[11.52,12.48],"Length1":[23.2,24.0],"Length2":[25.4,26.3],"Species": ["Bream","Bream"]}'`;
+const inputsZeroShotImageClassification = () => `"cats.jpg"`;
+const modelInputSnippets = {
+    "audio-to-audio": inputsAudioToAudio,
+    "audio-classification": inputsAudioClassification,
+    "automatic-speech-recognition": inputsAutomaticSpeechRecognition,
+    "document-question-answering": inputsVisualQuestionAnswering,
+    "feature-extraction": inputsFeatureExtraction,
+    "fill-mask": inputsFillMask,
+    "image-classification": inputsImageClassification,
+    "image-to-text": inputsImageToText,
+    "image-segmentation": inputsImageSegmentation,
+    "object-detection": inputsObjectDetection,
+    "question-answering": inputsQuestionAnswering,
+    "sentence-similarity": inputsSentenceSimilarity,
+    summarization: inputsSummarization,
+    "table-question-answering": inputsTableQuestionAnswering,
+    "tabular-regression": inputsTabularPrediction,
+    "tabular-classification": inputsTabularPrediction,
+    "text-classification": inputsTextClassification,
+    "text-generation": inputsTextGeneration,
+    "image-text-to-text": inputsTextGeneration,
+    "text-to-image": inputsTextToImage,
+    "text-to-speech": inputsTextToSpeech,
+    "text-to-audio": inputsTextToAudio,
+    "text2text-generation": inputsText2TextGeneration,
+    "token-classification": inputsTokenClassification,
+    translation: inputsTranslation,
+    "zero-shot-classification": inputsZeroShotClassification,
+    "zero-shot-image-classification": inputsZeroShotImageClassification,
+};
+// Use noWrap to put the whole snippet on a single line (removing new lines and tabulations)
+// Use noQuotes to strip quotes from start & end (example: "abc" -> abc)
+function getModelInputSnippet(model, noWrap = false, noQuotes = false) {
+    if (model.pipeline_tag) {
+        const inputs = modelInputSnippets[model.pipeline_tag];
+        if (inputs) {
+            let result = inputs(model);
+            if (typeof result === "string") {
+                if (noWrap) {
+                    result = result.replace(/(?:(?:\r?\n|\r)\t*)|\t+/g, " ");
+                }
+                if (noQuotes) {
+                    const REGEX_QUOTES = /^"(.+)"$/s;
+                    const match = result.match(REGEX_QUOTES);
+                    result = match ? match[1] : result;
+                }
+            }
+            return result;
+        }
+    }
+    return "No input example has been defined for this model task.";
+}

package/dist/commonjs/snippets/js.js

@@ -0,0 +1,278 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.jsSnippets = exports.snippetFile = exports.snippetTextToAudio = exports.snippetTextToImage = exports.snippetZeroShotClassification = exports.snippetTextGeneration = exports.snippetBasic = void 0;
+exports.getJsInferenceSnippet = getJsInferenceSnippet;
+exports.hasJsInferenceSnippet = hasJsInferenceSnippet;
+const common_js_1 = require("./common.js");
+const inputs_js_1 = require("./inputs.js");
+const snippetBasic = (model, accessToken) => ({
+    content: `async function query(data) {
+	const response = await fetch(
+		"https://api-inference.huggingface.co/models/${model.id}",
+		{
+			headers: {
+				Authorization: "Bearer ${accessToken || `{API_TOKEN}`}"
+				"Content-Type": "application/json",
+			},
+			method: "POST",
+			body: JSON.stringify(data),
+		}
+	);
+	const result = await response.json();
+	return result;
+}
+
+query({"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model)}}).then((response) => {
+	console.log(JSON.stringify(response));
+});`,
+});
+exports.snippetBasic = snippetBasic;
+const snippetTextGeneration = (model, accessToken, opts) => {
+    if (model.tags.includes("conversational")) {
+        // Conversational model detected, so we display a code snippet that features the Messages API
+        const streaming = opts?.streaming ?? true;
+        const exampleMessages = (0, inputs_js_1.getModelInputSnippet)(model);
+        const messages = opts?.messages ?? exampleMessages;
+        const messagesStr = (0, common_js_1.stringifyMessages)(messages, { indent: "\t" });
+        const config = {
+            ...(opts?.temperature ? { temperature: opts.temperature } : undefined),
+            max_tokens: opts?.max_tokens ?? 500,
+            ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
+        };
+        const configStr = (0, common_js_1.stringifyGenerationConfig)(config, {
+            indent: "\n\t",
+            attributeValueConnector: ": ",
+        });
+        if (streaming) {
+            return [
+                {
+                    client: "huggingface.js",
+                    content: `import { HfInference } from "@huggingface/inference";
+
+const client = new HfInference("${accessToken || `{API_TOKEN}`}");
+
+let out = "";
+
+const stream = client.chatCompletionStream({
+	model: "${model.id}",
+	messages: ${messagesStr},
+	${configStr}
+});
+
+for await (const chunk of stream) {
+	if (chunk.choices && chunk.choices.length > 0) {
+		const newContent = chunk.choices[0].delta.content;
+		out += newContent;
+		console.log(newContent);
+	}  
+}`,
+                },
+                {
+                    client: "openai",
+                    content: `import { OpenAI } from "openai";
+
+const client = new OpenAI({
+	baseURL: "https://api-inference.huggingface.co/v1/",
+    apiKey: "${accessToken || `{API_TOKEN}`}"
+});
+
+let out = "";
+
+const stream = await client.chat.completions.create({
+	model: "${model.id}",
+	messages: ${messagesStr},
+	${configStr},
+	stream: true,
+});
+
+for await (const chunk of stream) {
+	if (chunk.choices && chunk.choices.length > 0) {
+		const newContent = chunk.choices[0].delta.content;
+		out += newContent;
+		console.log(newContent);
+	}  
+}`,
+                },
+            ];
+        }
+        else {
+            return [
+                {
+                    client: "huggingface.js",
+                    content: `import { HfInference } from "@huggingface/inference";
+
+const client = new HfInference("${accessToken || `{API_TOKEN}`}");
+
+const chatCompletion = await client.chatCompletion({
+	model: "${model.id}",
+	messages: ${messagesStr},
+	${configStr}
+});
+
+console.log(chatCompletion.choices[0].message);`,
+                },
+                {
+                    client: "openai",
+                    content: `import { OpenAI } from "openai";
+
+const client = new OpenAI({
+    baseURL: "https://api-inference.huggingface.co/v1/",
+    apiKey: "${accessToken || `{API_TOKEN}`}"
+});
+
+const chatCompletion = await client.chat.completions.create({
+	model: "${model.id}",
+	messages: ${messagesStr},
+	${configStr}
+});
+
+console.log(chatCompletion.choices[0].message);`,
+                },
+            ];
+        }
+    }
+    else {
+        return (0, exports.snippetBasic)(model, accessToken);
+    }
+};
+exports.snippetTextGeneration = snippetTextGeneration;
+const snippetZeroShotClassification = (model, accessToken) => ({
+    content: `async function query(data) {
+	const response = await fetch(
+		"https://api-inference.huggingface.co/models/${model.id}",
+		{
+			headers: {
+				Authorization: "Bearer ${accessToken || `{API_TOKEN}`}"
+				"Content-Type": "application/json",
+			},
+			method: "POST",
+			body: JSON.stringify(data),
+		}
+	);
+	const result = await response.json();
+	return result;
+}
+
+query({"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}).then((response) => {
+	console.log(JSON.stringify(response));
+});`,
+});
+exports.snippetZeroShotClassification = snippetZeroShotClassification;
+const snippetTextToImage = (model, accessToken) => ({
+    content: `async function query(data) {
+	const response = await fetch(
+		"https://api-inference.huggingface.co/models/${model.id}",
+		{
+			headers: {
+				Authorization: "Bearer ${accessToken || `{API_TOKEN}`}"
+				"Content-Type": "application/json",
+			},
+			method: "POST",
+			body: JSON.stringify(data),
+		}
+	);
+	const result = await response.blob();
+	return result;
+}
+query({"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model)}}).then((response) => {
+	// Use image
+});`,
+});
+exports.snippetTextToImage = snippetTextToImage;
+const snippetTextToAudio = (model, accessToken) => {
+    const commonSnippet = `async function query(data) {
+		const response = await fetch(
+			"https://api-inference.huggingface.co/models/${model.id}",
+			{
+				headers: {
+					Authorization: "Bearer ${accessToken || `{API_TOKEN}`}"
+					"Content-Type": "application/json",
+				},
+				method: "POST",
+				body: JSON.stringify(data),
+			}
+		);`;
+    if (model.library_name === "transformers") {
+        return {
+            content: commonSnippet +
+                `
+			const result = await response.blob();
+			return result;
+		}
+		query({"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model)}}).then((response) => {
+			// Returns a byte object of the Audio wavform. Use it directly!
+		});`,
+        };
+    }
+    else {
+        return {
+            content: commonSnippet +
+                `
+			const result = await response.json();
+			return result;
+		}
+		
+		query({"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model)}}).then((response) => {
+			console.log(JSON.stringify(response));
+		});`,
+        };
+    }
+};
+exports.snippetTextToAudio = snippetTextToAudio;
+const snippetFile = (model, accessToken) => ({
+    content: `async function query(filename) {
+	const data = fs.readFileSync(filename);
+	const response = await fetch(
+		"https://api-inference.huggingface.co/models/${model.id}",
+		{
+			headers: {
+				Authorization: "Bearer ${accessToken || `{API_TOKEN}`}"
+				"Content-Type": "application/json",
+			},
+			method: "POST",
+			body: data,
+		}
+	);
+	const result = await response.json();
+	return result;
+}
+
+query(${(0, inputs_js_1.getModelInputSnippet)(model)}).then((response) => {
+	console.log(JSON.stringify(response));
+});`,
+});
+exports.snippetFile = snippetFile;
+exports.jsSnippets = {
+    // Same order as in js/src/lib/interfaces/Types.ts
+    "text-classification": exports.snippetBasic,
+    "token-classification": exports.snippetBasic,
+    "table-question-answering": exports.snippetBasic,
+    "question-answering": exports.snippetBasic,
+    "zero-shot-classification": exports.snippetZeroShotClassification,
+    translation: exports.snippetBasic,
+    summarization: exports.snippetBasic,
+    "feature-extraction": exports.snippetBasic,
+    "text-generation": exports.snippetTextGeneration,
+    "image-text-to-text": exports.snippetTextGeneration,
+    "text2text-generation": exports.snippetBasic,
+    "fill-mask": exports.snippetBasic,
+    "sentence-similarity": exports.snippetBasic,
+    "automatic-speech-recognition": exports.snippetFile,
+    "text-to-image": exports.snippetTextToImage,
+    "text-to-speech": exports.snippetTextToAudio,
+    "text-to-audio": exports.snippetTextToAudio,
+    "audio-to-audio": exports.snippetFile,
+    "audio-classification": exports.snippetFile,
+    "image-classification": exports.snippetFile,
+    "image-to-text": exports.snippetFile,
+    "object-detection": exports.snippetFile,
+    "image-segmentation": exports.snippetFile,
+};
+function getJsInferenceSnippet(model, accessToken, opts) {
+    return model.pipeline_tag && model.pipeline_tag in exports.jsSnippets
+        ? exports.jsSnippets[model.pipeline_tag]?.(model, accessToken, opts) ?? { content: "" }
+        : { content: "" };
+}
+function hasJsInferenceSnippet(model) {
+    return !!model.pipeline_tag && model.pipeline_tag in exports.jsSnippets;
+}