ai 3.3.33 → 3.3.35

package/dist/index.mjs

@@ -4112,8 +4112,9 @@ var DefaultStreamTextResult = class {
   }
   toDataStreamInternal({
     callbacks = {},
-    getErrorMessage: getErrorMessage4 = () => ""
+    getErrorMessage: getErrorMessage4 = () => "",
     // mask error messages for safety by default
+    sendUsage = true
   } = {}) {
     let aggregatedResponse = "";
     const callbackTransformer = new TransformStream({
@@ -4188,10 +4189,10 @@ var DefaultStreamTextResult = class {
             controller.enqueue(
               formatStreamPart("finish_roundtrip", {
                 finishReason: chunk.finishReason,
-                usage: {
+                usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
                   completionTokens: chunk.usage.completionTokens
-                }
+                } : void 0
               })
             );
             break;
@@ -4199,10 +4200,10 @@ var DefaultStreamTextResult = class {
             controller.enqueue(
               formatStreamPart("finish_message", {
                 finishReason: chunk.finishReason,
-                usage: {
+                usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
                   completionTokens: chunk.usage.completionTokens
-                }
+                } : void 0
               })
             );
             break;
@@ -4226,6 +4227,7 @@ var DefaultStreamTextResult = class {
     };
     const data = options == null ? void 0 : "data" in options ? options.data : void 0;
     const getErrorMessage4 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
+    const sendUsage = options == null ? void 0 : "sendUsage" in options ? options.sendUsage : void 0;
     writeToServerResponse({
       response,
       status: init == null ? void 0 : init.status,
@@ -4234,7 +4236,7 @@ var DefaultStreamTextResult = class {
         contentType: "text/plain; charset=utf-8",
         dataStreamVersion: "v1"
       }),
-      stream: this.toDataStream({ data, getErrorMessage: getErrorMessage4 })
+      stream: this.toDataStream({ data, getErrorMessage: getErrorMessage4, sendUsage })
     });
   }
   pipeTextStreamToResponse(response, init) {
@@ -4253,7 +4255,8 @@ var DefaultStreamTextResult = class {
   }
   toDataStream(options) {
     const stream = this.toDataStreamInternal({
-      getErrorMessage: options == null ? void 0 : options.getErrorMessage
+      getErrorMessage: options == null ? void 0 : options.getErrorMessage,
+      sendUsage: options == null ? void 0 : options.sendUsage
     });
     return (options == null ? void 0 : options.data) ? mergeStreams(options == null ? void 0 : options.data.stream, stream) : stream;
   }
@@ -4266,14 +4269,18 @@ var DefaultStreamTextResult = class {
     };
     const data = options == null ? void 0 : "data" in options ? options.data : void 0;
     const getErrorMessage4 = options == null ? void 0 : "getErrorMessage" in options ? options.getErrorMessage : void 0;
-    return new Response(this.toDataStream({ data, getErrorMessage: getErrorMessage4 }), {
-      status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
-      statusText: init == null ? void 0 : init.statusText,
-      headers: prepareResponseHeaders(init, {
-        contentType: "text/plain; charset=utf-8",
-        dataStreamVersion: "v1"
-      })
-    });
+    const sendUsage = options == null ? void 0 : "sendUsage" in options ? options.sendUsage : void 0;
+    return new Response(
+      this.toDataStream({ data, getErrorMessage: getErrorMessage4, sendUsage }),
+      {
+        status: (_a11 = init == null ? void 0 : init.status) != null ? _a11 : 200,
+        statusText: init == null ? void 0 : init.statusText,
+        headers: prepareResponseHeaders(init, {
+          contentType: "text/plain; charset=utf-8",
+          dataStreamVersion: "v1"
+        })
+      }
+    );
   }
   toTextStreamResponse(init) {
     var _a11;