ai 5.0.0-alpha.3 → 5.0.0-alpha.4

package/dist/index.mjs

@@ -1586,18 +1586,18 @@ var SerialJobExecutor = class {
 function shouldResubmitMessages({
   originalMaxToolInvocationStep,
   originalMessageCount,
-  maxSteps: maxSteps2,
+  maxSteps,
   messages
 }) {
   var _a17;
   const lastMessage = messages[messages.length - 1];
   return (
     // check if the feature is enabled:
-    maxSteps2 > 1 && // ensure there is a last message:
+    maxSteps > 1 && // ensure there is a last message:
     lastMessage != null && // ensure we actually have new steps (to prevent infinite loops in case of errors):
     (messages.length > originalMessageCount || extractMaxToolInvocationStep(getToolInvocations(lastMessage)) !== originalMaxToolInvocationStep) && // check that next step is possible:
     isAssistantMessageWithCompletedToolCalls(lastMessage) && // limit the number of automatic steps:
-    ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) < maxSteps2
+    ((_a17 = extractMaxToolInvocationStep(getToolInvocations(lastMessage))) != null ? _a17 : 0) < maxSteps
   );
 }
 function isAssistantMessageWithCompletedToolCalls(message) {
@@ -1637,7 +1637,7 @@ var ChatStore = class {
     chats = {},
     generateId: generateId3,
     transport,
-    maxSteps: maxSteps2 = 1,
+    maxSteps = 1,
     messageMetadataSchema,
     dataPartSchemas
   }) {
@@ -1653,7 +1653,7 @@ var ChatStore = class {
         }
       ])
     );
-    this.maxSteps = maxSteps2;
+    this.maxSteps = maxSteps;
     this.transport = transport;
     this.subscribers = /* @__PURE__ */ new Set();
     this.generateId = generateId3 != null ? generateId3 : generateIdFunc;
@@ -2244,7 +2244,7 @@ function defaultChatStore({
   generateId: generateId3 = generateIdFunc2,
   dataPartSchemas,
   messageMetadataSchema,
-  maxSteps: maxSteps2 = 1,
+  maxSteps = 1,
   chats
 }) {
   return new ChatStore({
@@ -2259,7 +2259,7 @@ function defaultChatStore({
     generateId: generateId3,
     messageMetadataSchema,
     dataPartSchemas,
-    maxSteps: maxSteps2,
+    maxSteps,
     chats
   });
 }
@@ -5403,6 +5403,11 @@ var DefaultSpeechResult = class {
 // core/generate-text/generate-text.ts
 import { createIdGenerator as createIdGenerator3 } from "@ai-sdk/provider-utils";
 
+// src/util/as-array.ts
+function asArray(value) {
+  return value === void 0 ? [] : Array.isArray(value) ? value : [value];
+}
+
 // core/prompt/prepare-tools-and-tool-choice.ts
 import { asSchema as asSchema2 } from "@ai-sdk/provider-utils";
 
@@ -5624,8 +5629,8 @@ var DefaultStepResult = class {
 };
 
 // core/generate-text/stop-condition.ts
-function maxSteps(maxSteps2) {
-  return ({ steps }) => steps.length >= maxSteps2;
+function stepCountIs(stepCount) {
+  return ({ steps }) => steps.length === stepCount;
 }
 function hasToolCall(toolName) {
   return ({ steps }) => {
@@ -5635,6 +5640,12 @@ function hasToolCall(toolName) {
     )) != null ? _c : false;
   };
 }
+async function isStopConditionMet({
+  stopConditions,
+  steps
+}) {
+  return (await Promise.all(stopConditions.map((condition) => condition({ steps })))).some((result) => result);
+}
 
 // core/generate-text/to-response-messages.ts
 function toResponseMessages({
@@ -5709,12 +5720,14 @@ async function generateText({
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
-  continueUntil = maxSteps(1),
+  stopWhen = stepCountIs(1),
   experimental_output: output,
   experimental_telemetry: telemetry,
   providerOptions,
-  experimental_activeTools: activeTools,
-  experimental_prepareStep: prepareStep,
+  experimental_activeTools,
+  activeTools = experimental_activeTools,
+  experimental_prepareStep,
+  prepareStep = experimental_prepareStep,
   experimental_repairToolCall: repairToolCall,
   _internal: {
     generateId: generateId3 = originalGenerateId3,
@@ -5723,6 +5736,7 @@ async function generateText({
   onStepFinish,
   ...settings
 }) {
+  const stopConditions = asArray(stopWhen);
   const { maxRetries, retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const callSettings = prepareCallSettings(settings);
   const baseTelemetryAttributes = getBaseTelemetryAttributes({
@@ -5786,7 +5800,7 @@ async function generateText({
         const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
           tools,
           toolChoice: (_b = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _b : toolChoice,
-          activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.experimental_activeTools) != null ? _c : activeTools
+          activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _c : activeTools
         });
         currentModelResponse = await retry(
           () => {
@@ -5931,8 +5945,8 @@ async function generateText({
       } while (
         // there are tool calls:
         currentToolCalls.length > 0 && // all current tool calls have results:
-        currentToolResults.length === currentToolCalls.length && // continue until the stop condition is met:
-        !await continueUntil({ steps })
+        currentToolResults.length === currentToolCalls.length && // continue until a stop condition is met:
+        !await isStopConditionMet({ stopConditions, steps })
       );
       span.setAttributes(
         selectTelemetryAttributes({
@@ -6282,11 +6296,6 @@ function smoothStream({
 // core/generate-text/stream-text.ts
 import { createIdGenerator as createIdGenerator4 } from "@ai-sdk/provider-utils";
 
-// src/util/as-array.ts
-function asArray(value) {
-  return value === void 0 ? [] : Array.isArray(value) ? value : [value];
-}
-
 // core/generate-text/run-tools-transformation.ts
 import { generateId } from "@ai-sdk/provider-utils";
 function runToolsTransformation({
@@ -6497,13 +6506,15 @@ function streamText({
   maxRetries,
   abortSignal,
   headers,
-  continueUntil = maxSteps(1),
+  stopWhen = stepCountIs(1),
   experimental_output: output,
   experimental_telemetry: telemetry,
+  prepareStep,
   providerOptions,
   experimental_toolCallStreaming = false,
   toolCallStreaming = experimental_toolCallStreaming,
-  experimental_activeTools: activeTools,
+  experimental_activeTools,
+  activeTools = experimental_activeTools,
   experimental_repairToolCall: repairToolCall,
   experimental_transform: transform,
   onChunk,
@@ -6533,9 +6544,10 @@ function streamText({
     transforms: asArray(transform),
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions: asArray(stopWhen),
     output,
     providerOptions,
+    prepareStep,
     onChunk,
     onError,
     onFinish,
@@ -6610,9 +6622,10 @@ var DefaultStreamTextResult = class {
     transforms,
     activeTools,
     repairToolCall,
-    continueUntil,
+    stopConditions,
     output,
     providerOptions,
+    prepareStep,
     now: now2,
     currentDate,
     generateId: generateId3,
@@ -6831,6 +6844,7 @@ var DefaultStreamTextResult = class {
           responseMessages,
           usage
         }) {
+          var _a17, _b, _c;
           stepFinish = new DelayedPromise();
           const initialPrompt = await standardizePrompt({
             system,
@@ -6841,6 +6855,11 @@ var DefaultStreamTextResult = class {
             ...initialPrompt.messages,
             ...responseMessages
           ];
+          const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
+            model,
+            steps: recordedSteps,
+            stepNumber: recordedSteps.length
+          }));
           const promptMessages = await convertToLanguageModelPrompt({
             prompt: {
               system: initialPrompt.system,
@@ -6848,9 +6867,12 @@ var DefaultStreamTextResult = class {
             },
             supportedUrls: await model.supportedUrls
           });
-          const toolsAndToolChoice = {
-            ...prepareToolsAndToolChoice({ tools, toolChoice, activeTools })
-          };
+          const stepModel = (_a17 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a17 : model;
+          const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
+            tools,
+            toolChoice: (_b = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _b : toolChoice,
+            activeTools: (_c = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _c : activeTools
+          });
           const {
             result: { stream: stream2, response, request },
             doStreamSpan,
@@ -6866,24 +6888,23 @@ var DefaultStreamTextResult = class {
                     telemetry
                   }),
                   ...baseTelemetryAttributes,
+                  // model:
+                  "ai.model.provider": stepModel.provider,
+                  "ai.model.id": stepModel.modelId,
+                  // prompt:
                   "ai.prompt.messages": {
                     input: () => JSON.stringify(promptMessages)
                   },
                   "ai.prompt.tools": {
                     // convert the language model level tools:
-                    input: () => {
-                      var _a17;
-                      return (_a17 = toolsAndToolChoice.tools) == null ? void 0 : _a17.map(
-                        (tool2) => JSON.stringify(tool2)
-                      );
-                    }
+                    input: () => stepTools == null ? void 0 : stepTools.map((tool2) => JSON.stringify(tool2))
                   },
                   "ai.prompt.toolChoice": {
-                    input: () => toolsAndToolChoice.toolChoice != null ? JSON.stringify(toolsAndToolChoice.toolChoice) : void 0
+                    input: () => stepToolChoice != null ? JSON.stringify(stepToolChoice) : void 0
                   },
                   // standardized gen-ai llm span attributes:
-                  "gen_ai.system": model.provider,
-                  "gen_ai.request.model": model.modelId,
+                  "gen_ai.system": stepModel.provider,
+                  "gen_ai.request.model": stepModel.modelId,
                   "gen_ai.request.frequency_penalty": callSettings.frequencyPenalty,
                   "gen_ai.request.max_tokens": callSettings.maxOutputTokens,
                   "gen_ai.request.presence_penalty": callSettings.presencePenalty,
@@ -6900,9 +6921,10 @@ var DefaultStreamTextResult = class {
                   startTimestampMs: now2(),
                   // get before the call
                   doStreamSpan: doStreamSpan2,
-                  result: await model.doStream({
+                  result: await stepModel.doStream({
                     ...callSettings,
-                    ...toolsAndToolChoice,
+                    tools: stepTools,
+                    toolChoice: stepToolChoice,
                     responseFormat: output == null ? void 0 : output.responseFormat,
                     prompt: promptMessages,
                     providerOptions,
@@ -6955,7 +6977,7 @@ var DefaultStreamTextResult = class {
             streamWithToolResults.pipeThrough(
               new TransformStream({
                 async transform(chunk, controller) {
-                  var _a17, _b, _c, _d;
+                  var _a18, _b2, _c2, _d;
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
                     return;
@@ -7018,9 +7040,9 @@ var DefaultStreamTextResult = class {
                     }
                     case "response-metadata": {
                       stepResponse = {
-                        id: (_a17 = chunk.id) != null ? _a17 : stepResponse.id,
-                        timestamp: (_b = chunk.timestamp) != null ? _b : stepResponse.timestamp,
-                        modelId: (_c = chunk.modelId) != null ? _c : stepResponse.modelId
+                        id: (_a18 = chunk.id) != null ? _a18 : stepResponse.id,
+                        timestamp: (_b2 = chunk.timestamp) != null ? _b2 : stepResponse.timestamp,
+                        modelId: (_c2 = chunk.modelId) != null ? _c2 : stepResponse.modelId
                       };
                       break;
                     }
@@ -7109,7 +7131,11 @@ var DefaultStreamTextResult = class {
                   const combinedUsage = addLanguageModelUsage(usage, stepUsage);
                   await stepFinish.promise;
                   if (stepToolCalls.length > 0 && // all current tool calls have results:
-                  stepToolResults.length === stepToolCalls.length && !await continueUntil({ steps: recordedSteps })) {
+                  stepToolResults.length === stepToolCalls.length && // continue until a stop condition is met:
+                  !await isStopConditionMet({
+                    stopConditions,
+                    steps: recordedSteps
+                  })) {
                     responseMessages.push(
                       ...toResponseMessages({
                         content: stepContent,
@@ -8601,7 +8627,6 @@ export {
   isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData,
   jsonSchema2 as jsonSchema,
-  maxSteps,
   modelMessageSchema,
   parsePartialJson,
   pipeTextStreamToResponse,
@@ -8610,6 +8635,7 @@ export {
   simulateReadableStream,
   simulateStreamingMiddleware,
   smoothStream,
+  stepCountIs,
   streamObject,
   streamText,
   systemModelMessageSchema,