Skip to content

Commit 5c02b51

Browse files
committed
chore: remove elicitation schema
1 parent b1b179f commit 5c02b51

File tree

7 files changed

+32
-450
lines changed

7 files changed

+32
-450
lines changed

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/WebFluxSseIntegrationTests.java

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
package io.modelcontextprotocol;
55

66
import java.time.Duration;
7-
import java.util.ArrayList;
87
import java.util.List;
98
import java.util.Map;
109
import java.util.concurrent.ConcurrentHashMap;
@@ -28,7 +27,6 @@
2827
import io.modelcontextprotocol.spec.McpError;
2928
import io.modelcontextprotocol.spec.McpSchema;
3029
import io.modelcontextprotocol.spec.McpSchema.*;
31-
import io.modelcontextprotocol.spec.McpSchema.ServerCapabilities.CompletionCapabilities;
3230
import org.junit.jupiter.api.AfterEach;
3331
import org.junit.jupiter.api.BeforeEach;
3432
import org.junit.jupiter.params.ParameterizedTest;
@@ -390,9 +388,8 @@ void testCreateElicitationSuccess(String clientType) {
390388

391389
var elicitationRequest = ElicitRequest.builder()
392390
.message("Test message")
393-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
394-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
395-
.build())
391+
.requestedSchema(
392+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
396393
.build();
397394

398395
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {
@@ -459,9 +456,8 @@ void testCreateElicitationWithRequestTimeoutSuccess(String clientType) {
459456

460457
var elicitationRequest = ElicitRequest.builder()
461458
.message("Test message")
462-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
463-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
464-
.build())
459+
.requestedSchema(
460+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
465461
.build();
466462

467463
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {
@@ -525,9 +521,8 @@ void testCreateElicitationWithRequestTimeoutFail(String clientType) {
525521

526522
var elicitationRequest = ElicitRequest.builder()
527523
.message("Test message")
528-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
529-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
530-
.build())
524+
.requestedSchema(
525+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
531526
.build();
532527

533528
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {

mcp-spring/mcp-spring-webmvc/src/test/java/io/modelcontextprotocol/server/WebMvcSseIntegrationTests.java

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -409,9 +409,8 @@ void testCreateElicitationSuccess() {
409409

410410
var elicitationRequest = McpSchema.ElicitRequest.builder()
411411
.message("Test message")
412-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
413-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
414-
.build())
412+
.requestedSchema(
413+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
415414
.build();
416415

417416
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {
@@ -477,9 +476,8 @@ void testCreateElicitationWithRequestTimeoutSuccess() {
477476

478477
var elicitationRequest = McpSchema.ElicitRequest.builder()
479478
.message("Test message")
480-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
481-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
482-
.build())
479+
.requestedSchema(
480+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
483481
.build();
484482

485483
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {
@@ -542,9 +540,8 @@ void testCreateElicitationWithRequestTimeoutFail() {
542540

543541
var elicitationRequest = McpSchema.ElicitRequest.builder()
544542
.message("Test message")
545-
.requestedSchema(McpSchema.PrimitiveSchemaDefinition.builder()
546-
.properties(Map.of("message", McpSchema.StringSchema.builder().build()))
547-
.build())
543+
.requestedSchema(
544+
Map.of("type", "object", "properties", Map.of("message", Map.of("type", "string"))))
548545
.build();
549546

550547
StepVerifier.create(exchange.createElicitation(elicitationRequest)).consumeNextWith(result -> {

0 commit comments

Comments
 (0)