Update tasks specs (automated commit)

Author: Wauplin

packages/tasks/src/tasks/feature-extraction/inference.ts

@@ -9,9 +9,14 @@ export type FeatureExtractionOutput = Array<number[]>;
  *
  * Auto-generated from TEI specs.
  * For more details, check out
- * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
+ * https://github.com/huggingface/huggingface.js/blob/main/packages/tasks-gen/scripts/inference-tei-import.ts.
  */
 export interface FeatureExtractionInput {
+	/**
+	 * The number of dimensions that the output embeddings should have. If not set, the original
+	 * shape of the representation will be returned instead.
+	 */
+	dimensions?: number;
 	/**
 	 * The text or list of texts to embed.
 	 */
@@ -38,4 +43,4 @@ export interface FeatureExtractionInput {
  * The text or list of texts to embed.
  */
 export type FeatureExtractionInputs = string[] | string;
-export type FeatureExtractionInputTruncationDirection = "left" | "right";
+export type FeatureExtractionInputTruncationDirection = "Left" | "Right";

packages/tasks/src/tasks/feature-extraction/spec/input.json

@@ -1,11 +1,19 @@
 {
 	"$id": "/inference/schemas/feature-extraction/input.json",
 	"$schema": "http://json-schema.org/draft-06/schema#",
-	"description": "Feature Extraction Input.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.",
+	"description": "Feature Extraction Input.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks-gen/scripts/inference-tei-import.ts.",
 	"title": "FeatureExtractionInput",
 	"type": "object",
 	"required": ["inputs"],
 	"properties": {
+		"dimensions": {
+			"type": "integer",
+			"description": "The number of dimensions that the output embeddings should have. If not set, the original\nshape of the representation will be returned instead.",
+			"default": "null",
+			"example": "null",
+			"nullable": true,
+			"minimum": 0
+		},
 		"inputs": {
 			"title": "FeatureExtractionInputs",
 			"description": "The text or list of texts to embed.",
@@ -51,7 +59,7 @@
 	"$defs": {
 		"FeatureExtractionInputTruncationDirection": {
 			"type": "string",
-			"enum": ["left", "right"],
+			"enum": ["Left", "Right"],
 			"title": "FeatureExtractionInputTruncationDirection"
 		}
 	}

packages/tasks/src/tasks/feature-extraction/spec/output.json

@@ -1,7 +1,7 @@
 {
 	"$id": "/inference/schemas/feature-extraction/output.json",
 	"$schema": "http://json-schema.org/draft-06/schema#",
-	"description": "Feature Extraction Output.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.",
+	"description": "Feature Extraction Output.\n\nAuto-generated from TEI specs.\nFor more details, check out https://github.com/huggingface/huggingface.js/blob/main/packages/tasks-gen/scripts/inference-tei-import.ts.",
 	"title": "FeatureExtractionOutput",
 	"type": "array",
 	"$defs": {},