ai 3.3.36 → 3.3.38

package/dist/index.mjs

@@ -1112,8 +1112,8 @@ function prepareCallSettings({
     if (typeof topK !== "number") {
       throw new InvalidArgumentError({
         parameter: "topK",
-        value: topP,
-        message: "topP must be a number"
+        value: topK,
+        message: "topK must be a number"
       });
     }
   }
@@ -1768,7 +1768,7 @@ function validateObjectGenerationInput({
 }
 
 // core/generate-object/generate-object.ts
-var originalGenerateId = createIdGenerator({ prefix: "aiobj-", length: 24 });
+var originalGenerateId = createIdGenerator({ prefix: "aiobj-", size: 24 });
 async function generateObject({
   model,
   enum: enumValues,
@@ -2257,7 +2257,7 @@ function writeToServerResponse({
 }
 
 // core/generate-object/stream-object.ts
-var originalGenerateId2 = createIdGenerator2({ prefix: "aiobj-", length: 24 });
+var originalGenerateId2 = createIdGenerator2({ prefix: "aiobj-", size: 24 });
 async function streamObject({
   model,
   schema: inputSchema,
@@ -2978,7 +2978,7 @@ function parseToolCall({
 }
 
 // core/generate-text/generate-text.ts
-var originalGenerateId3 = createIdGenerator3({ prefix: "aitxt-", length: 24 });
+var originalGenerateId3 = createIdGenerator3({ prefix: "aitxt-", size: 24 });
 async function generateText({
   model,
   tools,
@@ -3658,7 +3658,7 @@ function runToolsTransformation({
 }
 
 // core/generate-text/stream-text.ts
-var originalGenerateId4 = createIdGenerator4({ prefix: "aitxt-", length: 24 });
+var originalGenerateId4 = createIdGenerator4({ prefix: "aitxt-", size: 24 });
 async function streamText({
   model,
   tools,
@@ -3676,6 +3676,7 @@ async function streamText({
   experimental_toolCallStreaming: toolCallStreaming = false,
   onChunk,
   onFinish,
+  onStepFinish,
   _internal: {
     now: now2 = now,
     generateId: generateId3 = originalGenerateId4,
@@ -3808,6 +3809,7 @@ async function streamText({
         rawResponse,
         onChunk,
         onFinish,
+        onStepFinish,
         rootSpan,
         doStreamSpan,
         telemetry,
@@ -3830,6 +3832,7 @@ var DefaultStreamTextResult = class {
     rawResponse,
     onChunk,
     onFinish,
+    onStepFinish,
     rootSpan,
     doStreamSpan,
     telemetry,
@@ -3982,26 +3985,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 +3994,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 +4005,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 +4020,27 @@ 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
+              };
+              stepResults.push(stepResult);
+              await (onStepFinish == null ? void 0 : onStepFinish(stepResult));
               const combinedUsage = {
                 promptTokens: usage.promptTokens + stepUsage.promptTokens,
                 completionTokens: usage.completionTokens + stepUsage.completionTokens,
@@ -4093,7 +4098,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 +4106,7 @@ var DefaultStreamTextResult = class {
                       "ai.finishReason": stepFinishReason,
                       "ai.result.text": { output: () => stepText },
                       "ai.result.toolCalls": {
-                        output: () => telemetryToolCalls
+                        output: () => stepToolCallsJson
                       }
                     }
                   })
@@ -4119,6 +4124,7 @@ var DefaultStreamTextResult = class {
                 resolveSteps(stepResults);
                 await (onFinish == null ? void 0 : onFinish({
                   finishReason: stepFinishReason,
+                  logprobs: stepLogProbs,
                   usage: combinedUsage,
                   text: stepText,
                   toolCalls: stepToolCalls,