@huggingface/tasks 0.0.9 → 0.1.0

package/dist/index.d.ts

@@ -23,6 +23,7 @@ declare enum ModelLibrary {
     "pyannote-audio" = "pyannote.audio",
     "sample-factory" = "Sample Factory",
     "sentence-transformers" = "Sentence Transformers",
+    "setfit" = "SetFit",
     "sklearn" = "Scikit-learn",
     "spacy" = "spaCy",
     "span-marker" = "SpanMarker",
@@ -452,6 +453,9 @@ declare const PIPELINE_TYPES_SET: Set<"other" | "text-classification" | "token-c
  */
 declare const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLibraryKey, PipelineType[]>>;
 
+/**
+ * See default-widget-inputs.ts for the default widget inputs, this files only contains the types
+ */
 type TableData = Record<string, (string | number)[]>;
 type WidgetExampleOutputLabels = Array<{
     label: string;
@@ -526,6 +530,7 @@ interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput> ex
 type WidgetExample<TOutput = WidgetExampleOutput> = WidgetExampleTextInput<TOutput> | WidgetExampleTextAndContextInput<TOutput> | WidgetExampleTextAndTableInput<TOutput> | WidgetExampleAssetInput<TOutput> | WidgetExampleAssetAndPromptInput<TOutput> | WidgetExampleAssetAndTextInput<TOutput> | WidgetExampleAssetAndZeroShotInput<TOutput> | WidgetExampleStructuredDataInput<TOutput> | WidgetExampleTableDataInput<TOutput> | WidgetExampleZeroShotTextInput<TOutput> | WidgetExampleSentenceSimilarityInput<TOutput>;
 type KeysOfUnion<T> = T extends unknown ? keyof T : never;
 type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;
+
 declare enum InferenceDisplayability {
     /**
      * Yes

package/dist/index.js

@@ -66,6 +66,7 @@ var LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS = {
   peft: ["text-generation"],
   "pyannote-audio": ["automatic-speech-recognition"],
   "sentence-transformers": ["feature-extraction", "sentence-similarity"],
+  setfit: ["text-classification"],
   sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
   spacy: ["token-classification", "text-classification", "sentence-similarity"],
   "span-marker": ["token-classification"],
@@ -218,8 +219,8 @@ var paddlenlp = (model) => {
       [
         `from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = ${architecture}.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   } else {
@@ -228,8 +229,8 @@ var paddlenlp = (model) => {
         `# \u26A0\uFE0F Type of model unknown`,
         `from paddlenlp.transformers import AutoTokenizer, AutoModel`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = AutoModel.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   }
@@ -365,6 +366,11 @@ var sentenceTransformers = (model) => [
 
 model = SentenceTransformer("${model.id}")`
 ];
+var setfit = (model) => [
+  `from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`
+];
 var spacy = (model) => [
   `!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -637,6 +643,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
     snippets: sentenceTransformers
   },
+  setfit: {
+    btnLabel: "setfit",
+    repoName: "setfit",
+    repoUrl: "https://github.com/huggingface/setfit",
+    docsUrl: "https://huggingface.co/docs/hub/setfit",
+    snippets: setfit
+  },
   sklearn: {
     btnLabel: "Scikit-learn",
     repoName: "Scikit-learn",
@@ -4459,7 +4472,7 @@ var TASKS_MODEL_LIBRARIES = {
   "tabular-classification": ["sklearn"],
   "tabular-regression": ["sklearn"],
   "tabular-to-text": ["transformers"],
-  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+  "text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
   "text-generation": ["transformers", "transformers.js"],
   "text-retrieval": [],
   "text-to-image": ["diffusers"],
@@ -4566,6 +4579,7 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["pyannote-audio"] = "pyannote.audio";
   ModelLibrary2["sample-factory"] = "Sample Factory";
   ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["setfit"] = "SetFit";
   ModelLibrary2["sklearn"] = "Scikit-learn";
   ModelLibrary2["spacy"] = "spaCy";
   ModelLibrary2["span-marker"] = "SpanMarker";

package/dist/index.mjs

@@ -28,6 +28,7 @@ var LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS = {
   peft: ["text-generation"],
   "pyannote-audio": ["automatic-speech-recognition"],
   "sentence-transformers": ["feature-extraction", "sentence-similarity"],
+  setfit: ["text-classification"],
   sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
   spacy: ["token-classification", "text-classification", "sentence-similarity"],
   "span-marker": ["token-classification"],
@@ -180,8 +181,8 @@ var paddlenlp = (model) => {
       [
         `from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = ${architecture}.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   } else {
@@ -190,8 +191,8 @@ var paddlenlp = (model) => {
         `# \u26A0\uFE0F Type of model unknown`,
         `from paddlenlp.transformers import AutoTokenizer, AutoModel`,
         "",
-        `tokenizer = AutoTokenizer.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`,
-        `model = AutoModel.from_pretrained("${model.id}"${model.private ? ", use_auth_token=True" : ""}, from_hf_hub=True)`
+        `tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+        `model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`
       ].join("\n")
     ];
   }
@@ -327,6 +328,11 @@ var sentenceTransformers = (model) => [
 
 model = SentenceTransformer("${model.id}")`
 ];
+var setfit = (model) => [
+  `from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`
+];
 var spacy = (model) => [
   `!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -599,6 +605,13 @@ var MODEL_LIBRARIES_UI_ELEMENTS = {
     docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
     snippets: sentenceTransformers
   },
+  setfit: {
+    btnLabel: "setfit",
+    repoName: "setfit",
+    repoUrl: "https://github.com/huggingface/setfit",
+    docsUrl: "https://huggingface.co/docs/hub/setfit",
+    snippets: setfit
+  },
   sklearn: {
     btnLabel: "Scikit-learn",
     repoName: "Scikit-learn",
@@ -4421,7 +4434,7 @@ var TASKS_MODEL_LIBRARIES = {
   "tabular-classification": ["sklearn"],
   "tabular-regression": ["sklearn"],
   "tabular-to-text": ["transformers"],
-  "text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+  "text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
   "text-generation": ["transformers", "transformers.js"],
   "text-retrieval": [],
   "text-to-image": ["diffusers"],
@@ -4528,6 +4541,7 @@ var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
   ModelLibrary2["pyannote-audio"] = "pyannote.audio";
   ModelLibrary2["sample-factory"] = "Sample Factory";
   ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["setfit"] = "SetFit";
   ModelLibrary2["sklearn"] = "Scikit-learn";
   ModelLibrary2["spacy"] = "spaCy";
   ModelLibrary2["span-marker"] = "SpanMarker";

package/package.json

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

package/src/default-widget-inputs.ts

@@ -1,4 +1,4 @@
-import type { WidgetExample } from "./model-data";
+import type { WidgetExample } from "./widget-example";
 import type { PipelineType } from "./pipelines";
 
 type LanguageCode = string;

package/src/index.ts

@@ -16,9 +16,8 @@ export {
 } from "./pipelines";
 export { ModelLibrary, ALL_DISPLAY_MODEL_LIBRARY_KEYS } from "./model-libraries";
 export type { ModelLibraryKey } from "./model-libraries";
-export {
-	ModelData,
-	TransformersInfo,
+export type { ModelData, TransformersInfo } from "./model-data";
+export type {
 	WidgetExample,
 	WidgetExampleAttribute,
 	WidgetExampleAssetAndPromptInput,
@@ -37,7 +36,7 @@ export {
 	WidgetExampleOutputLabels,
 	WidgetExampleOutputAnswerScore,
 	WidgetExampleOutputText,
-} from "./model-data";
+} from "./widget-example";
 export { InferenceDisplayability } from "./model-data";
 
 export { TAG_NFAA_CONTENT, OTHER_TAGS_SUGGESTIONS, TAG_TEXT_GENERATION_INFERENCE, TAG_CUSTOM_CODE } from "./tags";

package/src/library-to-tasks.ts

@@ -31,6 +31,7 @@ export const LIBRARY_TASK_MAPPING_EXCLUDING_TRANSFORMERS: Partial<Record<ModelLi
 	peft: ["text-generation"],
 	"pyannote-audio": ["automatic-speech-recognition"],
 	"sentence-transformers": ["feature-extraction", "sentence-similarity"],
+	setfit: ["text-classification"],
 	sklearn: ["tabular-classification", "tabular-regression", "text-classification"],
 	spacy: ["token-classification", "text-classification", "sentence-similarity"],
 	"span-marker": ["token-classification"],

package/src/library-ui-elements.ts

@@ -187,12 +187,8 @@ const paddlenlp = (model: ModelData) => {
 			[
 				`from paddlenlp.transformers import AutoTokenizer, ${architecture}`,
 				"",
-				`tokenizer = AutoTokenizer.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
-				`model = ${architecture}.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
+				`tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+				`model = ${architecture}.from_pretrained("${model.id}", from_hf_hub=True)`,
 			].join("\n"),
 		];
 	} else {
@@ -201,12 +197,8 @@ const paddlenlp = (model: ModelData) => {
 				`# ⚠️ Type of model unknown`,
 				`from paddlenlp.transformers import AutoTokenizer, AutoModel`,
 				"",
-				`tokenizer = AutoTokenizer.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
-				`model = AutoModel.from_pretrained("${model.id}"${
-					model.private ? ", use_auth_token=True" : ""
-				}, from_hf_hub=True)`,
+				`tokenizer = AutoTokenizer.from_pretrained("${model.id}", from_hf_hub=True)`,
+				`model = AutoModel.from_pretrained("${model.id}", from_hf_hub=True)`,
 			].join("\n"),
 		];
 	}
@@ -358,6 +350,12 @@ const sentenceTransformers = (model: ModelData) => [
 model = SentenceTransformer("${model.id}")`,
 ];
 
+const setfit = (model: ModelData) => [
+	`from setfit import SetFitModel
+
+model = SetFitModel.from_pretrained("${model.id}")`,
+];
+
 const spacy = (model: ModelData) => [
 	`!pip install https://huggingface.co/${model.id}/resolve/main/${nameWithoutNamespace(model.id)}-any-py3-none-any.whl
 
@@ -661,6 +659,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS: Partial<Record<ModelLibraryKey, Librar
 		docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
 		snippets: sentenceTransformers,
 	},
+	setfit: {
+		btnLabel: "setfit",
+		repoName: "setfit",
+		repoUrl: "https://github.com/huggingface/setfit",
+		docsUrl: "https://huggingface.co/docs/hub/setfit",
+		snippets: setfit,
+	},
 	sklearn: {
 		btnLabel: "Scikit-learn",
 		repoName: "Scikit-learn",

package/src/model-data.ts

@@ -1,119 +1,5 @@
 import type { PipelineType } from "./pipelines";
-
-type TableData = Record<string, (string | number)[]>;
-
-//#region outputs
-export type WidgetExampleOutputLabels = Array<{ label: string; score: number }>;
-export interface WidgetExampleOutputAnswerScore {
-	answer: string;
-	score: number;
-}
-export interface WidgetExampleOutputText {
-	text: string;
-}
-export interface WidgetExampleOutputUrl {
-	url: string;
-}
-
-export type WidgetExampleOutput =
-	| WidgetExampleOutputLabels
-	| WidgetExampleOutputAnswerScore
-	| WidgetExampleOutputText
-	| WidgetExampleOutputUrl;
-//#endregion
-
-export interface WidgetExampleBase<TOutput> {
-	example_title?: string;
-	group?: string;
-	/**
-	 * Potential overrides to API parameters for this specific example
-	 * (takes precedences over the model card metadata's inference.parameters)
-	 */
-	parameters?: {
-		/// token-classification
-		aggregation_strategy?: string;
-		/// text-generation
-		top_k?: number;
-		top_p?: number;
-		temperature?: number;
-		max_new_tokens?: number;
-		do_sample?: boolean;
-		/// text-to-image
-		negative_prompt?: string;
-		guidance_scale?: number;
-		num_inference_steps?: number;
-	};
-	/**
-	 * Optional output
-	 */
-	output?: TOutput;
-}
-
-export interface WidgetExampleTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	text: string;
-}
-
-export interface WidgetExampleTextAndContextInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleTextInput<TOutput> {
-	context: string;
-}
-
-export interface WidgetExampleTextAndTableInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
-	table: TableData;
-}
-
-export interface WidgetExampleAssetInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	src: string;
-}
-export interface WidgetExampleAssetAndPromptInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleAssetInput<TOutput> {
-	prompt: string;
-}
-
-export type WidgetExampleAssetAndTextInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
-	WidgetExampleTextInput<TOutput>;
-
-export type WidgetExampleAssetAndZeroShotInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
-	WidgetExampleZeroShotTextInput<TOutput>;
-
-export interface WidgetExampleStructuredDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	structured_data: TableData;
-}
-
-export interface WidgetExampleTableDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
-	table: TableData;
-}
-
-export interface WidgetExampleZeroShotTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
-	text: string;
-	candidate_labels: string;
-	multi_class: boolean;
-}
-
-export interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput>
-	extends WidgetExampleBase<TOutput> {
-	source_sentence: string;
-	sentences: string[];
-}
-
-//#endregion
-
-export type WidgetExample<TOutput = WidgetExampleOutput> =
-	| WidgetExampleTextInput<TOutput>
-	| WidgetExampleTextAndContextInput<TOutput>
-	| WidgetExampleTextAndTableInput<TOutput>
-	| WidgetExampleAssetInput<TOutput>
-	| WidgetExampleAssetAndPromptInput<TOutput>
-	| WidgetExampleAssetAndTextInput<TOutput>
-	| WidgetExampleAssetAndZeroShotInput<TOutput>
-	| WidgetExampleStructuredDataInput<TOutput>
-	| WidgetExampleTableDataInput<TOutput>
-	| WidgetExampleZeroShotTextInput<TOutput>
-	| WidgetExampleSentenceSimilarityInput<TOutput>;
-
-type KeysOfUnion<T> = T extends unknown ? keyof T : never;
-
-export type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;
+import type { WidgetExample } from "./widget-example";
 
 export enum InferenceDisplayability {
 	/**

package/src/model-libraries.ts

@@ -23,6 +23,7 @@ export enum ModelLibrary {
 	"pyannote-audio" = "pyannote.audio",
 	"sample-factory" = "Sample Factory",
 	"sentence-transformers" = "Sentence Transformers",
+	"setfit" = "SetFit",
 	"sklearn" = "Scikit-learn",
 	"spacy" = "spaCy",
 	"span-marker" = "SpanMarker",

package/src/tasks/index.ts

@@ -69,7 +69,7 @@ export const TASKS_MODEL_LIBRARIES: Record<PipelineType, ModelLibraryKey[]> = {
 	"tabular-classification": ["sklearn"],
 	"tabular-regression": ["sklearn"],
 	"tabular-to-text": ["transformers"],
-	"text-classification": ["adapter-transformers", "spacy", "transformers", "transformers.js"],
+	"text-classification": ["adapter-transformers", "setfit", "spacy", "transformers", "transformers.js"],
 	"text-generation": ["transformers", "transformers.js"],
 	"text-retrieval": [],
 	"text-to-image": ["diffusers"],

package/src/widget-example.ts

@@ -0,0 +1,118 @@
+/**
+ * See default-widget-inputs.ts for the default widget inputs, this files only contains the types
+ */
+
+type TableData = Record<string, (string | number)[]>;
+
+//#region outputs
+export type WidgetExampleOutputLabels = Array<{ label: string; score: number }>;
+export interface WidgetExampleOutputAnswerScore {
+	answer: string;
+	score: number;
+}
+export interface WidgetExampleOutputText {
+	text: string;
+}
+export interface WidgetExampleOutputUrl {
+	url: string;
+}
+
+export type WidgetExampleOutput =
+	| WidgetExampleOutputLabels
+	| WidgetExampleOutputAnswerScore
+	| WidgetExampleOutputText
+	| WidgetExampleOutputUrl;
+//#endregion
+
+export interface WidgetExampleBase<TOutput> {
+	example_title?: string;
+	group?: string;
+	/**
+	 * Potential overrides to API parameters for this specific example
+	 * (takes precedences over the model card metadata's inference.parameters)
+	 */
+	parameters?: {
+		/// token-classification
+		aggregation_strategy?: string;
+		/// text-generation
+		top_k?: number;
+		top_p?: number;
+		temperature?: number;
+		max_new_tokens?: number;
+		do_sample?: boolean;
+		/// text-to-image
+		negative_prompt?: string;
+		guidance_scale?: number;
+		num_inference_steps?: number;
+	};
+	/**
+	 * Optional output
+	 */
+	output?: TOutput;
+}
+
+export interface WidgetExampleTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	text: string;
+}
+
+export interface WidgetExampleTextAndContextInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleTextInput<TOutput> {
+	context: string;
+}
+
+export interface WidgetExampleTextAndTableInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
+	table: TableData;
+}
+
+export interface WidgetExampleAssetInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	src: string;
+}
+export interface WidgetExampleAssetAndPromptInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleAssetInput<TOutput> {
+	prompt: string;
+}
+
+export type WidgetExampleAssetAndTextInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
+	WidgetExampleTextInput<TOutput>;
+
+export type WidgetExampleAssetAndZeroShotInput<TOutput = WidgetExampleOutput> = WidgetExampleAssetInput<TOutput> &
+	WidgetExampleZeroShotTextInput<TOutput>;
+
+export interface WidgetExampleStructuredDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	structured_data: TableData;
+}
+
+export interface WidgetExampleTableDataInput<TOutput = WidgetExampleOutput> extends WidgetExampleBase<TOutput> {
+	table: TableData;
+}
+
+export interface WidgetExampleZeroShotTextInput<TOutput = WidgetExampleOutput> extends WidgetExampleTextInput<TOutput> {
+	text: string;
+	candidate_labels: string;
+	multi_class: boolean;
+}
+
+export interface WidgetExampleSentenceSimilarityInput<TOutput = WidgetExampleOutput>
+	extends WidgetExampleBase<TOutput> {
+	source_sentence: string;
+	sentences: string[];
+}
+
+//#endregion
+
+export type WidgetExample<TOutput = WidgetExampleOutput> =
+	| WidgetExampleTextInput<TOutput>
+	| WidgetExampleTextAndContextInput<TOutput>
+	| WidgetExampleTextAndTableInput<TOutput>
+	| WidgetExampleAssetInput<TOutput>
+	| WidgetExampleAssetAndPromptInput<TOutput>
+	| WidgetExampleAssetAndTextInput<TOutput>
+	| WidgetExampleAssetAndZeroShotInput<TOutput>
+	| WidgetExampleStructuredDataInput<TOutput>
+	| WidgetExampleTableDataInput<TOutput>
+	| WidgetExampleZeroShotTextInput<TOutput>
+	| WidgetExampleSentenceSimilarityInput<TOutput>;
+
+type KeysOfUnion<T> = T extends unknown ? keyof T : never;
+
+export type WidgetExampleAttribute = KeysOfUnion<WidgetExample>;