@ai-sdk/workflow 1.0.70 → 2.0.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/workflow",
-  "version": "1.0.70",
+  "version": "2.0.0",
   "type": "module",
   "description": "WorkflowAgent for building AI agents with AI SDK",
   "license": "Apache-2.0",
@@ -27,21 +27,22 @@
   },
   "dependencies": {
     "ajv": "^8.20.0",
-    "@ai-sdk/provider-utils": "5.0.27",
-    "ai": "7.0.69",
-    "@ai-sdk/provider": "4.0.7"
+    "@ai-sdk/provider": "4.0.7",
+    "@ai-sdk/provider-utils": "5.0.28",
+    "ai": "7.0.70"
   },
   "devDependencies": {
     "@types/node": "22.19.19",
-    "@workflow/vitest": "4.0.5",
+    "@workflow/vitest": "5.0.0-beta.42",
     "tsup": "^8.5.1",
     "typescript": "5.8.3",
     "vitest": "4.1.6",
-    "workflow": "4.2.4",
-    "zod": "3.25.76",
+    "workflow": "5.0.0-beta.42",
+    "zod": "4.4.3",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {
+    "workflow": "^5.0.0-beta.42",
     "zod": "^3.25.76 || ^4.1.8"
   },
   "engines": {

package/src/do-stream-step.ts

@@ -6,7 +6,7 @@ import { isAbortError } from '@ai-sdk/provider-utils';
 import {
   experimental_streamLanguageModelCall as streamModelCall,
   gateway,
-  type Experimental_LanguageModelStreamPart as ModelCallStreamPart,
+  type Experimental_LanguageModelStreamPart,
   type FinishReason,
   type LanguageModel,
   type LanguageModelUsage,
@@ -22,7 +22,10 @@ import {
   resolveSerializableTools,
   type SerializableToolDef,
 } from './serializable-schema.js';
-export type { Experimental_LanguageModelStreamPart as ModelCallStreamPart } from 'ai';
+
+export type ModelCallStreamPart<TTools extends ToolSet = ToolSet> =
+  | Experimental_LanguageModelStreamPart<TTools>
+  | { type: 'reset-step' };
 
 export type ModelStopCondition = StopCondition<NoInfer<ToolSet>, any>;
 
@@ -248,6 +251,11 @@ export async function doStreamStep(
   const writer = writable?.getWriter();
 
   try {
+    // A workflow step can be retried after already writing partial output.
+    // Reset the current UI step before every attempt so a retry invalidates
+    // chunks left behind by an earlier execution.
+    await writer?.write({ type: 'reset-step' });
+
     for await (const part of modelStream) {
       switch (part.type) {
         case 'text-delta':

package/src/normalize-ui-message-stream.ts

@@ -115,6 +115,16 @@ export async function* normalizeUIMessageStreamParts(
 
   for await (const chunk of source) {
     switch (chunk.type) {
+      case 'reset-step':
+        // A retried model-call step starts a new frame. Forget parts from the
+        // invalidated attempt so reused ids are framed normally.
+        text.open.clear();
+        text.ended.clear();
+        reasoning.open.clear();
+        reasoning.ended.clear();
+        yield chunk;
+        break;
+
       case 'finish-step':
         // The consumer clears its active-part maps here, so part ids may be
         // legitimately reused in the next step. Reset to match.

package/src/stream-text-iterator.ts

@@ -7,7 +7,6 @@ import type {
 import type { Context } from '@ai-sdk/provider-utils';
 import {
   experimental_filterActiveTools as filterActiveTools,
-  type Experimental_LanguageModelStreamPart as ModelCallStreamPart,
   type Experimental_SandboxSession as SandboxSession,
   type Instructions,
   type LanguageModel,
@@ -22,6 +21,7 @@ import { createRestrictedTelemetryDispatcher } from 'ai/internal';
 import {
   type DoStreamStepRawResult,
   doStreamStep,
+  type ModelCallStreamPart,
   type ModelStopCondition,
   type ParsedToolCall,
   type ProviderExecutedToolResult,

package/src/to-ui-message-chunk.ts

@@ -1,8 +1,5 @@
-import type {
-  Experimental_LanguageModelStreamPart as ModelCallStreamPart,
-  ToolSet,
-  UIMessageChunk,
-} from 'ai';
+import type { ToolSet, UIMessageChunk } from 'ai';
+import type { ModelCallStreamPart } from './do-stream-step.js';
 
 /**
  * Convert a single ModelCallStreamPart to a UIMessageChunk.
@@ -12,6 +9,9 @@ export function toUIMessageChunk(
   part: ModelCallStreamPart<ToolSet>,
 ): UIMessageChunk | undefined {
   switch (part.type) {
+    case 'reset-step':
+      return { type: 'reset-step' };
+
     case 'text-start':
       return {
         type: 'text-start',

package/src/workflow-agent.ts

@@ -20,7 +20,6 @@ import {
   type FinishReason,
   type LanguageModelResponseMetadata,
   type LanguageModelUsage,
-  type Experimental_LanguageModelStreamPart as ModelCallStreamPart,
   type ModelMessage,
   type StepResult,
   type StopCondition,
@@ -45,6 +44,7 @@ import {
   validateApprovedToolApprovals,
 } from 'ai/internal';
 import { createLanguageModelToolResultOutput } from './create-language-model-tool-result-output.js';
+import type { ModelCallStreamPart } from './do-stream-step.js';
 import { streamTextIterator } from './stream-text-iterator.js';
 
 // Re-export for consumers

package/src/workflow-chat-transport.ts

@@ -59,6 +59,10 @@ function createOrphanFilter(): OrphanFilter {
 
   function shouldDrop(chunk: UIMessageChunk): boolean {
     switch (chunk.type) {
+      case 'reset-step':
+        seenStartedIds.clear();
+        seenStartedToolCallIds.clear();
+        return false;
       case 'text-start':
       case 'reasoning-start':
         seenStartedIds.add(chunk.id);