Skip to content

Commit 77e7259

Browse files
committed
code review
1 parent 9aefc3f commit 77e7259

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

mcp/src/main/java/io/modelcontextprotocol/client/transport/StreamableHttpClientTransport.java

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,7 @@ public static class Builder {
9797
.version(HttpClient.Version.HTTP_1_1)
9898
.connectTimeout(Duration.ofSeconds(10));
9999

100-
private final HttpRequest.Builder requestBuilder = HttpRequest.newBuilder()
101-
.header("Accept", "application/json, text/event-stream");
100+
private final HttpRequest.Builder requestBuilder = HttpRequest.newBuilder();
102101

103102
private ObjectMapper objectMapper = new ObjectMapper();
104103

@@ -174,7 +173,10 @@ public Mono<Void> connect(final Function<Mono<McpSchema.JSONRPCMessage>, Mono<Mc
174173
}
175174

176175
return Mono.defer(() -> Mono.fromFuture(() -> {
177-
final HttpRequest.Builder builder = requestBuilder.copy().GET().uri(uri);
176+
final HttpRequest.Builder builder = requestBuilder.copy()
177+
.GET()
178+
.header("Accept", "text/event-stream")
179+
.uri(uri);
178180
final String lastId = lastEventId.get();
179181
if (lastId != null) {
180182
builder.header("Last-Event-ID", lastId);
@@ -216,6 +218,8 @@ public Mono<Void> sendMessage(final McpSchema.JSONRPCMessage message,
216218
return serializeJson(message).flatMap(json -> {
217219
final HttpRequest request = requestBuilder.copy()
218220
.POST(HttpRequest.BodyPublishers.ofString(json))
221+
.header("Accept", "application/json, text/event-stream")
222+
.header("Content-Type", "application/json")
219223
.uri(uri)
220224
.build();
221225
return Mono.fromFuture(httpClient.sendAsync(request, HttpResponse.BodyHandlers.ofInputStream()))

0 commit comments

Comments
 (0)