ai 3.2.36 → 3.2.38

package/dist/index.mjs

@@ -887,12 +887,18 @@ function calculateCompletionTokenUsage(usage) {
 }
 
 // core/util/prepare-response-headers.ts
-function prepareResponseHeaders(init, { contentType }) {
+function prepareResponseHeaders(init, {
+  contentType,
+  dataStreamVersion
+}) {
   var _a;
   const headers = new Headers((_a = init == null ? void 0 : init.headers) != null ? _a : {});
   if (!headers.has("Content-Type")) {
     headers.set("Content-Type", contentType);
   }
+  if (dataStreamVersion !== void 0) {
+    headers.set("X-Vercel-AI-Data-Stream", dataStreamVersion);
+  }
   return headers;
 }
 
@@ -985,7 +991,6 @@ async function generateObject({
     },
     tracer,
     fn: async (span) => {
-      var _a2, _b;
       const retry = retryWithExponentialBackoff({ maxRetries });
       if (mode === "auto" || mode == null) {
         mode = model.defaultObjectGenerationMode;
@@ -1030,20 +1035,20 @@ async function generateObject({
                   abortSignal,
                   headers
                 });
+                if (result2.text === void 0) {
+                  throw new NoObjectGeneratedError();
+                }
                 span2.setAttributes({
                   "ai.finishReason": result2.finishReason,
                   "ai.usage.promptTokens": result2.usage.promptTokens,
                   "ai.usage.completionTokens": result2.usage.completionTokens,
-                  "ai.result.text": result2.text
+                  "ai.result.object": result2.text
                 });
-                return result2;
+                return { ...result2, objectText: result2.text };
               }
             })
           );
-          if (generateResult.text === void 0) {
-            throw new NoObjectGeneratedError();
-          }
-          result = generateResult.text;
+          result = generateResult.objectText;
           finishReason = generateResult.finishReason;
           usage = generateResult.usage;
           warnings = generateResult.warnings;
@@ -1073,6 +1078,7 @@ async function generateObject({
               },
               tracer,
               fn: async (span2) => {
+                var _a2, _b;
                 const result2 = await model.doGenerate({
                   mode: {
                     type: "object-tool",
@@ -1089,22 +1095,21 @@ async function generateObject({
                   abortSignal,
                   headers
                 });
+                const objectText = (_b = (_a2 = result2.toolCalls) == null ? void 0 : _a2[0]) == null ? void 0 : _b.args;
+                if (objectText === void 0) {
+                  throw new NoObjectGeneratedError();
+                }
                 span2.setAttributes({
                   "ai.finishReason": result2.finishReason,
                   "ai.usage.promptTokens": result2.usage.promptTokens,
                   "ai.usage.completionTokens": result2.usage.completionTokens,
-                  "ai.result.text": result2.text,
-                  "ai.result.toolCalls": JSON.stringify(result2.toolCalls)
+                  "ai.result.object": objectText
                 });
-                return result2;
+                return { ...result2, objectText };
               }
             })
           );
-          const functionArgs = (_b = (_a2 = generateResult.toolCalls) == null ? void 0 : _a2[0]) == null ? void 0 : _b.args;
-          if (functionArgs === void 0) {
-            throw new NoObjectGeneratedError();
-          }
-          result = functionArgs;
+          result = generateResult.objectText;
           finishReason = generateResult.finishReason;
           usage = generateResult.usage;
           warnings = generateResult.warnings;
@@ -2483,6 +2488,15 @@ var DefaultStreamTextResult = class {
             );
             break;
           case "finish":
+            controller.enqueue(
+              formatStreamPart("finish_message", {
+                finishReason: chunk.finishReason,
+                usage: {
+                  promptTokens: chunk.usage.promptTokens,
+                  completionTokens: chunk.usage.completionTokens
+                }
+              })
+            );
             break;
           default: {
             const exhaustiveCheck = chunkType;
@@ -2494,6 +2508,9 @@ var DefaultStreamTextResult = class {
     return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
   pipeAIStreamToResponse(response, init) {
+    return this.pipeDataStreamToResponse(response, init);
+  }
+  pipeDataStreamToResponse(response, init) {
     var _a;
     response.writeHead((_a = init == null ? void 0 : init.status) != null ? _a : 200, {
       "Content-Type": "text/plain; charset=utf-8",
@@ -2540,6 +2557,9 @@ var DefaultStreamTextResult = class {
     read();
   }
   toAIStreamResponse(options) {
+    return this.toDataStreamResponse(options);
+  }
+  toDataStreamResponse(options) {
     var _a;
     const init = options == null ? void 0 : "init" in options ? options.init : {
       headers: "headers" in options ? options.headers : void 0,
@@ -2552,7 +2572,8 @@ var DefaultStreamTextResult = class {
       status: (_a = init == null ? void 0 : init.status) != null ? _a : 200,
       statusText: init == null ? void 0 : init.statusText,
       headers: prepareResponseHeaders(init, {
-        contentType: "text/plain; charset=utf-8"
+        contentType: "text/plain; charset=utf-8",
+        dataStreamVersion: "v1"
       })
     });
   }
@@ -3390,9 +3411,13 @@ function InkeepStream(res, callbacks) {
 // streams/langchain-adapter.ts
 var langchain_adapter_exports = {};
 __export(langchain_adapter_exports, {
-  toAIStream: () => toAIStream
+  toAIStream: () => toAIStream,
+  toDataStream: () => toDataStream
 });
 function toAIStream(stream, callbacks) {
+  return toDataStream(stream, callbacks);
+}
+function toDataStream(stream, callbacks) {
   return stream.pipeThrough(
     new TransformStream({
       transform: async (value, controller) => {