@huggingface/tasks 0.0.5 → 0.0.6

package/dist/index.d.ts

@@ -40,6 +40,7 @@ declare enum ModelLibrary {
     "mindspore" = "MindSpore"
 }
 type ModelLibraryKey = keyof typeof ModelLibrary;
+declare const ALL_DISPLAY_MODEL_LIBRARY_KEYS: string[];
 
 declare const MODALITIES: readonly ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
 type Modality = (typeof MODALITIES)[number];
@@ -410,6 +411,8 @@ declare const PIPELINE_DATA: {
 };
 type PipelineType = keyof typeof PIPELINE_DATA;
 declare const PIPELINE_TYPES: ("other" | "text-classification" | "token-classification" | "table-question-answering" | "question-answering" | "zero-shot-classification" | "translation" | "summarization" | "conversational" | "feature-extraction" | "text-generation" | "text2text-generation" | "fill-mask" | "sentence-similarity" | "text-to-speech" | "text-to-audio" | "automatic-speech-recognition" | "audio-to-audio" | "audio-classification" | "voice-activity-detection" | "depth-estimation" | "image-classification" | "object-detection" | "image-segmentation" | "text-to-image" | "image-to-text" | "image-to-image" | "unconditional-image-generation" | "video-classification" | "reinforcement-learning" | "robotics" | "tabular-classification" | "tabular-regression" | "tabular-to-text" | "table-to-text" | "multiple-choice" | "text-retrieval" | "time-series-forecasting" | "text-to-video" | "visual-question-answering" | "document-question-answering" | "zero-shot-image-classification" | "graph-ml")[];
+declare const SUBTASK_TYPES: string[];
+declare const PIPELINE_TYPES_SET: Set<"other" | "text-classification" | "token-classification" | "table-question-answering" | "question-answering" | "zero-shot-classification" | "translation" | "summarization" | "conversational" | "feature-extraction" | "text-generation" | "text2text-generation" | "fill-mask" | "sentence-similarity" | "text-to-speech" | "text-to-audio" | "automatic-speech-recognition" | "audio-to-audio" | "audio-classification" | "voice-activity-detection" | "depth-estimation" | "image-classification" | "object-detection" | "image-segmentation" | "text-to-image" | "image-to-text" | "image-to-image" | "unconditional-image-generation" | "video-classification" | "reinforcement-learning" | "robotics" | "tabular-classification" | "tabular-regression" | "tabular-to-text" | "table-to-text" | "multiple-choice" | "text-retrieval" | "time-series-forecasting" | "text-to-video" | "visual-question-answering" | "document-question-answering" | "zero-shot-image-classification" | "graph-ml">;
 
 interface ExampleRepo {
     description: string;
@@ -464,4 +467,9 @@ interface TaskData {
 
 declare const TASKS_DATA: Record<PipelineType, TaskData | undefined>;
 
-export { ExampleRepo, MODALITIES, MODALITY_LABELS, Modality, ModelLibrary, ModelLibraryKey, PIPELINE_DATA, PIPELINE_TYPES, PipelineData, PipelineType, TASKS_DATA, TaskData, TaskDemo, TaskDemoEntry };
+declare const TAG_NFAA_CONTENT = "not-for-all-audiences";
+declare const OTHER_TAGS_SUGGESTIONS: string[];
+declare const TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
+declare const TAG_CUSTOM_CODE = "custom_code";
+
+export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ExampleRepo, MODALITIES, MODALITY_LABELS, Modality, ModelLibrary, ModelLibraryKey, OTHER_TAGS_SUGGESTIONS, PIPELINE_DATA, PIPELINE_TYPES, PIPELINE_TYPES_SET, PipelineData, PipelineType, SUBTASK_TYPES, TAG_CUSTOM_CODE, TAG_NFAA_CONTENT, TAG_TEXT_GENERATION_INFERENCE, TASKS_DATA, TaskData, TaskDemo, TaskDemoEntry };

package/dist/index.js

@@ -20,11 +20,18 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
+  ALL_DISPLAY_MODEL_LIBRARY_KEYS: () => ALL_DISPLAY_MODEL_LIBRARY_KEYS,
   MODALITIES: () => MODALITIES,
   MODALITY_LABELS: () => MODALITY_LABELS,
   ModelLibrary: () => ModelLibrary,
+  OTHER_TAGS_SUGGESTIONS: () => OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA: () => PIPELINE_DATA,
   PIPELINE_TYPES: () => PIPELINE_TYPES,
+  PIPELINE_TYPES_SET: () => PIPELINE_TYPES_SET,
+  SUBTASK_TYPES: () => SUBTASK_TYPES,
+  TAG_CUSTOM_CODE: () => TAG_CUSTOM_CODE,
+  TAG_NFAA_CONTENT: () => TAG_NFAA_CONTENT,
+  TAG_TEXT_GENERATION_INFERENCE: () => TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA: () => TASKS_DATA
 });
 module.exports = __toCommonJS(src_exports);
@@ -586,6 +593,8 @@ var PIPELINE_DATA = {
   }
 };
 var PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
+var SUBTASK_TYPES = Object.values(PIPELINE_DATA).flatMap((data) => "subtasks" in data ? data.subtasks : []).map((s) => s.type);
+var PIPELINE_TYPES_SET = new Set(PIPELINE_TYPES);
 
 // src/audio-classification/data.ts
 var taskData = {
@@ -3144,12 +3153,39 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["mindspore"] = "MindSpore";
   return ModelLibrary2;
 })(ModelLibrary || {});
+var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+  (k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
+);
+
+// src/tags.ts
+var TAG_NFAA_CONTENT = "not-for-all-audiences";
+var OTHER_TAGS_SUGGESTIONS = [
+  "chemistry",
+  "biology",
+  "finance",
+  "legal",
+  "music",
+  "art",
+  "code",
+  "climate",
+  "medical",
+  TAG_NFAA_CONTENT
+];
+var TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
+var TAG_CUSTOM_CODE = "custom_code";
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ALL_DISPLAY_MODEL_LIBRARY_KEYS,
   MODALITIES,
   MODALITY_LABELS,
   ModelLibrary,
+  OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA,
   PIPELINE_TYPES,
+  PIPELINE_TYPES_SET,
+  SUBTASK_TYPES,
+  TAG_CUSTOM_CODE,
+  TAG_NFAA_CONTENT,
+  TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA
 });

package/dist/index.mjs

@@ -555,6 +555,8 @@ var PIPELINE_DATA = {
   }
 };
 var PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
+var SUBTASK_TYPES = Object.values(PIPELINE_DATA).flatMap((data) => "subtasks" in data ? data.subtasks : []).map((s) => s.type);
+var PIPELINE_TYPES_SET = new Set(PIPELINE_TYPES);
 
 // src/audio-classification/data.ts
 var taskData = {
@@ -3113,11 +3115,38 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["mindspore"] = "MindSpore";
   return ModelLibrary2;
 })(ModelLibrary || {});
+var ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+  (k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
+);
+
+// src/tags.ts
+var TAG_NFAA_CONTENT = "not-for-all-audiences";
+var OTHER_TAGS_SUGGESTIONS = [
+  "chemistry",
+  "biology",
+  "finance",
+  "legal",
+  "music",
+  "art",
+  "code",
+  "climate",
+  "medical",
+  TAG_NFAA_CONTENT
+];
+var TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
+var TAG_CUSTOM_CODE = "custom_code";
 export {
+  ALL_DISPLAY_MODEL_LIBRARY_KEYS,
   MODALITIES,
   MODALITY_LABELS,
   ModelLibrary,
+  OTHER_TAGS_SUGGESTIONS,
   PIPELINE_DATA,
   PIPELINE_TYPES,
+  PIPELINE_TYPES_SET,
+  SUBTASK_TYPES,
+  TAG_CUSTOM_CODE,
+  TAG_NFAA_CONTENT,
+  TAG_TEXT_GENERATION_INFERENCE,
   TASKS_DATA
 };

package/package.json

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

package/src/index.ts

@@ -8,6 +8,10 @@ export {
 	type Modality,
 	MODALITIES,
 	MODALITY_LABELS,
+	SUBTASK_TYPES,
+	PIPELINE_TYPES_SET,
 } from "./pipelines";
-export { ModelLibrary } from "./modelLibraries";
+export { ModelLibrary, ALL_DISPLAY_MODEL_LIBRARY_KEYS } from "./modelLibraries";
 export type { ModelLibraryKey } from "./modelLibraries";
+
+export { TAG_NFAA_CONTENT, OTHER_TAGS_SUGGESTIONS, TAG_TEXT_GENERATION_INFERENCE, TAG_CUSTOM_CODE } from "./tags";

package/src/modelLibraries.ts

@@ -41,3 +41,7 @@ export enum ModelLibrary {
 }
 
 export type ModelLibraryKey = keyof typeof ModelLibrary;
+
+export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.keys(ModelLibrary).filter(
+	(k) => !["doctr", "k2", "mindspore", "tensorflowtts"].includes(k)
+);

package/src/pipelines.ts

@@ -616,4 +616,11 @@ export const PIPELINE_DATA = {
 } satisfies Record<string, PipelineData>;
 
 export type PipelineType = keyof typeof PIPELINE_DATA;
+
 export const PIPELINE_TYPES = Object.keys(PIPELINE_DATA) as PipelineType[];
+
+export const SUBTASK_TYPES = Object.values(PIPELINE_DATA)
+	.flatMap((data) => ("subtasks" in data ? data.subtasks : []))
+	.map((s) => s.type);
+
+export const PIPELINE_TYPES_SET = new Set(PIPELINE_TYPES);

package/src/tags.ts

@@ -0,0 +1,15 @@
+export const TAG_NFAA_CONTENT = "not-for-all-audiences";
+export const OTHER_TAGS_SUGGESTIONS = [
+	"chemistry",
+	"biology",
+	"finance",
+	"legal",
+	"music",
+	"art",
+	"code",
+	"climate",
+	"medical",
+	TAG_NFAA_CONTENT,
+];
+export const TAG_TEXT_GENERATION_INFERENCE = "text-generation-inference";
+export const TAG_CUSTOM_CODE = "custom_code";