@ai-sdk/workflow 1.0.0-beta.2 → 1.0.0-beta.21

package/dist/index.mjs

@@ -1,6 +1,7 @@
 // src/workflow-agent.ts
 import {
-  Output
+  Output,
+  experimental_filterActiveTools as filterActiveTools2
 } from "ai";
 import {
   convertToLanguageModelPrompt,
@@ -9,6 +10,11 @@ import {
   standardizePrompt
 } from "ai/internal";
 
+// src/stream-text-iterator.ts
+import {
+  experimental_filterActiveTools as filterActiveTools
+} from "ai";
+
 // src/do-stream-step.ts
 import {
   experimental_streamLanguageModelCall as streamModelCall
@@ -77,18 +83,7 @@ function resolveSerializableTools(tools) {
 async function doStreamStep(conversationPrompt, modelInit, writable, serializedTools, options) {
   "use step";
   var _a, _b, _c, _d, _e, _f, _g, _h, _i;
-  let model;
-  if (typeof modelInit === "string") {
-    model = gateway.languageModel(modelInit);
-  } else if (typeof modelInit === "function") {
-    model = await modelInit();
-  } else if (typeof modelInit === "object" && modelInit !== null && "modelId" in modelInit) {
-    model = modelInit;
-  } else {
-    throw new Error(
-      'Invalid "model initialization" argument. Must be a string, a LanguageModel instance, or a function that returns a LanguageModel instance.'
-    );
-  }
+  const model = typeof modelInit === "string" ? gateway.languageModel(modelInit) : modelInit;
   const tools = serializedTools ? resolveSerializableTools(serializedTools) : void 0;
   const { stream: modelStream } = await streamModelCall({
     model,
@@ -198,7 +193,8 @@ async function doStreamStep(conversationPrompt, modelInit, writable, serializedT
     },
     functionId: void 0,
     metadata: void 0,
-    context: void 0,
+    runtimeContext: void 0,
+    toolsContext: {},
     content: [
       ...text ? [{ type: "text", text }] : [],
       ...toolCalls.filter((tc) => !tc.invalid).map((tc) => ({
@@ -276,7 +272,6 @@ async function* streamTextIterator({
   writable,
   model,
   stopConditions,
-  maxSteps,
   onStepFinish,
   onStepStart,
   onError,
@@ -289,7 +284,7 @@ async function* streamTextIterator({
   repairToolCall,
   responseFormat
 }) {
-  var _a;
+  var _a, _b;
   let conversationPrompt = [...prompt];
   let currentModel = model;
   let currentGenerationSettings = generationSettings != null ? generationSettings : {};
@@ -302,11 +297,7 @@ async function* streamTextIterator({
   let stepNumber = 0;
   let lastStep;
   let lastStepWasToolCalls = false;
-  const effectiveMaxSteps = maxSteps != null ? maxSteps : Infinity;
   while (!done) {
-    if (stepNumber >= effectiveMaxSteps) {
-      break;
-    }
     if ((_a = currentGenerationSettings.abortSignal) == null ? void 0 : _a.aborted) {
       break;
     }
@@ -417,11 +408,15 @@ async function* streamTextIterator({
       await onStepStart({
         stepNumber,
         model: currentModel,
-        messages: conversationPrompt
+        messages: conversationPrompt,
+        steps: [...steps]
       });
     }
     try {
-      const effectiveTools = currentActiveTools && currentActiveTools.length > 0 ? filterToolSet(tools, currentActiveTools) : tools;
+      const effectiveTools = currentActiveTools && currentActiveTools.length > 0 ? (_b = filterActiveTools({
+        tools,
+        activeTools: currentActiveTools
+      })) != null ? _b : tools : tools;
       const serializedTools = serializeToolSet(effectiveTools);
       const { toolCalls, finish, step, providerExecutedToolResults } = await doStreamStep(
         conversationPrompt,
@@ -525,15 +520,6 @@ async function* streamTextIterator({
   }
   return conversationPrompt;
 }
-function filterToolSet(tools, activeTools) {
-  const filtered = {};
-  for (const toolName of activeTools) {
-    if (toolName in tools) {
-      filtered[toolName] = tools[toolName];
-    }
-  }
-  return filtered;
-}
 function sanitizeProviderMetadataForToolCall(metadata) {
   if (metadata == null) return void 0;
   const meta = metadata;
@@ -559,12 +545,18 @@ function sanitizeProviderMetadataForToolCall(metadata) {
 var WorkflowAgent = class {
   constructor(options) {
     var _a, _b;
+    this.id = options.id;
     this.model = options.model;
     this.tools = (_a = options.tools) != null ? _a : {};
     this.instructions = (_b = options.instructions) != null ? _b : options.system;
     this.toolChoice = options.toolChoice;
     this.telemetry = options.experimental_telemetry;
     this.experimentalContext = options.experimental_context;
+    this.stopWhen = options.stopWhen;
+    this.activeTools = options.activeTools;
+    this.output = options.output;
+    this.experimentalRepairToolCall = options.experimental_repairToolCall;
+    this.experimentalDownload = options.experimental_download;
     this.prepareStep = options.prepareStep;
     this.constructorOnStepFinish = options.onStepFinish;
     this.constructorOnFinish = options.onFinish;
@@ -592,14 +584,16 @@ var WorkflowAgent = class {
     throw new Error("Not implemented");
   }
   async stream(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x, _y, _z, _A, _B;
     let effectiveModel = this.model;
     let effectiveInstructions = (_a = options.system) != null ? _a : this.instructions;
+    let effectivePrompt = options.prompt;
     let effectiveMessages = options.messages;
     let effectiveGenerationSettings = { ...this.generationSettings };
     let effectiveExperimentalContext = (_b = options.experimental_context) != null ? _b : this.experimentalContext;
     let effectiveToolChoiceFromPrepare = (_c = options.toolChoice) != null ? _c : this.toolChoice;
     let effectiveTelemetryFromPrepare = (_d = options.experimental_telemetry) != null ? _d : this.telemetry;
+    const resolvedMessagesForPrepareCall = (_e = effectiveMessages != null ? effectiveMessages : typeof effectivePrompt === "string" ? [{ role: "user", content: effectivePrompt }] : effectivePrompt) != null ? _e : [];
     if (this.prepareCall) {
       const prepared = await this.prepareCall({
         model: effectiveModel,
@@ -608,14 +602,16 @@ var WorkflowAgent = class {
         toolChoice: effectiveToolChoiceFromPrepare,
         experimental_telemetry: effectiveTelemetryFromPrepare,
         experimental_context: effectiveExperimentalContext,
-        messages: effectiveMessages,
+        messages: resolvedMessagesForPrepareCall,
         ...effectiveGenerationSettings
       });
       if (prepared.model !== void 0) effectiveModel = prepared.model;
       if (prepared.instructions !== void 0)
         effectiveInstructions = prepared.instructions;
-      if (prepared.messages !== void 0)
+      if (prepared.messages !== void 0) {
         effectiveMessages = prepared.messages;
+        effectivePrompt = void 0;
+      }
       if (prepared.experimental_context !== void 0)
         effectiveExperimentalContext = prepared.experimental_context;
       if (prepared.toolChoice !== void 0)
@@ -645,7 +641,7 @@ var WorkflowAgent = class {
     }
     const prompt = await standardizePrompt({
       system: effectiveInstructions,
-      messages: effectiveMessages
+      ...effectivePrompt != null ? { prompt: effectivePrompt } : { messages: effectiveMessages }
     });
     const { approvedToolApprovals, deniedToolApprovals } = collectToolApprovalsFromMessages(prompt.messages);
     if (approvedToolApprovals.length > 0 || deniedToolApprovals.length > 0) {
@@ -741,10 +737,10 @@ var WorkflowAgent = class {
     const modelPrompt = await convertToLanguageModelPrompt({
       prompt,
       supportedUrls: {},
-      download: options.experimental_download
+      download: (_f = options.experimental_download) != null ? _f : this.experimentalDownload
     });
     const effectiveAbortSignal = mergeAbortSignals(
-      (_e = options.abortSignal) != null ? _e : effectiveGenerationSettings.abortSignal,
+      (_g = options.abortSignal) != null ? _g : effectiveGenerationSettings.abortSignal,
       options.timeout != null ? AbortSignal.timeout(options.timeout) : void 0
     );
     const mergedGenerationSettings = {
@@ -804,7 +800,11 @@ var WorkflowAgent = class {
     );
     const effectiveToolChoice = effectiveToolChoiceFromPrepare;
     const effectiveTelemetry = effectiveTelemetryFromPrepare;
-    const effectiveTools = options.activeTools && options.activeTools.length > 0 ? filterTools(this.tools, options.activeTools) : this.tools;
+    const effectiveActiveTools = (_h = options.activeTools) != null ? _h : this.activeTools;
+    const effectiveTools = effectiveActiveTools && effectiveActiveTools.length > 0 ? (_i = filterActiveTools2({
+      tools: this.tools,
+      activeTools: effectiveActiveTools
+    })) != null ? _i : this.tools : this.tools;
     let experimentalContext = effectiveExperimentalContext;
     const steps = [];
     let lastStepToolCalls = [];
@@ -812,61 +812,68 @@ var WorkflowAgent = class {
     if (mergedOnStart) {
       await mergedOnStart({
         model: effectiveModel,
-        messages: effectiveMessages
+        messages: prompt.messages
       });
     }
-    const executeToolWithCallbacks = async (toolCall, tools, messages2, context) => {
+    const executeToolWithCallbacks = async (toolCall, tools, messages2, context, currentStepNumber = 0) => {
+      const toolCallEvent = {
+        type: "tool-call",
+        toolCallId: toolCall.toolCallId,
+        toolName: toolCall.toolName,
+        input: toolCall.input
+      };
       if (mergedOnToolCallStart) {
         await mergedOnToolCallStart({
-          toolCall: {
-            type: "tool-call",
-            toolCallId: toolCall.toolCallId,
-            toolName: toolCall.toolName,
-            input: toolCall.input
-          }
+          toolCall: toolCallEvent,
+          stepNumber: currentStepNumber
         });
       }
+      const startTime = Date.now();
       let result;
       try {
         result = await executeTool(toolCall, tools, messages2, context);
       } catch (err) {
+        const durationMs2 = Date.now() - startTime;
         if (mergedOnToolCallFinish) {
           await mergedOnToolCallFinish({
-            toolCall: {
-              type: "tool-call",
-              toolCallId: toolCall.toolCallId,
-              toolName: toolCall.toolName,
-              input: toolCall.input
-            },
+            toolCall: toolCallEvent,
+            stepNumber: currentStepNumber,
+            durationMs: durationMs2,
+            success: false,
             error: err
           });
         }
         throw err;
       }
+      const durationMs = Date.now() - startTime;
       if (mergedOnToolCallFinish) {
         const isError = result.output && "type" in result.output && (result.output.type === "error-text" || result.output.type === "error-json");
-        await mergedOnToolCallFinish({
-          toolCall: {
-            type: "tool-call",
-            toolCallId: toolCall.toolCallId,
-            toolName: toolCall.toolName,
-            input: toolCall.input
-          },
-          ...isError ? {
+        if (isError) {
+          await mergedOnToolCallFinish({
+            toolCall: toolCallEvent,
+            stepNumber: currentStepNumber,
+            durationMs,
+            success: false,
             error: "value" in result.output ? result.output.value : void 0
-          } : {
-            result: result.output && "value" in result.output ? result.output.value : void 0
-          }
-        });
+          });
+        } else {
+          await mergedOnToolCallFinish({
+            toolCall: toolCallEvent,
+            stepNumber: currentStepNumber,
+            durationMs,
+            success: true,
+            output: result.output && "value" in result.output ? result.output.value : void 0
+          });
+        }
       }
       return result;
     };
-    if ((_f = mergedGenerationSettings.abortSignal) == null ? void 0 : _f.aborted) {
+    if ((_j = mergedGenerationSettings.abortSignal) == null ? void 0 : _j.aborted) {
       if (options.onAbort) {
         await options.onAbort({ steps });
       }
       return {
-        messages: options.messages,
+        messages: prompt.messages,
         steps,
         toolCalls: [],
         toolResults: [],
@@ -878,19 +885,18 @@ var WorkflowAgent = class {
       tools: effectiveTools,
       writable: options.writable,
       prompt: modelPrompt,
-      stopConditions: options.stopWhen,
-      maxSteps: options.maxSteps,
+      stopConditions: (_k = options.stopWhen) != null ? _k : this.stopWhen,
       onStepFinish: mergedOnStepFinish,
       onStepStart: mergedOnStepStart,
       onError: options.onError,
-      prepareStep: (_g = options.prepareStep) != null ? _g : this.prepareStep,
+      prepareStep: (_l = options.prepareStep) != null ? _l : this.prepareStep,
       generationSettings: mergedGenerationSettings,
       toolChoice: effectiveToolChoice,
       experimental_context: experimentalContext,
       experimental_telemetry: effectiveTelemetry,
-      includeRawChunks: (_h = options.includeRawChunks) != null ? _h : false,
-      repairToolCall: options.experimental_repairToolCall,
-      responseFormat: await ((_i = options.output) == null ? void 0 : _i.responseFormat)
+      includeRawChunks: (_m = options.includeRawChunks) != null ? _m : false,
+      repairToolCall: (_n = options.experimental_repairToolCall) != null ? _n : this.experimentalRepairToolCall,
+      responseFormat: await ((_p = (_o = options.output) != null ? _o : this.output) == null ? void 0 : _p.responseFormat)
     });
     let finalMessages;
     let encounteredError;
@@ -898,7 +904,7 @@ var WorkflowAgent = class {
     try {
       let result = await iterator.next();
       while (!result.done) {
-        if ((_j = mergedGenerationSettings.abortSignal) == null ? void 0 : _j.aborted) {
+        if ((_q = mergedGenerationSettings.abortSignal) == null ? void 0 : _q.aborted) {
           wasAborted = true;
           if (options.onAbort) {
             await options.onAbort({ steps });
@@ -912,6 +918,7 @@ var WorkflowAgent = class {
           context,
           providerExecutedToolResults
         } = result.value;
+        const currentStepNumber = steps.length;
         if (step) {
           steps.push(step);
         }
@@ -952,7 +959,8 @@ var WorkflowAgent = class {
                   toolCall,
                   effectiveTools,
                   iterMessages,
-                  experimentalContext
+                  experimentalContext,
+                  currentStepNumber
                 )
               )
             );
@@ -991,8 +999,8 @@ var WorkflowAgent = class {
               await mergedOnFinish({
                 steps,
                 messages: messages2,
-                text: (_k = lastStep == null ? void 0 : lastStep.text) != null ? _k : "",
-                finishReason: (_l = lastStep == null ? void 0 : lastStep.finishReason) != null ? _l : "other",
+                text: (_r = lastStep == null ? void 0 : lastStep.text) != null ? _r : "",
+                finishReason: (_s = lastStep == null ? void 0 : lastStep.finishReason) != null ? _s : "other",
                 totalUsage: aggregateUsage(steps),
                 experimental_context: experimentalContext,
                 output: void 0
@@ -1016,8 +1024,8 @@ var WorkflowAgent = class {
               }
             }
             if (options.writable) {
-              const sendFinish = (_m = options.sendFinish) != null ? _m : true;
-              const preventClose = (_n = options.preventClose) != null ? _n : false;
+              const sendFinish = (_t = options.sendFinish) != null ? _t : true;
+              const preventClose = (_u = options.preventClose) != null ? _u : false;
               if (sendFinish || !preventClose) {
                 await closeStream(options.writable, preventClose, sendFinish);
               }
@@ -1036,7 +1044,8 @@ var WorkflowAgent = class {
                 toolCall,
                 effectiveTools,
                 iterMessages,
-                experimentalContext
+                experimentalContext,
+                currentStepNumber
               )
             )
           );
@@ -1110,14 +1119,15 @@ var WorkflowAgent = class {
         await options.onError({ error });
       }
     }
-    const messages = finalMessages != null ? finalMessages : options.messages;
+    const messages = finalMessages != null ? finalMessages : prompt.messages;
+    const effectiveOutput = (_v = options.output) != null ? _v : this.output;
     let experimentalOutput = void 0;
-    if (options.output && steps.length > 0) {
+    if (effectiveOutput && steps.length > 0) {
       const lastStep = steps[steps.length - 1];
       const text = lastStep.text;
       if (text) {
         try {
-          experimentalOutput = await options.output.parseCompleteOutput(
+          experimentalOutput = await effectiveOutput.parseCompleteOutput(
             { text },
             {
               response: lastStep.response,
@@ -1137,8 +1147,8 @@ var WorkflowAgent = class {
       await mergedOnFinish({
         steps,
         messages,
-        text: (_o = lastStep == null ? void 0 : lastStep.text) != null ? _o : "",
-        finishReason: (_p = lastStep == null ? void 0 : lastStep.finishReason) != null ? _p : "other",
+        text: (_w = lastStep == null ? void 0 : lastStep.text) != null ? _w : "",
+        finishReason: (_x = lastStep == null ? void 0 : lastStep.finishReason) != null ? _x : "other",
         totalUsage: aggregateUsage(steps),
         experimental_context: experimentalContext,
         output: experimentalOutput
@@ -1146,8 +1156,8 @@ var WorkflowAgent = class {
     }
     if (encounteredError) {
       if (options.writable) {
-        const sendFinish = (_q = options.sendFinish) != null ? _q : true;
-        const preventClose = (_r = options.preventClose) != null ? _r : false;
+        const sendFinish = (_y = options.sendFinish) != null ? _y : true;
+        const preventClose = (_z = options.preventClose) != null ? _z : false;
         if (sendFinish || !preventClose) {
           await closeStream(options.writable, preventClose, sendFinish);
         }
@@ -1155,8 +1165,8 @@ var WorkflowAgent = class {
       throw encounteredError;
     }
     if (options.writable) {
-      const sendFinish = (_s = options.sendFinish) != null ? _s : true;
-      const preventClose = (_t = options.preventClose) != null ? _t : false;
+      const sendFinish = (_A = options.sendFinish) != null ? _A : true;
+      const preventClose = (_B = options.preventClose) != null ? _B : false;
       if (sendFinish || !preventClose) {
         await closeStream(options.writable, preventClose, sendFinish);
       }
@@ -1257,15 +1267,6 @@ function aggregateUsage(steps) {
     totalTokens: inputTokens + outputTokens
   };
 }
-function filterTools(tools, activeTools) {
-  const filtered = {};
-  for (const toolName of activeTools) {
-    if (toolName in tools) {
-      filtered[toolName] = tools[toolName];
-    }
-  }
-  return filtered;
-}
 function getErrorMessage(error) {
   if (error == null) {
     return "unknown error";