rely on conversational tag

Author: Wauplin

packages/tasks/src/snippets/curl.ts

@@ -10,7 +10,7 @@ export const snippetBasic = (model: ModelDataMinimal, accessToken: string): stri
 	-H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}"`;
 
 export const snippetTextGeneration = (model: ModelDataMinimal, accessToken: string): string => {
-	if (model.config?.tokenizer_config?.chat_template) {
+	if (model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
 -H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}" \\
@@ -28,7 +28,7 @@ export const snippetTextGeneration = (model: ModelDataMinimal, accessToken: stri
 };
 
 export const snippetImageTextToTextGeneration = (model: ModelDataMinimal, accessToken: string): string => {
-	if (model.config?.tokenizer_config?.chat_template) {
+	if (model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
 -H "Authorization: Bearer ${accessToken || `{API_TOKEN}`}" \\

packages/tasks/src/snippets/js.ts

@@ -24,7 +24,7 @@ query({"inputs": ${getModelInputSnippet(model)}}).then((response) => {
 });`;
 
 export const snippetTextGeneration = (model: ModelDataMinimal, accessToken: string): string => {
-	if (model.config?.tokenizer_config?.chat_template) {
+	if (model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return `import { HfInference } from "@huggingface/inference";
 
@@ -43,7 +43,7 @@ for await (const chunk of inference.chatCompletionStream({
 };
 
 export const snippetImageTextToTextGeneration = (model: ModelDataMinimal, accessToken: string): string => {
-	if (model.config?.tokenizer_config?.chat_template) {
+	if (model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return `import { HfInference } from "@huggingface/inference";
 

packages/tasks/src/snippets/python.ts

@@ -177,10 +177,10 @@ export const pythonSnippets: Partial<Record<PipelineType, (model: ModelDataMinim
 };
 
 export function getPythonInferenceSnippet(model: ModelDataMinimal, accessToken: string): string {
-	if (model.pipeline_tag === "text-generation" && model.config?.tokenizer_config?.chat_template) {
+	if (model.pipeline_tag === "text-generation" && model.tags.includes("conversational")) {
 		// Conversational model detected, so we display a code snippet that features the Messages API
 		return snippetConversational(model, accessToken);
-	} else if (model.pipeline_tag === "image-text-to-text" && model.config?.tokenizer_config?.chat_template) {
+	} 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 {

packages/tasks/src/snippets/types.ts

@@ -5,4 +5,7 @@ import type { ModelData } from "../model-data";
  *
  * Add more fields as needed.
  */
-export type ModelDataMinimal = Pick<ModelData, "id" | "pipeline_tag" | "mask_token" | "library_name" | "config">;
+export type ModelDataMinimal = Pick<
+	ModelData,
+	"id" | "pipeline_tag" | "mask_token" | "library_name" | "config" | "tags"
+>;