ai 3.4.31 → 3.4.33

package/dist/index.mjs

@@ -3333,7 +3333,7 @@ async function generateText({
     }),
     tracer,
     fn: async (span) => {
-      var _a11, _b, _c, _d, _e, _f, _g;
+      var _a11, _b, _c, _d, _e, _f;
       const retry = retryWithExponentialBackoff({ maxRetries });
       const mode = {
         type: "regular",
@@ -3487,12 +3487,15 @@ async function generateText({
             nextStepType = "tool-result";
           }
         }
-        const stepText = nextStepType === "continue" ? removeTextAfterLastWhitespace((_b = currentModelResponse.text) != null ? _b : "") : (_c = currentModelResponse.text) != null ? _c : "";
+        const originalText = (_b = currentModelResponse.text) != null ? _b : "";
+        const stepTextLeadingWhitespaceTrimmed = stepType === "continue" && // only for continue steps
+        text.trimEnd() !== text ? originalText.trimStart() : originalText;
+        const stepText = nextStepType === "continue" ? removeTextAfterLastWhitespace(stepTextLeadingWhitespaceTrimmed) : stepTextLeadingWhitespaceTrimmed;
         text = nextStepType === "continue" || stepType === "continue" ? text + stepText : stepText;
         if (stepType === "continue") {
           const lastMessage = responseMessages[responseMessages.length - 1];
           if (typeof lastMessage.content === "string") {
-            lastMessage.content = text;
+            lastMessage.content += stepText;
           } else {
             lastMessage.content.push({
               text: stepText,
@@ -3518,10 +3521,10 @@ async function generateText({
           usage: currentUsage,
           warnings: currentModelResponse.warnings,
           logprobs: currentModelResponse.logprobs,
-          request: (_d = currentModelResponse.request) != null ? _d : {},
+          request: (_c = currentModelResponse.request) != null ? _c : {},
           response: {
             ...currentModelResponse.response,
-            headers: (_e = currentModelResponse.rawResponse) == null ? void 0 : _e.headers,
+            headers: (_d = currentModelResponse.rawResponse) == null ? void 0 : _d.headers,
             // deep clone msgs to avoid mutating past messages in multi-step:
             messages: JSON.parse(JSON.stringify(responseMessages))
           },
@@ -3563,10 +3566,10 @@ async function generateText({
         finishReason: currentModelResponse.finishReason,
         usage,
         warnings: currentModelResponse.warnings,
-        request: (_f = currentModelResponse.request) != null ? _f : {},
+        request: (_e = currentModelResponse.request) != null ? _e : {},
         response: {
           ...currentModelResponse.response,
-          headers: (_g = currentModelResponse.rawResponse) == null ? void 0 : _g.headers,
+          headers: (_f = currentModelResponse.rawResponse) == null ? void 0 : _f.headers,
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,
@@ -4266,7 +4269,8 @@ var DefaultStreamTextResult = class {
       },
       stepType,
       previousStepText = "",
-      stepRequest
+      stepRequest,
+      hasLeadingWhitespace
     }) {
       const stepToolCalls = [];
       const stepToolResults = [];
@@ -4288,6 +4292,8 @@ var DefaultStreamTextResult = class {
       };
       let chunkBuffer = "";
       let chunkTextPublished = false;
+      let inWhitespacePrefix = true;
+      let hasWhitespaceSuffix = false;
       async function publishTextChunk({
         controller,
         chunk
@@ -4296,6 +4302,7 @@ var DefaultStreamTextResult = class {
         stepText += chunk.textDelta;
         fullStepText += chunk.textDelta;
         chunkTextPublished = true;
+        hasWhitespaceSuffix = chunk.textDelta.trimEnd() !== chunk.textDelta;
         await (onChunk == null ? void 0 : onChunk({ chunk }));
       }
       addStream(
@@ -4324,7 +4331,12 @@ var DefaultStreamTextResult = class {
               switch (chunkType) {
                 case "text-delta": {
                   if (continueSteps) {
-                    chunkBuffer += chunk.textDelta;
+                    const trimmedChunkText = inWhitespacePrefix && hasLeadingWhitespace ? chunk.textDelta.trimStart() : chunk.textDelta;
+                    if (trimmedChunkText.length === 0) {
+                      break;
+                    }
+                    inWhitespacePrefix = false;
+                    chunkBuffer += trimmedChunkText;
                     const split = splitOnLastWhitespace(chunkBuffer);
                     if (split != null) {
                       chunkBuffer = split.suffix;
@@ -4467,7 +4479,7 @@ var DefaultStreamTextResult = class {
               if (stepType === "continue") {
                 const lastMessage = responseMessages[responseMessages.length - 1];
                 if (typeof lastMessage.content === "string") {
-                  lastMessage.content = stepText;
+                  lastMessage.content += stepText;
                 } else {
                   lastMessage.content.push({
                     text: stepText,
@@ -4528,7 +4540,8 @@ var DefaultStreamTextResult = class {
                   usage: combinedUsage,
                   stepType: nextStepType,
                   previousStepText: fullStepText,
-                  stepRequest: result.request
+                  stepRequest: result.request,
+                  hasLeadingWhitespace: hasWhitespaceSuffix
                 });
                 return;
               }
@@ -4573,7 +4586,8 @@ var DefaultStreamTextResult = class {
                 resolveRequest(stepRequest);
                 resolveResponse({
                   ...stepResponse,
-                  headers: rawResponse == null ? void 0 : rawResponse.headers
+                  headers: rawResponse == null ? void 0 : rawResponse.headers,
+                  messages: responseMessages
                 });
                 resolveSteps(stepResults);
                 resolveResponseMessages(responseMessages);
@@ -4618,7 +4632,8 @@ var DefaultStreamTextResult = class {
       responseMessages: [],
       usage: void 0,
       stepType: "initial",
-      stepRequest: request
+      stepRequest: request,
+      hasLeadingWhitespace: false
     });
   }
   /**