From b79fa38f0be898b1992d7ddf939a2e615d267dad Mon Sep 17 00:00:00 2001 From: lambochen Date: Thu, 24 Apr 2025 16:29:12 +0800 Subject: [PATCH] mcp server support baseUrl --- .../server/transport/WebFluxSseServerTransportProvider.java | 4 ++-- .../server/transport/WebMvcSseServerTransportProvider.java | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/server/transport/WebFluxSseServerTransportProvider.java b/mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/server/transport/WebFluxSseServerTransportProvider.java index 62264d9a..886c2b08 100644 --- a/mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/server/transport/WebFluxSseServerTransportProvider.java +++ b/mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/server/transport/WebFluxSseServerTransportProvider.java @@ -159,8 +159,8 @@ public WebFluxSseServerTransportProvider(ObjectMapper objectMapper, String baseU this.messageEndpoint = messageEndpoint; this.sseEndpoint = sseEndpoint; this.routerFunction = RouterFunctions.route() - .GET(this.sseEndpoint, this::handleSseConnection) - .POST(this.messageEndpoint, this::handleMessage) + .GET(this.baseUrl + this.sseEndpoint, this::handleSseConnection) + .POST(this.baseUrl + this.messageEndpoint, this::handleMessage) .build(); } diff --git a/mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransportProvider.java b/mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransportProvider.java index fc86cfaa..af0551a0 100644 --- a/mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransportProvider.java +++ b/mcp-spring/mcp-spring-webmvc/src/main/java/io/modelcontextprotocol/server/transport/WebMvcSseServerTransportProvider.java @@ -159,8 +159,8 @@ public WebMvcSseServerTransportProvider(ObjectMapper objectMapper, String baseUr this.messageEndpoint = messageEndpoint; this.sseEndpoint = sseEndpoint; this.routerFunction = RouterFunctions.route() - .GET(this.sseEndpoint, this::handleSseConnection) - .POST(this.messageEndpoint, this::handleMessage) + .GET(this.baseUrl + this.sseEndpoint, this::handleSseConnection) + .POST(this.baseUrl + this.messageEndpoint, this::handleMessage) .build(); }