@ai-sdk/workflow 2.0.70 → 2.0.71

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/workflow",
-  "version": "2.0.70",
+  "version": "2.0.71",
   "type": "module",
   "description": "WorkflowAgent for building AI agents with AI SDK",
   "license": "Apache-2.0",
@@ -38,11 +38,11 @@
   "dependencies": {
     "@ai-sdk/provider": "4.0.27",
     "@ai-sdk/provider-utils": "5.0.59",
-    "ai": "7.0.138",
+    "ai": "7.0.139",
     "ajv": "^8.20.0"
   },
   "devDependencies": {
-    "@ai-sdk/otel": "1.0.138",
+    "@ai-sdk/otel": "1.0.139",
     "@opentelemetry/api": "1.9.1",
     "@opentelemetry/sdk-trace-base": "2.7.1",
     "@types/node": "22.19.19",

package/src/do-generate-step.ts

@@ -7,6 +7,7 @@ import {
 } from 'ai';
 import {
   asLanguageModelUsage,
+  convertToLanguageModelPrompt,
   parseToolCall,
   prepareLanguageModelCallOptions,
   prepareRetries,
@@ -14,6 +15,7 @@ import {
   prepareTools,
   resolveLanguageModel,
 } from 'ai/internal';
+import { languageModelPromptToModelMessages } from './language-model-prompt-to-model-messages.js';
 import type {
   ModelCallOptions,
   ModelCallRawContentPart,
@@ -90,6 +92,16 @@ async function generateModelCall(
           AbortSignal.timeout(remaining),
         ]);
   const model = resolveLanguageModel(modelInit);
+  const modelPrompt = await convertToLanguageModelPrompt({
+    prompt: {
+      instructions: undefined,
+      messages: languageModelPromptToModelMessages(prompt),
+    },
+    supportedUrls: await model.supportedUrls,
+    download: options.download,
+    abortSignal,
+    provider: model.provider.split('.')[0],
+  });
   const tools = resolveSerializableTools(serializedTools);
   const toolChoice = prepareToolChoice({ toolChoice: options.toolChoice });
   const modelTools = await prepareTools({ tools });
@@ -125,7 +137,7 @@ async function generateModelCall(
       execute: () =>
         model.doGenerate({
           ...settings,
-          prompt: [...prompt],
+          prompt: modelPrompt,
           tools: modelTools,
           toolChoice,
           responseFormat: options.responseFormat,
@@ -159,7 +171,7 @@ async function generateModelCall(
           tools,
           repairToolCall: options.repairToolCall,
           instructions:
-            prompt
+            modelPrompt
               .filter(message => message.role === 'system')
               .map(message => message.content)
               .join('\n') || undefined,

package/src/do-stream-step.ts

@@ -10,10 +10,10 @@ import {
   NoOutputGeneratedError,
   wrapLanguageModel,
   type LanguageModel,
-  type ModelMessage,
   type ToolSet,
 } from 'ai';
 import { isOutputChunk, prepareRetries } from 'ai/internal';
+import { languageModelPromptToModelMessages } from './language-model-prompt-to-model-messages.js';
 import type { StreamTextTransform } from './workflow-agent.js';
 import {
   resolveSerializableTools,
@@ -178,21 +178,7 @@ async function streamModelStep(
           // streamModelCall expects Prompt (ModelMessage[]) but we pass the
           // pre-converted LanguageModelV4Prompt. standardizePrompt inside
           // streamModelCall handles both formats.
-          messages: conversationPrompt.map(message =>
-            message.role !== 'tool'
-              ? message
-              : {
-                  ...message,
-                  // Provider prompt approval responses have already been filtered by
-                  // convertToLanguageModelPrompt. Restore the marker expected by the
-                  // model-call helper when it converts these messages again.
-                  content: message.content.map(part =>
-                    part.type === 'tool-approval-response'
-                      ? { ...part, providerExecuted: true }
-                      : part,
-                  ),
-                },
-          ) as unknown as ModelMessage[],
+          messages: languageModelPromptToModelMessages(conversationPrompt),
           allowSystemInMessages: true,
           tools,
           toolChoice: options?.toolChoice,
@@ -211,6 +197,7 @@ async function streamModelStep(
           stopSequences: options?.stopSequences,
           seed: options?.seed,
           repairToolCall: options?.repairToolCall,
+          download: options?.download,
           ...languageModelCallTelemetry,
         }),
       );

package/src/language-model-prompt-to-model-messages.ts

@@ -0,0 +1,28 @@
+import type { LanguageModelV4Prompt } from '@ai-sdk/provider';
+import type { ModelMessage } from 'ai';
+
+/**
+ * Prepares a workflow-side `LanguageModelV4Prompt` for a second, model-aware
+ * `convertToLanguageModelPrompt` pass inside a model step.
+ *
+ * The workflow-side conversion preserves URLs (it cannot download them) and
+ * filters out approval responses for tools that are not provider-executed.
+ * The remaining approval responses are restored with the `providerExecuted`
+ * marker that the second conversion expects.
+ */
+export function languageModelPromptToModelMessages(
+  prompt: LanguageModelV4Prompt,
+): ModelMessage[] {
+  return prompt.map(message =>
+    message.role !== 'tool'
+      ? message
+      : {
+          ...message,
+          content: message.content.map(part =>
+            part.type === 'tool-approval-response'
+              ? { ...part, providerExecuted: true }
+              : part,
+          ),
+        },
+  ) as unknown as ModelMessage[];
+}

package/src/model-call-iterator.ts

@@ -143,6 +143,7 @@ export async function* modelCallIterator({
   includeRawChunks = false,
   timeoutAt,
   repairToolCall,
+  download,
   responseFormat,
   experimental_transform,
   experimental_sandbox: sandbox,
@@ -170,6 +171,7 @@ export async function* modelCallIterator({
   includeRawChunks?: boolean;
   timeoutAt?: number;
   repairToolCall?: ToolCallRepairFunction<ToolSet>;
+  download?: ModelCallOptions['download'];
   responseFormat?: LanguageModelV4CallOptions['responseFormat'];
   experimental_transform?:
     | StreamTextTransform<ToolSet>
@@ -380,6 +382,7 @@ export async function* modelCallIterator({
         includeRawChunks,
         timeoutAt,
         repairToolCall,
+        download,
         responseFormat,
         include,
         experimental_transform,

package/src/model-call.ts

@@ -15,7 +15,11 @@ import type {
   ToolChoice,
   ToolSet,
 } from 'ai';
-import type { ProviderOptions, StreamTextTransform } from './workflow-agent.js';
+import type {
+  DownloadFunction,
+  ProviderOptions,
+  StreamTextTransform,
+} from './workflow-agent.js';
 
 export type ModelCallStreamPart<TTools extends ToolSet = ToolSet> =
   | Experimental_LanguageModelStreamPart<TTools>
@@ -62,6 +66,7 @@ export interface ModelCallOptions {
   toolChoice?: ToolChoice<ToolSet>;
   includeRawChunks?: boolean;
   repairToolCall?: ToolCallRepairFunction<ToolSet>;
+  download?: DownloadFunction;
   include?: {
     requestBody?: boolean;
     responseBody?: boolean;

package/src/workflow-agent.ts

@@ -578,6 +578,7 @@ export type WorkflowAgentOptions<
 
     /**
      * Default custom download function to use for URLs.
+     * Mark the function with `'use step'` when used in a workflow.
      *
      * Per-stream `experimental_download` values passed to `stream()` override this default.
      */
@@ -1074,6 +1075,7 @@ export type WorkflowAgentCallOptions<
     /**
      * Custom download function to use for URLs.
      * By default, files are downloaded if the model does not support the URL for the given media type.
+     * Mark the function with `'use step'` when used in a workflow.
      */
     experimental_download?: DownloadFunction;
 
@@ -1834,7 +1836,8 @@ export class WorkflowAgent<
         ? { prompt: effectivePrompt }
         : { messages: effectiveMessages! }),
     } as Prompt);
-    const download = effectiveDownloadFromPrepare;
+    const workflowDownload: DownloadFunction = async requests =>
+      requests.map(() => null);
     const sandbox = options.experimental_sandbox ?? this.experimentalSandbox;
     // Model steps enforce the absolute deadline below. Avoid creating a native
     // timeout signal in the workflow VM, where timer APIs are unavailable.
@@ -1872,7 +1875,8 @@ export class WorkflowAgent<
       effectiveTelemetry,
       telemetryDispatcher,
       prompt,
-      download,
+      download: effectiveDownloadFromPrepare,
+      workflowDownload,
       sandbox,
       effectiveAbortSignal,
       timeoutAt,
@@ -1909,6 +1913,7 @@ export class WorkflowAgent<
       telemetryDispatcher,
       prompt,
       download,
+      workflowDownload,
       sandbox,
       effectiveAbortSignal,
       timeoutAt,
@@ -2003,7 +2008,7 @@ export class WorkflowAgent<
                 tool,
                 errorMode: 'text',
                 supportedUrls: {},
-                download,
+                download: workflowDownload,
               }),
             });
             continue;
@@ -2061,7 +2066,7 @@ export class WorkflowAgent<
                 tool,
                 errorMode: 'text',
                 supportedUrls: {},
-                download,
+                download: workflowDownload,
               }),
             });
             continue;
@@ -2183,7 +2188,7 @@ export class WorkflowAgent<
     const modelPrompt = await convertToLanguageModelPrompt({
       prompt,
       supportedUrls: {},
-      download,
+      download: workflowDownload,
     });
 
     // Merge generation settings: constructor defaults < prepareCall < stream options
@@ -2368,7 +2373,7 @@ export class WorkflowAgent<
             messages,
             resolvedContext,
             effectiveAbortSignal,
-            download,
+            workflowDownload,
             stepSandbox,
           );
         result =
@@ -2546,6 +2551,7 @@ export class WorkflowAgent<
       repairToolCall: (options.repairToolCall ??
         options.experimental_repairToolCall ??
         this.repairToolCall) as ToolCallRepairFunction<ToolSet> | undefined,
+      download,
       responseFormat: await (options.output ?? this.output)?.responseFormat,
       experimental_transform: options.experimental_transform as
         | StreamTextTransform<ToolSet>
@@ -2735,7 +2741,7 @@ export class WorkflowAgent<
                     toolCall,
                     capturedProviderToolResults,
                     stepTools,
-                    download,
+                    workflowDownload,
                   ),
                 })),
             );
@@ -2942,7 +2948,7 @@ export class WorkflowAgent<
                 toolCall,
                 capturedProviderToolResults,
                 stepTools,
-                download,
+                workflowDownload,
               ),
             })),
           );