feat(mcp): Add SSE endpoint parameter to WebFlux and WebMvc transport providers

Add serverProperties.getSseEndpoint() parameter to both WebFluxSseServerTransportProvider
and WebMvcSseServerTransportProvider constructors.

Signed-off-by: liujunjie <smartliujunjie@gmail.com>
This commit is contained in:
LiujunjieALiling
2025-04-22 23:47:58 +08:00
committed by Christian Tzolov
parent 25e52b8c94
commit 4e313a9dc6
2 changed files with 4 additions and 2 deletions

View File

@@ -77,7 +77,8 @@ public class McpWebFluxServerAutoConfiguration {
public WebFluxSseServerTransportProvider webFluxTransport(ObjectProvider<ObjectMapper> objectMapperProvider,
McpServerProperties serverProperties) {
ObjectMapper objectMapper = objectMapperProvider.getIfAvailable(ObjectMapper::new);
return new WebFluxSseServerTransportProvider(objectMapper, serverProperties.getSseMessageEndpoint());
return new WebFluxSseServerTransportProvider(objectMapper, serverProperties.getSseMessageEndpoint(),
serverProperties.getSseEndpoint());
}
// Router function for SSE transport used by Spring WebFlux to start an HTTP server.

View File

@@ -72,7 +72,8 @@ public class McpWebMvcServerAutoConfiguration {
public WebMvcSseServerTransportProvider webMvcSseServerTransportProvider(
ObjectProvider<ObjectMapper> objectMapperProvider, McpServerProperties serverProperties) {
ObjectMapper objectMapper = objectMapperProvider.getIfAvailable(ObjectMapper::new);
return new WebMvcSseServerTransportProvider(objectMapper, serverProperties.getSseMessageEndpoint());
return new WebMvcSseServerTransportProvider(objectMapper, serverProperties.getSseMessageEndpoint(),
serverProperties.getSseEndpoint());
}
@Bean