modelcontextprotocol/java-sdk
Watch
1
0
Fork
You've already forked java-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve MpcSchema structure

Browse source
This commit is contained in:
Christian Tzolov 2024-12-12 14:00:31 +01:00
commit 7f25221d42
1 changed files with 40 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

57
spring-ai-mcp-core/src/main/java/org/springframework/ai/mcp/spec/McpSchema.java
View file

@ -451,16 +451,22 @@ public class McpSchema {
// Sampling and Message Creation // Sampling and Message Creation
@JsonInclude(JsonInclude.Include.NON_ABSENT) @JsonInclude(JsonInclude.Include.NON_ABSENT)
public record CreateMessageRequest(List<SamplingMessage> messages, ModelPreferences modelPreferences, public record CreateMessageRequest(// @formatter:off
String systemPrompt, ContextInclusionStrategy includeContext, Double temperature, int maxTokens, @JsonProperty("messages") List<SamplingMessage> messages,
List<String> stopSequences, Map<String, Object> metadata) implements Request { @JsonProperty("modelPreferences") ModelPreferences modelPreferences,
@JsonProperty("systemPrompt") String systemPrompt,
@JsonProperty("includeContext") ContextInclusionStrategy includeContext,
@JsonProperty("temperature") Double temperature,
@JsonProperty("maxTokens") int maxTokens,
@JsonProperty("stopSequences") List<String> stopSequences,
@JsonProperty("metadata") Map<String, Object> metadata) implements Request {
public enum ContextInclusionStrategy {// @formatter:off public enum ContextInclusionStrategy {
@JsonProperty("none") NONE, @JsonProperty("none") NONE,
@JsonProperty("this_server") THIS_SERVER, @JsonProperty("this_server") THIS_SERVER,
@JsonProperty("all_server") ALL_SERVERS @JsonProperty("all_server") ALL_SERVERS
} // @formatter:on }
} }// @formatter:on
@JsonInclude(JsonInclude.Include.NON_ABSENT) @JsonInclude(JsonInclude.Include.NON_ABSENT)
public record CreateMessageResult(Role role, Content content, String model, StopReason stopReason) { public record CreateMessageResult(Role role, Content content, String model, StopReason stopReason) {
@ -499,11 +505,17 @@ public class McpSchema {
// --------------------------- // ---------------------------
// Progress and Logging // Progress and Logging
// --------------------------- // ---------------------------
public record ProgressNotification(String progressToken, double progress, Double total) { public record ProgressNotification(// @formatter:off
} @JsonProperty("progressToken") String progressToken,
@JsonProperty("progress") double progress,
@JsonProperty("total") Double total) {
}// @formatter:on
public record LoggingMessageNotification(LoggingLevel level, String logger, Object data) { public record LoggingMessageNotification(// @formatter:off
} @JsonProperty("level") LoggingLevel level,
@JsonProperty("logger") String logger,
@JsonProperty("data") Object data) {
}// @formatter:on
// --------------------------- // ---------------------------
// Autocomplete // Autocomplete
@ -515,19 +527,28 @@ public class McpSchema {
} }
public record PromptReference(String type, String name) implements PromptOrResourceReference { public record PromptReference(// @formatter:off
} @JsonProperty("type") String type,
@JsonProperty("name") String name) implements PromptOrResourceReference {
}// @formatter:on
public record ResourceReference(String type, String uri) implements PromptOrResourceReference { public record ResourceReference(// @formatter:off
} @JsonProperty("type") String type,
@JsonProperty("uri") String uri) implements PromptOrResourceReference {
}// @formatter:on
public record CompleteArgument(String name, String value) { public record CompleteArgument(// @formatter:off
} @JsonProperty("name") String name,
@JsonProperty("value") String value) {
}// @formatter:on
} }
public record CompleteResult(CompleteCompletion completion) { public record CompleteResult(CompleteCompletion completion) {
public record CompleteCompletion(List<String> values, Integer total, Boolean hasMore) { public record CompleteCompletion(// @formatter:off
} @JsonProperty("values") List<String> values,
@JsonProperty("total") Integer total,
@JsonProperty("hasMore") Boolean hasMore) {
}// @formatter:on
} }
// --------------------------- // ---------------------------