@bitfab/sdk 0.30.2 → 0.31.0

package/dist/index.cjs

@@ -524,6 +524,17 @@ __export(replay_exports, {
   replay: () => replay,
   reportReplayProgress: () => reportReplayProgress
 });
+function resolveDbBranchSettings(dbBranch) {
+  if (!dbBranch) {
+    return void 0;
+  }
+  const settings = {
+    ...dbBranch.minCu === void 0 ? {} : { minCu: dbBranch.minCu },
+    ...dbBranch.maxCu === void 0 ? {} : { maxCu: dbBranch.maxCu },
+    ...dbBranch.warmupSql === void 0 ? {} : { warmupSql: dbBranch.warmupSql }
+  };
+  return Object.keys(settings).length === 0 ? void 0 : settings;
+}
 function reportReplayProgress(progress) {
   const stderr = typeof process !== "undefined" ? process.stderr : void 0;
   if (!stderr) {
@@ -586,6 +597,7 @@ function buildMockTree(rootNode) {
 }
 async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy, resolvedOverrides, replayedTraceId, environment, adaptInputs) {
   const lease = environment ? serverItem.dbBranchLease : void 0;
+  const leaseError = environment ? serverItem.dbBranchLeaseError : void 0;
   let inputs = [];
   let originalOutput;
   let result;
@@ -594,6 +606,11 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
   const originalTraceId = serverItem.originalTraceId ?? serverItem.sourceTraceId;
   const originalSpanId = serverItem.originalSpanId ?? serverItem.sourceSpanId;
   try {
+    if (leaseError) {
+      throw new BitfabError(
+        `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 spanData = span.rawData?.span_data ?? {};
     inputs = deserializeInputs(spanData);
@@ -767,7 +784,8 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
     // includeDbBranchLease
     options?.experimentGroupId,
     options?.datasetId,
-    options?.graderIds
+    options?.graderIds,
+    resolveDbBranchSettings(options?.dbBranch)
   );
   const mockStrategy = options?.mock ?? "marked";
   const maxConcurrency = options?.maxConcurrency ?? 10;
@@ -954,7 +972,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/version.generated.ts
-var __version__ = "0.30.2";
+var __version__ = "0.31.0";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1313,7 +1331,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, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId, graderIds) {
+  async startReplay(traceFunctionKey, limit, traceIds, name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, experimentGroupId, datasetId, graderIds, dbBranchSettings) {
     const payload = { traceFunctionKey };
     if (limit !== void 0) {
       payload.limit = limit;
@@ -1342,6 +1360,9 @@ var HttpClient = class {
     if (graderIds !== void 0) {
       payload.graderIds = graderIds;
     }
+    if (dbBranchSettings !== void 0) {
+      payload.dbBranchSettings = dbBranchSettings;
+    }
     const timeout = includeDbBranchLease ? 18e4 : 3e4;
     return this.request("/api/sdk/replay/start", payload, {
       timeout
@@ -3072,6 +3093,13 @@ var ReplayEnvironment = class {
   get readOnly() {
     return this.require().readOnly;
   }
+  /**
+   * The branch's region, e.g. `aws-us-east-1`. A compute runs in its project's
+   * region, so a replay runner elsewhere pays that round trip on every query.
+   */
+  get region() {
+    return this.require().region;
+  }
   /** The historical trace ID that produced the input for this replay item. */
   get traceId() {
     return this.require().traceId;
@@ -3116,6 +3144,7 @@ var ReplayEnvironment = class {
       expiresAt: lease.expiresAt,
       providerConsoleUrl: lease.providerConsoleUrl,
       readOnly: lease.readOnly,
+      region: lease.region,
       traceId
     };
   }