@bitfab/sdk 0.32.0 → 0.33.1

package/dist/{chunk-YZZO475T.js → chunk-7UHJ67MH.js}

@@ -1,5 +1,8 @@
 import {
   BitfabError,
+  __privateAdd,
+  __privateGet,
+  __privateSet,
   asyncStorageReady,
   createAsyncLocalStorage,
   deserializeValue,
@@ -11,10 +14,10 @@ import {
   toJsonSafe,
   toJsonSafeReport,
   warnOnce
-} from "./chunk-NMNVPON4.js";
+} from "./chunk-ADUWXWWI.js";
 
 // src/version.generated.ts
-var __version__ = "0.32.0";
+var __version__ = "0.33.1";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -401,7 +404,7 @@ var HttpClient = class {
     if (dbBranchSettings !== void 0) {
       payload.dbBranchSettings = dbBranchSettings;
     }
-    const timeout = includeDbBranchLease ? 18e4 : 3e4;
+    const timeout = includeDbBranchLease ? 3e5 : 3e4;
     return this.request("/api/sdk/replay/start", payload, {
       timeout
     });
@@ -809,7 +812,7 @@ var BitfabClaudeAgentHandler = class {
     }
   }
   // ── hook callbacks ───────────────────────────────────────────
-  async preToolUseHook(inputData, toolUseId, _context) {
+  async preToolUseHook(inputData, toolUseId, _context2) {
     try {
       const sid = inputData.tool_use_id ?? toolUseId ?? randomUuid();
       const toolName = inputData.tool_name ?? "tool";
@@ -821,7 +824,7 @@ var BitfabClaudeAgentHandler = class {
     }
     return {};
   }
-  async postToolUseHook(inputData, toolUseId, _context) {
+  async postToolUseHook(inputData, toolUseId, _context2) {
     try {
       const sid = inputData.tool_use_id ?? toolUseId ?? "";
       const toolResponse = inputData.tool_response;
@@ -830,7 +833,7 @@ var BitfabClaudeAgentHandler = class {
     }
     return {};
   }
-  async postToolUseFailureHook(inputData, toolUseId, _context) {
+  async postToolUseFailureHook(inputData, toolUseId, _context2) {
     try {
       const sid = inputData.tool_use_id ?? toolUseId ?? "";
       const error = String(inputData.error ?? "Unknown error");
@@ -839,7 +842,7 @@ var BitfabClaudeAgentHandler = class {
     }
     return {};
   }
-  async subagentStartHook(inputData, _toolUseId, _context) {
+  async subagentStartHook(inputData, _toolUseId, _context2) {
     try {
       const agentId = inputData.agent_id ?? randomUuid();
       const agentType = inputData.agent_type ?? "subagent";
@@ -857,7 +860,7 @@ var BitfabClaudeAgentHandler = class {
     }
     return {};
   }
-  async subagentStopHook(inputData, _toolUseId, _context) {
+  async subagentStopHook(inputData, _toolUseId, _context2) {
     try {
       const agentId = inputData.agent_id ?? "";
       const spanId = this.activeSubagentSpans.get(agentId);
@@ -2086,101 +2089,42 @@ var BitfabOpenAIAgentHandler = class {
   }
 };
 
-// src/replayEnvironment.ts
-var ReplayEnvironment = class {
-  constructor(options = {}) {
-    this.dbBranch = { ...options };
+// src/replayBranch.ts
+var _url, _context;
+var ReplayBranch = class {
+  /** @internal Built by `getCurrentReplayBranch()`; never constructed by callers. */
+  constructor(lease, traceId, context) {
+    // Genuine JS private fields, not TypeScript `private`: these must be
+    // non-enumerable so neither the connection string nor the whole replay
+    // 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);
+    __privateSet(this, _context, context);
   }
   /**
-   * The per-trace branch URL for the item currently being replayed.
-   * Throws if read outside a replay item.
+   * Connection string for this item's branch. Point your database client at it
+   * instead of the live database for the duration of the replayed call.
+   *
+   * Reading it records on the trace that the replayed code obtained the branch
+   * URL, which is what separates "a branch was provisioned" from "the branch
+   * was actually used". The other fields inspect the lease without exposing the
+   * connection string, so they deliberately do not record anything. That is
+   * also why this is a getter and not a plain field: the URL is absent from
+   * `JSON.stringify(branch)` and from logging the object.
    */
   get databaseUrl() {
-    const snapshot = this.require();
-    this.markAccessed();
-    return snapshot.databaseUrl;
-  }
-  /** When the per-trace branch URL stops being valid. ISO-8601. */
-  get expiresAt() {
-    return this.require().expiresAt;
-  }
-  /** Deep link to the branch in the provider console, if available. */
-  get providerConsoleUrl() {
-    return this.require().providerConsoleUrl;
-  }
-  /**
-   * True if the branch is read-only. Customer code can use this to skip
-   * write operations during replay when the provider returned a read-only
-   * lease.
-   */
-  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;
-  }
-  /** True when read inside a replay item that has a resolved branch. */
-  get active() {
-    return this.read() !== null;
-  }
-  /** Non-throwing variant for callers that handle the inactive case. */
-  snapshot() {
-    const snapshot = this.read();
-    if (snapshot) {
-      this.markAccessed();
-    }
-    return snapshot;
-  }
-  /**
-   * Record on the replay context that customer code obtained the branch
-   * URL. Only `databaseUrl` and `snapshot()` count - `active`, `readOnly`
-   * and friends inspect the lease without exposing the connection string,
-   * so they don't prove the replayed code could have connected to the
-   * branch.
-   */
-  markAccessed() {
-    const ctx = getReplayContext();
-    if (ctx?.dbBranchLease) {
-      ctx.dbSnapshotAccessed = true;
-    }
-  }
-  read() {
-    const ctx = getReplayContext();
-    if (!ctx?.dbBranchLease) {
-      return null;
-    }
-    const traceId = ctx.sourceBitfabTraceId ?? ctx.inputSourceTraceId;
-    if (!traceId) {
-      return null;
-    }
-    const lease = ctx.dbBranchLease;
-    return {
-      databaseUrl: lease.databaseUrl,
-      expiresAt: lease.expiresAt,
-      providerConsoleUrl: lease.providerConsoleUrl,
-      readOnly: lease.readOnly,
-      region: lease.region,
-      traceId
-    };
-  }
-  require() {
-    const snapshot = this.read();
-    if (!snapshot) {
-      throw new Error(
-        "ReplayEnvironment accessed outside of a replay item. Pass it to bitfab.replay({ environment }) and only read it inside the replayed function."
-      );
-    }
-    return snapshot;
+    __privateGet(this, _context).dbSnapshotAccessed = true;
+    return __privateGet(this, _url);
   }
 };
+_url = new WeakMap();
+_context = new WeakMap();
 
 // src/tracing.ts
 var BitfabOpenAITracingProcessor = class {
@@ -2774,6 +2718,17 @@ function getCurrentSpan() {
     }
   };
 }
+function getCurrentReplayBranch() {
+  const ctx = getReplayContext();
+  if (!ctx?.dbBranchLease) {
+    return null;
+  }
+  const traceId = ctx.sourceBitfabTraceId ?? ctx.inputSourceTraceId;
+  if (!traceId) {
+    return null;
+  }
+  return new ReplayBranch(ctx.dbBranchLease, traceId, ctx);
+}
 function getCurrentTrace() {
   const stack = getSpanStack();
   const current = stack[stack.length - 1];
@@ -3837,7 +3792,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-DJWVOJUR.js");
+    const { replay: doReplay } = await import("./replay-EQPISJKC.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3848,12 +3803,6 @@ var Bitfab = class {
     );
   }
 };
-/**
- * Per-trace environment for `replay({ environment })`. Construct one,
- * pass it to replay, and read `env.databaseUrl` inside the replayed
- * function to pick up the per-trace branch URL.
- */
-Bitfab.ReplayEnvironment = ReplayEnvironment;
 var BitfabFunction = class {
   constructor(client, traceFunctionKey) {
     this.client = client;
@@ -4054,13 +4003,13 @@ export {
   SUPPORTED_PROVIDERS,
   BitfabLangGraphCallbackHandler,
   BitfabOpenAIAgentHandler,
-  ReplayEnvironment,
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   getCurrentSpan,
+  getCurrentReplayBranch,
   getCurrentTrace,
   Bitfab,
   BitfabFunction,
   finalizers
 };
-//# sourceMappingURL=chunk-YZZO475T.js.map
+//# sourceMappingURL=chunk-7UHJ67MH.js.map