diff --git a/langchain4j-anthropic-spring-boot-starter/pom.xml b/langchain4j-anthropic-spring-boot-starter/pom.xml index 1b556303..c25d79a3 100644 --- a/langchain4j-anthropic-spring-boot-starter/pom.xml +++ b/langchain4j-anthropic-spring-boot-starter/pom.xml @@ -53,6 +53,13 @@ test + + org.junit-pioneer + junit-pioneer + 1.9.1 + test + + diff --git a/langchain4j-anthropic-spring-boot-starter/src/test/java/dev/langchain4j/anthropic/spring/AutoConfigIT.java b/langchain4j-anthropic-spring-boot-starter/src/test/java/dev/langchain4j/anthropic/spring/AutoConfigIT.java index c38d24da..2cf95d96 100644 --- a/langchain4j-anthropic-spring-boot-starter/src/test/java/dev/langchain4j/anthropic/spring/AutoConfigIT.java +++ b/langchain4j-anthropic-spring-boot-starter/src/test/java/dev/langchain4j/anthropic/spring/AutoConfigIT.java @@ -8,7 +8,7 @@ import dev.langchain4j.model.chat.StreamingChatLanguageModel; import dev.langchain4j.model.output.Response; import org.junit.jupiter.api.AfterEach; -import org.junit.jupiter.api.Test; +import org.junitpioneer.jupiter.RetryingTest; import org.springframework.boot.autoconfigure.AutoConfigurations; import org.springframework.boot.test.context.runner.ApplicationContextRunner; @@ -29,7 +29,7 @@ void afterEach() throws InterruptedException { Thread.sleep(10_000); // to avoid hitting rate limits } - @Test + @RetryingTest(2) void should_provide_chat_model() { contextRunner .withPropertyValues( @@ -46,7 +46,7 @@ void should_provide_chat_model() { }); } - @Test + @RetryingTest(2) void should_provide_streaming_chat_model() { contextRunner .withPropertyValues(