bitfab 0.53.1 → 0.53.2

package/dist/{chunk-ZEGNMZU7.js → chunk-3LSPOHL7.js}

@@ -1,6 +1,6 @@
 import {
   BitfabError
-} from "./chunk-ZNY2HZT3.js";
+} from "./chunk-S4UK5HNW.js";
 
 // src/codeChange.ts
 var NUL = String.fromCharCode(0);
@@ -207,7 +207,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-7QLUPC6F.js");
+  const { flushTraces: flushTraces2 } = await import("./http-LOZSAC23.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -744,4 +744,4 @@ export {
   runRegistryMain,
   runIfEntrypoint
 };
-//# sourceMappingURL=chunk-ZEGNMZU7.js.map
+//# sourceMappingURL=chunk-3LSPOHL7.js.map

package/dist/{chunk-ZCOZHNAA.js → chunk-6SPWEVFD.js}

@@ -19,7 +19,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-MMT5MWG5.js";
+} from "./chunk-V3KD7B3B.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -35,7 +35,7 @@ import {
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-ASMHSKFS.js";
+} from "./chunk-JZAKQDGR.js";
 import {
   __privateAdd,
   __privateGet,
@@ -2482,6 +2482,15 @@ function runWithSeedContext(ctx, fn) {
   return fn();
 }
 
+// src/streamFinalizationError.ts
+var StreamFinalizationError = class extends Error {
+  constructor(message, output) {
+    super(message);
+    this.output = output;
+    this.name = "StreamFinalizationError";
+  }
+};
+
 // src/tracing.ts
 var BitfabOpenAITracingProcessor = class {
   /**
@@ -2749,52 +2758,110 @@ function summarizeGenerate(result, model) {
   }
   return summary;
 }
-function accumulateStream(onComplete, model) {
+function accumulateStream(source, onComplete, model) {
+  const reader = source.getReader();
   let text = "";
   const toolCalls = [];
   let usage;
   let finishReason;
   let completed = false;
-  const complete = () => {
+  let streamFailure;
+  const complete = (failure = streamFailure) => {
     if (completed) {
       return;
     }
     completed = true;
-    const summary = {
+    let error;
+    if (failure !== void 0) {
+      try {
+        error = failure.error instanceof Error ? failure.error.message : String(
+          failure.error ?? (failure.cancelled === true ? "Stream cancelled" : "Stream failed")
+        );
+      } catch {
+        error = "Stream failed";
+      }
+    }
+    onComplete({
       text,
       toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
       usage,
-      finishReason
-    };
-    if (model) {
-      summary.model = model;
-    }
-    onComplete(summary);
+      finishReason,
+      ...model !== void 0 && { model },
+      ...failure !== void 0 && { error },
+      ...failure?.cancelled === true && { cancelled: true }
+    });
   };
-  return new TransformStream({
-    transform(part, controller) {
-      try {
-        if (part?.type === "text-delta") {
-          text += part.delta ?? part.textDelta ?? "";
-        } else if (part?.type === "tool-call") {
-          toolCalls.push({
-            toolCallId: part.toolCallId,
-            toolName: part.toolName,
-            input: part.input ?? part.args
-          });
-        } else if (part?.type === "finish") {
-          usage = part.usage;
-          finishReason = part.finishReason;
-          complete();
+  return new ReadableStream(
+    {
+      start(controller) {
+        void reader.closed.then(
+          () => {
+            queueMicrotask(() => {
+              if (!completed) {
+                complete();
+                reader.releaseLock();
+                controller.close();
+              }
+            });
+          },
+          (error) => {
+            if (!completed) {
+              complete({ error });
+              controller.error(error);
+              reader.releaseLock();
+            }
+          }
+        );
+      },
+      async pull(controller) {
+        try {
+          const { done, value: part } = await reader.read();
+          if (completed) {
+            return;
+          }
+          if (done) {
+            complete();
+            reader.releaseLock();
+            controller.close();
+            return;
+          }
+          try {
+            if (part?.type === "text-delta") {
+              text += part.delta ?? part.textDelta ?? "";
+            } else if (part?.type === "tool-call") {
+              toolCalls.push({
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                input: part.input ?? part.args
+              });
+            } else if (part?.type === "error") {
+              streamFailure = { error: part.error };
+            } else if (part?.type === "finish") {
+              usage = part.usage;
+              finishReason = part.finishReason;
+            }
+          } catch {
+          }
+          controller.enqueue(part);
+        } catch (error) {
+          if (!completed) {
+            complete({ error });
+            reader.releaseLock();
+            controller.error(error);
+          }
+        }
+      },
+      async cancel(reason) {
+        complete({ error: reason, cancelled: true });
+        try {
+          await reader.cancel(reason);
+        } finally {
+          reader.releaseLock();
         }
-      } catch {
       }
-      controller.enqueue(part);
     },
-    flush() {
-      complete();
-    }
-  });
+    { highWaterMark: 0 }
+  );
 }
 var BitfabVercelAiHandler = class {
   constructor(config) {
@@ -2835,14 +2902,22 @@ var BitfabVercelAiHandler = class {
           // awaits the summary the accumulator resolves once the stream drains.
           {
             type: "llm",
-            finalize: () => summary,
+            finalize: async () => {
+              const output = await summary;
+              if (output.error !== void 0) {
+                throw new StreamFinalizationError(output.error, output);
+              }
+              return output;
+            },
             surface: "inherit",
             instrumentation: "vercel-ai"
           },
           async () => {
             const result = await doStream();
-            const stream = result.stream.pipeThrough(
-              accumulateStream(resolveSummary, label)
+            const stream = accumulateStream(
+              result.stream,
+              resolveSummary,
+              label
             );
             return { ...result, stream };
           }
@@ -4336,8 +4411,8 @@ var Bitfab = class {
             void self.httpClient.trackDeferred(
               Promise.resolve().then(() => options.finalize(result)).then((output) => sendSpan({ result: output })).catch(
                 (error) => sendSpan({
-                  result: void 0,
-                  error: error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
+                  result: error instanceof StreamFinalizationError ? error.output : void 0,
+                  error: error instanceof StreamFinalizationError ? error.message : error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
                 })
               )
             );
@@ -4983,7 +5058,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-2OOKOZQ3.js");
+        const { flushTraces: flushTraces2 } = await import("./http-PUEEHMKU.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -5042,7 +5117,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-KJKIGXRH.js");
+    const { replay: doReplay } = await import("./replay-SSJ72WEI.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -5326,7 +5401,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-2OOKOZQ3.js");
+  const { flushTraces: flushTraces2 } = await import("./http-PUEEHMKU.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -5363,4 +5438,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-ZCOZHNAA.js.map
+//# sourceMappingURL=chunk-6SPWEVFD.js.map