client task

Signed-off-by: He-Pin <hepin1989@gmail.com>

Author: He-Pin

mcp-core/src/main/java/io/modelcontextprotocol/spec/McpSchema.java

@@ -172,8 +172,24 @@ public interface Meta {
 
 	}
 
-	public sealed interface Request extends Meta
-			permits InitializeRequest, CallToolRequest, CreateMessageRequest, ElicitRequest, CompleteRequest,
+	public sealed interface TaskArgumentedRequest extends Request
+			permits CallToolRequest, CreateMessageRequest, ElicitRequest {
+
+		/**
+		 * If specified, the caller is requesting task-augmented execution for this
+		 * request. The request will return a CreateTaskResult immediately, and the actual
+		 * result can be retrieved later via tasks/result.
+		 * <p>
+		 * Task augmentation is subject to capability negotiation - receivers MUST declare
+		 * support for task augmentation of specific request types in their capabilities.
+		 */
+		default TaskMetaData task() {
+			return null;
+		}
+
+	}
+
+	public sealed interface Request extends Meta permits InitializeRequest, TaskArgumentedRequest, CompleteRequest,
 			GetPromptRequest, ReadResourceRequest, SubscribeRequest, UnsubscribeRequest, PaginatedRequest {
 
 		default Object progressToken() {
@@ -1769,7 +1785,7 @@ public record CallToolRequest( // @formatter:off
 		@JsonProperty("name") String name,
 		@JsonProperty("arguments") Map<String, Object> arguments,
         @JsonProperty("task") TaskMetaData task,
-		@JsonProperty("_meta") Map<String, Object> meta) implements Request { // @formatter:on
+		@JsonProperty("_meta") Map<String, Object> meta) implements TaskArgumentedRequest { // @formatter:on
 
 		public CallToolRequest(McpJsonMapper jsonMapper, String name, String jsonArguments) {
 			this(name, parseJsonArguments(jsonMapper, jsonArguments), null, null);
@@ -2198,6 +2214,7 @@ public record SamplingMessage( // @formatter:off
 	 * @param maxTokens The maximum number of tokens to sample, as requested by the
 	 * server. The client MAY choose to sample fewer tokens than requested
 	 * @param stopSequences Optional stop sequences for sampling
+	 * @param task Optional task metadata
 	 * @param metadata Optional metadata to pass through to the LLM provider. The format
 	 * of this metadata is provider-specific
 	 * @param meta See specification for notes on _meta usage
@@ -2212,17 +2229,26 @@ public record CreateMessageRequest( // @formatter:off
 		@JsonProperty("temperature") Double temperature,
 		@JsonProperty("maxTokens") Integer maxTokens,
 		@JsonProperty("stopSequences") List<String> stopSequences,
-		@JsonProperty("metadata") Map<String, Object> metadata,
-		@JsonProperty("_meta") Map<String, Object> meta) implements Request { // @formatter:on
+        @JsonProperty("task") TaskMetaData task,
+        @JsonProperty("metadata") Map<String, Object> metadata,
+		@JsonProperty("_meta") Map<String, Object> meta) implements TaskArgumentedRequest { // @formatter:on
 
 		// backwards compatibility constructor
 		public CreateMessageRequest(List<SamplingMessage> messages, ModelPreferences modelPreferences,
 				String systemPrompt, ContextInclusionStrategy includeContext, Double temperature, Integer maxTokens,
 				List<String> stopSequences, Map<String, Object> metadata) {
-			this(messages, modelPreferences, systemPrompt, includeContext, temperature, maxTokens, stopSequences,
+			this(messages, modelPreferences, systemPrompt, includeContext, temperature, maxTokens, stopSequences, null,
 					metadata, null);
 		}
 
+		// backwards compatibility constructor
+		public CreateMessageRequest(List<SamplingMessage> messages, ModelPreferences modelPreferences,
+				String systemPrompt, ContextInclusionStrategy includeContext, Double temperature, Integer maxTokens,
+				List<String> stopSequences, Map<String, Object> metadata, Map<String, Object> meta) {
+			this(messages, modelPreferences, systemPrompt, includeContext, temperature, maxTokens, stopSequences, null,
+					metadata, meta);
+		}
+
 		public enum ContextInclusionStrategy {
 
 			// @formatter:off
@@ -2251,6 +2277,8 @@ public static class Builder {
 
 			private List<String> stopSequences;
 
+			private TaskMetaData task;
+
 			private Map<String, Object> metadata;
 
 			private Map<String, Object> meta;
@@ -2290,6 +2318,11 @@ public Builder stopSequences(List<String> stopSequences) {
 				return this;
 			}
 
+			public Builder task(TaskMetaData task) {
+				this.task = task;
+				return this;
+			}
+
 			public Builder metadata(Map<String, Object> metadata) {
 				this.metadata = metadata;
 				return this;
@@ -2310,7 +2343,7 @@ public Builder progressToken(Object progressToken) {
 
 			public CreateMessageRequest build() {
 				return new CreateMessageRequest(messages, modelPreferences, systemPrompt, includeContext, temperature,
-						maxTokens, stopSequences, metadata, meta);
+						maxTokens, stopSequences, task, metadata, meta);
 			}
 
 		}
@@ -2435,13 +2468,19 @@ public CreateMessageResult build() {
 	public record ElicitRequest( // @formatter:off
 		@JsonProperty("message") String message,
 		@JsonProperty("requestedSchema") Map<String, Object> requestedSchema,
-		@JsonProperty("_meta") Map<String, Object> meta) implements Request { // @formatter:on
+        @JsonProperty("task") TaskMetaData task,
+        @JsonProperty("_meta") Map<String, Object> meta) implements TaskArgumentedRequest { // @formatter:on
 
 		// backwards compatibility constructor
 		public ElicitRequest(String message, Map<String, Object> requestedSchema) {
 			this(message, requestedSchema, null);
 		}
 
+		// backwards compatibility constructor
+		public ElicitRequest(String message, Map<String, Object> requestedSchema, Map<String, Object> meta) {
+			this(message, requestedSchema, null, meta);
+		}
+
 		public static Builder builder() {
 			return new Builder();
 		}
@@ -2452,6 +2491,8 @@ public static class Builder {
 
 			private Map<String, Object> requestedSchema;
 
+			private TaskMetaData task;
+
 			private Map<String, Object> meta;
 
 			public Builder message(String message) {
@@ -2464,6 +2505,11 @@ public Builder requestedSchema(Map<String, Object> requestedSchema) {
 				return this;
 			}
 
+			public Builder task(TaskMetaData task) {
+				this.task = task;
+				return this;
+			}
+
 			public Builder meta(Map<String, Object> meta) {
 				this.meta = meta;
 				return this;
@@ -2478,7 +2524,7 @@ public Builder progressToken(Object progressToken) {
 			}
 
 			public ElicitRequest build() {
-				return new ElicitRequest(message, requestedSchema, meta);
+				return new ElicitRequest(message, requestedSchema, task, meta);
 			}
 
 		}