@zenning/ai 5.3.0 → 5.3.2

package/dist/index.mjs

@@ -711,7 +711,7 @@ import {
 } from "@zenning/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.3.0" : "0.0.0-test";
+var VERSION = true ? "5.3.2" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -3439,7 +3439,8 @@ function createStreamingUIMessageState({
     activeReasoningParts: {},
     partialToolCalls: {},
     isFinalized: false,
-    messageQueue: []
+    messageQueue: [],
+    seenAnnotations: /* @__PURE__ */ new Set()
   };
 }
 function processUIMessageStream({
@@ -3455,7 +3456,7 @@ function processUIMessageStream({
     new TransformStream({
       async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          var _a17, _b, _c, _d;
+          var _a17, _b;
           function getToolInvocation(toolCallId) {
             const toolInvocations = state.message.parts.filter(isToolUIPart);
             const toolInvocation = toolInvocations.find(
@@ -3559,12 +3560,39 @@ function processUIMessageStream({
               state.message.metadata = mergedMetadata;
             }
           }
+          function filterDuplicateAnnotations(providerMetadata) {
+            var _a18;
+            if (!((_a18 = providerMetadata == null ? void 0 : providerMetadata.openai) == null ? void 0 : _a18.annotations) || !state.seenAnnotations) {
+              return providerMetadata;
+            }
+            const filtered = { ...providerMetadata };
+            if (filtered.openai) {
+              filtered.openai = { ...filtered.openai };
+              const uniqueAnnotations = filtered.openai.annotations.filter((annotation) => {
+                const key = JSON.stringify(annotation);
+                if (state.seenAnnotations.has(key)) {
+                  return false;
+                }
+                state.seenAnnotations.add(key);
+                return true;
+              });
+              if (uniqueAnnotations.length > 0) {
+                filtered.openai.annotations = uniqueAnnotations;
+              } else {
+                delete filtered.openai.annotations;
+                if (Object.keys(filtered.openai).length === 0) {
+                  delete filtered.openai;
+                }
+              }
+            }
+            return Object.keys(filtered).length > 0 ? filtered : void 0;
+          }
           switch (chunk.type) {
             case "text-start": {
               const textPart = {
                 type: "text",
                 text: "",
-                providerMetadata: chunk.providerMetadata,
+                providerMetadata: filterDuplicateAnnotations(chunk.providerMetadata),
                 state: "streaming"
               };
               state.activeTextParts[chunk.id] = textPart;
@@ -3575,14 +3603,20 @@ function processUIMessageStream({
             case "text-delta": {
               const textPart = state.activeTextParts[chunk.id];
               textPart.text += chunk.delta;
-              textPart.providerMetadata = (_a17 = chunk.providerMetadata) != null ? _a17 : textPart.providerMetadata;
+              const filteredMetadata = filterDuplicateAnnotations(chunk.providerMetadata);
+              if (filteredMetadata && !textPart.providerMetadata) {
+                textPart.providerMetadata = filteredMetadata;
+              }
               write();
               break;
             }
             case "text-end": {
               const textPart = state.activeTextParts[chunk.id];
               textPart.state = "done";
-              textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              const filteredMetadata = filterDuplicateAnnotations(chunk.providerMetadata);
+              if (filteredMetadata && !textPart.providerMetadata) {
+                textPart.providerMetadata = filteredMetadata;
+              }
               delete state.activeTextParts[chunk.id];
               write();
               break;
@@ -3602,13 +3636,13 @@ function processUIMessageStream({
             case "reasoning-delta": {
               const reasoningPart = state.activeReasoningParts[chunk.id];
               reasoningPart.text += chunk.delta;
-              reasoningPart.providerMetadata = (_c = chunk.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
+              reasoningPart.providerMetadata = (_a17 = chunk.providerMetadata) != null ? _a17 : reasoningPart.providerMetadata;
               write();
               break;
             }
             case "reasoning-end": {
               const reasoningPart = state.activeReasoningParts[chunk.id];
-              reasoningPart.providerMetadata = (_d = chunk.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
+              reasoningPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : reasoningPart.providerMetadata;
               reasoningPart.state = "done";
               delete state.activeReasoningParts[chunk.id];
               write();
@@ -3624,38 +3658,53 @@ function processUIMessageStream({
               break;
             }
             case "source-url": {
-              state.message.parts.push({
-                type: "source-url",
-                sourceId: chunk.sourceId,
-                url: chunk.url,
-                title: chunk.title,
-                providerMetadata: chunk.providerMetadata
-              });
-              write();
+              const existingSource = state.message.parts.find(
+                (part) => part.type === "source-url" && part.sourceId === chunk.sourceId
+              );
+              if (!existingSource) {
+                state.message.parts.push({
+                  type: "source-url",
+                  sourceId: chunk.sourceId,
+                  url: chunk.url,
+                  title: chunk.title,
+                  providerMetadata: chunk.providerMetadata
+                });
+                write();
+              }
               break;
             }
             case "source-execution-file": {
-              state.message.parts.push({
-                type: "source-execution-file",
-                sourceId: chunk.sourceId,
-                providerMetadata: chunk.providerMetadata
-              });
-              write();
+              const existingSource = state.message.parts.find(
+                (part) => part.type === "source-execution-file" && part.sourceId === chunk.sourceId
+              );
+              if (!existingSource) {
+                state.message.parts.push({
+                  type: "source-execution-file",
+                  sourceId: chunk.sourceId,
+                  providerMetadata: chunk.providerMetadata
+                });
+                write();
+              }
               break;
             }
             case "source-document": {
-              state.message.parts.push({
-                type: "source-document",
-                sourceId: chunk.sourceId,
-                mediaType: chunk.mediaType,
-                title: chunk.title,
-                filename: chunk.filename,
-                fileId: chunk.fileId,
-                startIndex: chunk.startIndex,
-                endIndex: chunk.endIndex,
-                providerMetadata: chunk.providerMetadata
-              });
-              write();
+              const existingSource = state.message.parts.find(
+                (part) => part.type === "source-document" && part.sourceId === chunk.sourceId
+              );
+              if (!existingSource) {
+                state.message.parts.push({
+                  type: "source-document",
+                  sourceId: chunk.sourceId,
+                  mediaType: chunk.mediaType,
+                  title: chunk.title,
+                  filename: chunk.filename,
+                  fileId: chunk.fileId,
+                  startIndex: chunk.startIndex,
+                  endIndex: chunk.endIndex,
+                  providerMetadata: chunk.providerMetadata
+                });
+                write();
+              }
               break;
             }
             case "tool-input-start": {
@@ -3841,6 +3890,7 @@ function processUIMessageStream({
                   state.activeReasoningParts = {};
                   state.partialToolCalls = {};
                   state.isFinalized = false;
+                  state.seenAnnotations = /* @__PURE__ */ new Set();
                   write();
                 } else if (!state.message.id) {
                   state.message.id = chunk.messageId;