@bitfab/sdk 0.52.2 → 0.52.4

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.52.2";
+    __version__ = "0.52.4";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -385,15 +385,18 @@ var init_payloadBudget = __esm({
 });
 
 // src/warnOnce.ts
+function warn(message) {
+  try {
+    console.warn(`[bitfab] ${message}`);
+  } catch {
+  }
+}
 function warnOnce(key, message) {
   if (warned.has(key)) {
     return;
   }
   warned.add(key);
-  try {
-    console.warn(`[bitfab] ${message}`);
-  } catch {
-  }
+  warn(message);
 }
 var warned;
 var init_warnOnce = __esm({
@@ -535,6 +538,9 @@ var init_serializePayload = __esm({
 });
 
 // src/traceMetadata.ts
+function findRecord(traceId) {
+  return activeRecords.get(traceId) ?? completedRecords.get(traceId);
+}
 function recordCallerTraceMetadata(traceId, metadata) {
   if (typeof traceId !== "string" || traceId === "") {
     return;
@@ -545,15 +551,52 @@ function recordCallerTraceMetadata(traceId, metadata) {
   if (Object.keys(metadata).length === 0) {
     return;
   }
-  callerMetadata.set(traceId, { ...callerMetadata.get(traceId), ...metadata });
+  const existing = findRecord(traceId);
+  if (existing) {
+    Object.assign(existing.caller, metadata);
+    return;
+  }
+  activeRecords.set(traceId, {
+    caller: { ...metadata },
+    derived: {},
+    warnedKeys: /* @__PURE__ */ new Set()
+  });
 }
 function callerTraceMetadata(traceId) {
-  const recorded = callerMetadata.get(traceId);
-  return recorded ? { ...recorded } : void 0;
+  const record = findRecord(traceId);
+  if (!record || Object.keys(record.caller).length === 0) {
+    return void 0;
+  }
+  return { ...record.caller };
 }
-function forgetTraceMetadata(traceId) {
-  callerMetadata.delete(traceId);
-  derivedMetadata.delete(traceId);
+function retireTraceMetadata(traceId) {
+  const record = activeRecords.get(traceId);
+  if (!record) {
+    return;
+  }
+  activeRecords.delete(traceId);
+  completedRecords.set(traceId, record);
+  while (completedRecords.size > RETAINED_COMPLETED_TRACES) {
+    const oldest = completedRecords.keys().next();
+    if (oldest.done) {
+      return;
+    }
+    completedRecords.delete(oldest.value);
+  }
+}
+function warnAboutShadowedKeys(traceId, record) {
+  const shadowed = Object.keys(record.derived).filter(
+    (key) => key in record.caller && record.caller[key] !== record.derived[key] && !record.warnedKeys.has(key)
+  ).sort();
+  if (shadowed.length === 0) {
+    return;
+  }
+  for (const key of shadowed) {
+    record.warnedKeys.add(key);
+  }
+  warn(
+    `trace metadata key(s) ${shadowed.join(", ")} on trace ${traceId} were set both by the caller and by an integration's own trace export; the caller's value is the one kept on the trace.`
+  );
 }
 function mergeCallerMetadataIntoTracePayload(payload) {
   const externalTrace = payload.externalTrace;
@@ -565,35 +608,31 @@ function mergeCallerMetadataIntoTracePayload(payload) {
   if (typeof traceId !== "string" || traceId === "") {
     return payload;
   }
-  const caller = callerMetadata.get(traceId);
-  if (!caller) {
+  const record = findRecord(traceId);
+  if (!record || Object.keys(record.caller).length === 0) {
     return payload;
   }
-  const derived = derivedMetadata.get(traceId) ?? {};
   const exported = external.metadata;
   if (typeof exported === "object" && exported !== null) {
-    Object.assign(derived, exported);
-  }
-  derivedMetadata.set(traceId, derived);
-  const shadowed = Object.keys(derived).filter((key) => key in caller && caller[key] !== derived[key]).sort();
-  if (shadowed.length > 0) {
-    warnOnce(
-      `trace-metadata-shadowed:${shadowed.join(",")}`,
-      `trace metadata key(s) ${shadowed.join(", ")} were set both by the caller and by an integration's own trace export; the caller's value is the one kept on the trace.`
-    );
+    Object.assign(record.derived, exported);
   }
+  warnAboutShadowedKeys(traceId, record);
   return {
     ...payload,
-    externalTrace: { ...external, metadata: { ...derived, ...caller } }
+    externalTrace: {
+      ...external,
+      metadata: { ...record.derived, ...record.caller }
+    }
   };
 }
-var callerMetadata, derivedMetadata;
+var RETAINED_COMPLETED_TRACES, activeRecords, completedRecords;
 var init_traceMetadata = __esm({
   "src/traceMetadata.ts"() {
     "use strict";
     init_warnOnce();
-    callerMetadata = /* @__PURE__ */ new Map();
-    derivedMetadata = /* @__PURE__ */ new Map();
+    RETAINED_COMPLETED_TRACES = 128;
+    activeRecords = /* @__PURE__ */ new Map();
+    completedRecords = /* @__PURE__ */ new Map();
   }
 });
 
@@ -2076,6 +2115,18 @@ var init_http = __esm({
           { timeout: REPLAY_COMPLETE_REQUEST_TIMEOUT_MS }
         );
       }
+      async getReseedSource(traceId) {
+        return this.get(
+          `/api/sdk/traces/${encodeURIComponent(traceId)}/reseedSource`
+        );
+      }
+      async reseedTrace(traceId, runTraceId) {
+        return this.request(
+          `/api/sdk/traces/${encodeURIComponent(traceId)}/reseed`,
+          { runTraceId },
+          { timeout: 3e4 }
+        );
+      }
       /**
        * Ask the server to materialize a per-trace DB branch lease from a
        * captured `dbSnapshotRef`. Blocking - the resolver creates a Neon
@@ -2520,6 +2571,7 @@ __export(replay_exports, {
   BITFAB_PROGRESS_PREFIX: () => BITFAB_PROGRESS_PREFIX,
   DbBranchReplayError: () => DbBranchReplayError,
   ReplayError: () => ReplayError,
+  deserializeInputs: () => deserializeInputs,
   replay: () => replay,
   reportReplayProgress: () => reportReplayProgress,
   serializeReplayResult: () => serializeReplayResult,
@@ -3370,6 +3422,7 @@ __export(index_exports, {
   getCurrentSpan: () => getCurrentSpan,
   getCurrentTrace: () => getCurrentTrace,
   reportReplayProgress: () => reportReplayProgress,
+  reseedFromRegistry: () => reseedFromRegistry,
   seedFromRegistry: () => seedFromRegistry,
   serializeReplayResult: () => serializeReplayResult
 });
@@ -5679,6 +5732,7 @@ var BitfabOpenAIAgentHandler = class {
 
 // src/client.ts
 init_randomUuid();
+init_replay();
 
 // src/replayBranch.ts
 var _url, _context;
@@ -7458,7 +7512,7 @@ var Bitfab = class {
                 }
               });
               activeTraceStates.delete(traceId);
-              forgetTraceMetadata(traceId);
+              retireTraceMetadata(traceId);
             }
           } catch {
           }
@@ -7651,7 +7705,7 @@ var Bitfab = class {
       } catch (setupError) {
         if (registeredTraceId) {
           activeTraceStates.delete(registeredTraceId);
-          forgetTraceMetadata(registeredTraceId);
+          retireTraceMetadata(registeredTraceId);
         }
         if (setupError instanceof MixedTracingError) {
           throw setupError;
@@ -8050,7 +8104,7 @@ var Bitfab = class {
       });
     } finally {
       activeTraceStates.delete(traceId);
-      forgetTraceMetadata(traceId);
+      retireTraceMetadata(traceId);
     }
     return traceId;
   }
@@ -8096,7 +8150,7 @@ var Bitfab = class {
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
-        forgetTraceMetadata(traceId);
+        retireTraceMetadata(traceId);
       }
     }
     if (unrecorded) {
@@ -8106,6 +8160,36 @@ var Bitfab = class {
     }
     return traceId;
   }
+  async reseedTrace(traceFunctionKey, fn, options) {
+    const source = await this.httpClient.getReseedSource(options.traceId);
+    if (source.traceFunctionKey !== traceFunctionKey) {
+      throw new BitfabError(
+        `Trace ${options.traceId} belongs to '${source.traceFunctionKey}', not '${traceFunctionKey}'.`
+      );
+    }
+    const args = deserializeInputs({
+      input: source.input,
+      input_meta: source.inputMeta
+    });
+    const runTraceId = await this.seedTraceByRunning(
+      traceFunctionKey,
+      fn,
+      {
+        args,
+        metadata: source.metadata,
+        ...source.sessionId === null ? {} : { sessionId: source.sessionId },
+        ...source.name === null ? {} : { name: source.name }
+      }
+    );
+    const adopted = await this.httpClient.reseedTrace(
+      options.traceId,
+      runTraceId
+    );
+    return {
+      traceId: adopted.traceId,
+      previousRunTraceId: adopted.previousRunTraceId
+    };
+  }
   async replay(traceFunctionKey, fn, options) {
     const wrappedKey = fn._bitfabTraceFunctionKey;
     let replayFn = fn;
@@ -8367,6 +8451,24 @@ init_replay();
 function defineReplayRegistry(registry) {
   return registry;
 }
+async function reseedFromRegistry(registry, pipeline, traceIds) {
+  const registration = registry[pipeline];
+  if (registration === void 0) {
+    throw new BitfabError(
+      `Unknown pipeline '${pipeline}'. Registered: ${Object.keys(registry).join(", ")}`
+    );
+  }
+  const traceFunctionKey = resolveTraceFunctionKey(registration);
+  const reseeded = [];
+  for (const traceId of traceIds) {
+    reseeded.push(
+      await registration.client.reseedTrace(traceFunctionKey, registration.fn, {
+        traceId
+      })
+    );
+  }
+  return { pipeline, traceFunctionKey, reseeded };
+}
 async function seedFromRegistry(registry, pipeline, cases, options = {}) {
   const registration = registry[pipeline];
   if (registration === void 0) {
@@ -8443,6 +8545,7 @@ function resolveTraceFunctionKey(registration) {
   getCurrentSpan,
   getCurrentTrace,
   reportReplayProgress,
+  reseedFromRegistry,
   seedFromRegistry,
   serializeReplayResult
 });