@bitfab/sdk 0.19.1 → 0.21.0

package/dist/{chunk-WZ72P5SX.js → chunk-UO3CIQ7R.js}

@@ -10,7 +10,7 @@ import {
 } from "./chunk-EQI6ZJC3.js";
 
 // src/version.generated.ts
-var __version__ = "0.19.1";
+var __version__ = "0.21.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -287,7 +287,7 @@ var HttpClient = class {
    * Start a replay session by fetching historical traces.
    * Blocking call — creates a test run and returns lightweight item references.
    */
-  async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId) {
+  async startReplay(traceFunctionKey, limit, traceIds, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId) {
     const payload = { traceFunctionKey };
     if (limit !== void 0) {
       payload.limit = limit;
@@ -307,6 +307,9 @@ var HttpClient = class {
     if (experimentGroupId !== void 0) {
       payload.experimentGroupId = experimentGroupId;
     }
+    if (datasetId !== void 0) {
+      payload.datasetId = datasetId;
+    }
     const timeout = includeDbBranchLease ? 18e4 : 3e4;
     return this.request("/api/sdk/replay/start", payload, {
       timeout
@@ -2651,11 +2654,11 @@ var Bitfab = class {
         startedAt,
         spanType: options.type ?? "custom"
       };
-      const sendSpan = async (params) => {
+      const sendSpan = async (params, spanOpts) => {
         const replayCtx = getReplayContext();
         const persistenceCollector = isRootSpan ? replayCtx?.pendingPersistence : void 0;
         let resolvePersistence;
-        if (persistenceCollector) {
+        if (persistenceCollector && !spanOpts?.skipPersistenceRegistration) {
           persistenceCollector.push(
             new Promise((resolve) => {
               resolvePersistence = resolve;
@@ -2755,11 +2758,41 @@ var Bitfab = class {
           }
         }
       }
+      const recordSpan = (result) => {
+        if (options.finalize) {
+          const replayCtx = getReplayContext();
+          const persistenceCollector = isRootSpan ? replayCtx?.pendingPersistence : void 0;
+          let resolvePersistence;
+          if (persistenceCollector) {
+            persistenceCollector.push(
+              new Promise((resolve) => {
+                resolvePersistence = resolve;
+              })
+            );
+          }
+          void Promise.resolve().then(() => options.finalize(result)).then(
+            (output) => sendSpan(
+              { result: output },
+              { skipPersistenceRegistration: true }
+            )
+          ).catch(
+            (error) => sendSpan(
+              {
+                result: void 0,
+                error: error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
+              },
+              { skipPersistenceRegistration: true }
+            )
+          ).finally(() => resolvePersistence?.());
+        } else {
+          void sendSpan({ result });
+        }
+      };
       const executeWithContext = () => {
         const result = fn(...args);
         if (result instanceof Promise) {
           return result.then((resolvedResult) => {
-            void sendSpan({ result: resolvedResult });
+            recordSpan(resolvedResult);
             return resolvedResult;
           }).catch((error) => {
             void sendSpan({
@@ -2772,7 +2805,7 @@ var Bitfab = class {
         if (isAsyncGenerator(result)) {
           return wrapAsyncGenerator(result, newStack, sendSpan);
         }
-        void sendSpan({ result });
+        recordSpan(result);
         return result;
       };
       return runWithSpanStack(newStack, executeWithContext);
@@ -2991,7 +3024,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-QAWGVRCZ.js");
+    const { replay: doReplay } = await import("./replay-V6RPJYXZ.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3056,6 +3089,54 @@ var BitfabFunction = class {
   }
 };
 
+// src/finalizers.ts
+async function settle(value) {
+  try {
+    return await value;
+  } catch {
+    return void 0;
+  }
+}
+async function aiSdk(result) {
+  const r = result ?? {};
+  const [text, usage, totalUsage, finishReason, toolCalls, toolResults] = await Promise.all([
+    settle(r.text),
+    settle(r.usage),
+    settle(r.totalUsage),
+    settle(r.finishReason),
+    settle(r.toolCalls),
+    settle(r.toolResults)
+  ]);
+  return {
+    text,
+    usage: totalUsage ?? usage,
+    finishReason,
+    toolCalls,
+    toolResults
+  };
+}
+async function readableStream(stream, onLive) {
+  const [live, copy] = stream.tee();
+  onLive(live);
+  const chunks = [];
+  const reader = copy.getReader();
+  try {
+    for (; ; ) {
+      const { done, value } = await reader.read();
+      if (done) {
+        break;
+      }
+      chunks.push(value);
+    }
+  } catch {
+  }
+  return { chunks };
+}
+var finalizers = {
+  aiSdk,
+  readableStream
+};
+
 export {
   __version__,
   DEFAULT_SERVICE_URL,
@@ -3068,6 +3149,7 @@ export {
   getCurrentSpan,
   getCurrentTrace,
   Bitfab,
-  BitfabFunction
+  BitfabFunction,
+  finalizers
 };
-//# sourceMappingURL=chunk-WZ72P5SX.js.map
+//# sourceMappingURL=chunk-UO3CIQ7R.js.map