ai 3.0.27 → 3.0.29

package/dist/index.mjs

@@ -375,6 +375,8 @@ async function experimental_generateObject({
   let finishReason;
   let usage;
   let warnings;
+  let rawResponse;
+  let logprobs;
   switch (mode) {
     case "json": {
       const validatedPrompt = getValidatedPrompt({
@@ -398,6 +400,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case "grammar": {
@@ -422,6 +426,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case "tool": {
@@ -455,6 +461,8 @@ async function experimental_generateObject({
       finishReason = generateResult.finishReason;
       usage = generateResult.usage;
       warnings = generateResult.warnings;
+      rawResponse = generateResult.rawResponse;
+      logprobs = generateResult.logprobs;
       break;
     }
     case void 0: {
@@ -473,7 +481,9 @@ async function experimental_generateObject({
     object: parseResult.value,
     finishReason,
     usage: calculateTokenUsage(usage),
-    warnings
+    warnings,
+    rawResponse,
+    logprobs
   });
 }
 var GenerateObjectResult = class {
@@ -482,6 +492,8 @@ var GenerateObjectResult = class {
     this.finishReason = options.finishReason;
     this.usage = options.usage;
     this.warnings = options.warnings;
+    this.rawResponse = options.rawResponse;
+    this.logprobs = options.logprobs;
   }
 };
 
@@ -913,6 +925,7 @@ async function experimental_streamObject({
             case "text-delta":
               controller.enqueue(chunk.textDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -940,6 +953,7 @@ async function experimental_streamObject({
             case "text-delta":
               controller.enqueue(chunk.textDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -975,6 +989,7 @@ async function experimental_streamObject({
             case "tool-call-delta":
               controller.enqueue(chunk.argsTextDelta);
               break;
+            case "finish":
             case "error":
               controller.enqueue(chunk);
               break;
@@ -994,16 +1009,19 @@ async function experimental_streamObject({
   const result = await retry(() => model.doStream(callOptions));
   return new StreamObjectResult({
     stream: result.stream.pipeThrough(new TransformStream(transformer)),
-    warnings: result.warnings
+    warnings: result.warnings,
+    rawResponse: result.rawResponse
   });
 }
 var StreamObjectResult = class {
   constructor({
     stream,
-    warnings
+    warnings,
+    rawResponse
   }) {
     this.originalStream = stream;
     this.warnings = warnings;
+    this.rawResponse = rawResponse;
   }
   get partialObjectStream() {
     let accumulatedText = "";
@@ -1019,13 +1037,42 @@ var StreamObjectResult = class {
             latestObject = currentObject;
             controller.enqueue(currentObject);
           }
-        }
-        if (typeof chunk === "object" && chunk.type === "error") {
+        } else if (chunk.type === "error") {
           throw chunk.error;
         }
       }
     });
   }
+  get fullStream() {
+    let accumulatedText = "";
+    let latestObject = void 0;
+    return createAsyncIterableStream(this.originalStream, {
+      transform(chunk, controller) {
+        if (typeof chunk === "string") {
+          accumulatedText += chunk;
+          const currentObject = parsePartialJson(
+            accumulatedText
+          );
+          if (!isDeepEqualData(latestObject, currentObject)) {
+            latestObject = currentObject;
+            controller.enqueue({ type: "object", object: currentObject });
+          }
+        } else {
+          switch (chunk.type) {
+            case "finish":
+              controller.enqueue({
+                ...chunk,
+                usage: calculateTokenUsage(chunk.usage)
+              });
+              break;
+            default:
+              controller.enqueue(chunk);
+              break;
+          }
+        }
+      }
+    });
+  }
 };
 
 // core/generate-text/tool-call.ts
@@ -1113,7 +1160,9 @@ async function experimental_generateText({
     toolResults,
     finishReason: modelResponse.finishReason,
     usage: calculateTokenUsage(modelResponse.usage),
-    warnings: modelResponse.warnings
+    warnings: modelResponse.warnings,
+    rawResponse: modelResponse.rawResponse,
+    logprobs: modelResponse.logprobs
   });
 }
 async function executeTools({
@@ -1147,6 +1196,8 @@ var GenerateTextResult = class {
     this.finishReason = options.finishReason;
     this.usage = options.usage;
     this.warnings = options.warnings;
+    this.rawResponse = options.rawResponse;
+    this.logprobs = options.logprobs;
   }
 };
 
@@ -1247,11 +1298,8 @@ function runToolsTransformation({
           controller.enqueue({
             type: "finish",
             finishReason: chunk.finishReason,
-            usage: {
-              promptTokens: chunk.usage.promptTokens,
-              completionTokens: chunk.usage.completionTokens,
-              totalTokens: chunk.usage.promptTokens + chunk.usage.completionTokens
-            }
+            logprobs: chunk.logprobs,
+            usage: calculateTokenUsage(chunk.usage)
           });
           break;
         }
@@ -1309,7 +1357,7 @@ async function experimental_streamText({
 }) {
   const retry = retryWithExponentialBackoff({ maxRetries });
   const validatedPrompt = getValidatedPrompt({ system, prompt, messages });
-  const { stream, warnings } = await retry(
+  const { stream, warnings, rawResponse } = await retry(
     () => model.doStream({
       mode: {
         type: "regular",
@@ -1331,16 +1379,19 @@ async function experimental_streamText({
       tools,
       generatorStream: stream
     }),
-    warnings
+    warnings,
+    rawResponse
   });
 }
 var StreamTextResult = class {
   constructor({
     stream,
-    warnings
+    warnings,
+    rawResponse
   }) {
     this.originalStream = stream;
     this.warnings = warnings;
+    this.rawResponse = rawResponse;
   }
   /**
   A text stream that returns only the generated text deltas. You can use it