From 47b09165c59874d8833271c0d82c9b42eb2f6313 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Mon, 30 Dec 2024 09:08:56 +0100 Subject: [PATCH] refactor(mcp) Increase the server stdio shutdown timeout --- .../ai/mcp/server/transport/SseServerTransport.java | 1 - .../ai/mcp/server/transport/StdioServerTransport.java | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) 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 index 8125f4d..88bda2e 100644 --- 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.List; 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 index dff9727..93d6c23 100644 --- 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 class StdioServerTransport implements McpTransport { // 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