@ai-sdk/workflow 2.0.37 → 2.0.38

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/workflow",
-  "version": "2.0.37",
+  "version": "2.0.38",
   "type": "module",
   "description": "WorkflowAgent for building AI agents with AI SDK",
   "license": "Apache-2.0",
@@ -37,8 +37,8 @@
   },
   "dependencies": {
     "@ai-sdk/provider": "4.0.17",
-    "@ai-sdk/provider-utils": "5.0.44",
-    "ai": "7.0.106",
+    "@ai-sdk/provider-utils": "5.0.45",
+    "ai": "7.0.107",
     "ajv": "^8.20.0"
   },
   "devDependencies": {

package/src/stream-text-iterator.ts

@@ -19,7 +19,10 @@ import {
   type ToolChoice,
   type ToolSet,
 } from 'ai';
-import { createRestrictedTelemetryDispatcher } from 'ai/internal';
+import {
+  createRestrictedTelemetryDispatcher,
+  createToolSearchState,
+} from 'ai/internal';
 import {
   type DoStreamStepRawResult,
   doStreamStep,
@@ -86,6 +89,8 @@ function mergePrepareStepGenerationSettings(
 export interface StreamTextIteratorYieldValue {
   /** The tool calls requested by the model (parsed with typed inputs) */
   toolCalls: ParsedToolCall[];
+  /** The tools available for execution in the current step. */
+  tools?: ToolSet;
   /** The conversation messages up to (and including) the tool call request */
   messages: LanguageModelV4Prompt;
   /** The step result from the current step */
@@ -186,6 +191,10 @@ export async function* streamTextIterator({
   let wasAborted = false;
   let terminalError: unknown;
   let hasTerminalError = false;
+  const prepareToolSearch = createToolSearchState({
+    tools,
+    toolCallers: undefined,
+  });
 
   // TODO(#12164): replace this AI-core telemetry bridge with a
   // WorkflowAgent-specific typed dispatcher. `streamTextIterator` widens
@@ -306,14 +315,15 @@ export async function* streamTextIterator({
     });
 
     try {
-      // Filter tools if activeTools is specified
+      const stepActiveTools = filterActiveTools({
+        tools,
+        activeTools: currentActiveTools,
+      });
       const effectiveTools =
-        currentActiveTools !== undefined
-          ? (filterActiveTools({
-              tools,
-              activeTools: currentActiveTools,
-            }) ?? tools)
-          : tools;
+        prepareToolSearch(stepActiveTools, {
+          toolsContext: currentToolsContext as never,
+          experimental_sandbox: stepSandbox,
+        }) ?? {};
 
       // Serialize tools before crossing the step boundary — zod schemas
       // contain functions that can't be serialized by the workflow runtime.
@@ -484,6 +494,7 @@ export async function* streamTextIterator({
         // Also include provider-executed tool results so they can be used instead of local execution
         const toolResults = yield {
           toolCalls,
+          tools: effectiveTools,
           messages: conversationPrompt,
           step,
           runtimeContext: currentRuntimeContext,

package/src/workflow-agent.ts

@@ -2350,6 +2350,7 @@ export class WorkflowAgent<
 
         const {
           toolCalls,
+          tools: stepTools = effectiveTools as ToolSet,
           messages: iterMessages,
           step,
           runtimeContext: yieldedRuntimeContext,
@@ -2415,7 +2416,7 @@ export class WorkflowAgent<
           // Check which tools need approval (can be async)
           const approvalNeeded = await Promise.all(
             nonProviderToolCalls.map(async tc => {
-              const tool = (effectiveTools as ToolSet)[tc.toolName];
+              const tool = stepTools[tc.toolName];
               if (!tool) return false;
               if (tool.needsApproval == null) return false;
               if (typeof tool.needsApproval === 'boolean')
@@ -2439,14 +2440,14 @@ export class WorkflowAgent<
           // - paused: no execute function (client-side) OR needs approval
           // Note: missing tools (!tool) are left to executeTool which will throw.
           const executableToolCalls = nonProviderToolCalls.filter((tc, i) => {
-            const tool = (effectiveTools as ToolSet)[tc.toolName];
+            const tool = stepTools[tc.toolName];
             return (
               (!tool || typeof tool.execute === 'function') &&
               !approvalNeeded[i]
             );
           });
           const pausedToolCalls = nonProviderToolCalls.filter((tc, i) => {
-            const tool = (effectiveTools as ToolSet)[tc.toolName];
+            const tool = stepTools[tc.toolName];
             return (
               (tool && typeof tool.execute !== 'function') || approvalNeeded[i]
             );
@@ -2463,7 +2464,7 @@ export class WorkflowAgent<
                 (toolCall): Promise<WorkflowToolExecutionResult> =>
                   executeToolWithCallbacks(
                     toolCall,
-                    effectiveTools as ToolSet,
+                    stepTools,
                     iterMessages,
                     toolsContext,
                     currentStepNumber,
@@ -2479,7 +2480,7 @@ export class WorkflowAgent<
                 result: await resolveProviderToolResult(
                   toolCall,
                   capturedProviderToolResults,
-                  effectiveTools as ToolSet,
+                  stepTools,
                   download,
                 ),
               })),
@@ -2657,7 +2658,7 @@ export class WorkflowAgent<
               (toolCall): Promise<WorkflowToolExecutionResult> =>
                 executeToolWithCallbacks(
                   toolCall,
-                  effectiveTools as ToolSet,
+                  stepTools,
                   iterMessages,
                   toolsContext,
                   currentStepNumber,
@@ -2673,7 +2674,7 @@ export class WorkflowAgent<
               result: await resolveProviderToolResult(
                 toolCall,
                 capturedProviderToolResults,
-                effectiveTools as ToolSet,
+                stepTools,
                 download,
               ),
             })),