Update tasks specs (automated commit)

Author: Wauplin

packages/tasks/src/tasks/chat-completion/inference.ts

@@ -18,6 +18,10 @@ export interface ChatCompletionInput {
 	 * decreasing the model's likelihood to repeat the same line verbatim.
 	 */
 	frequency_penalty?: number;
+	/**
+	 * A guideline to be used in the chat_template
+	 */
+	guideline?: string;
 	/**
 	 * UNUSED
 	 * Modify the likelihood of specified tokens appearing in the completion. Accepts a JSON
@@ -154,10 +158,20 @@ export interface ChatCompletionInputStreamOptions {
 	[property: string]: unknown;
 }
 
-export type ChatCompletionInputTool = ChatCompletionInputToolType | string;
+/**
+ * Controls which (if any) tool is called by the model.
+ */
+export type ChatCompletionInputTool = ChatCompletionInputToolChoiceEnum | ChatCompletionInputToolChoiceObject;
+
+/**
+ * Means the model can pick between generating a message or calling one or more tools.
+ *
+ * Means the model will not call any tool and instead generates a message.
+ */
+export type ChatCompletionInputToolChoiceEnum = "auto" | "none";
 
-export interface ChatCompletionInputToolType {
-	function?: ChatCompletionInputFunctionName;
+export interface ChatCompletionInputToolChoiceObject {
+	function: ChatCompletionInputFunctionName;
 	[property: string]: unknown;
 }
 

packages/tasks/src/tasks/chat-completion/spec/input.json

@@ -13,6 +13,13 @@
 			"example": "1.0",
 			"nullable": true
 		},
+		"guideline": {
+			"type": "string",
+			"description": "A guideline to be used in the chat_template",
+			"default": "null",
+			"example": "null",
+			"nullable": true
+		},
 		"logit_bias": {
 			"type": "array",
 			"items": {
@@ -283,12 +290,14 @@
 		"ChatCompletionInputToolType": {
 			"oneOf": [
 				{
-					"type": "object",
-					"default": null,
-					"nullable": true
+					"type": "string",
+					"description": "Means the model can pick between generating a message or calling one or more tools.",
+					"enum": ["auto"]
 				},
 				{
-					"type": "string"
+					"type": "string",
+					"description": "Means the model will not call any tool and instead generates a message.",
+					"enum": ["none"]
 				},
 				{
 					"type": "object",
@@ -298,13 +307,10 @@
 							"$ref": "#/$defs/ChatCompletionInputFunctionName"
 						}
 					}
-				},
-				{
-					"type": "object",
-					"default": null,
-					"nullable": true
 				}
 			],
+			"description": "Controls which (if any) tool is called by the model.",
+			"example": "auto",
 			"title": "ChatCompletionInputToolType"
 		},
 		"ChatCompletionInputFunctionName": {

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

@@ -23,7 +23,7 @@ export interface FeatureExtractionInput {
 	 * The name of the prompt that should be used by for encoding. If not set, no prompt
 	 * will be applied.
 	 *
-	 * Must be a key in the `Sentence Transformers` configuration `prompts` dictionary.
+	 * Must be a key in the `sentence-transformers` configuration `prompts` dictionary.
 	 *
 	 * For example if ``prompt_name`` is "query" and the ``prompts`` is {"query": "query: ",
 	 * ...},

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

@@ -17,7 +17,7 @@
 		},
 		"prompt_name": {
 			"type": "string",
-			"description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `Sentence Transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
+			"description": "The name of the prompt that should be used by for encoding. If not set, no prompt\nwill be applied.\n\nMust be a key in the `sentence-transformers` configuration `prompts` dictionary.\n\nFor example if ``prompt_name`` is \"query\" and the ``prompts`` is {\"query\": \"query: \", ...},\nthen the sentence \"What is the capital of France?\" will be encoded as\n\"query: What is the capital of France?\" because the prompt text will be prepended before\nany text to encode.",
 			"default": "null",
 			"example": "null",
 			"nullable": true