ai 3.3.37 → 3.3.39

package/dist/index.mjs

@@ -2998,6 +2998,7 @@ async function generateText({
     generateId: generateId3 = originalGenerateId3,
     currentDate = () => /* @__PURE__ */ new Date()
   } = {},
+  onStepFinish,
   ...settings
 }) {
   var _a11;
@@ -3159,7 +3160,7 @@ async function generateText({
         usage.completionTokens += currentUsage.completionTokens;
         usage.promptTokens += currentUsage.promptTokens;
         usage.totalTokens += currentUsage.totalTokens;
-        steps.push({
+        const currentStep = {
           text: (_b = currentModelResponse.text) != null ? _b : "",
           toolCalls: currentToolCalls,
           toolResults: currentToolResults,
@@ -3170,8 +3171,11 @@ async function generateText({
           response: {
             ...currentModelResponse.response,
             headers: (_c = currentModelResponse.rawResponse) == null ? void 0 : _c.headers
-          }
-        });
+          },
+          experimental_providerMetadata: currentModelResponse.providerMetadata
+        };
+        steps.push(currentStep);
+        await (onStepFinish == null ? void 0 : onStepFinish(currentStep));
         const newResponseMessages = toResponseMessages({
           text: currentModelResponse.text,
           toolCalls: currentToolCalls,
@@ -3676,6 +3680,7 @@ async function streamText({
   experimental_toolCallStreaming: toolCallStreaming = false,
   onChunk,
   onFinish,
+  onStepFinish,
   _internal: {
     now: now2 = now,
     generateId: generateId3 = originalGenerateId4,
@@ -3808,6 +3813,7 @@ async function streamText({
         rawResponse,
         onChunk,
         onFinish,
+        onStepFinish,
         rootSpan,
         doStreamSpan,
         telemetry,
@@ -3830,6 +3836,7 @@ var DefaultStreamTextResult = class {
     rawResponse,
     onChunk,
     onFinish,
+    onStepFinish,
     rootSpan,
     doStreamSpan,
     telemetry,
@@ -3982,26 +3989,7 @@ var DefaultStreamTextResult = class {
             },
             // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
             async flush(controller) {
-              controller.enqueue({
-                type: "step-finish",
-                finishReason: stepFinishReason,
-                usage: stepUsage,
-                experimental_providerMetadata: stepProviderMetadata,
-                logprobs: stepLogProbs,
-                response: stepResponse
-              });
-              stepResults.push({
-                text: stepText,
-                toolCalls: stepToolCalls,
-                toolResults: stepToolResults,
-                finishReason: stepFinishReason,
-                usage: stepUsage,
-                warnings: self.warnings,
-                logprobs: stepLogProbs,
-                response: stepResponse,
-                rawResponse: self.rawResponse
-              });
-              const telemetryToolCalls = stepToolCalls.length > 0 ? JSON.stringify(stepToolCalls) : void 0;
+              const stepToolCallsJson = stepToolCalls.length > 0 ? JSON.stringify(stepToolCalls) : void 0;
               try {
                 doStreamSpan2.setAttributes(
                   selectTelemetryAttributes({
@@ -4010,7 +3998,7 @@ var DefaultStreamTextResult = class {
                       "ai.response.finishReason": stepFinishReason,
                       "ai.response.text": { output: () => stepText },
                       "ai.response.toolCalls": {
-                        output: () => telemetryToolCalls
+                        output: () => stepToolCallsJson
                       },
                       "ai.response.id": stepResponse.id,
                       "ai.response.model": stepResponse.modelId,
@@ -4021,7 +4009,7 @@ var DefaultStreamTextResult = class {
                       "ai.finishReason": stepFinishReason,
                       "ai.result.text": { output: () => stepText },
                       "ai.result.toolCalls": {
-                        output: () => telemetryToolCalls
+                        output: () => stepToolCallsJson
                       },
                       // standardized gen-ai llm span attributes:
                       "gen_ai.response.finish_reasons": [stepFinishReason],
@@ -4036,6 +4024,28 @@ var DefaultStreamTextResult = class {
               } finally {
                 doStreamSpan2.end();
               }
+              controller.enqueue({
+                type: "step-finish",
+                finishReason: stepFinishReason,
+                usage: stepUsage,
+                experimental_providerMetadata: stepProviderMetadata,
+                logprobs: stepLogProbs,
+                response: stepResponse
+              });
+              const stepResult = {
+                text: stepText,
+                toolCalls: stepToolCalls,
+                toolResults: stepToolResults,
+                finishReason: stepFinishReason,
+                usage: stepUsage,
+                warnings: self.warnings,
+                logprobs: stepLogProbs,
+                response: stepResponse,
+                rawResponse: self.rawResponse,
+                experimental_providerMetadata: stepProviderMetadata
+              };
+              stepResults.push(stepResult);
+              await (onStepFinish == null ? void 0 : onStepFinish(stepResult));
               const combinedUsage = {
                 promptTokens: usage.promptTokens + stepUsage.promptTokens,
                 completionTokens: usage.completionTokens + stepUsage.completionTokens,
@@ -4093,7 +4103,7 @@ var DefaultStreamTextResult = class {
                       "ai.response.finishReason": stepFinishReason,
                       "ai.response.text": { output: () => stepText },
                       "ai.response.toolCalls": {
-                        output: () => telemetryToolCalls
+                        output: () => stepToolCallsJson
                       },
                       "ai.usage.promptTokens": combinedUsage.promptTokens,
                       "ai.usage.completionTokens": combinedUsage.completionTokens,
@@ -4101,7 +4111,7 @@ var DefaultStreamTextResult = class {
                       "ai.finishReason": stepFinishReason,
                       "ai.result.text": { output: () => stepText },
                       "ai.result.toolCalls": {
-                        output: () => telemetryToolCalls
+                        output: () => stepToolCallsJson
                       }
                     }
                   })
@@ -4119,6 +4129,7 @@ var DefaultStreamTextResult = class {
                 resolveSteps(stepResults);
                 await (onFinish == null ? void 0 : onFinish({
                   finishReason: stepFinishReason,
+                  logprobs: stepLogProbs,
                   usage: combinedUsage,
                   text: stepText,
                   toolCalls: stepToolCalls,