added retries to anthropic ITs

langchain4j · May 24, 2024 · e34b592 · e34b592
1 parent 102050e
commit e34b592
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/langchain4j-anthropic-spring-boot-starter/pom.xml b/langchain4j-anthropic-spring-boot-starter/pom.xml
@@ -53,6 +53,13 @@
             <scope>test</scope>
         </dependency>
 
+        <dependency>
+            <groupId>org.junit-pioneer</groupId>
+            <artifactId>junit-pioneer</artifactId>
+            <version>1.9.1</version>
+            <scope>test</scope>
+        </dependency>
+
     </dependencies>
 
     <licenses>

diff --git a/...opic-spring-boot-starter/src/test/java/dev/langchain4j/anthropic/spring/AutoConfigIT.java b/...opic-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(