Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -91,11 +91,11 @@ public class McpAsyncServer {

private static final Logger logger = LoggerFactory.getLogger(McpAsyncServer.class);

private final McpServerTransportProviderBase mcpTransportProvider;
protected final McpServerTransportProviderBase mcpTransportProvider;

private final McpJsonMapper jsonMapper;
protected final McpJsonMapper jsonMapper;

private final JsonSchemaValidator jsonSchemaValidator;
protected final JsonSchemaValidator jsonSchemaValidator;

private final McpSchema.ServerCapabilities serverCapabilities;

Expand Down Expand Up @@ -128,7 +128,7 @@ public class McpAsyncServer {
* @param features The MCP server supported features.
* @param jsonMapper The JsonMapper to use for JSON serialization/deserialization
*/
McpAsyncServer(McpServerTransportProvider mcpTransportProvider, McpJsonMapper jsonMapper,
public McpAsyncServer(McpServerTransportProvider mcpTransportProvider, McpJsonMapper jsonMapper,
McpServerFeatures.Async features, Duration requestTimeout,
McpUriTemplateManagerFactory uriTemplateManagerFactory, JsonSchemaValidator jsonSchemaValidator) {
this.mcpTransportProvider = mcpTransportProvider;
Expand All @@ -153,7 +153,7 @@ public class McpAsyncServer {
requestTimeout, transport, this::asyncInitializeRequestHandler, requestHandlers, notificationHandlers));
}

McpAsyncServer(McpStreamableServerTransportProvider mcpTransportProvider, McpJsonMapper jsonMapper,
public McpAsyncServer(McpStreamableServerTransportProvider mcpTransportProvider, McpJsonMapper jsonMapper,
McpServerFeatures.Async features, Duration requestTimeout,
McpUriTemplateManagerFactory uriTemplateManagerFactory, JsonSchemaValidator jsonSchemaValidator) {
this.mcpTransportProvider = mcpTransportProvider;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ public class McpServerFeatures {
* roots list changes
* @param instructions The server instructions text
*/
record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpServerFeatures.AsyncToolSpecification> tools, Map<String, AsyncResourceSpecification> resources,
Map<String, McpServerFeatures.AsyncResourceTemplateSpecification> resourceTemplates,
Map<String, McpServerFeatures.AsyncPromptSpecification> prompts,
Expand All @@ -59,7 +59,7 @@ record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities s
* the roots list changes
* @param instructions The server instructions text
*/
Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpServerFeatures.AsyncToolSpecification> tools, Map<String, AsyncResourceSpecification> resources,
Map<String, McpServerFeatures.AsyncResourceTemplateSpecification> resourceTemplates,
Map<String, McpServerFeatures.AsyncPromptSpecification> prompts,
Expand Down Expand Up @@ -101,7 +101,7 @@ record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities s
* @return a specification which is protected from blocking calls specified by the
* user.
*/
static Async fromSync(Sync syncSpec, boolean immediateExecution) {
public static Async fromSync(Sync syncSpec, boolean immediateExecution) {
List<McpServerFeatures.AsyncToolSpecification> tools = new ArrayList<>();
for (var tool : syncSpec.tools()) {
tools.add(AsyncToolSpecification.fromSync(tool, immediateExecution));
Expand Down Expand Up @@ -153,7 +153,7 @@ static Async fromSync(Sync syncSpec, boolean immediateExecution) {
* roots list changes
* @param instructions The server instructions text
*/
record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpServerFeatures.SyncToolSpecification> tools,
Map<String, McpServerFeatures.SyncResourceSpecification> resources,
Map<String, McpServerFeatures.SyncResourceTemplateSpecification> resourceTemplates,
Expand All @@ -173,7 +173,7 @@ record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities se
* the roots list changes
* @param instructions The server instructions text
*/
Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpServerFeatures.SyncToolSpecification> tools,
Map<String, McpServerFeatures.SyncResourceSpecification> resources,
Map<String, McpServerFeatures.SyncResourceTemplateSpecification> resourceTemplates,
Expand Down Expand Up @@ -354,7 +354,8 @@ public static Builder builder() {
public record AsyncResourceSpecification(McpSchema.Resource resource,
BiFunction<McpAsyncServerExchange, McpSchema.ReadResourceRequest, Mono<McpSchema.ReadResourceResult>> readHandler) {

static AsyncResourceSpecification fromSync(SyncResourceSpecification resource, boolean immediateExecution) {
public static AsyncResourceSpecification fromSync(SyncResourceSpecification resource,
boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (resource == null) {
return null;
Expand Down Expand Up @@ -394,7 +395,7 @@ static AsyncResourceSpecification fromSync(SyncResourceSpecification resource, b
public record AsyncResourceTemplateSpecification(McpSchema.ResourceTemplate resourceTemplate,
BiFunction<McpAsyncServerExchange, McpSchema.ReadResourceRequest, Mono<McpSchema.ReadResourceResult>> readHandler) {

static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecification resource,
public static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecification resource,
boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (resource == null) {
Expand Down Expand Up @@ -442,7 +443,7 @@ static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecifica
public record AsyncPromptSpecification(McpSchema.Prompt prompt,
BiFunction<McpAsyncServerExchange, McpSchema.GetPromptRequest, Mono<McpSchema.GetPromptResult>> promptHandler) {

static AsyncPromptSpecification fromSync(SyncPromptSpecification prompt, boolean immediateExecution) {
public static AsyncPromptSpecification fromSync(SyncPromptSpecification prompt, boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (prompt == null) {
return null;
Expand Down Expand Up @@ -482,7 +483,7 @@ public record AsyncCompletionSpecification(McpSchema.CompleteReference reference
* @return an asynchronous wrapper of the provided sync specification, or
* {@code null} if input is null
*/
static AsyncCompletionSpecification fromSync(SyncCompletionSpecification completion,
public static AsyncCompletionSpecification fromSync(SyncCompletionSpecification completion,
boolean immediateExecution) {
if (completion == null) {
return null;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,9 +51,9 @@ public class McpStatelessAsyncServer {

private static final Logger logger = LoggerFactory.getLogger(McpStatelessAsyncServer.class);

private final McpStatelessServerTransport mcpTransportProvider;
protected final McpStatelessServerTransport mcpTransportProvider;

private final McpJsonMapper jsonMapper;
protected final McpJsonMapper jsonMapper;

private final McpSchema.ServerCapabilities serverCapabilities;

Expand All @@ -77,7 +77,7 @@ public class McpStatelessAsyncServer {

private final JsonSchemaValidator jsonSchemaValidator;

McpStatelessAsyncServer(McpStatelessServerTransport mcpTransport, McpJsonMapper jsonMapper,
public McpStatelessAsyncServer(McpStatelessServerTransport mcpTransport, McpJsonMapper jsonMapper,
McpStatelessServerFeatures.Async features, Duration requestTimeout,
McpUriTemplateManagerFactory uriTemplateManagerFactory, JsonSchemaValidator jsonSchemaValidator) {
this.mcpTransportProvider = mcpTransport;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ public class McpStatelessServerFeatures {
* @param prompts The map of prompt specifications
* @param instructions The server instructions text
*/
record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpStatelessServerFeatures.AsyncToolSpecification> tools,
Map<String, AsyncResourceSpecification> resources,
Map<String, McpStatelessServerFeatures.AsyncResourceTemplateSpecification> resourceTemplates,
Expand All @@ -56,7 +56,7 @@ record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities s
* @param prompts The map of prompt specifications
* @param instructions The server instructions text
*/
Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpStatelessServerFeatures.AsyncToolSpecification> tools,
Map<String, AsyncResourceSpecification> resources,
Map<String, McpStatelessServerFeatures.AsyncResourceTemplateSpecification> resourceTemplates,
Expand Down Expand Up @@ -94,7 +94,7 @@ record Async(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities s
* @return a specification which is protected from blocking calls specified by the
* user.
*/
static Async fromSync(Sync syncSpec, boolean immediateExecution) {
public static Async fromSync(Sync syncSpec, boolean immediateExecution) {
List<McpStatelessServerFeatures.AsyncToolSpecification> tools = new ArrayList<>();
for (var tool : syncSpec.tools()) {
tools.add(AsyncToolSpecification.fromSync(tool, immediateExecution));
Expand Down Expand Up @@ -136,7 +136,7 @@ static Async fromSync(Sync syncSpec, boolean immediateExecution) {
* @param prompts The map of prompt specifications
* @param instructions The server instructions text
*/
record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpStatelessServerFeatures.SyncToolSpecification> tools,
Map<String, McpStatelessServerFeatures.SyncResourceSpecification> resources,
Map<String, McpStatelessServerFeatures.SyncResourceTemplateSpecification> resourceTemplates,
Expand All @@ -154,7 +154,7 @@ record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities se
* @param prompts The map of prompt specifications
* @param instructions The server instructions text
*/
Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
public Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities serverCapabilities,
List<McpStatelessServerFeatures.SyncToolSpecification> tools,
Map<String, McpStatelessServerFeatures.SyncResourceSpecification> resources,
Map<String, McpStatelessServerFeatures.SyncResourceTemplateSpecification> resourceTemplates,
Expand Down Expand Up @@ -199,11 +199,11 @@ record Sync(McpSchema.Implementation serverInfo, McpSchema.ServerCapabilities se
public record AsyncToolSpecification(McpSchema.Tool tool,
BiFunction<McpTransportContext, CallToolRequest, Mono<McpSchema.CallToolResult>> callHandler) {

static AsyncToolSpecification fromSync(SyncToolSpecification syncToolSpec) {
public static AsyncToolSpecification fromSync(SyncToolSpecification syncToolSpec) {
return fromSync(syncToolSpec, false);
}

static AsyncToolSpecification fromSync(SyncToolSpecification syncToolSpec, boolean immediate) {
public static AsyncToolSpecification fromSync(SyncToolSpecification syncToolSpec, boolean immediate) {

// FIXME: This is temporary, proper validation should be implemented
if (syncToolSpec == null) {
Expand Down Expand Up @@ -291,7 +291,8 @@ public static Builder builder() {
public record AsyncResourceSpecification(McpSchema.Resource resource,
BiFunction<McpTransportContext, McpSchema.ReadResourceRequest, Mono<McpSchema.ReadResourceResult>> readHandler) {

static AsyncResourceSpecification fromSync(SyncResourceSpecification resource, boolean immediateExecution) {
public static AsyncResourceSpecification fromSync(SyncResourceSpecification resource,
boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (resource == null) {
return null;
Expand Down Expand Up @@ -330,7 +331,7 @@ static AsyncResourceSpecification fromSync(SyncResourceSpecification resource, b
public record AsyncResourceTemplateSpecification(McpSchema.ResourceTemplate resourceTemplate,
BiFunction<McpTransportContext, McpSchema.ReadResourceRequest, Mono<McpSchema.ReadResourceResult>> readHandler) {

static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecification resource,
public static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecification resource,
boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (resource == null) {
Expand Down Expand Up @@ -362,7 +363,7 @@ static AsyncResourceTemplateSpecification fromSync(SyncResourceTemplateSpecifica
public record AsyncPromptSpecification(McpSchema.Prompt prompt,
BiFunction<McpTransportContext, McpSchema.GetPromptRequest, Mono<McpSchema.GetPromptResult>> promptHandler) {

static AsyncPromptSpecification fromSync(SyncPromptSpecification prompt, boolean immediateExecution) {
public static AsyncPromptSpecification fromSync(SyncPromptSpecification prompt, boolean immediateExecution) {
// FIXME: This is temporary, proper validation should be implemented
if (prompt == null) {
return null;
Expand Down Expand Up @@ -400,7 +401,7 @@ public record AsyncCompletionSpecification(McpSchema.CompleteReference reference
* @return an asynchronous wrapper of the provided sync specification, or
* {@code null} if input is null
*/
static AsyncCompletionSpecification fromSync(SyncCompletionSpecification completion,
public static AsyncCompletionSpecification fromSync(SyncCompletionSpecification completion,
boolean immediateExecution) {
if (completion == null) {
return null;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ public class McpStatelessSyncServer {

private final boolean immediateExecution;

McpStatelessSyncServer(McpStatelessAsyncServer asyncServer, boolean immediateExecution) {
public McpStatelessSyncServer(McpStatelessAsyncServer asyncServer, boolean immediateExecution) {
this.asyncServer = asyncServer;
this.immediateExecution = immediateExecution;
}
Expand Down
Loading