From 75db25b13ddec60f7343ac5a84d9a5f56d590d36 Mon Sep 17 00:00:00 2001 From: Youngrae Cho Date: Thu, 18 Jul 2024 21:11:21 +0900 Subject: [PATCH] Fix RetryTemplate retry condition Resolves #846 --- .../ai/autoconfigure/retry/SpringAiRetryAutoConfiguration.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/retry/SpringAiRetryAutoConfiguration.java b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/retry/SpringAiRetryAutoConfiguration.java index 00c32ed67..295ce7ca4 100644 --- a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/retry/SpringAiRetryAutoConfiguration.java +++ b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/retry/SpringAiRetryAutoConfiguration.java @@ -53,7 +53,7 @@ public class SpringAiRetryAutoConfiguration { public RetryTemplate retryTemplate(SpringAiRetryProperties properties) { return RetryTemplate.builder() .maxAttempts(properties.getMaxAttempts()) - .retryOn(NonTransientAiException.class) + .retryOn(TransientAiException.class) .exponentialBackoff(properties.getBackoff().getInitialInterval(), properties.getBackoff().getMultiplier(), properties.getBackoff().getMaxInterval()) .withListener(new RetryListener() {