@huggingface/tasks 0.12.26 → 0.12.27

package/dist/index.cjs

@@ -6329,7 +6329,6 @@ __export(curl_exports, {
   hasCurlInferenceSnippet: () => hasCurlInferenceSnippet,
   snippetBasic: () => snippetBasic,
   snippetFile: () => snippetFile,
-  snippetImageTextToTextGeneration: () => snippetImageTextToTextGeneration,
   snippetTextGeneration: () => snippetTextGeneration,
   snippetZeroShotClassification: () => snippetZeroShotClassification
 });
@@ -6377,9 +6376,21 @@ var snippetBasic = (model, accessToken) => ({
 var snippetTextGeneration = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const messages = opts?.messages ?? [
-      { role: "user", content: "What is the capital of France?" }
+    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          },
+          { type: "text", text: "Describe this image in one sentence." }
+        ]
+      }
     ];
+    const messages = opts?.messages ?? exampleMessages;
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
       max_tokens: opts?.max_tokens ?? 500,
@@ -6414,32 +6425,6 @@ var snippetTextGeneration = (model, accessToken, opts) => {
     return snippetBasic(model, accessToken);
   }
 };
-var snippetImageTextToTextGeneration = (model, accessToken) => {
-  if (model.tags.includes("conversational")) {
-    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' \\
--d '{
-	"model": "${model.id}",
-	"messages": [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"}},
-				{"type": "text", "text": "Describe this image in one sentence."}
-			]
-		}
-	],
-	"max_tokens": 500,
-	"stream": false
-}'
-`
-    };
-  } else {
-    return snippetBasic(model, accessToken);
-  }
-};
 var snippetZeroShotClassification = (model, accessToken) => ({
   content: `curl https://api-inference.huggingface.co/models/${model.id} \\
 	-X POST \\
@@ -6464,7 +6449,7 @@ var curlSnippets = {
   summarization: snippetBasic,
   "feature-extraction": snippetBasic,
   "text-generation": snippetTextGeneration,
-  "image-text-to-text": snippetImageTextToTextGeneration,
+  "image-text-to-text": snippetTextGeneration,
   "text2text-generation": snippetBasic,
   "fill-mask": snippetBasic,
   "sentence-similarity": snippetBasic,
@@ -6494,7 +6479,6 @@ __export(python_exports, {
   pythonSnippets: () => pythonSnippets,
   snippetBasic: () => snippetBasic2,
   snippetConversational: () => snippetConversational,
-  snippetConversationalWithImage: () => snippetConversationalWithImage,
   snippetDocumentQuestionAnswering: () => snippetDocumentQuestionAnswering,
   snippetFile: () => snippetFile2,
   snippetTabular: () => snippetTabular,
@@ -6505,9 +6489,21 @@ __export(python_exports, {
 });
 var snippetConversational = (model, accessToken, opts) => {
   const streaming = opts?.streaming ?? true;
-  const messages = opts?.messages ?? [
-    { role: "user", content: "What is the capital of France?" }
+  const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+    {
+      role: "user",
+      content: [
+        {
+          type: "image_url",
+          image_url: {
+            url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+          }
+        },
+        { type: "text", text: "Describe this image in one sentence." }
+      ]
+    }
   ];
+  const messages = opts?.messages ?? exampleMessages;
   const messagesStr = stringifyMessages(messages, {
     sep: ",\n	",
     start: `[
@@ -6609,29 +6605,6 @@ print(completion.choices[0].message)`
     ];
   }
 };
-var snippetConversationalWithImage = (model, accessToken) => ({
-  content: `from huggingface_hub import InferenceClient
-
-client = InferenceClient(api_key="${accessToken || "{API_TOKEN}"}")
-
-image_url = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-
-for message in client.chat_completion(
-	model="${model.id}",
-	messages=[
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": image_url}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens=500,
-	stream=True,
-):
-	print(message.choices[0].delta.content, end="")`
-});
 var snippetZeroShotClassification2 = (model) => ({
   content: `def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -6749,7 +6722,7 @@ var pythonSnippets = {
   "feature-extraction": snippetBasic2,
   "text-generation": snippetBasic2,
   "text2text-generation": snippetBasic2,
-  "image-text-to-text": snippetConversationalWithImage,
+  "image-text-to-text": snippetConversational,
   "fill-mask": snippetBasic2,
   "sentence-similarity": snippetBasic2,
   "automatic-speech-recognition": snippetFile2,
@@ -6768,10 +6741,8 @@ var pythonSnippets = {
   "zero-shot-image-classification": snippetZeroShotImageClassification
 };
 function getPythonInferenceSnippet(model, accessToken, opts) {
-  if (model.pipeline_tag === "text-generation" && model.tags.includes("conversational")) {
+  if (model.tags.includes("conversational")) {
     return snippetConversational(model, accessToken, opts);
-  } else if (model.pipeline_tag === "image-text-to-text" && model.tags.includes("conversational")) {
-    return snippetConversationalWithImage(model, accessToken);
   } else {
     let snippets = model.pipeline_tag && model.pipeline_tag in pythonSnippets ? pythonSnippets[model.pipeline_tag]?.(model, accessToken) ?? { content: "" } : { content: "" };
     snippets = Array.isArray(snippets) ? snippets : [snippets];
@@ -6800,7 +6771,6 @@ __export(js_exports, {
   jsSnippets: () => jsSnippets,
   snippetBasic: () => snippetBasic3,
   snippetFile: () => snippetFile3,
-  snippetImageTextToTextGeneration: () => snippetImageTextToTextGeneration2,
   snippetTextGeneration: () => snippetTextGeneration2,
   snippetTextToAudio: () => snippetTextToAudio2,
   snippetTextToImage: () => snippetTextToImage2,
@@ -6830,9 +6800,21 @@ query({"inputs": ${getModelInputSnippet(model)}}).then((response) => {
 var snippetTextGeneration2 = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const messages = opts?.messages ?? [
-      { role: "user", content: "What is the capital of France?" }
+    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          },
+          { type: "text", text: "Describe this image in one sentence." }
+        ]
+      }
     ];
+    const messages = opts?.messages ?? exampleMessages;
     const messagesStr = stringifyMessages(messages, { sep: ",\n		", start: "[\n		", end: "\n	]" });
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
@@ -6935,34 +6917,6 @@ console.log(chatCompletion.choices[0].message);`
     return snippetBasic3(model, accessToken);
   }
 };
-var snippetImageTextToTextGeneration2 = (model, accessToken) => {
-  if (model.tags.includes("conversational")) {
-    return {
-      content: `import { HfInference } from "@huggingface/inference";
-
-const inference = new HfInference("${accessToken || `{API_TOKEN}`}");
-const imageUrl = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg";
-
-for await (const chunk of inference.chatCompletionStream({
-	model: "${model.id}",
-	messages: [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": imageUrl}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens: 500,
-})) {
-	process.stdout.write(chunk.choices[0]?.delta?.content || "");
-}`
-    };
-  } else {
-    return snippetBasic3(model, accessToken);
-  }
-};
 var snippetZeroShotClassification3 = (model, accessToken) => ({
   content: `async function query(data) {
 	const response = await fetch(
@@ -7075,7 +7029,7 @@ var jsSnippets = {
   summarization: snippetBasic3,
   "feature-extraction": snippetBasic3,
   "text-generation": snippetTextGeneration2,
-  "image-text-to-text": snippetImageTextToTextGeneration2,
+  "image-text-to-text": snippetTextGeneration2,
   "text2text-generation": snippetBasic3,
   "fill-mask": snippetBasic3,
   "sentence-similarity": snippetBasic3,

package/dist/index.js

@@ -6291,7 +6291,6 @@ __export(curl_exports, {
   hasCurlInferenceSnippet: () => hasCurlInferenceSnippet,
   snippetBasic: () => snippetBasic,
   snippetFile: () => snippetFile,
-  snippetImageTextToTextGeneration: () => snippetImageTextToTextGeneration,
   snippetTextGeneration: () => snippetTextGeneration,
   snippetZeroShotClassification: () => snippetZeroShotClassification
 });
@@ -6339,9 +6338,21 @@ var snippetBasic = (model, accessToken) => ({
 var snippetTextGeneration = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const messages = opts?.messages ?? [
-      { role: "user", content: "What is the capital of France?" }
+    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          },
+          { type: "text", text: "Describe this image in one sentence." }
+        ]
+      }
     ];
+    const messages = opts?.messages ?? exampleMessages;
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
       max_tokens: opts?.max_tokens ?? 500,
@@ -6376,32 +6387,6 @@ var snippetTextGeneration = (model, accessToken, opts) => {
     return snippetBasic(model, accessToken);
   }
 };
-var snippetImageTextToTextGeneration = (model, accessToken) => {
-  if (model.tags.includes("conversational")) {
-    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' \\
--d '{
-	"model": "${model.id}",
-	"messages": [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"}},
-				{"type": "text", "text": "Describe this image in one sentence."}
-			]
-		}
-	],
-	"max_tokens": 500,
-	"stream": false
-}'
-`
-    };
-  } else {
-    return snippetBasic(model, accessToken);
-  }
-};
 var snippetZeroShotClassification = (model, accessToken) => ({
   content: `curl https://api-inference.huggingface.co/models/${model.id} \\
 	-X POST \\
@@ -6426,7 +6411,7 @@ var curlSnippets = {
   summarization: snippetBasic,
   "feature-extraction": snippetBasic,
   "text-generation": snippetTextGeneration,
-  "image-text-to-text": snippetImageTextToTextGeneration,
+  "image-text-to-text": snippetTextGeneration,
   "text2text-generation": snippetBasic,
   "fill-mask": snippetBasic,
   "sentence-similarity": snippetBasic,
@@ -6456,7 +6441,6 @@ __export(python_exports, {
   pythonSnippets: () => pythonSnippets,
   snippetBasic: () => snippetBasic2,
   snippetConversational: () => snippetConversational,
-  snippetConversationalWithImage: () => snippetConversationalWithImage,
   snippetDocumentQuestionAnswering: () => snippetDocumentQuestionAnswering,
   snippetFile: () => snippetFile2,
   snippetTabular: () => snippetTabular,
@@ -6467,9 +6451,21 @@ __export(python_exports, {
 });
 var snippetConversational = (model, accessToken, opts) => {
   const streaming = opts?.streaming ?? true;
-  const messages = opts?.messages ?? [
-    { role: "user", content: "What is the capital of France?" }
+  const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+    {
+      role: "user",
+      content: [
+        {
+          type: "image_url",
+          image_url: {
+            url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+          }
+        },
+        { type: "text", text: "Describe this image in one sentence." }
+      ]
+    }
   ];
+  const messages = opts?.messages ?? exampleMessages;
   const messagesStr = stringifyMessages(messages, {
     sep: ",\n	",
     start: `[
@@ -6571,29 +6567,6 @@ print(completion.choices[0].message)`
     ];
   }
 };
-var snippetConversationalWithImage = (model, accessToken) => ({
-  content: `from huggingface_hub import InferenceClient
-
-client = InferenceClient(api_key="${accessToken || "{API_TOKEN}"}")
-
-image_url = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-
-for message in client.chat_completion(
-	model="${model.id}",
-	messages=[
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": image_url}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens=500,
-	stream=True,
-):
-	print(message.choices[0].delta.content, end="")`
-});
 var snippetZeroShotClassification2 = (model) => ({
   content: `def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -6711,7 +6684,7 @@ var pythonSnippets = {
   "feature-extraction": snippetBasic2,
   "text-generation": snippetBasic2,
   "text2text-generation": snippetBasic2,
-  "image-text-to-text": snippetConversationalWithImage,
+  "image-text-to-text": snippetConversational,
   "fill-mask": snippetBasic2,
   "sentence-similarity": snippetBasic2,
   "automatic-speech-recognition": snippetFile2,
@@ -6730,10 +6703,8 @@ var pythonSnippets = {
   "zero-shot-image-classification": snippetZeroShotImageClassification
 };
 function getPythonInferenceSnippet(model, accessToken, opts) {
-  if (model.pipeline_tag === "text-generation" && model.tags.includes("conversational")) {
+  if (model.tags.includes("conversational")) {
     return snippetConversational(model, accessToken, opts);
-  } else if (model.pipeline_tag === "image-text-to-text" && model.tags.includes("conversational")) {
-    return snippetConversationalWithImage(model, accessToken);
   } else {
     let snippets = model.pipeline_tag && model.pipeline_tag in pythonSnippets ? pythonSnippets[model.pipeline_tag]?.(model, accessToken) ?? { content: "" } : { content: "" };
     snippets = Array.isArray(snippets) ? snippets : [snippets];
@@ -6762,7 +6733,6 @@ __export(js_exports, {
   jsSnippets: () => jsSnippets,
   snippetBasic: () => snippetBasic3,
   snippetFile: () => snippetFile3,
-  snippetImageTextToTextGeneration: () => snippetImageTextToTextGeneration2,
   snippetTextGeneration: () => snippetTextGeneration2,
   snippetTextToAudio: () => snippetTextToAudio2,
   snippetTextToImage: () => snippetTextToImage2,
@@ -6792,9 +6762,21 @@ query({"inputs": ${getModelInputSnippet(model)}}).then((response) => {
 var snippetTextGeneration2 = (model, accessToken, opts) => {
   if (model.tags.includes("conversational")) {
     const streaming = opts?.streaming ?? true;
-    const messages = opts?.messages ?? [
-      { role: "user", content: "What is the capital of France?" }
+    const exampleMessages = model.pipeline_tag === "text-generation" ? [{ role: "user", content: "What is the capital of France?" }] : [
+      {
+        role: "user",
+        content: [
+          {
+            type: "image_url",
+            image_url: {
+              url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
+            }
+          },
+          { type: "text", text: "Describe this image in one sentence." }
+        ]
+      }
     ];
+    const messages = opts?.messages ?? exampleMessages;
     const messagesStr = stringifyMessages(messages, { sep: ",\n		", start: "[\n		", end: "\n	]" });
     const config = {
       ...opts?.temperature ? { temperature: opts.temperature } : void 0,
@@ -6897,34 +6879,6 @@ console.log(chatCompletion.choices[0].message);`
     return snippetBasic3(model, accessToken);
   }
 };
-var snippetImageTextToTextGeneration2 = (model, accessToken) => {
-  if (model.tags.includes("conversational")) {
-    return {
-      content: `import { HfInference } from "@huggingface/inference";
-
-const inference = new HfInference("${accessToken || `{API_TOKEN}`}");
-const imageUrl = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg";
-
-for await (const chunk of inference.chatCompletionStream({
-	model: "${model.id}",
-	messages: [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": imageUrl}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens: 500,
-})) {
-	process.stdout.write(chunk.choices[0]?.delta?.content || "");
-}`
-    };
-  } else {
-    return snippetBasic3(model, accessToken);
-  }
-};
 var snippetZeroShotClassification3 = (model, accessToken) => ({
   content: `async function query(data) {
 	const response = await fetch(
@@ -7037,7 +6991,7 @@ var jsSnippets = {
   summarization: snippetBasic3,
   "feature-extraction": snippetBasic3,
   "text-generation": snippetTextGeneration2,
-  "image-text-to-text": snippetImageTextToTextGeneration2,
+  "image-text-to-text": snippetTextGeneration2,
   "text2text-generation": snippetBasic3,
   "fill-mask": snippetBasic3,
   "sentence-similarity": snippetBasic3,

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

@@ -9,7 +9,6 @@ export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToke
     max_tokens?: GenerationParameters["max_tokens"];
     top_p?: GenerationParameters["top_p"];
 }) => InferenceSnippet;
-export declare const snippetImageTextToTextGeneration: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetFile: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>) => InferenceSnippet>>;

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

@@ -1 +1 @@
-{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,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,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,gBAuCF,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBA0B/F,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM3F,CAAC;AAEH,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAKzE,CAAC;AAEH,eAAO,MAAM,YAAY,EAAE,OAAO,CACjC,MAAM,CACL,YAAY,EACZ,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,gBAAgB,CAClG,CA0BD,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,gBAAgB,CAItG;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,cAAc,CAAC,GAAG,OAAO,CAE9F"}
+{"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,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,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,gBAsDF,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM3F,CAAC;AAEH,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAKzE,CAAC;AAEH,eAAO,MAAM,YAAY,EAAE,OAAO,CACjC,MAAM,CACL,YAAY,EACZ,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,gBAAgB,CAClG,CA0BD,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,GAAG,gBAAgB,CAItG;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,cAAc,CAAC,GAAG,OAAO,CAE9F"}

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

@@ -9,7 +9,6 @@ export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToke
     max_tokens?: GenerationParameters["max_tokens"];
     top_p?: GenerationParameters["top_p"];
 }) => InferenceSnippet | InferenceSnippet[];
-export declare const snippetImageTextToTextGeneration: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetTextToImage: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetTextToAudio: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;

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

@@ -1 +1 @@
-{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,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,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAoB1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,GAAG,gBAAgB,EA8GrC,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBA4B/F,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAsB3F,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAmBhF,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAuCjF,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAqBzE,CAAC;AAEH,eAAO,MAAM,UAAU,EAAE,OAAO,CAC/B,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,GAAG,gBAAgB,EAAE,CAC1C,CA0BD,CAAC;AAEF,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,GACjB,gBAAgB,GAAG,gBAAgB,EAAE,CAIvC;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAEtE"}
+{"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,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,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAoB1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,GAAG,gBAAgB,EA6HrC,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAsB3F,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAmBhF,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAuCjF,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAqBzE,CAAC;AAEH,eAAO,MAAM,UAAU,EAAE,OAAO,CAC/B,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,GAAG,gBAAgB,EAAE,CAC1C,CA0BD,CAAC;AAEF,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,GACjB,gBAAgB,GAAG,gBAAgB,EAAE,CAIvC;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAEtE"}

package/dist/src/snippets/python.d.ts

@@ -8,7 +8,6 @@ export declare const snippetConversational: (model: ModelDataMinimal, accessToke
     max_tokens?: GenerationParameters["max_tokens"];
     top_p?: GenerationParameters["top_p"];
 }) => InferenceSnippet[];
-export declare const snippetConversationalWithImage: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
 export declare const snippetZeroShotClassification: (model: ModelDataMinimal) => InferenceSnippet;
 export declare const snippetZeroShotImageClassification: (model: ModelDataMinimal) => InferenceSnippet;
 export declare const snippetBasic: (model: ModelDataMinimal) => InferenceSnippet;

package/dist/src/snippets/python.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"python.d.ts","sourceRoot":"","sources":["../../../src/snippets/python.ts"],"names":[],"mappings":"AAAA,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,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,EAyGlB,CAAC;AAEF,eAAO,MAAM,8BAA8B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAsB5F,CAAC;AAEH,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,KAAG,gBAStE,CAAC;AAEH,eAAO,MAAM,kCAAkC,UAAW,gBAAgB,KAAG,gBAe3E,CAAC;AAEH,eAAO,MAAM,YAAY,UAAW,gBAAgB,KAAG,gBAQrD,CAAC;AAEH,eAAO,MAAM,WAAW,UAAW,gBAAgB,KAAG,gBAQpD,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,gBAW3D,CAAC;AAEH,eAAO,MAAM,cAAc,UAAW,gBAAgB,KAAG,gBAOvD,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,gBA+B5D,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,KAAG,gBAWzE,CAAC;AAEH,eAAO,MAAM,cAAc,EAAE,OAAO,CACnC,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,GAAG,gBAAgB,EAAE,CAC1C,CA8BD,CAAC;AAEF,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,GAAG,gBAAgB,EAAE,CA2BvC;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAE1E"}
+{"version":3,"file":"python.d.ts","sourceRoot":"","sources":["../../../src/snippets/python.ts"],"names":[],"mappings":"AAAA,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,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;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,EAwHlB,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,KAAG,gBAStE,CAAC;AAEH,eAAO,MAAM,kCAAkC,UAAW,gBAAgB,KAAG,gBAe3E,CAAC;AAEH,eAAO,MAAM,YAAY,UAAW,gBAAgB,KAAG,gBAQrD,CAAC;AAEH,eAAO,MAAM,WAAW,UAAW,gBAAgB,KAAG,gBAQpD,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,gBAW3D,CAAC;AAEH,eAAO,MAAM,cAAc,UAAW,gBAAgB,KAAG,gBAOvD,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,KAAG,gBA+B5D,CAAC;AAEF,eAAO,MAAM,gCAAgC,UAAW,gBAAgB,KAAG,gBAWzE,CAAC;AAEH,eAAO,MAAM,cAAc,EAAE,OAAO,CACnC,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,GAAG,gBAAgB,EAAE,CAC1C,CA8BD,CAAC;AAEF,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,GAAG,gBAAgB,EAAE,CAwBvC;AAED,wBAAgB,yBAAyB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAE1E"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@huggingface/tasks",
   "packageManager": "pnpm@8.10.5",
-  "version": "0.12.26",
+  "version": "0.12.27",
   "description": "List of ML tasks for huggingface.co/tasks",
   "repository": "https://github.com/huggingface/huggingface.js.git",
   "publishConfig": {

package/src/snippets/curl.ts

@@ -26,9 +26,24 @@ export const snippetTextGeneration = (
 	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 messages: ChatCompletionInputMessage[] = opts?.messages ?? [
-			{ role: "user", content: "What is the capital of France?" },
-		];
+		const exampleMessages: ChatCompletionInputMessage[] =
+			model.pipeline_tag === "text-generation"
+				? [{ role: "user", content: "What is the capital of France?" }]
+				: [
+						{
+							role: "user",
+							content: [
+								{
+									type: "image_url",
+									image_url: {
+										url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg",
+									},
+								},
+								{ type: "text", text: "Describe this image in one sentence." },
+							],
+						},
+				  ];
+		const messages = opts?.messages ?? exampleMessages;
 
 		const config = {
 			...(opts?.temperature ? { temperature: opts.temperature } : undefined),
@@ -63,34 +78,6 @@ export const snippetTextGeneration = (
 	}
 };
 
-export const snippetImageTextToTextGeneration = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => {
-	if (model.tags.includes("conversational")) {
-		// Conversational model detected, so we display a code snippet that features the Messages API
-		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' \\
--d '{
-	"model": "${model.id}",
-	"messages": [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"}},
-				{"type": "text", "text": "Describe this image in one sentence."}
-			]
-		}
-	],
-	"max_tokens": 500,
-	"stream": false
-}'
-`,
-		};
-	} else {
-		return snippetBasic(model, accessToken);
-	}
-};
-
 export const snippetZeroShotClassification = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
 	content: `curl https://api-inference.huggingface.co/models/${model.id} \\
 	-X POST \\
@@ -122,7 +109,7 @@ export const curlSnippets: Partial<
 	summarization: snippetBasic,
 	"feature-extraction": snippetBasic,
 	"text-generation": snippetTextGeneration,
-	"image-text-to-text": snippetImageTextToTextGeneration,
+	"image-text-to-text": snippetTextGeneration,
 	"text2text-generation": snippetBasic,
 	"fill-mask": snippetBasic,
 	"sentence-similarity": snippetBasic,

package/src/snippets/js.ts

@@ -40,9 +40,24 @@ export const snippetTextGeneration = (
 	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 messages: ChatCompletionInputMessage[] = opts?.messages ?? [
-			{ role: "user", content: "What is the capital of France?" },
-		];
+		const exampleMessages: ChatCompletionInputMessage[] =
+			model.pipeline_tag === "text-generation"
+				? [{ role: "user", content: "What is the capital of France?" }]
+				: [
+						{
+							role: "user",
+							content: [
+								{
+									type: "image_url",
+									image_url: {
+										url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg",
+									},
+								},
+								{ type: "text", text: "Describe this image in one sentence." },
+							],
+						},
+				  ];
+		const messages = opts?.messages ?? exampleMessages;
 		const messagesStr = stringifyMessages(messages, { sep: ",\n\t\t", start: "[\n\t\t", end: "\n\t]" });
 
 		const config = {
@@ -148,36 +163,6 @@ console.log(chatCompletion.choices[0].message);`,
 	}
 };
 
-export const snippetImageTextToTextGeneration = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => {
-	if (model.tags.includes("conversational")) {
-		// Conversational model detected, so we display a code snippet that features the Messages API
-		return {
-			content: `import { HfInference } from "@huggingface/inference";
-
-const inference = new HfInference("${accessToken || `{API_TOKEN}`}");
-const imageUrl = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg";
-
-for await (const chunk of inference.chatCompletionStream({
-	model: "${model.id}",
-	messages: [
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": imageUrl}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens: 500,
-})) {
-	process.stdout.write(chunk.choices[0]?.delta?.content || "");
-}`,
-		};
-	} else {
-		return snippetBasic(model, accessToken);
-	}
-};
-
 export const snippetZeroShotClassification = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
 	content: `async function query(data) {
 	const response = await fetch(
@@ -307,7 +292,7 @@ export const jsSnippets: Partial<
 	summarization: snippetBasic,
 	"feature-extraction": snippetBasic,
 	"text-generation": snippetTextGeneration,
-	"image-text-to-text": snippetImageTextToTextGeneration,
+	"image-text-to-text": snippetTextGeneration,
 	"text2text-generation": snippetBasic,
 	"fill-mask": snippetBasic,
 	"sentence-similarity": snippetBasic,

package/src/snippets/python.ts

@@ -16,9 +16,24 @@ export const snippetConversational = (
 	}
 ): InferenceSnippet[] => {
 	const streaming = opts?.streaming ?? true;
-	const messages: ChatCompletionInputMessage[] = opts?.messages ?? [
-		{ role: "user", content: "What is the capital of France?" },
-	];
+	const exampleMessages: ChatCompletionInputMessage[] =
+		model.pipeline_tag === "text-generation"
+			? [{ role: "user", content: "What is the capital of France?" }]
+			: [
+					{
+						role: "user",
+						content: [
+							{
+								type: "image_url",
+								image_url: {
+									url: "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg",
+								},
+							},
+							{ type: "text", text: "Describe this image in one sentence." },
+						],
+					},
+			  ];
+	const messages = opts?.messages ?? exampleMessages;
 	const messagesStr = stringifyMessages(messages, {
 		sep: ",\n\t",
 		start: `[\n\t`,
@@ -121,30 +136,6 @@ print(completion.choices[0].message)`,
 	}
 };
 
-export const snippetConversationalWithImage = (model: ModelDataMinimal, accessToken: string): InferenceSnippet => ({
-	content: `from huggingface_hub import InferenceClient
-
-client = InferenceClient(api_key="${accessToken || "{API_TOKEN}"}")
-
-image_url = "https://cdn.britannica.com/61/93061-050-99147DCE/Statue-of-Liberty-Island-New-York-Bay.jpg"
-
-for message in client.chat_completion(
-	model="${model.id}",
-	messages=[
-		{
-			"role": "user",
-			"content": [
-				{"type": "image_url", "image_url": {"url": image_url}},
-				{"type": "text", "text": "Describe this image in one sentence."},
-			],
-		}
-	],
-	max_tokens=500,
-	stream=True,
-):
-	print(message.choices[0].delta.content, end="")`,
-});
-
 export const snippetZeroShotClassification = (model: ModelDataMinimal): InferenceSnippet => ({
 	content: `def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
@@ -282,7 +273,7 @@ export const pythonSnippets: Partial<
 	"feature-extraction": snippetBasic,
 	"text-generation": snippetBasic,
 	"text2text-generation": snippetBasic,
-	"image-text-to-text": snippetConversationalWithImage,
+	"image-text-to-text": snippetConversational,
 	"fill-mask": snippetBasic,
 	"sentence-similarity": snippetBasic,
 	"automatic-speech-recognition": snippetFile,
@@ -306,12 +297,9 @@ export function getPythonInferenceSnippet(
 	accessToken: string,
 	opts?: Record<string, unknown>
 ): InferenceSnippet | InferenceSnippet[] {
-	if (model.pipeline_tag === "text-generation" && model.tags.includes("conversational")) {
+	if (model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return snippetConversational(model, accessToken, opts);
-	} else if (model.pipeline_tag === "image-text-to-text" && model.tags.includes("conversational")) {
-		// Example sending an image to the Message API
-		return snippetConversationalWithImage(model, accessToken);
 	} else {
 		let snippets =
 			model.pipeline_tag && model.pipeline_tag in pythonSnippets