bitfab 0.36.0 → 0.36.2

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.36.0";
+    __version__ = "0.36.2";
   }
 });
 
@@ -1530,9 +1530,11 @@ var init_http = __esm({
       /**
        * Fetch an external span by ID.
        * Blocking GET request.
+       * The replay view limits rawData to input/output serialization fields.
        */
-      async getExternalSpan(spanId) {
-        const url = `${this.serviceUrl}/api/sdk/externalSpans/${spanId}`;
+      async getExternalSpan(spanId, options) {
+        const query = options?.view === "replay" ? "?view=replay" : "";
+        const url = `${this.serviceUrl}/api/sdk/externalSpans/${spanId}${query}`;
         const controller = new AbortController();
         const timeoutId = setTimeout(() => controller.abort(), 3e4);
         try {
@@ -1570,9 +1572,18 @@ var init_http = __esm({
        * Pass `includeOutputs: false` for a payload-free tree (structure +
        * `externalSpanId` only), so recorded outputs are fetched lazily per mocked
        * span instead of all up front. Omit it (default eager) for `mock: "all"`.
+       * Pass `includeRootOutput: false` when the root was already fetched.
        */
       async getSpanTree(externalSpanId, options) {
-        const query = options?.includeOutputs === false ? "?includeOutputs=false" : "";
+        const searchParams = new URLSearchParams();
+        if (options?.includeOutputs === false) {
+          searchParams.set("includeOutputs", "false");
+        }
+        if (options?.includeRootOutput === false) {
+          searchParams.set("includeRootOutput", "false");
+        }
+        const encodedQuery = searchParams.toString();
+        const query = encodedQuery ? `?${encodedQuery}` : "";
         const url = `${this.serviceUrl}/api/sdk/replay/spanTree/${externalSpanId}${query}`;
         const controller = new AbortController();
         const timeoutId = setTimeout(() => controller.abort(), 3e4);
@@ -2155,7 +2166,9 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
         `Replay requested a database branch for trace ${originalTraceId} but it could not be resolved (${leaseError.code}): ${leaseError.message}. The function was not run, because replaying it against the live database would produce a result that looks valid but did not use the historical data you asked for.`
       );
     }
-    const span = await httpClient.getExternalSpan(originalSpanId);
+    const span = await httpClient.getExternalSpan(originalSpanId, {
+      view: "replay"
+    });
     const spanData = span.rawData?.span_data ?? {};
     inputs = deserializeInputs(spanData);
     originalOutput = deserializeOutput(spanData);
@@ -2175,7 +2188,8 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     if (needTree) {
       try {
         const treeResponse = await httpClient.getSpanTree(originalSpanId, {
-          includeOutputs
+          includeOutputs,
+          includeRootOutput: false
         });
         if (treeResponse.root) {
           mockTree = buildMockTree(treeResponse.root);
@@ -2197,7 +2211,7 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     const fetchSpanOutput = mockTree && !includeOutputs ? (externalSpanId) => {
       let pending = outputCache.get(externalSpanId);
       if (!pending) {
-        pending = httpClient.getExternalSpan(externalSpanId).then(
+        pending = httpClient.getExternalSpan(externalSpanId, { view: "replay" }).then(
           (s) => deserializeOutput(
             s.rawData?.span_data ?? {}
           )
@@ -4190,12 +4204,20 @@ var ReplayBranch = class {
     // context can ride along into a log line or a serialized payload.
     __privateAdd(this, _url);
     __privateAdd(this, _context);
-    this.expiresAt = lease.expiresAt;
-    this.providerConsoleUrl = lease.providerConsoleUrl;
-    this.readOnly = lease.readOnly;
-    this.region = lease.region;
-    this.traceId = traceId;
-    __privateSet(this, _url, lease.databaseUrl);
+    const { databaseUrl, ...exposed } = lease;
+    for (const [key, value] of Object.entries(exposed)) {
+      Object.defineProperty(this, key, {
+        value,
+        enumerable: true,
+        configurable: true
+      });
+    }
+    Object.defineProperty(this, "traceId", {
+      value: traceId,
+      enumerable: true,
+      configurable: true
+    });
+    __privateSet(this, _url, databaseUrl);
     __privateSet(this, _context, context);
   }
   /**
@@ -5535,6 +5557,7 @@ var Bitfab = class {
                   dbSnapshotUsage: {
                     neonBranchId: replayCtx.dbBranchLease.neonBranchId,
                     snapshotTimestamp: replayCtx.dbBranchLease.snapshotTimestamp,
+                    region: replayCtx.dbBranchLease.region,
                     originalTraceId: replayCtx.sourceBitfabTraceId,
                     accessed: replayCtx.dbSnapshotAccessed === true
                   }
@@ -5810,6 +5833,9 @@ var Bitfab = class {
         ...params.dbSnapshotUsage.snapshotTimestamp && {
           snapshot_timestamp: params.dbSnapshotUsage.snapshotTimestamp
         },
+        ...params.dbSnapshotUsage.region && {
+          region: params.dbSnapshotUsage.region
+        },
         ...params.dbSnapshotUsage.originalTraceId && {
           original_trace_id: params.dbSnapshotUsage.originalTraceId,
           // Deprecated wire alias, kept so this SDK still reports usage