request retry policy aligned with the one used by Python SDK (#18)

xemantic · Nov 17, 2024 · 9025660 · 9025660
1 parent 7132763
commit 9025660
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/commonMain/kotlin/Anthropic.kt b/src/commonMain/kotlin/Anthropic.kt
@@ -116,6 +116,13 @@ class Anthropic internal constructor(
 
   private val client = HttpClient {
 
+    val retriableResponses = setOf<HttpStatusCode>(
+      HttpStatusCode.RequestTimeout,
+      HttpStatusCode.Conflict,
+      HttpStatusCode.TooManyRequests,
+      HttpStatusCode.InternalServerError
+    )
+
     install(ContentNegotiation) {
       json(anthropicJson)
     }
@@ -129,12 +136,10 @@ class Anthropic internal constructor(
     }
 
     install(HttpRequestRetry) {
-      retryOnServerErrors(maxRetries = 5)
       exponentialDelay()
       maxRetries = 5
       retryIf { _, response ->
-        response.status == HttpStatusCode.TooManyRequests
-            || response.status.value == 529 // Overloaded
+        response.status in retriableResponses || response.status.value >= 500
       }
     }