refactor(mcp) Increase the server stdio shutdown timeout

tzolov · tzolov · commit 47b09165c598 · 2024-12-30T09:08:56.000+01:00
diff --git a/mcp/src/main/java/org/springframework/ai/mcp/server/transport/SseServerTransport.java b/mcp/src/main/java/org/springframework/ai/mcp/server/transport/SseServerTransport.java
@@ -6,7 +6,6 @@
 import java.util.UUID;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.function.Function;
-import java.util.stream.Collectors;
 
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
diff --git a/mcp/src/main/java/org/springframework/ai/mcp/server/transport/StdioServerTransport.java b/mcp/src/main/java/org/springframework/ai/mcp/server/transport/StdioServerTransport.java
@@ -252,7 +252,7 @@ public Mono<Void> closeGracefully() {
 			// First complete the sinks to stop processing
 			inboundSink.tryEmitComplete();
 			outboundSink.tryEmitComplete();
-			return Mono.delay(Duration.ofMillis(100));
+			return Mono.delay(Duration.ofMillis(200));
 		})).then(Mono.fromRunnable(() -> {
 			try {
 				// Dispose schedulers first
