@bitfab/sdk 0.53.1 → 0.53.2

package/dist/index.d.cts

@@ -1069,9 +1069,9 @@ declare class BitfabClaudeAgentHandler {
  * The span records the call parameters (the prompt/messages) as its input and a
  * serializable summary (`{ text, toolCalls, usage, finishReason }`) as its
  * output. Streaming is handled by passing the model's stream through a
- * transform that accumulates the assembled text/usage as the caller consumes
- * it: the live stream is handed back unchanged (first-byte latency untouched)
- * and the span is finalized once the stream completes.
+ * reader that accumulates text and usage as the caller consumes it. Parts,
+ * upstream errors, and cancellation propagate to the caller unchanged; the
+ * span finalizes on completion, error, or cancellation with partial output.
  *
  * The middleware is fully duck-typed (no static or dynamic `ai` import), so it
  * adds no dependency and is browser-safe. It works with `ai` v5 and v6.
@@ -1112,6 +1112,7 @@ interface VercelGenerateResult {
 /** Duck-typed subset of a single streaming part from `doStream`. */
 interface VercelStreamPart {
     type: string;
+    error?: unknown;
     delta?: string;
     textDelta?: string;
     toolCallId?: string;
@@ -3343,7 +3344,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.53.1";
+declare const __version__ = "0.53.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -1069,9 +1069,9 @@ declare class BitfabClaudeAgentHandler {
  * The span records the call parameters (the prompt/messages) as its input and a
  * serializable summary (`{ text, toolCalls, usage, finishReason }`) as its
  * output. Streaming is handled by passing the model's stream through a
- * transform that accumulates the assembled text/usage as the caller consumes
- * it: the live stream is handed back unchanged (first-byte latency untouched)
- * and the span is finalized once the stream completes.
+ * reader that accumulates text and usage as the caller consumes it. Parts,
+ * upstream errors, and cancellation propagate to the caller unchanged; the
+ * span finalizes on completion, error, or cancellation with partial output.
  *
  * The middleware is fully duck-typed (no static or dynamic `ai` import), so it
  * adds no dependency and is browser-safe. It works with `ai` v5 and v6.
@@ -1112,6 +1112,7 @@ interface VercelGenerateResult {
 /** Duck-typed subset of a single streaming part from `doStream`. */
 interface VercelStreamPart {
     type: string;
+    error?: unknown;
     delta?: string;
     textDelta?: string;
     toolCallId?: string;
@@ -3343,7 +3344,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.53.1";
+declare const __version__ = "0.53.2";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -19,7 +19,7 @@ import {
   getCurrentTrace,
   reseedFromRegistry,
   seedFromRegistry
-} from "./chunk-UWXRK2KJ.js";
+} from "./chunk-VRGVD6LA.js";
 import "./chunk-E5BLYL5X.js";
 import {
   BITFAB_PROGRESS_PREFIX,
@@ -28,7 +28,7 @@ import {
   ReplayError,
   reportReplayProgress,
   serializeReplayResult
-} from "./chunk-LIHAOYN5.js";
+} from "./chunk-6J5LDOLY.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -36,7 +36,7 @@ import {
   MixedTracingError,
   __version__,
   flushTraces
-} from "./chunk-XPIB2WYS.js";
+} from "./chunk-W76KGDVL.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BITFAB_PROGRESS_PREFIX,

package/dist/node.cjs

@@ -88,7 +88,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.53.1";
+    __version__ = "0.53.2";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -6724,6 +6724,17 @@ function runWithSeedContext(ctx, fn) {
 init_serialize();
 init_simulationPlan();
 init_spanOrigin();
+
+// src/streamFinalizationError.ts
+var StreamFinalizationError = class extends Error {
+  constructor(message, output) {
+    super(message);
+    this.output = output;
+    this.name = "StreamFinalizationError";
+  }
+};
+
+// src/client.ts
 init_traceMetadata();
 
 // src/tracing.ts
@@ -6997,52 +7008,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) {
@@ -7083,14 +7152,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 };
           }
@@ -8585,8 +8662,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)}`
                 })
               )
             );