ai 5.0.0-beta.19 → 5.0.0-beta.20

package/dist/index.mjs

@@ -95,7 +95,33 @@ var RetryError = class extends AISDKError3 {
 _a3 = symbol3;
 
 // src/util/retry-with-exponential-backoff.ts
-var retryWithExponentialBackoff = ({
+function getRetryDelay(error, exponentialBackoffDelay) {
+  const headers = error.responseHeaders;
+  if (!headers)
+    return exponentialBackoffDelay;
+  let timeoutMillis;
+  const retryAfterMs = headers["retry-after-ms"];
+  if (retryAfterMs) {
+    const timeoutMs = parseFloat(retryAfterMs);
+    if (!Number.isNaN(timeoutMs)) {
+      timeoutMillis = timeoutMs;
+    }
+  }
+  const retryAfter = headers["retry-after"];
+  if (retryAfter && timeoutMillis === void 0) {
+    const timeoutSeconds = parseFloat(retryAfter);
+    if (!Number.isNaN(timeoutSeconds)) {
+      timeoutMillis = timeoutSeconds * 1e3;
+    } else {
+      timeoutMillis = Date.parse(retryAfter) - Date.now();
+    }
+  }
+  if (timeoutMillis !== void 0 && 0 <= timeoutMillis && timeoutMillis < 60 * 1e3) {
+    return timeoutMillis;
+  }
+  return exponentialBackoffDelay;
+}
+var retryWithExponentialBackoffRespectingRetryHeaders = ({
   maxRetries = 2,
   initialDelayInMs = 2e3,
   backoffFactor = 2
@@ -129,7 +155,8 @@ async function _retryWithExponentialBackoff(f, {
       });
     }
     if (error instanceof Error && APICallError.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
-      await delay(delayInMs);
+      const actualDelay = getRetryDelay(error, delayInMs);
+      await delay(actualDelay);
       return _retryWithExponentialBackoff(
         f,
         { maxRetries, delayInMs: backoffFactor * delayInMs, backoffFactor },
@@ -170,7 +197,9 @@ function prepareRetries({
   const maxRetriesResult = maxRetries != null ? maxRetries : 2;
   return {
     maxRetries: maxRetriesResult,
-    retry: retryWithExponentialBackoff({ maxRetries: maxRetriesResult })
+    retry: retryWithExponentialBackoffRespectingRetryHeaders({
+      maxRetries: maxRetriesResult
+    })
   };
 }