ai 7.0.153 → 7.0.154

package/src/telemetry/create-telemetry-dispatcher.ts

@@ -1,5 +1,6 @@
 import { asArray, type Context } from '@ai-sdk/provider-utils';
 import type { Callback } from '../util/callback';
+import { VERSION } from '../version';
 import {
   filterIncludedContext,
   filterToolContext,
@@ -61,6 +62,7 @@ function augmentEvent<EVENT>(
       recordInputs: telemetry.recordInputs,
       recordOutputs: telemetry.recordOutputs,
       functionId: telemetry.functionId,
+      aiVersion: VERSION,
     },
   );
 
@@ -224,6 +226,7 @@ export function createTelemetryDispatcher({
       onStepEnd == null && onStepFinish == null
         ? undefined
         : mergeCallbacks(onStepEnd, onStepFinish),
+    onStepError: mergeTelemetryCallback('onStepError'),
     onObjectStepStart: mergeTelemetryCallback('onObjectStepStart'),
     onObjectStepEnd: mergeTelemetryCallback('onObjectStepEnd'),
     onEmbedStart: mergeTelemetryCallback('onEmbedStart'),

package/src/telemetry/telemetry.ts

@@ -22,6 +22,7 @@ import type {
   GenerateTextEndEvent,
   GenerateTextStartEvent,
   GenerateTextStepEndEvent,
+  GenerateTextStepErrorEvent,
   GenerateTextStepStartEvent,
 } from '../generate-text/generate-text-events';
 import type {
@@ -59,7 +60,13 @@ export type InferTelemetryEvent<EVENT> = EVENT &
   Omit<
     TelemetryOptions,
     'integrations' | 'isEnabled' | 'includeRuntimeContext'
-  >;
+  > & {
+    /**
+     * The version of the `ai` package that emitted this event.
+     * Absent in older releases.
+     */
+    readonly aiVersion?: string;
+  };
 
 type BivariantCallback<EVENT> = {
   bivarianceHack(event: EVENT): PromiseLike<void> | void;
@@ -110,6 +117,7 @@ export interface TelemetryDispatcher {
   onStepEnd?: Callback<GenerateTextStepEndEvent>;
   /** @deprecated Use `onStepEnd` instead. */
   onStepFinish?: Callback<GenerateTextStepEndEvent>;
+  onStepError?: Callback<GenerateTextStepErrorEvent>;
   onObjectStepStart?: Callback<GenerateObjectStepStartEvent>;
   onObjectStepEnd?: Callback<GenerateObjectStepEndEvent>;
   onEmbedStart?: Callback<EmbeddingModelCallStartEvent>;
@@ -202,6 +210,12 @@ export interface Telemetry {
    */
   onStepFinish?: Callback<InferTelemetryEvent<GenerateTextStepEndEvent>>;
 
+  /**
+   * Called when a step that started fails before it produces a step result.
+   * Use it to close step-level spans. The call then ends with `onError`.
+   */
+  onStepError?: Callback<InferTelemetryEvent<GenerateTextStepErrorEvent>>;
+
   /**
    * Called when an object generation step (single LLM invocation) begins.
    * For generateObject/streamObject there is always exactly one step.

package/src/util/is-caused-by-abort.ts

@@ -0,0 +1,20 @@
+import { isAbortError } from '@ai-sdk/provider-utils';
+
+/**
+ * Checks if the error is an abort or timeout error, or is caused by one.
+ * Providers can wrap the abort or timeout error, e.g. as the cause of an
+ * API call error.
+ */
+export function isCausedByAbort(error: unknown): boolean {
+  let current = error;
+
+  for (let depth = 0; current != null && depth < 5; depth++) {
+    if (isAbortError(current)) {
+      return true;
+    }
+
+    current = (current as { cause?: unknown }).cause;
+  }
+
+  return false;
+}