Update tasks specs (automated commit)

huggingface · Mar 1, 2025 · 04ebb1c · 04ebb1c
1 parent 822ab9e
commit 04ebb1c
Show file tree

Hide file tree

Showing 4 changed files with 99 additions and 36 deletions.
diff --git a/packages/tasks/src/tasks/chat-completion/inference.ts b/packages/tasks/src/tasks/chat-completion/inference.ts
@@ -105,9 +105,10 @@ export interface ChatCompletionInput {
 	[property: string]: unknown;
 }
 export interface ChatCompletionInputMessage {
-	content: ChatCompletionInputMessageContent;
+	content?: ChatCompletionInputMessageContent;
 	name?: string;
 	role: string;
+	tool_calls?: ChatCompletionInputToolCall[];
 	[property: string]: unknown;
 }
 export type ChatCompletionInputMessageContent = ChatCompletionInputMessageChunk[] | string;
@@ -122,6 +123,18 @@ export interface ChatCompletionInputURL {
 	[property: string]: unknown;
 }
 export type ChatCompletionInputMessageChunkType = "text" | "image_url";
+export interface ChatCompletionInputToolCall {
+	function: ChatCompletionInputFunctionDefinition;
+	id: string;
+	type: string;
+	[property: string]: unknown;
+}
+export interface ChatCompletionInputFunctionDefinition {
+	arguments: unknown;
+	description?: string;
+	name: string;
+	[property: string]: unknown;
+}
 export interface ChatCompletionInputGrammarType {
 	type: ChatCompletionInputGrammarTypeType;
 	/**
@@ -170,12 +183,6 @@ export interface ChatCompletionInputTool {
 	type: string;
 	[property: string]: unknown;
 }
-export interface ChatCompletionInputFunctionDefinition {
-	arguments: unknown;
-	description?: string;
-	name: string;
-	[property: string]: unknown;
-}
 /**
  * Chat Completion Output.
  *
@@ -217,6 +224,7 @@ export interface ChatCompletionOutputTopLogprob {
 export interface ChatCompletionOutputMessage {
 	content?: string;
 	role: string;
+	tool_call_id?: string;
 	tool_calls?: ChatCompletionOutputToolCall[];
 	[property: string]: unknown;
 }
@@ -264,6 +272,7 @@ export interface ChatCompletionStreamOutputChoice {
 export interface ChatCompletionStreamOutputDelta {
 	content?: string;
 	role: string;
+	tool_call_id?: string;
 	tool_calls?: ChatCompletionStreamOutputDeltaToolCall;
 	[property: string]: unknown;
 }

diff --git a/packages/tasks/src/tasks/chat-completion/spec/input.json b/packages/tasks/src/tasks/chat-completion/spec/input.json
@@ -151,24 +151,54 @@
 	},
 	"$defs": {
 		"ChatCompletionInputMessage": {
-			"type": "object",
-			"required": ["role", "content"],
-			"properties": {
-				"content": {
-					"$ref": "#/$defs/ChatCompletionInputMessageContent"
-				},
-				"name": {
-					"type": "string",
-					"example": "\"David\"",
-					"nullable": true
+			"allOf": [
+				{
+					"$ref": "#/$defs/ChatCompletionInputMessageBody"
 				},
-				"role": {
-					"type": "string",
-					"example": "user"
+				{
+					"type": "object",
+					"required": ["role"],
+					"properties": {
+						"name": {
+							"type": "string",
+							"example": "\"David\"",
+							"nullable": true
+						},
+						"role": {
+							"type": "string",
+							"example": "user"
+						}
+					}
 				}
-			},
+			],
 			"title": "ChatCompletionInputMessage"
 		},
+		"ChatCompletionInputMessageBody": {
+			"oneOf": [
+				{
+					"type": "object",
+					"required": ["content"],
+					"properties": {
+						"content": {
+							"$ref": "#/$defs/ChatCompletionInputMessageContent"
+						}
+					}
+				},
+				{
+					"type": "object",
+					"required": ["tool_calls"],
+					"properties": {
+						"tool_calls": {
+							"type": "array",
+							"items": {
+								"$ref": "#/$defs/ChatCompletionInputToolCall"
+							}
+						}
+					}
+				}
+			],
+			"title": "ChatCompletionInputMessageBody"
+		},
 		"ChatCompletionInputMessageContent": {
 			"oneOf": [
 				{
@@ -227,6 +257,37 @@
 			},
 			"title": "ChatCompletionInputUrl"
 		},
+		"ChatCompletionInputToolCall": {
+			"type": "object",
+			"required": ["id", "type", "function"],
+			"properties": {
+				"function": {
+					"$ref": "#/$defs/ChatCompletionInputFunctionDefinition"
+				},
+				"id": {
+					"type": "string"
+				},
+				"type": {
+					"type": "string"
+				}
+			},
+			"title": "ChatCompletionInputToolCall"
+		},
+		"ChatCompletionInputFunctionDefinition": {
+			"type": "object",
+			"required": ["name", "arguments"],
+			"properties": {
+				"arguments": {},
+				"description": {
+					"type": "string",
+					"nullable": true
+				},
+				"name": {
+					"type": "string"
+				}
+			},
+			"title": "ChatCompletionInputFunctionDefinition"
+		},
 		"ChatCompletionInputGrammarType": {
 			"oneOf": [
 				{
@@ -326,21 +387,6 @@
 				}
 			},
 			"title": "ChatCompletionInputTool"
-		},
-		"ChatCompletionInputFunctionDefinition": {
-			"type": "object",
-			"required": ["name", "arguments"],
-			"properties": {
-				"arguments": {},
-				"description": {
-					"type": "string",
-					"nullable": true
-				},
-				"name": {
-					"type": "string"
-				}
-			},
-			"title": "ChatCompletionInputFunctionDefinition"
 		}
 	}
 }
diff --git a/packages/tasks/src/tasks/chat-completion/spec/output.json b/packages/tasks/src/tasks/chat-completion/spec/output.json
@@ -128,6 +128,10 @@
 				"role": {
 					"type": "string",
 					"example": "user"
+				},
+				"tool_call_id": {
+					"type": "string",
+					"nullable": true
 				}
 			},
 			"title": "ChatCompletionOutputTextMessage"

diff --git a/packages/tasks/src/tasks/chat-completion/spec/stream_output.json b/packages/tasks/src/tasks/chat-completion/spec/stream_output.json
@@ -87,6 +87,10 @@
 				"role": {
 					"type": "string",
 					"example": "user"
+				},
+				"tool_call_id": {
+					"type": "string",
+					"nullable": true
 				}
 			},
 			"title": "ChatCompletionStreamOutputTextMessage"