ai 3.2.22 → 3.2.24

package/dist/index.mjs

@@ -1339,7 +1339,7 @@ async function generateText({
     },
     tracer,
     fn: async (span) => {
-      var _a2, _b, _c;
+      var _a2, _b, _c, _d;
       const retry = retryWithExponentialBackoff({ maxRetries });
       const validatedPrompt = getValidatedPrompt({
         system,
@@ -1355,10 +1355,11 @@ async function generateText({
       let currentModelResponse;
       let currentToolCalls = [];
       let currentToolResults = [];
-      let roundtrips = 0;
+      let roundtripCount = 0;
       const responseMessages = [];
+      const roundtrips = [];
       do {
-        const currentInputFormat = roundtrips === 0 ? validatedPrompt.type : "messages";
+        const currentInputFormat = roundtripCount === 0 ? validatedPrompt.type : "messages";
         currentModelResponse = await retry(
           () => recordSpan({
             name: "ai.generateText.doGenerate",
@@ -1396,9 +1397,18 @@ async function generateText({
           tools,
           tracer
         });
-        const newResponseMessages = toResponseMessages({
+        roundtrips.push({
           text: (_b = currentModelResponse.text) != null ? _b : "",
           toolCalls: currentToolCalls,
+          toolResults: currentToolResults,
+          finishReason: currentModelResponse.finishReason,
+          usage: calculateCompletionTokenUsage(currentModelResponse.usage),
+          warnings: currentModelResponse.warnings,
+          logprobs: currentModelResponse.logprobs
+        });
+        const newResponseMessages = toResponseMessages({
+          text: (_c = currentModelResponse.text) != null ? _c : "",
+          toolCalls: currentToolCalls,
           toolResults: currentToolResults
         });
         responseMessages.push(...newResponseMessages);
@@ -1409,7 +1419,7 @@ async function generateText({
         // there are tool calls:
         currentToolCalls.length > 0 && // all current tool calls have results:
         currentToolResults.length === currentToolCalls.length && // the number of roundtrips is less than the maximum:
-        roundtrips++ < maxToolRoundtrips
+        roundtripCount++ < maxToolRoundtrips
       );
       span.setAttributes({
         "ai.finishReason": currentModelResponse.finishReason,
@@ -1422,7 +1432,7 @@ async function generateText({
         // Always return a string so that the caller doesn't have to check for undefined.
         // If they need to check if the model did not return any text,
         // they can check the length of the string:
-        text: (_c = currentModelResponse.text) != null ? _c : "",
+        text: (_d = currentModelResponse.text) != null ? _d : "",
         toolCalls: currentToolCalls,
         toolResults: currentToolResults,
         finishReason: currentModelResponse.finishReason,
@@ -1430,7 +1440,8 @@ async function generateText({
         warnings: currentModelResponse.warnings,
         rawResponse: currentModelResponse.rawResponse,
         logprobs: currentModelResponse.logprobs,
-        responseMessages
+        responseMessages,
+        roundtrips
       });
     }
   });
@@ -1488,6 +1499,7 @@ var GenerateTextResult = class {
     this.rawResponse = options.rawResponse;
     this.logprobs = options.logprobs;
     this.responseMessages = options.responseMessages;
+    this.roundtrips = options.roundtrips;
   }
 };
 function toResponseMessages({