ai 2.2.31 → 2.2.32

package/dist/index.mjs

@@ -114,6 +114,16 @@ var toolCallStreamPart = {
     };
   }
 };
+var messageAnnotationsStreamPart = {
+  code: "8",
+  name: "message_annotations",
+  parse: (value) => {
+    if (!Array.isArray(value)) {
+      throw new Error('"message_annotations" parts expect an array value.');
+    }
+    return { type: "message_annotations", value };
+  }
+};
 var streamParts = [
   textStreamPart,
   functionCallStreamPart,
@@ -122,7 +132,8 @@ var streamParts = [
   assistantMessageStreamPart,
   assistantControlDataStreamPart,
   dataMessageStreamPart,
-  toolCallStreamPart
+  toolCallStreamPart,
+  messageAnnotationsStreamPart
 ];
 var streamPartsByCode = {
   [textStreamPart.code]: textStreamPart,
@@ -132,7 +143,8 @@ var streamPartsByCode = {
   [assistantMessageStreamPart.code]: assistantMessageStreamPart,
   [assistantControlDataStreamPart.code]: assistantControlDataStreamPart,
   [dataMessageStreamPart.code]: dataMessageStreamPart,
-  [toolCallStreamPart.code]: toolCallStreamPart
+  [toolCallStreamPart.code]: toolCallStreamPart,
+  [messageAnnotationsStreamPart.code]: messageAnnotationsStreamPart
 };
 var StreamStringPrefixes = {
   [textStreamPart.name]: textStreamPart.code,
@@ -142,7 +154,8 @@ var StreamStringPrefixes = {
   [assistantMessageStreamPart.name]: assistantMessageStreamPart.code,
   [assistantControlDataStreamPart.name]: assistantControlDataStreamPart.code,
   [dataMessageStreamPart.name]: dataMessageStreamPart.code,
-  [toolCallStreamPart.name]: toolCallStreamPart.code
+  [toolCallStreamPart.name]: toolCallStreamPart.code,
+  [messageAnnotationsStreamPart.name]: messageAnnotationsStreamPart.code
 };
 var validCodes = streamParts.map((part) => part.code);
 var parseStreamPart = (line) => {
@@ -208,7 +221,9 @@ function createEventStreamTransformer(customParser) {
             return;
           }
           if ("data" in event) {
-            const parsedMessage = customParser ? customParser(event.data) : event.data;
+            const parsedMessage = customParser ? customParser(event.data, {
+              event: event.event
+            }) : event.data;
             if (parsedMessage)
               controller.enqueue(parsedMessage);
           }
@@ -320,6 +335,7 @@ var experimental_StreamData = class {
     this.isClosed = false;
     // array to store appended data
     this.data = [];
+    this.messageAnnotations = [];
     this.isClosedPromise = new Promise((resolve) => {
       this.isClosedPromiseResolver = resolve;
     });
@@ -336,6 +352,12 @@ var experimental_StreamData = class {
           self.data = [];
           controller.enqueue(encodedData);
         }
+        if (self.messageAnnotations.length) {
+          const encodedmessageAnnotations = self.encoder.encode(
+            formatStreamPart("message_annotations", self.messageAnnotations)
+          );
+          controller.enqueue(encodedmessageAnnotations);
+        }
         controller.enqueue(chunk);
       },
       async flush(controller) {
@@ -354,6 +376,12 @@ var experimental_StreamData = class {
           );
           controller.enqueue(encodedData);
         }
+        if (self.messageAnnotations.length) {
+          const encodedData = self.encoder.encode(
+            formatStreamPart("message_annotations", self.messageAnnotations)
+          );
+          controller.enqueue(encodedData);
+        }
       }
     });
   }
@@ -374,6 +402,12 @@ var experimental_StreamData = class {
     }
     this.data.push(value);
   }
+  appendMessageAnnotation(value) {
+    if (this.isClosed) {
+      throw new Error("Data Stream has already been closed.");
+    }
+    this.messageAnnotations.push(value);
+  }
 };
 function createStreamDataTransformer(experimental_streamData) {
   if (!experimental_streamData) {
@@ -1059,6 +1093,22 @@ async function parseComplexResponse({
         };
       }
     }
+    if (type == "message_annotations") {
+      if (prefixMap["text"]) {
+        prefixMap["text"] = {
+          ...prefixMap["text"],
+          annotations: [...prefixMap["text"].annotations || [], ...value]
+        };
+      } else {
+        prefixMap["text"] = {
+          id: generateId(),
+          role: "assistant",
+          content: "",
+          annotations: [...value],
+          createdAt
+        };
+      }
+    }
     let functionCallMessage = null;
     if (type === "function_call") {
       prefixMap["function_call"] = {