ai 3.2.27 → 3.2.29

package/dist/index.mjs

@@ -748,6 +748,44 @@ function createAsyncIterableStream(source, transformer) {
   return transformedStream;
 }
 
+// core/util/delayed-promise.ts
+var DelayedPromise = class {
+  constructor() {
+    this.status = { type: "pending" };
+    this._resolve = void 0;
+    this._reject = void 0;
+  }
+  get value() {
+    if (this.promise) {
+      return this.promise;
+    }
+    this.promise = new Promise((resolve, reject) => {
+      if (this.status.type === "resolved") {
+        resolve(this.status.value);
+      } else if (this.status.type === "rejected") {
+        reject(this.status.error);
+      }
+      this._resolve = resolve;
+      this._reject = reject;
+    });
+    return this.promise;
+  }
+  resolve(value) {
+    var _a;
+    this.status = { type: "resolved", value };
+    if (this.promise) {
+      (_a = this._resolve) == null ? void 0 : _a.call(this, value);
+    }
+  }
+  reject(error) {
+    var _a;
+    this.status = { type: "rejected", error };
+    if (this.promise) {
+      (_a = this._reject) == null ? void 0 : _a.call(this, error);
+    }
+  }
+};
+
 // core/generate-object/stream-object.ts
 async function streamObject({
   model,
@@ -892,12 +930,7 @@ var StreamObjectResult = class {
   }) {
     this.warnings = warnings;
     this.rawResponse = rawResponse;
-    let resolveObject;
-    let rejectObject;
-    this.object = new Promise((resolve, reject) => {
-      resolveObject = resolve;
-      rejectObject = reject;
-    });
+    this.objectPromise = new DelayedPromise();
     let resolveUsage;
     this.usage = new Promise((resolve) => {
       resolveUsage = resolve;
@@ -908,6 +941,7 @@ var StreamObjectResult = class {
     let accumulatedText = "";
     let delta = "";
     let latestObject = void 0;
+    const self = this;
     this.originalStream = stream.pipeThrough(
       new TransformStream({
         async transform(chunk, controller) {
@@ -948,10 +982,10 @@ var StreamObjectResult = class {
               });
               if (validationResult.success) {
                 object = validationResult.value;
-                resolveObject(object);
+                self.objectPromise.resolve(object);
               } else {
                 error = validationResult.error;
-                rejectObject(error);
+                self.objectPromise.reject(error);
               }
               break;
             }
@@ -983,6 +1017,12 @@ var StreamObjectResult = class {
     );
   }
   /**
+  The generated object (typed according to the schema). Resolved when the response is finished.
+     */
+  get object() {
+    return this.objectPromise.value;
+  }
+  /**
   Stream of partial objects. It gets more complete as the stream progresses.
     
   Note that the partial object is not validated. 
@@ -1547,12 +1587,101 @@ function toResponseMessages({
 }
 var experimental_generateText = generateText;
 
+// core/util/merge-streams.ts
+function mergeStreams(stream1, stream2) {
+  const reader1 = stream1.getReader();
+  const reader2 = stream2.getReader();
+  let lastRead1 = void 0;
+  let lastRead2 = void 0;
+  let stream1Done = false;
+  let stream2Done = false;
+  async function readStream1(controller) {
+    try {
+      if (lastRead1 == null) {
+        lastRead1 = reader1.read();
+      }
+      const result = await lastRead1;
+      lastRead1 = void 0;
+      if (!result.done) {
+        controller.enqueue(result.value);
+      } else {
+        controller.close();
+      }
+    } catch (error) {
+      controller.error(error);
+    }
+  }
+  async function readStream2(controller) {
+    try {
+      if (lastRead2 == null) {
+        lastRead2 = reader2.read();
+      }
+      const result = await lastRead2;
+      lastRead2 = void 0;
+      if (!result.done) {
+        controller.enqueue(result.value);
+      } else {
+        controller.close();
+      }
+    } catch (error) {
+      controller.error(error);
+    }
+  }
+  return new ReadableStream({
+    async pull(controller) {
+      try {
+        if (stream1Done) {
+          await readStream2(controller);
+          return;
+        }
+        if (stream2Done) {
+          await readStream1(controller);
+          return;
+        }
+        if (lastRead1 == null) {
+          lastRead1 = reader1.read();
+        }
+        if (lastRead2 == null) {
+          lastRead2 = reader2.read();
+        }
+        const { result, reader } = await Promise.race([
+          lastRead1.then((result2) => ({ result: result2, reader: reader1 })),
+          lastRead2.then((result2) => ({ result: result2, reader: reader2 }))
+        ]);
+        if (!result.done) {
+          controller.enqueue(result.value);
+        }
+        if (reader === reader1) {
+          lastRead1 = void 0;
+          if (result.done) {
+            await readStream2(controller);
+            stream1Done = true;
+          }
+        } else {
+          lastRead2 = void 0;
+          if (result.done) {
+            stream2Done = true;
+            await readStream1(controller);
+          }
+        }
+      } catch (error) {
+        controller.error(error);
+      }
+    },
+    cancel() {
+      reader1.cancel();
+      reader2.cancel();
+    }
+  });
+}
+
 // core/generate-text/run-tools-transformation.ts
 import { NoSuchToolError as NoSuchToolError2 } from "@ai-sdk/provider";
 import { generateId } from "@ai-sdk/ui-utils";
 function runToolsTransformation({
   tools,
   generatorStream,
+  toolCallStreaming,
   tracer
 }) {
   let canClose = false;
@@ -1563,6 +1692,7 @@ function runToolsTransformation({
       toolResultsStreamController = controller;
     }
   });
+  const activeToolCalls = {};
   const forwardStream = new TransformStream({
     transform(chunk, controller) {
       const chunkType = chunk.type;
@@ -1572,6 +1702,25 @@ function runToolsTransformation({
           controller.enqueue(chunk);
           break;
         }
+        case "tool-call-delta": {
+          if (toolCallStreaming) {
+            if (!activeToolCalls[chunk.toolCallId]) {
+              controller.enqueue({
+                type: "tool-call-streaming-start",
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName
+              });
+              activeToolCalls[chunk.toolCallId] = true;
+            }
+            controller.enqueue({
+              type: "tool-call-delta",
+              toolCallId: chunk.toolCallId,
+              toolName: chunk.toolName,
+              argsTextDelta: chunk.argsTextDelta
+            });
+          }
+          break;
+        }
         case "tool-call": {
           const toolName = chunk.toolName;
           if (tools == null) {
@@ -1657,9 +1806,6 @@ function runToolsTransformation({
           });
           break;
         }
-        case "tool-call-delta": {
-          break;
-        }
         default: {
           const _exhaustiveCheck = chunkType;
           throw new Error(`Unhandled chunk type: ${_exhaustiveCheck}`);
@@ -1712,6 +1858,7 @@ async function streamText({
   abortSignal,
   headers,
   experimental_telemetry: telemetry,
+  experimental_toolCallStreaming: toolCallStreaming = false,
   onFinish,
   ...settings
 }) {
@@ -1772,6 +1919,7 @@ async function streamText({
         stream: runToolsTransformation({
           tools,
           generatorStream: stream,
+          toolCallStreaming,
           tracer
         }),
         warnings,
@@ -1849,6 +1997,8 @@ var StreamTextResult = class {
               resolveText(text);
               resolveToolCalls(toolCalls);
               break;
+            case "tool-call-streaming-start":
+            case "tool-call-delta":
             case "error":
               break;
             default: {
@@ -1990,12 +2140,29 @@ var StreamTextResult = class {
           await callbacks.onFinal(aggregatedResponse);
       }
     });
-    const streamDataTransformer = new TransformStream({
+    const streamPartsTransformer = new TransformStream({
       transform: async (chunk, controller) => {
-        switch (chunk.type) {
+        const chunkType = chunk.type;
+        switch (chunkType) {
           case "text-delta":
             controller.enqueue(formatStreamPart("text", chunk.textDelta));
             break;
+          case "tool-call-streaming-start":
+            controller.enqueue(
+              formatStreamPart("tool_call_streaming_start", {
+                toolCallId: chunk.toolCallId,
+                toolName: chunk.toolName
+              })
+            );
+            break;
+          case "tool-call-delta":
+            controller.enqueue(
+              formatStreamPart("tool_call_delta", {
+                toolCallId: chunk.toolCallId,
+                argsTextDelta: chunk.argsTextDelta
+              })
+            );
+            break;
           case "tool-call":
             controller.enqueue(
               formatStreamPart("tool_call", {
@@ -2020,10 +2187,16 @@ var StreamTextResult = class {
               formatStreamPart("error", JSON.stringify(chunk.error))
             );
             break;
+          case "finish":
+            break;
+          default: {
+            const exhaustiveCheck = chunkType;
+            throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);
+          }
         }
       }
     });
-    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamDataTransformer).pipeThrough(new TextEncoderStream());
+    return this.fullStream.pipeThrough(callbackTransformer).pipeThrough(streamPartsTransformer).pipeThrough(new TextEncoderStream());
   }
   /**
   Writes stream data output to a Node.js response-like object.
@@ -2091,12 +2264,27 @@ var StreamTextResult = class {
   Converts the result to a streamed response object with a stream data part stream.
   It can be used with the `useChat` and `useCompletion` hooks.
   
-  @param init Optional headers.
+  @param options An object with an init property (ResponseInit) and a data property. 
+  You can also pass in a ResponseInit directly (deprecated).
   
   @return A response object.
      */
-  toAIStreamResponse(init) {
-    return new StreamingTextResponse(this.toAIStream(), init);
+  toAIStreamResponse(options) {
+    var _a;
+    const init = options == null ? void 0 : "init" in options ? options.init : {
+      headers: "headers" in options ? options.headers : void 0,
+      status: "status" in options ? options.status : void 0,
+      statusText: "statusText" in options ? options.statusText : void 0
+    };
+    const data = options == null ? void 0 : "data" in options ? options.data : void 0;
+    const stream = data ? mergeStreams(data.stream, this.toAIStream()) : this.toAIStream();
+    return new Response(stream, {
+      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"
+      })
+    });
   }
   /**
   Creates a simple text stream response.
@@ -2182,6 +2370,13 @@ function convertToCoreMessages(messages) {
     experimental_attachments
   } of messages) {
     switch (role) {
+      case "system": {
+        coreMessages.push({
+          role: "system",
+          content
+        });
+        break;
+      }
       case "user": {
         coreMessages.push({
           role: "user",
@@ -2533,7 +2728,7 @@ function readableFromAsyncIterable(iterable) {
 
 // streams/stream-data.ts
 import { formatStreamPart as formatStreamPart2 } from "@ai-sdk/ui-utils";
-var StreamData = class {
+var StreamData2 = class {
   constructor() {
     this.encoder = new TextEncoder();
     this.controller = null;
@@ -2604,7 +2799,7 @@ function createStreamDataTransformer() {
     }
   });
 }
-var experimental_StreamData = class extends StreamData {
+var experimental_StreamData = class extends StreamData2 {
 };
 
 // streams/anthropic-stream.ts
@@ -3374,94 +3569,6 @@ async function ReplicateStream(res, cb, options) {
   );
 }
 
-// core/util/merge-streams.ts
-function mergeStreams(stream1, stream2) {
-  const reader1 = stream1.getReader();
-  const reader2 = stream2.getReader();
-  let lastRead1 = void 0;
-  let lastRead2 = void 0;
-  let stream1Done = false;
-  let stream2Done = false;
-  async function readStream1(controller) {
-    try {
-      if (lastRead1 == null) {
-        lastRead1 = reader1.read();
-      }
-      const result = await lastRead1;
-      lastRead1 = void 0;
-      if (!result.done) {
-        controller.enqueue(result.value);
-      } else {
-        controller.close();
-      }
-    } catch (error) {
-      controller.error(error);
-    }
-  }
-  async function readStream2(controller) {
-    try {
-      if (lastRead2 == null) {
-        lastRead2 = reader2.read();
-      }
-      const result = await lastRead2;
-      lastRead2 = void 0;
-      if (!result.done) {
-        controller.enqueue(result.value);
-      } else {
-        controller.close();
-      }
-    } catch (error) {
-      controller.error(error);
-    }
-  }
-  return new ReadableStream({
-    async pull(controller) {
-      try {
-        if (stream1Done) {
-          readStream2(controller);
-          return;
-        }
-        if (stream2Done) {
-          readStream1(controller);
-          return;
-        }
-        if (lastRead1 == null) {
-          lastRead1 = reader1.read();
-        }
-        if (lastRead2 == null) {
-          lastRead2 = reader2.read();
-        }
-        const { result, reader } = await Promise.race([
-          lastRead1.then((result2) => ({ result: result2, reader: reader1 })),
-          lastRead2.then((result2) => ({ result: result2, reader: reader2 }))
-        ]);
-        if (!result.done) {
-          controller.enqueue(result.value);
-        }
-        if (reader === reader1) {
-          lastRead1 = void 0;
-          if (result.done) {
-            readStream2(controller);
-            stream1Done = true;
-          }
-        } else {
-          lastRead2 = void 0;
-          if (result.done) {
-            stream2Done = true;
-            readStream1(controller);
-          }
-        }
-      } catch (error) {
-        controller.error(error);
-      }
-    },
-    cancel() {
-      reader1.cancel();
-      reader2.cancel();
-    }
-  });
-}
-
 // streams/stream-to-response.ts
 function streamToResponse(res, response, init, data) {
   var _a;
@@ -3545,7 +3652,7 @@ export {
   OpenAIStream,
   ReplicateStream,
   RetryError2 as RetryError,
-  StreamData,
+  StreamData2 as StreamData,
   StreamObjectResult,
   StreamTextResult,
   StreamingTextResponse,