Skip to content

Commit 46751fd

Browse files
committed
Merge branch 'refs/heads/main' into feat/header
# Conflicts: # mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/client/transport/WebClientStreamableHttpTransport.java
2 parents df10fd4 + 9ebff0c commit 46751fd

30 files changed

+1405
-223
lines changed

mcp-spring/mcp-spring-webflux/src/main/java/io/modelcontextprotocol/client/transport/WebClientStreamableHttpTransport.java

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,12 @@
11
package io.modelcontextprotocol.client.transport;
22

3-
import com.fasterxml.jackson.core.type.TypeReference;
4-
import com.fasterxml.jackson.databind.ObjectMapper;
5-
import io.modelcontextprotocol.spec.DefaultMcpTransportSession;
6-
import io.modelcontextprotocol.spec.DefaultMcpTransportStream;
7-
import io.modelcontextprotocol.spec.McpClientTransport;
8-
import io.modelcontextprotocol.spec.McpError;
9-
import io.modelcontextprotocol.spec.McpSchema;
10-
import io.modelcontextprotocol.spec.McpTransportSessionNotFoundException;
11-
import io.modelcontextprotocol.spec.McpTransportSession;
12-
import io.modelcontextprotocol.spec.McpTransportStream;
13-
import io.modelcontextprotocol.util.Assert;
3+
import java.io.IOException;
4+
import java.util.List;
5+
import java.util.Optional;
6+
import java.util.concurrent.atomic.AtomicReference;
7+
import java.util.function.Consumer;
8+
import java.util.function.Function;
9+
1410
import org.reactivestreams.Publisher;
1511
import org.slf4j.Logger;
1612
import org.slf4j.LoggerFactory;
@@ -21,20 +17,25 @@
2117
import org.springframework.web.reactive.function.client.ClientResponse;
2218
import org.springframework.web.reactive.function.client.WebClient;
2319
import org.springframework.web.reactive.function.client.WebClientResponseException;
20+
21+
import com.fasterxml.jackson.core.type.TypeReference;
22+
import com.fasterxml.jackson.databind.ObjectMapper;
23+
24+
import io.modelcontextprotocol.spec.DefaultMcpTransportSession;
25+
import io.modelcontextprotocol.spec.DefaultMcpTransportStream;
26+
import io.modelcontextprotocol.spec.McpClientTransport;
27+
import io.modelcontextprotocol.spec.McpError;
28+
import io.modelcontextprotocol.spec.McpSchema;
29+
import io.modelcontextprotocol.spec.McpTransportSession;
30+
import io.modelcontextprotocol.spec.McpTransportSessionNotFoundException;
31+
import io.modelcontextprotocol.spec.McpTransportStream;
32+
import io.modelcontextprotocol.util.Assert;
2433
import reactor.core.Disposable;
2534
import reactor.core.publisher.Flux;
2635
import reactor.core.publisher.Mono;
2736
import reactor.util.function.Tuple2;
2837
import reactor.util.function.Tuples;
2938

30-
import java.io.IOException;
31-
import java.util.List;
32-
import java.util.Optional;
33-
import java.util.concurrent.atomic.AtomicReference;
34-
import java.util.function.Consumer;
35-
import java.util.function.Function;
36-
import java.util.function.Supplier;
37-
3839
import static io.modelcontextprotocol.spec.McpSchema.Headers.LAST_EVENT_ID;
3940
import static io.modelcontextprotocol.spec.McpSchema.Headers.MCP_SESSION_ID;
4041

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/WebClientStreamableHttpAsyncClientTests.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,12 @@
11
package io.modelcontextprotocol.client;
22

3-
import com.fasterxml.jackson.databind.ObjectMapper;
43
import io.modelcontextprotocol.client.transport.WebClientStreamableHttpTransport;
54
import io.modelcontextprotocol.spec.McpClientTransport;
65
import org.junit.jupiter.api.Timeout;
7-
import org.springframework.web.reactive.function.client.WebClient;
86
import org.testcontainers.containers.GenericContainer;
97
import org.testcontainers.containers.wait.strategy.Wait;
10-
import org.testcontainers.images.builder.ImageFromDockerfile;
8+
9+
import org.springframework.web.reactive.function.client.WebClient;
1110

1211
@Timeout(15)
1312
public class WebClientStreamableHttpAsyncClientTests extends AbstractMcpAsyncClientTests {

mcp-spring/mcp-spring-webflux/src/test/java/io/modelcontextprotocol/client/WebFluxSseMcpAsyncClientTests.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,13 @@
66

77
import java.time.Duration;
88

9-
import io.modelcontextprotocol.client.transport.WebFluxSseClientTransport;
10-
import io.modelcontextprotocol.spec.McpClientTransport;
119
import org.junit.jupiter.api.Timeout;
10+
import org.springframework.web.reactive.function.client.WebClient;
1211
import org.testcontainers.containers.GenericContainer;
1312
import org.testcontainers.containers.wait.strategy.Wait;
1413

15-
import org.springframework.web.reactive.function.client.WebClient;
14+
import io.modelcontextprotocol.client.transport.WebFluxSseClientTransport;
15+
import io.modelcontextprotocol.spec.McpClientTransport;
1616

1717
/**
1818
* Tests for the {@link McpAsyncClient} with {@link WebFluxSseClientTransport}.

mcp-test/pom.xml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,11 @@
5454
<artifactId>junit-jupiter-api</artifactId>
5555
<version>${junit.version}</version>
5656
</dependency>
57+
<dependency>
58+
<groupId>org.junit.jupiter</groupId>
59+
<artifactId>junit-jupiter-params</artifactId>
60+
<version>${junit.version}</version>
61+
</dependency>
5762
<dependency>
5863
<groupId>org.mockito</groupId>
5964
<artifactId>mockito-core</artifactId>

mcp-test/src/main/java/io/modelcontextprotocol/client/AbstractMcpAsyncClientResiliencyTests.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
/*
2+
* Copyright 2024-2024 the original author or authors.
3+
*/
14
package io.modelcontextprotocol.client;
25

36
import eu.rekawek.toxiproxy.Proxy;

0 commit comments

Comments
 (0)