@ai-sdk/devtools 1.0.4 → 1.0.5

package/dist/index.d.ts

@@ -19,6 +19,13 @@ import { Telemetry } from 'ai';
  */
 declare const devToolsMiddleware: () => LanguageModelV4Middleware;
 
+interface DevToolsTelemetryOptions {
+    /**
+     * Identifier used to group multiple AI SDK calls into one DevTools run.
+     * Reuse the same value when resuming an interrupted interaction.
+     */
+    runId?: string;
+}
 /**
  * Creates a devtools telemetry integration that logs all AI SDK operations
  * to the devtools viewer.
@@ -34,6 +41,6 @@ declare const devToolsMiddleware: () => LanguageModelV4Middleware;
  * Telemetry is enabled by default — no need to set `telemetry`
  * unless you want to configure `functionId`, `recordInputs`, or `recordOutputs`.
  */
-declare function DevToolsTelemetry(): Telemetry;
+declare function DevToolsTelemetry(options?: DevToolsTelemetryOptions): Telemetry;
 
-export { DevToolsTelemetry, devToolsMiddleware };
+export { DevToolsTelemetry, type DevToolsTelemetryOptions, devToolsMiddleware };

package/dist/index.js

@@ -117,6 +117,10 @@ var updateStepResult = async (stepId, result) => {
     notifyServer("step-update");
   }
 };
+var getStepsForRun = async (runId) => {
+  const db = getDb();
+  return db.steps.filter((s) => s.run_id === runId).sort((a, b) => a.step_number - b.step_number);
+};
 
 // src/middleware.ts
 var generateId = () => crypto.randomUUID();
@@ -433,7 +437,7 @@ function getOperationType(operationId) {
   }
   return "generate";
 }
-function DevToolsTelemetry() {
+function DevToolsTelemetry(options = {}) {
   if (process.env.NODE_ENV === "production") {
     throw new Error(
       "@ai-sdk/devtools should not be used in production. Remove DevToolsTelemetry from your telemetry configuration for production builds."
@@ -476,7 +480,8 @@ function DevToolsTelemetry() {
         frequencyPenalty: event.frequencyPenalty,
         seed: event.seed
       },
-      stepStates: /* @__PURE__ */ new Map()
+      stepStates: /* @__PURE__ */ new Map(),
+      stepNumberOffset: 0
     };
     callStates.set(callId, state);
     return state;
@@ -494,6 +499,10 @@ function DevToolsTelemetry() {
         operationId,
         startEvent
       );
+      if (options.runId != null) {
+        state.runId = options.runId;
+        state.stepNumberOffset = (await getStepsForRun(options.runId)).length;
+      }
       await createRun(state.runId, parentInfo, state.functionId);
     },
     onStepStart: async (event) => {
@@ -513,7 +522,7 @@ function DevToolsTelemetry() {
       await createStep({
         id: stepId,
         run_id: state.runId,
-        step_number: stepNumber + 1,
+        step_number: state.stepNumberOffset + stepNumber + 1,
         type: state.operationType,
         model_id: stepStartEvent.modelId,
         provider: stepStartEvent.provider ?? null,
@@ -552,7 +561,7 @@ function DevToolsTelemetry() {
       await createStep({
         id: stepId,
         run_id: state.runId,
-        step_number: stepStartEvent.stepNumber + 1,
+        step_number: state.stepNumberOffset + stepStartEvent.stepNumber + 1,
         type: state.operationType,
         model_id: stepStartEvent.modelId,
         provider: stepStartEvent.provider ?? null,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/devtools",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -55,7 +55,7 @@
     "vite": "^6.4.3",
     "vitest": "^4.1.6",
     "zod": "3.25.76",
-    "ai": "7.0.20"
+    "ai": "7.0.26"
   },
   "publishConfig": {
     "access": "public",

package/src/index.ts

@@ -1,2 +1,5 @@
 export { devToolsMiddleware } from './middleware.js';
-export { DevToolsTelemetry } from './integration.js';
+export {
+  DevToolsTelemetry,
+  type DevToolsTelemetryOptions,
+} from './integration.js';

package/src/integration.ts

@@ -11,6 +11,7 @@ import type {
 import {
   createRun,
   createStep,
+  getStepsForRun,
   updateStepResult,
   notifyServerAsync,
 } from './db.js';
@@ -28,6 +29,15 @@ interface CallState {
   functionId: string | undefined;
   settings: Record<string, unknown>;
   stepStates: Map<number, StepState>;
+  stepNumberOffset: number;
+}
+
+export interface DevToolsTelemetryOptions {
+  /**
+   * Identifier used to group multiple AI SDK calls into one DevTools run.
+   * Reuse the same value when resuming an interrupted interaction.
+   */
+  runId?: string;
 }
 
 const activeSteps = new Map<string, StepState>();
@@ -99,7 +109,9 @@ function getOperationType(operationId: string): OperationType {
  * Telemetry is enabled by default — no need to set `telemetry`
  * unless you want to configure `functionId`, `recordInputs`, or `recordOutputs`.
  */
-export function DevToolsTelemetry(): Telemetry {
+export function DevToolsTelemetry(
+  options: DevToolsTelemetryOptions = {},
+): Telemetry {
   if (process.env.NODE_ENV === 'production') {
     throw new Error(
       '@ai-sdk/devtools should not be used in production. ' +
@@ -179,6 +191,7 @@ export function DevToolsTelemetry(): Telemetry {
         seed: event.seed,
       },
       stepStates: new Map(),
+      stepNumberOffset: 0,
     };
     callStates.set(callId, state);
     return state;
@@ -211,6 +224,11 @@ export function DevToolsTelemetry(): Telemetry {
         startEvent,
       );
 
+      if (options.runId != null) {
+        state.runId = options.runId;
+        state.stepNumberOffset = (await getStepsForRun(options.runId)).length;
+      }
+
       await createRun(state.runId, parentInfo, state.functionId);
     },
 
@@ -238,7 +256,7 @@ export function DevToolsTelemetry(): Telemetry {
       await createStep({
         id: stepId,
         run_id: state.runId,
-        step_number: stepNumber + 1,
+        step_number: state.stepNumberOffset + stepNumber + 1,
         type: state.operationType,
         model_id: stepStartEvent.modelId,
         provider: stepStartEvent.provider ?? null,
@@ -288,7 +306,7 @@ export function DevToolsTelemetry(): Telemetry {
       await createStep({
         id: stepId,
         run_id: state.runId,
-        step_number: stepStartEvent.stepNumber + 1,
+        step_number: state.stepNumberOffset + stepStartEvent.stepNumber + 1,
         type: state.operationType,
         model_id: stepStartEvent.modelId,
         provider: stepStartEvent.provider ?? null,