fix (ai): support abort during retry waits (#7661)

## Background

Abort was delayed during the delays when retrying.

## Summary

* add abort signal support to delay
* pass abort signal into delay from the retry function

## Related Issues

Fixes #7656

Author: lgrammel

.changeset/three-humans-report.md

@@ -0,0 +1,6 @@
+---
+'@ai-sdk/provider-utils': patch
+'ai': patch
+---
+
+fix (ai): support abort during retry waits

packages/ai/src/embed/embed-many.ts

@@ -92,7 +92,10 @@ Only applicable for HTTP-based providers.
     });
   }
 
-  const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { maxRetries, retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
     model,

packages/ai/src/embed/embed.ts

@@ -79,7 +79,10 @@ Only applicable for HTTP-based providers.
     });
   }
 
-  const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { maxRetries, retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
     model,

packages/ai/src/generate-image/generate-image.ts

@@ -122,7 +122,10 @@ Only applicable for HTTP-based providers.
     });
   }
 
-  const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   // default to 1 if the model has not specified limits on
   // how many images can be generated in a single call

packages/ai/src/generate-object/generate-object.ts

@@ -243,7 +243,10 @@ Default and recommended: 'auto' (best mode for the model).
     enumValues,
   });
 
-  const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { maxRetries, retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   const outputStrategy = getOutputStrategy({
     output,

packages/ai/src/generate-object/stream-object.ts

@@ -427,6 +427,7 @@ class DefaultStreamObjectResult<PARTIAL, RESULT, ELEMENT_STREAM>
 
     const { maxRetries, retry } = prepareRetries({
       maxRetries: maxRetriesArg,
+      abortSignal,
     });
 
     const callSettings = prepareCallSettings(settings);

packages/ai/src/generate-speech/generate-speech.ts

@@ -121,7 +121,10 @@ Only applicable for HTTP-based providers.
     });
   }
 
-  const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   const result = await retry(() =>
     model.doGenerate({

packages/ai/src/generate-text/generate-text.ts

@@ -226,7 +226,10 @@ A function that attempts to repair a tool call that failed to parse.
   }): Promise<GenerateTextResult<TOOLS, OUTPUT>> {
   const model = resolveLanguageModel(modelArg);
   const stopConditions = asArray(stopWhen);
-  const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { maxRetries, retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
 
   const callSettings = prepareCallSettings(settings);
 

packages/ai/src/generate-text/stream-text.ts

@@ -944,6 +944,7 @@ class DefaultStreamTextResult<TOOLS extends ToolSet, OUTPUT, PARTIAL_OUTPUT>
 
     const { maxRetries, retry } = prepareRetries({
       maxRetries: maxRetriesArg,
+      abortSignal,
     });
 
     const tracer = getTracer(telemetry);

packages/ai/src/transcribe/transcribe.ts

@@ -87,7 +87,11 @@ Only applicable for HTTP-based providers.
     });
   }
 
-  const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
+  const { retry } = prepareRetries({
+    maxRetries: maxRetriesArg,
+    abortSignal,
+  });
+
   const audioData =
     audio instanceof URL
       ? (await download({ url: audio })).data