ai 2.2.31 → 2.2.33

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) {
@@ -699,7 +733,35 @@ function parseOpenAIStream() {
 }
 async function* streamable3(stream) {
   const extract = chunkToText();
-  for await (const chunk of stream) {
+  for await (let chunk of stream) {
+    if ("promptFilterResults" in chunk) {
+      chunk = {
+        id: chunk.id,
+        created: chunk.created.getDate(),
+        object: chunk.object,
+        // not exposed by Azure API
+        model: chunk.model,
+        // not exposed by Azure API
+        choices: chunk.choices.map((choice) => {
+          var _a, _b, _c, _d, _e, _f, _g;
+          return {
+            delta: {
+              content: (_a = choice.delta) == null ? void 0 : _a.content,
+              function_call: (_b = choice.delta) == null ? void 0 : _b.functionCall,
+              role: (_c = choice.delta) == null ? void 0 : _c.role,
+              tool_calls: ((_e = (_d = choice.delta) == null ? void 0 : _d.toolCalls) == null ? void 0 : _e.length) ? (_g = (_f = choice.delta) == null ? void 0 : _f.toolCalls) == null ? void 0 : _g.map((toolCall, index) => ({
+                index,
+                id: toolCall.id,
+                function: toolCall.function,
+                type: toolCall.type
+              })) : void 0
+            },
+            finish_reason: choice.finishReason,
+            index: choice.index
+          };
+        })
+      };
+    }
     const text = extract(chunk);
     if (text)
       yield text;
@@ -1059,6 +1121,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"] = {