@huggingface/tasks 0.16.6 → 0.17.0

package/src/tasks/chat-completion/spec/input.json

@@ -151,24 +151,54 @@
 	},
 	"$defs": {
 		"ChatCompletionInputMessage": {
-			"type": "object",
-			"required": ["role", "content"],
-			"properties": {
-				"content": {
-					"$ref": "#/$defs/ChatCompletionInputMessageContent"
-				},
-				"name": {
-					"type": "string",
-					"example": "\"David\"",
-					"nullable": true
+			"allOf": [
+				{
+					"$ref": "#/$defs/ChatCompletionInputMessageBody"
 				},
-				"role": {
-					"type": "string",
-					"example": "user"
+				{
+					"type": "object",
+					"required": ["role"],
+					"properties": {
+						"name": {
+							"type": "string",
+							"example": "\"David\"",
+							"nullable": true
+						},
+						"role": {
+							"type": "string",
+							"example": "user"
+						}
+					}
 				}
-			},
+			],
 			"title": "ChatCompletionInputMessage"
 		},
+		"ChatCompletionInputMessageBody": {
+			"oneOf": [
+				{
+					"type": "object",
+					"required": ["content"],
+					"properties": {
+						"content": {
+							"$ref": "#/$defs/ChatCompletionInputMessageContent"
+						}
+					}
+				},
+				{
+					"type": "object",
+					"required": ["tool_calls"],
+					"properties": {
+						"tool_calls": {
+							"type": "array",
+							"items": {
+								"$ref": "#/$defs/ChatCompletionInputToolCall"
+							}
+						}
+					}
+				}
+			],
+			"title": "ChatCompletionInputMessageBody"
+		},
 		"ChatCompletionInputMessageContent": {
 			"oneOf": [
 				{
@@ -227,6 +257,37 @@
 			},
 			"title": "ChatCompletionInputUrl"
 		},
+		"ChatCompletionInputToolCall": {
+			"type": "object",
+			"required": ["id", "type", "function"],
+			"properties": {
+				"function": {
+					"$ref": "#/$defs/ChatCompletionInputFunctionDefinition"
+				},
+				"id": {
+					"type": "string"
+				},
+				"type": {
+					"type": "string"
+				}
+			},
+			"title": "ChatCompletionInputToolCall"
+		},
+		"ChatCompletionInputFunctionDefinition": {
+			"type": "object",
+			"required": ["name", "arguments"],
+			"properties": {
+				"arguments": {},
+				"description": {
+					"type": "string",
+					"nullable": true
+				},
+				"name": {
+					"type": "string"
+				}
+			},
+			"title": "ChatCompletionInputFunctionDefinition"
+		},
 		"ChatCompletionInputGrammarType": {
 			"oneOf": [
 				{
@@ -326,21 +387,6 @@
 				}
 			},
 			"title": "ChatCompletionInputTool"
-		},
-		"ChatCompletionInputFunctionDefinition": {
-			"type": "object",
-			"required": ["name", "arguments"],
-			"properties": {
-				"arguments": {},
-				"description": {
-					"type": "string",
-					"nullable": true
-				},
-				"name": {
-					"type": "string"
-				}
-			},
-			"title": "ChatCompletionInputFunctionDefinition"
 		}
 	}
 }

package/src/tasks/chat-completion/spec/output.json

@@ -128,6 +128,10 @@
 				"role": {
 					"type": "string",
 					"example": "user"
+				},
+				"tool_call_id": {
+					"type": "string",
+					"nullable": true
 				}
 			},
 			"title": "ChatCompletionOutputTextMessage"

package/src/tasks/chat-completion/spec/stream_output.json

@@ -87,6 +87,10 @@
 				"role": {
 					"type": "string",
 					"example": "user"
+				},
+				"tool_call_id": {
+					"type": "string",
+					"nullable": true
 				}
 			},
 			"title": "ChatCompletionStreamOutputTextMessage"

package/src/tasks/text-to-video/data.ts

@@ -99,7 +99,7 @@ const taskData: TaskDataCustom = {
 	],
 	summary:
 		"Text-to-video models can be used in any application that requires generating consistent sequence of images from text. ",
-	widgetModels: ["tencent/HunyuanVideo"],
+	widgetModels: ["Wan-AI/Wan2.1-T2V-14B"],
 	youtubeId: undefined,
 };
 

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

@@ -1,17 +0,0 @@
-import { type SnippetInferenceProvider } from "../inference-providers.js";
-import type { PipelineType } from "../pipelines.js";
-import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
-import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
-export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: {
-    streaming?: boolean;
-    messages?: ChatCompletionInputMessage[];
-    temperature?: GenerationParameters["temperature"];
-    max_tokens?: GenerationParameters["max_tokens"];
-    top_p?: GenerationParameters["top_p"];
-}) => InferenceSnippet[];
-export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetFile: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: Record<string, unknown>) => InferenceSnippet[]>>;
-export declare function getCurlInferenceSnippet(model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: Record<string, unknown>): InferenceSnippet[];
-//# sourceMappingURL=curl.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmC,KAAK,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AAC3G,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAErE,eAAO,MAAM,YAAY,UACjB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAelB,CAAC;AAEF,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,YACT,wBAAwB,oBAChB,MAAM,SACjB;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAAgB,EA2ClB,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAClC,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAclB,CAAC;AAEF,eAAO,MAAM,WAAW,UAChB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAalB,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,OAAO,CACjC,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,wBAAwB,EAClC,eAAe,CAAC,EAAE,MAAM,EACxB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,EAAE,CACvB,CA0BD,CAAC;AAEF,wBAAgB,uBAAuB,CACtC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,wBAAwB,EAClC,eAAe,CAAC,EAAE,MAAM,EACxB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,EAAE,CAIpB"}

package/dist/commonjs/snippets/curl.js

@@ -1,129 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.curlSnippets = exports.snippetFile = exports.snippetZeroShotClassification = exports.snippetTextGeneration = exports.snippetBasic = void 0;
-exports.getCurlInferenceSnippet = getCurlInferenceSnippet;
-const inference_providers_js_1 = require("../inference-providers.js");
-const common_js_1 = require("./common.js");
-const inputs_js_1 = require("./inputs.js");
-const snippetBasic = (model, accessToken, provider) => {
-    if (provider !== "hf-inference") {
-        return [];
-    }
-    return [
-        {
-            client: "curl",
-            content: `\
-curl https://router.huggingface.co/hf-inference/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, provider, providerModelId, opts) => {
-    if (model.tags.includes("conversational")) {
-        const baseUrl = provider === "hf-inference"
-            ? `https://router.huggingface.co/hf-inference/models/${model.id}/v1/chat/completions`
-            : inference_providers_js_1.HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider) + "/v1/chat/completions";
-        const modelId = providerModelId ?? model.id;
-        // 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 [
-            {
-                client: "curl",
-                content: `curl '${baseUrl}' \\
--H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}' \\
--H 'Content-Type: application/json' \\
---data '{
-    "model": "${modelId}",
-    "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, provider);
-    }
-};
-exports.snippetTextGeneration = snippetTextGeneration;
-const snippetZeroShotClassification = (model, accessToken, provider) => {
-    if (provider !== "hf-inference") {
-        return [];
-    }
-    return [
-        {
-            client: "curl",
-            content: `curl https://router.huggingface.co/hf-inference/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, provider) => {
-    if (provider !== "hf-inference") {
-        return [];
-    }
-    return [
-        {
-            client: "curl",
-            content: `curl https://router.huggingface.co/hf-inference/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 tasks/src/pipelines.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, provider, providerModelId, opts) {
-    return model.pipeline_tag && model.pipeline_tag in exports.curlSnippets
-        ? exports.curlSnippets[model.pipeline_tag]?.(model, accessToken, provider, providerModelId, opts) ?? []
-        : [];
-}

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

@@ -1,21 +0,0 @@
-import { type SnippetInferenceProvider } from "../inference-providers.js";
-import type { PipelineType } from "../pipelines.js";
-import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
-import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
-export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: {
-    streaming?: boolean;
-    messages?: ChatCompletionInputMessage[];
-    temperature?: GenerationParameters["temperature"];
-    max_tokens?: GenerationParameters["max_tokens"];
-    top_p?: GenerationParameters["top_p"];
-}) => InferenceSnippet[];
-export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet[];
-export declare const snippetTextToImage: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetTextToVideo: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetTextToAudio: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetAutomaticSpeechRecognition: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const snippetFile: (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider) => InferenceSnippet[];
-export declare const jsSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: Record<string, unknown>) => InferenceSnippet[]>>;
-export declare function getJsInferenceSnippet(model: ModelDataMinimal, accessToken: string, provider: SnippetInferenceProvider, providerModelId?: string, opts?: Record<string, unknown>): InferenceSnippet[];
-//# sourceMappingURL=js.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,wBAAwB,EAAE,MAAM,2BAA2B,CAAC;AACzF,OAAO,KAAK,EAAE,YAAY,EAAc,MAAM,iBAAiB,CAAC;AAChE,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAgBrE,eAAO,MAAM,YAAY,UACjB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EA8ClB,CAAC;AAEF,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,YACT,wBAAwB,oBAChB,MAAM,SACjB;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAAgB,EA+GlB,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAAgB,EA2B5G,CAAC;AAEF,eAAO,MAAM,kBAAkB,UACvB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EA4ClB,CAAC;AAEF,eAAO,MAAM,kBAAkB,UACvB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAqBlB,CAAC;AAEF,eAAO,MAAM,kBAAkB,UACvB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAgDlB,CAAC;AAEF,eAAO,MAAM,iCAAiC,UACtC,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EAsBlB,CAAC;AAEF,eAAO,MAAM,WAAW,UAChB,gBAAgB,eACV,MAAM,YACT,wBAAwB,KAChC,gBAAgB,EA6BlB,CAAC;AAEF,eAAO,MAAM,UAAU,EAAE,OAAO,CAC/B,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,wBAAwB,EAClC,eAAe,CAAC,EAAE,MAAM,EACxB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,EAAE,CACvB,CA2BD,CAAC;AAEF,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,wBAAwB,EAClC,eAAe,CAAC,EAAE,MAAM,EACxB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,EAAE,CAIpB"}