Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix checkstyle

Browse Source 
Signed-off-by: Christian Tzolov <christian.tzolov@broadcom.com>
This commit is contained in:
Christian Tzolov
2025-05-16 15:14:52 +02:00
committed by Mark Pollack
parent 327cf40e97
commit 4bde57790d
2 changed files with 88 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

90
auto-configurations/mcp/spring-ai-autoconfigure-mcp-server/src/main/java/org/springframework/ai/mcp/server/autoconfigure/McpServerProperties.java
View File

@@ -134,52 +134,8 @@ public class McpServerProperties {
private Capabilities capabilities = new Capabilities();
public static class Capabilities {
private boolean resource = true;
private boolean tool = true;
private boolean prompt = true;
private boolean completion = true;
public boolean isResource() {
return resource;
}
public void setResource(boolean resource) {
this.resource = resource;
}
public boolean isTool() {
return tool;
}
public void setTool(boolean tool) {
this.tool = tool;
}
public boolean isPrompt() {
return prompt;
}
public void setPrompt(boolean prompt) {
this.prompt = prompt;
}
public boolean isCompletion() {
return completion;
}
public void setCompletion(boolean completion) {
this.completion = completion;
}
}
public Capabilities getCapabilities() {
return capabilities;
return this.capabilities;
}
/**
@@ -310,4 +266,48 @@ public class McpServerProperties {
return this.toolResponseMimeType;
}
public static class Capabilities {
private boolean resource = true;
private boolean tool = true;
private boolean prompt = true;
private boolean completion = true;
public boolean isResource() {
return this.resource;
}
public void setResource(boolean resource) {
this.resource = resource;
}
public boolean isTool() {
return this.tool;
}
public void setTool(boolean tool) {
this.tool = tool;
}
public boolean isPrompt() {
return this.prompt;
}
public void setPrompt(boolean prompt) {
this.prompt = prompt;
}
public boolean isCompletion() {
return this.completion;
}
public void setCompletion(boolean completion) {
this.completion = completion;
}
}
}

92
auto-configurations/mcp/spring-ai-autoconfigure-mcp-server/src/test/java/org/springframework/ai/mcp/server/autoconfigure/McpServerAutoConfigurationIT.java
View File

@@ -241,42 +241,6 @@ public class McpServerAutoConfigurationIT {
});
}
@Configuration
static class TestResourceConfiguration {
@Bean
List<SyncResourceSpecification> testResources() {
return List.of();
}
}
@Configuration
static class TestPromptConfiguration {
@Bean
List<SyncPromptSpecification> testPrompts() {
return List.of();
}
}
@Configuration
static class CustomCapabilitiesConfiguration {
@Bean
McpSchema.ServerCapabilities.Builder customCapabilitiesBuilder() {
return new CustomCapabilitiesBuilder();
}
}
static class CustomCapabilitiesBuilder extends McpSchema.ServerCapabilities.Builder {
// Custom implementation for testing
}
@Test
void capabilitiesConfiguration() {
this.contextRunner.withPropertyValues("spring.ai.mcp.server.capabilities.tool=false",
@@ -329,9 +293,44 @@ public class McpServerAutoConfigurationIT {
@Test
void toolCallbackProviderConfiguration() {
this.contextRunner.withUserConfiguration(TestToolCallbackProviderConfiguration.class).run(context -> {
assertThat(context).hasSingleBean(ToolCallbackProvider.class);
});
this.contextRunner.withUserConfiguration(TestToolCallbackProviderConfiguration.class)
.run(context -> assertThat(context).hasSingleBean(ToolCallbackProvider.class));
}
@Configuration
static class TestResourceConfiguration {
@Bean
List<SyncResourceSpecification> testResources() {
return List.of();
}
}
@Configuration
static class TestPromptConfiguration {
@Bean
List<SyncPromptSpecification> testPrompts() {
return List.of();
}
}
@Configuration
static class CustomCapabilitiesConfiguration {
@Bean
McpSchema.ServerCapabilities.Builder customCapabilitiesBuilder() {
return new CustomCapabilitiesBuilder();
}
}
static class CustomCapabilitiesBuilder extends McpSchema.ServerCapabilities.Builder {
// Custom implementation for testing
}
@Configuration
@@ -379,11 +378,8 @@ public class McpServerAutoConfigurationIT {
List<SyncCompletionSpecification> testCompletions() {
BiFunction<McpSyncServerExchange, McpSchema.CompleteRequest, McpSchema.CompleteResult> completionHandler = (
exchange, request) -> {
// Test implementation
return new McpSchema.CompleteResult(
new McpSchema.CompleteResult.CompleteCompletion(List.of(), 0, false));
};
exchange, request) -> new McpSchema.CompleteResult(
new McpSchema.CompleteResult.CompleteCompletion(List.of(), 0, false));
return List.of(new McpServerFeatures.SyncCompletionSpecification(
new McpSchema.PromptReference("ref/prompt", "code_review"), completionHandler));
@@ -397,11 +393,9 @@ public class McpServerAutoConfigurationIT {
@Bean
List<AsyncCompletionSpecification> testAsyncCompletions() {
BiFunction<McpAsyncServerExchange, McpSchema.CompleteRequest, Mono<McpSchema.CompleteResult>> completionHandler = (
exchange, request) -> {
// Test implementation
return Mono.just(new McpSchema.CompleteResult(
new McpSchema.CompleteResult.CompleteCompletion(List.of(), 0, false)));
};
exchange, request) -> Mono.just(new McpSchema.CompleteResult(
new McpSchema.CompleteResult.CompleteCompletion(List.of(), 0, false)));
return List.of(new McpServerFeatures.AsyncCompletionSpecification(
new McpSchema.PromptReference("ref/prompt", "code_review"), completionHandler));
}