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() {