@zenning/ai 6.0.29 → 6.0.31

package/dist/index.mjs

@@ -898,7 +898,7 @@ import {
 } from "@zenning/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.29" : "0.0.0-test";
+var VERSION = true ? "6.0.31" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -4761,16 +4761,43 @@ function processUIMessageStream({
               break;
             }
             case "text-delta": {
-              const textPart = state.activeTextParts[chunk.id];
+              let textPart = state.activeTextParts[chunk.id];
+              if (!textPart) {
+                const existing = state.message.parts.findLast(
+                  (p) => p.type === "text" && p.state !== "done"
+                );
+                if (existing) {
+                  textPart = existing;
+                } else {
+                  textPart = {
+                    type: "text",
+                    text: "",
+                    state: "streaming"
+                  };
+                  state.message.parts.push(textPart);
+                }
+                state.activeTextParts[chunk.id] = textPart;
+              }
               textPart.text += chunk.delta;
               textPart.providerMetadata = (_a16 = chunk.providerMetadata) != null ? _a16 : textPart.providerMetadata;
               write();
               break;
             }
             case "text-end": {
-              const textPart = state.activeTextParts[chunk.id];
-              textPart.state = "done";
-              textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              let textPart = state.activeTextParts[chunk.id];
+              if (!textPart) {
+                const existing = state.message.parts.findLast(
+                  (p) => p.type === "text" && p.state !== "done"
+                );
+                if (existing) {
+                  textPart = existing;
+                  state.activeTextParts[chunk.id] = textPart;
+                }
+              }
+              if (textPart) {
+                textPart.state = "done";
+                textPart.providerMetadata = (_b = chunk.providerMetadata) != null ? _b : textPart.providerMetadata;
+              }
               delete state.activeTextParts[chunk.id];
               write();
               break;
@@ -4788,16 +4815,43 @@ function processUIMessageStream({
               break;
             }
             case "reasoning-delta": {
-              const reasoningPart = state.activeReasoningParts[chunk.id];
+              let reasoningPart = state.activeReasoningParts[chunk.id];
+              if (!reasoningPart) {
+                const existing = state.message.parts.findLast(
+                  (p) => p.type === "reasoning" && p.state !== "done"
+                );
+                if (existing) {
+                  reasoningPart = existing;
+                } else {
+                  reasoningPart = {
+                    type: "reasoning",
+                    text: "",
+                    state: "streaming"
+                  };
+                  state.message.parts.push(reasoningPart);
+                }
+                state.activeReasoningParts[chunk.id] = reasoningPart;
+              }
               reasoningPart.text += chunk.delta;
               reasoningPart.providerMetadata = (_c = chunk.providerMetadata) != null ? _c : reasoningPart.providerMetadata;
               write();
               break;
             }
             case "reasoning-end": {
-              const reasoningPart = state.activeReasoningParts[chunk.id];
-              reasoningPart.providerMetadata = (_d = chunk.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
-              reasoningPart.state = "done";
+              let reasoningPart = state.activeReasoningParts[chunk.id];
+              if (!reasoningPart) {
+                const existing = state.message.parts.findLast(
+                  (p) => p.type === "reasoning" && p.state !== "done"
+                );
+                if (existing) {
+                  reasoningPart = existing;
+                  state.activeReasoningParts[chunk.id] = reasoningPart;
+                }
+              }
+              if (reasoningPart) {
+                reasoningPart.providerMetadata = (_d = chunk.providerMetadata) != null ? _d : reasoningPart.providerMetadata;
+                reasoningPart.state = "done";
+              }
               delete state.activeReasoningParts[chunk.id];
               write();
               break;
@@ -4866,7 +4920,24 @@ function processUIMessageStream({
               break;
             }
             case "tool-input-delta": {
-              const partialToolCall = state.partialToolCalls[chunk.toolCallId];
+              let partialToolCall = state.partialToolCalls[chunk.toolCallId];
+              if (!partialToolCall) {
+                const existingPart = state.message.parts.find(
+                  (p) => isToolUIPart(p) && p.toolCallId === chunk.toolCallId
+                );
+                if (existingPart) {
+                  const isDynamic = existingPart.type === "dynamic-tool";
+                  partialToolCall = {
+                    text: "",
+                    toolName: getToolName(existingPart),
+                    index: state.message.parts.filter(isStaticToolUIPart).indexOf(existingPart),
+                    dynamic: isDynamic
+                  };
+                  state.partialToolCalls[chunk.toolCallId] = partialToolCall;
+                } else {
+                  break;
+                }
+              }
               partialToolCall.text += chunk.inputTextDelta;
               const { value: partialArgs } = await parsePartialJson(
                 partialToolCall.text
@@ -5445,7 +5516,8 @@ function runToolsTransformation({
         case "source":
         case "response-metadata":
         case "error":
-        case "raw": {
+        case "raw":
+        case "compaction": {
           controller.enqueue(chunk);
           break;
         }
@@ -6512,6 +6584,10 @@ var DefaultStreamTextResult = class {
                       }
                       break;
                     }
+                    case "compaction": {
+                      controller.enqueue(chunk);
+                      break;
+                    }
                     default: {
                       const exhaustiveCheck = chunkType;
                       throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);
@@ -7048,6 +7124,14 @@ var DefaultStreamTextResult = class {
             case "raw": {
               break;
             }
+            case "compaction": {
+              controller.enqueue({
+                type: "compaction",
+                id: part.id,
+                encrypted_content: part.encrypted_content
+              });
+              break;
+            }
             default: {
               const exhaustiveCheck = partType;
               throw new Error(`Unknown chunk type: ${exhaustiveCheck}`);