braintrust 0.3.8 → 0.4.0

package/dist/browser.js

@@ -87,6 +87,44 @@ var Queue = (_class = class {
   }
 }, _class);
 
+// src/id-gen.ts
+
+function generateHexId(bytes) {
+  let result = "";
+  for (let i = 0; i < bytes; i++) {
+    result += Math.floor(Math.random() * 256).toString(16).padStart(2, "0");
+  }
+  return result;
+}
+var IDGenerator = class {
+};
+var UUIDGenerator = class extends IDGenerator {
+  getSpanId() {
+    return _uuid.v4.call(void 0, );
+  }
+  getTraceId() {
+    return _uuid.v4.call(void 0, );
+  }
+  shareRootSpanId() {
+    return true;
+  }
+};
+var OTELIDGenerator = class extends IDGenerator {
+  getSpanId() {
+    return generateHexId(8);
+  }
+  getTraceId() {
+    return generateHexId(16);
+  }
+  shareRootSpanId() {
+    return false;
+  }
+};
+function getIdGenerator() {
+  const useOtel = typeof process !== "undefined" && _optionalChain([process, 'access', _2 => _2.env, 'optionalAccess', _3 => _3.BRAINTRUST_OTEL_COMPAT, 'optionalAccess', _4 => _4.toLowerCase, 'call', _5 => _5()]) === "true";
+  return useOtel ? new OTELIDGenerator() : new UUIDGenerator();
+}
+
 // util/db_fields.ts
 var TRANSACTION_ID_FIELD = "_xact_id";
 var IS_MERGE_FIELD = "_is_merge";
@@ -237,7 +275,7 @@ var SpanComponentsV1 = class _SpanComponentsV1 {
     return {
       objectType: this.objectType,
       objectId: this.objectId,
-      rowIds: _optionalChain([this, 'access', _2 => _2.rowIds, 'optionalAccess', _3 => _3.toObject, 'call', _4 => _4()])
+      rowIds: _optionalChain([this, 'access', _6 => _6.rowIds, 'optionalAccess', _7 => _7.toObject, 'call', _8 => _8()])
     };
   }
 };
@@ -451,7 +489,7 @@ var SpanComponentsV2 = class _SpanComponentsV2 {
       objectType: this.objectType,
       objectId: this.objectId,
       computeObjectMetadataArgs: this.computeObjectMetadataArgs,
-      rowIds: _optionalChain([this, 'access', _5 => _5.rowIds, 'optionalAccess', _6 => _6.toObject, 'call', _7 => _7()])
+      rowIds: _optionalChain([this, 'access', _9 => _9.rowIds, 'optionalAccess', _10 => _10.toObject, 'call', _11 => _11()])
     };
   }
 };
@@ -783,14 +821,14 @@ function depthFirstSearch(args) {
   while (events.length) {
     const { eventType, vertex, extras } = events.pop();
     if (eventType === "last") {
-      _optionalChain([lastVisitF, 'optionalCall', _8 => _8(vertex)]);
+      _optionalChain([lastVisitF, 'optionalCall', _12 => _12(vertex)]);
       continue;
     }
     if (firstVisitedVertices.has(vertex)) {
       continue;
     }
     firstVisitedVertices.add(vertex);
-    _optionalChain([firstVisitF, 'optionalCall', _9 => _9(vertex, { parentVertex: extras.parentVertex })]);
+    _optionalChain([firstVisitF, 'optionalCall', _13 => _13(vertex, { parentVertex: extras.parentVertex })]);
     events.push({ eventType: "last", vertex, extras: {} });
     mapAt(graph, vertex).forEach((child) => {
       events.push({
@@ -812,7 +850,7 @@ function undirectedConnectedComponents(graph) {
   let labelCounter = 0;
   const vertexLabels = /* @__PURE__ */ new Map();
   const firstVisitF = (vertex, args) => {
-    const label = _optionalChain([args, 'optionalAccess', _10 => _10.parentVertex]) !== void 0 ? mapAt(vertexLabels, _optionalChain([args, 'optionalAccess', _11 => _11.parentVertex])) : labelCounter++;
+    const label = _optionalChain([args, 'optionalAccess', _14 => _14.parentVertex]) !== void 0 ? mapAt(vertexLabels, _optionalChain([args, 'optionalAccess', _15 => _15.parentVertex])) : labelCounter++;
     vertexLabels.set(vertex, label);
   };
   depthFirstSearch({ graph: directedGraph, firstVisitF });
@@ -1034,6 +1072,41 @@ function _urljoin(...parts) {
   ).filter((x) => x.trim() !== "").join("/");
 }
 
+// util/span_identifier_v4.ts
+
+var ENCODING_VERSION_NUMBER_V4 = 4;
+var INVALID_ENCODING_ERRMSG_V4 = `SpanComponents string is not properly encoded. This library only supports encoding versions up to ${ENCODING_VERSION_NUMBER_V4}. Please make sure the SDK library used to decode the SpanComponents is at least as new as any library used to encode it.`;
+var spanComponentsV4Schema = _v3.z.object({
+  object_type: spanObjectTypeV3EnumSchema,
+  propagated_event: _v3.z.record(_v3.z.unknown()).nullish()
+}).and(
+  _v3.z.union([
+    // Must provide one or the other.
+    _v3.z.object({
+      object_id: _v3.z.string().nullish(),
+      compute_object_metadata_args: _v3.z.optional(_v3.z.null())
+    }),
+    _v3.z.object({
+      object_id: _v3.z.optional(_v3.z.null()),
+      compute_object_metadata_args: _v3.z.record(_v3.z.unknown())
+    })
+  ])
+).and(
+  _v3.z.union([
+    // Either all of these must be provided or none.
+    _v3.z.object({
+      row_id: _v3.z.string(),
+      span_id: _v3.z.string(),
+      root_span_id: _v3.z.string()
+    }),
+    _v3.z.object({
+      row_id: _v3.z.optional(_v3.z.null()),
+      span_id: _v3.z.optional(_v3.z.null()),
+      root_span_id: _v3.z.optional(_v3.z.null())
+    })
+  ])
+);
+
 // util/git_fields.ts
 function mergeGitMetadataSettings(s1, s2) {
   if (s1.collect === "all") {
@@ -3034,7 +3107,7 @@ var NoopSpan = (_class4 = class {
     return this;
   }
   end(args) {
-    return _nullishCoalesce(_optionalChain([args, 'optionalAccess', _12 => _12.endTime]), () => ( getCurrentUnixTimestamp()));
+    return _nullishCoalesce(_optionalChain([args, 'optionalAccess', _16 => _16.endTime]), () => ( getCurrentUnixTimestamp()));
   }
   async export() {
     return "";
@@ -3087,7 +3160,7 @@ var loginSchema = _zod.z.strictObject({
 });
 var stateNonce = 0;
 var BraintrustState = (_class5 = class _BraintrustState {
-  constructor(loginParams) {;_class5.prototype.__init7.call(this);_class5.prototype.__init8.call(this);_class5.prototype.__init9.call(this);_class5.prototype.__init10.call(this);_class5.prototype.__init11.call(this);_class5.prototype.__init12.call(this);_class5.prototype.__init13.call(this);_class5.prototype.__init14.call(this);_class5.prototype.__init15.call(this);_class5.prototype.__init16.call(this);_class5.prototype.__init17.call(this);_class5.prototype.__init18.call(this);_class5.prototype.__init19.call(this);
+  constructor(loginParams) {;_class5.prototype.__init7.call(this);_class5.prototype.__init8.call(this);_class5.prototype.__init9.call(this);_class5.prototype.__init10.call(this);_class5.prototype.__init11.call(this);_class5.prototype.__init12.call(this);_class5.prototype.__init13.call(this);_class5.prototype.__init14.call(this);_class5.prototype.__init15.call(this);_class5.prototype.__init16.call(this);_class5.prototype.__init17.call(this);_class5.prototype.__init18.call(this);_class5.prototype.__init19.call(this);_class5.prototype.__init20.call(this);
     this.loginParams = loginParams;
     this.id = `${(/* @__PURE__ */ new Date()).toLocaleString()}-${stateNonce++}`;
     this.currentExperiment = void 0;
@@ -3141,6 +3214,7 @@ var BraintrustState = (_class5 = class _BraintrustState {
   __init18() {this._apiConn = null}
   __init19() {this._proxyConn = null}
   
+  __init20() {this._idGenerator = null}
   resetLoginInfo() {
     this.appUrl = null;
     this.appPublicUrl = null;
@@ -3155,6 +3229,15 @@ var BraintrustState = (_class5 = class _BraintrustState {
     this._apiConn = null;
     this._proxyConn = null;
   }
+  resetIdGenState() {
+    this._idGenerator = null;
+  }
+  get idGenerator() {
+    if (this._idGenerator === null) {
+      this._idGenerator = getIdGenerator();
+    }
+    return this._idGenerator;
+  }
   copyLoginInfo(other) {
     this.appUrl = other.appUrl;
     this.appPublicUrl = other.appPublicUrl;
@@ -3222,8 +3305,8 @@ var BraintrustState = (_class5 = class _BraintrustState {
   setFetch(fetch2) {
     this.loginParams.fetch = fetch2;
     this.fetch = fetch2;
-    _optionalChain([this, 'access', _13 => _13._apiConn, 'optionalAccess', _14 => _14.setFetch, 'call', _15 => _15(fetch2)]);
-    _optionalChain([this, 'access', _16 => _16._appConn, 'optionalAccess', _17 => _17.setFetch, 'call', _18 => _18(fetch2)]);
+    _optionalChain([this, 'access', _17 => _17._apiConn, 'optionalAccess', _18 => _18.setFetch, 'call', _19 => _19(fetch2)]);
+    _optionalChain([this, 'access', _20 => _20._appConn, 'optionalAccess', _21 => _21.setFetch, 'call', _22 => _22(fetch2)]);
   }
   setMaskingFunction(maskingFunction) {
     this.bgLogger().setMaskingFunction(maskingFunction);
@@ -3335,7 +3418,7 @@ function useTestBackgroundLogger() {
   return logger;
 }
 function clearTestBackgroundLogger() {
-  _optionalChain([_internalGetGlobalState, 'call', _19 => _19(), 'optionalAccess', _20 => _20.setOverrideBgLogger, 'call', _21 => _21(null)]);
+  _optionalChain([_internalGetGlobalState, 'call', _23 => _23(), 'optionalAccess', _24 => _24.setOverrideBgLogger, 'call', _25 => _25(null)]);
 }
 function initTestExperiment(experimentName, projectName) {
   setInitialTestState();
@@ -4008,9 +4091,9 @@ async function permalink(slug, opts) {
   if (slug === "") {
     return NOOP_SPAN_PERMALINK;
   }
-  const state = _nullishCoalesce(_optionalChain([opts, 'optionalAccess', _22 => _22.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([opts, 'optionalAccess', _26 => _26.state]), () => ( _globalState));
   const getOrgName = async () => {
-    if (_optionalChain([opts, 'optionalAccess', _23 => _23.orgName])) {
+    if (_optionalChain([opts, 'optionalAccess', _27 => _27.orgName])) {
       return opts.orgName;
     }
     await state.login({});
@@ -4020,7 +4103,7 @@ async function permalink(slug, opts) {
     return state.orgName;
   };
   const getAppUrl = async () => {
-    if (_optionalChain([opts, 'optionalAccess', _24 => _24.appUrl])) {
+    if (_optionalChain([opts, 'optionalAccess', _28 => _28.appUrl])) {
       return opts.appUrl;
     }
     await state.login({});
@@ -4107,8 +4190,8 @@ var Logger = (_class6 = class {
   
   
   // For type identification.
-  __init20() {this.kind = "logger"}
-  constructor(state, lazyMetadata, logOptions = {}) {;_class6.prototype.__init20.call(this);
+  __init21() {this.kind = "logger"}
+  constructor(state, lazyMetadata, logOptions = {}) {;_class6.prototype.__init21.call(this);
     this.lazyMetadata = lazyMetadata;
     this._asyncFlush = logOptions.asyncFlush;
     this.computeMetadataArgs = logOptions.computeMetadataArgs;
@@ -4150,7 +4233,7 @@ var Logger = (_class6 = class {
    * @returns The `id` of the logged event.
    */
   log(event, options) {
-    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _25 => _25.allowConcurrentWithSpans])) {
+    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _29 => _29.allowConcurrentWithSpans])) {
       throw new Error(
         "Cannot run toplevel `log` method while using spans. To log to the span, call `logger.traced` and then log with `span.log`"
       );
@@ -4218,12 +4301,12 @@ var Logger = (_class6 = class {
       state: this.state,
       ...startSpanParentArgs({
         state: this.state,
-        parent: _optionalChain([args, 'optionalAccess', _26 => _26.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _30 => _30.parent]),
         parentObjectType: this.parentObjectType(),
         parentObjectId: this.lazyId,
         parentComputeObjectMetadataArgs: this.computeMetadataArgs,
-        parentSpanIds: _optionalChain([args, 'optionalAccess', _27 => _27.parentSpanIds]),
-        propagatedEvent: _optionalChain([args, 'optionalAccess', _28 => _28.propagatedEvent])
+        parentSpanIds: _optionalChain([args, 'optionalAccess', _31 => _31.parentSpanIds]),
+        propagatedEvent: _optionalChain([args, 'optionalAccess', _32 => _32.propagatedEvent])
       }),
       defaultRootType: "task" /* TASK */
     });
@@ -4297,9 +4380,9 @@ function constructLogs3Data(items) {
 function now() {
   return (/* @__PURE__ */ new Date()).getTime();
 }
-var TestBackgroundLogger = (_class7 = class {constructor() { _class7.prototype.__init21.call(this);_class7.prototype.__init22.call(this); }
-  __init21() {this.items = []}
-  __init22() {this.maskingFunction = null}
+var TestBackgroundLogger = (_class7 = class {constructor() { _class7.prototype.__init22.call(this);_class7.prototype.__init23.call(this); }
+  __init22() {this.items = []}
+  __init23() {this.maskingFunction = null}
   log(items) {
     this.items.push(items);
   }
@@ -4353,26 +4436,26 @@ var BACKGROUND_LOGGER_BASE_SLEEP_TIME_S = 1;
 var HTTPBackgroundLogger = (_class8 = class _HTTPBackgroundLogger {
   
   
-  __init23() {this.activeFlush = Promise.resolve()}
-  __init24() {this.activeFlushResolved = true}
-  __init25() {this.activeFlushError = void 0}
+  __init24() {this.activeFlush = Promise.resolve()}
+  __init25() {this.activeFlushResolved = true}
+  __init26() {this.activeFlushError = void 0}
   
-  __init26() {this.maskingFunction = null}
-  __init27() {this.syncFlush = false}
+  __init27() {this.maskingFunction = null}
+  __init28() {this.syncFlush = false}
   // 6 MB for the AWS lambda gateway (from our own testing).
-  __init28() {this.maxRequestSize = 6 * 1024 * 1024}
-  __init29() {this.defaultBatchSize = 100}
-  __init30() {this.numTries = 3}
-  __init31() {this.queueDropExceedingMaxsize = DEFAULT_QUEUE_SIZE}
-  __init32() {this.queueDropLoggingPeriod = 60}
-  __init33() {this.failedPublishPayloadsDir = void 0}
-  __init34() {this.allPublishPayloadsDir = void 0}
-  __init35() {this._disabled = false}
-  __init36() {this.queueDropLoggingState = {
+  __init29() {this.maxRequestSize = 6 * 1024 * 1024}
+  __init30() {this.defaultBatchSize = 100}
+  __init31() {this.numTries = 3}
+  __init32() {this.queueDropExceedingMaxsize = DEFAULT_QUEUE_SIZE}
+  __init33() {this.queueDropLoggingPeriod = 60}
+  __init34() {this.failedPublishPayloadsDir = void 0}
+  __init35() {this.allPublishPayloadsDir = void 0}
+  __init36() {this._disabled = false}
+  __init37() {this.queueDropLoggingState = {
     numDropped: 0,
     lastLoggedTimestamp: 0
   }}
-  constructor(apiConn, opts) {;_class8.prototype.__init23.call(this);_class8.prototype.__init24.call(this);_class8.prototype.__init25.call(this);_class8.prototype.__init26.call(this);_class8.prototype.__init27.call(this);_class8.prototype.__init28.call(this);_class8.prototype.__init29.call(this);_class8.prototype.__init30.call(this);_class8.prototype.__init31.call(this);_class8.prototype.__init32.call(this);_class8.prototype.__init33.call(this);_class8.prototype.__init34.call(this);_class8.prototype.__init35.call(this);_class8.prototype.__init36.call(this);
+  constructor(apiConn, opts) {;_class8.prototype.__init24.call(this);_class8.prototype.__init25.call(this);_class8.prototype.__init26.call(this);_class8.prototype.__init27.call(this);_class8.prototype.__init28.call(this);_class8.prototype.__init29.call(this);_class8.prototype.__init30.call(this);_class8.prototype.__init31.call(this);_class8.prototype.__init32.call(this);_class8.prototype.__init33.call(this);_class8.prototype.__init34.call(this);_class8.prototype.__init35.call(this);_class8.prototype.__init36.call(this);_class8.prototype.__init37.call(this);
     opts = _nullishCoalesce(opts, () => ( {}));
     this.apiConn = apiConn;
     const syncFlushEnv = Number(isomorph_default.getEnv("BRAINTRUST_SYNC_FLUSH"));
@@ -4461,7 +4544,7 @@ var HTTPBackgroundLogger = (_class8 = class _HTTPBackgroundLogger {
       this.queue.clear();
       return;
     }
-    const batchSize = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _29 => _29.batchSize]), () => ( this.defaultBatchSize));
+    const batchSize = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _33 => _33.batchSize]), () => ( this.defaultBatchSize));
     const wrappedItems = this.queue.drain();
     const [allItems, attachments] = await this.unwrapLazyValues(wrappedItems);
     if (allItems.length === 0) {
@@ -4710,10 +4793,10 @@ Error: ${errorText}`;
         } catch (err) {
           if (err instanceof AggregateError) {
             for (const e of err.errors) {
-              _optionalChain([this, 'access', _30 => _30.onFlushError, 'optionalCall', _31 => _31(e)]);
+              _optionalChain([this, 'access', _34 => _34.onFlushError, 'optionalCall', _35 => _35(e)]);
             }
           } else {
-            _optionalChain([this, 'access', _32 => _32.onFlushError, 'optionalCall', _33 => _33(err)]);
+            _optionalChain([this, 'access', _36 => _36.onFlushError, 'optionalCall', _37 => _37(err)]);
           }
           this.activeFlushError = err;
         } finally {
@@ -5326,24 +5409,24 @@ async function summarize(options = {}) {
   return await e.summarize(options);
 }
 function currentExperiment(options) {
-  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _34 => _34.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _38 => _38.state]), () => ( _globalState));
   return state.currentExperiment;
 }
 function currentLogger(options) {
-  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _35 => _35.state]), () => ( _globalState));
-  return castLogger(state.currentLogger, _optionalChain([options, 'optionalAccess', _36 => _36.asyncFlush]));
+  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _39 => _39.state]), () => ( _globalState));
+  return castLogger(state.currentLogger, _optionalChain([options, 'optionalAccess', _40 => _40.asyncFlush]));
 }
 function currentSpan(options) {
-  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _37 => _37.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _41 => _41.state]), () => ( _globalState));
   return _nullishCoalesce(state.currentSpan.getStore(), () => ( NOOP_SPAN));
 }
 function getSpanParentObject(options) {
-  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _38 => _38.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _42 => _42.state]), () => ( _globalState));
   const parentSpan = currentSpan({ state });
   if (!Object.is(parentSpan, NOOP_SPAN)) {
     return parentSpan;
   }
-  const parentStr = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _39 => _39.parent]), () => ( state.currentParent.getStore()));
+  const parentStr = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _43 => _43.parent]), () => ( state.currentParent.getStore()));
   if (parentStr) return SpanComponentsV3.fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
@@ -5372,7 +5455,7 @@ function traced(callback, args) {
   const { span, isSyncFlushLogger } = startSpanAndIsLogger(args);
   const ret = runCatchFinally(
     () => {
-      if (_nullishCoalesce(_optionalChain([args, 'optionalAccess', _40 => _40.setCurrent]), () => ( true))) {
+      if (_nullishCoalesce(_optionalChain([args, 'optionalAccess', _44 => _44.setCurrent]), () => ( true))) {
         return withCurrent(span, callback);
       } else {
         return callback(span);
@@ -5384,7 +5467,7 @@ function traced(callback, args) {
     },
     () => span.end()
   );
-  if (_optionalChain([args, 'optionalAccess', _41 => _41.asyncFlush]) === void 0 || _optionalChain([args, 'optionalAccess', _42 => _42.asyncFlush])) {
+  if (_optionalChain([args, 'optionalAccess', _45 => _45.asyncFlush]) === void 0 || _optionalChain([args, 'optionalAccess', _46 => _46.asyncFlush])) {
     return ret;
   } else {
     return (async () => {
@@ -5508,14 +5591,14 @@ function wrapTraced(fn, args) {
   };
   const hasExplicitInput = args && args.event && "input" in args.event && args.event.input !== void 0;
   const hasExplicitOutput = args && args.event && args.event.output !== void 0;
-  const noTraceIO = _optionalChain([args, 'optionalAccess', _43 => _43.noTraceIO]) || hasExplicitInput || hasExplicitOutput;
+  const noTraceIO = _optionalChain([args, 'optionalAccess', _47 => _47.noTraceIO]) || hasExplicitInput || hasExplicitOutput;
   if (isGeneratorFunction(fn)) {
     return wrapTracedSyncGenerator(fn, spanArgs, !!noTraceIO);
   }
   if (isAsyncGeneratorFunction(fn)) {
     return wrapTracedAsyncGenerator(fn, spanArgs, !!noTraceIO);
   }
-  if (_optionalChain([args, 'optionalAccess', _44 => _44.asyncFlush])) {
+  if (_optionalChain([args, 'optionalAccess', _48 => _48.asyncFlush])) {
     return (...fnArgs) => traced((span) => {
       if (!hasExplicitInput) {
         span.log({ input: fnArgs });
@@ -5553,17 +5636,17 @@ function startSpan(args) {
   return startSpanAndIsLogger(args).span;
 }
 async function flush(options) {
-  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _45 => _45.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _49 => _49.state]), () => ( _globalState));
   return await state.bgLogger().flush();
 }
 function setFetch(fetch2) {
   _globalState.setFetch(fetch2);
 }
 function startSpanAndIsLogger(args) {
-  const state = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _46 => _46.state]), () => ( _globalState));
+  const state = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _50 => _50.state]), () => ( _globalState));
   const parentObject = getSpanParentObject({
-    asyncFlush: _optionalChain([args, 'optionalAccess', _47 => _47.asyncFlush]),
-    parent: _optionalChain([args, 'optionalAccess', _48 => _48.parent]),
+    asyncFlush: _optionalChain([args, 'optionalAccess', _51 => _51.asyncFlush]),
+    parent: _optionalChain([args, 'optionalAccess', _52 => _52.parent]),
     state
   });
   if (parentObject instanceof SpanComponentsV3) {
@@ -5580,14 +5663,14 @@ function startSpanAndIsLogger(args) {
       ),
       parentComputeObjectMetadataArgs: _nullishCoalesce(parentObject.data.compute_object_metadata_args, () => ( void 0)),
       parentSpanIds,
-      propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _49 => _49.propagatedEvent]), () => ( // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+      propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _53 => _53.propagatedEvent]), () => ( // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
       (_nullishCoalesce(parentObject.data.propagated_event, () => ( void 0)))))
     });
     return {
       span,
       isSyncFlushLogger: parentObject.data.object_type === 2 /* PROJECT_LOGS */ && // Since there's no parent logger here, we're free to choose the async flush
       // behavior, and therefore propagate along whatever we get from the arguments
-      _optionalChain([args, 'optionalAccess', _50 => _50.asyncFlush]) === false
+      _optionalChain([args, 'optionalAccess', _54 => _54.asyncFlush]) === false
     };
   } else {
     const span = parentObject.startSpan(args);
@@ -5776,10 +5859,10 @@ function extractAttachments(event, attachments) {
       event[key] = value.reference;
       continue;
     }
-    if (_optionalChain([value, 'optionalAccess', _51 => _51.type]) === BRAINTRUST_ATTACHMENT && value.key && !value.uploader) {
+    if (_optionalChain([value, 'optionalAccess', _55 => _55.type]) === BRAINTRUST_ATTACHMENT && value.key && !value.uploader) {
       continue;
     }
-    if (_optionalChain([value, 'optionalAccess', _52 => _52.reference, 'optionalAccess', _53 => _53.type]) === BRAINTRUST_ATTACHMENT && _optionalChain([value, 'optionalAccess', _54 => _54.uploader])) {
+    if (_optionalChain([value, 'optionalAccess', _56 => _56.reference, 'optionalAccess', _57 => _57.type]) === BRAINTRUST_ATTACHMENT && _optionalChain([value, 'optionalAccess', _58 => _58.uploader])) {
       const attachment = new Attachment({
         data: value.dataDebugString,
         filename: value.reference.filename,
@@ -5846,13 +5929,13 @@ function validateAndSanitizeExperimentLogFullArgs(event, hasDataset) {
 var INTERNAL_BTQL_LIMIT = 1e3;
 var MAX_BTQL_ITERATIONS = 1e4;
 var ObjectFetcher = (_class9 = class {
-  constructor(objectType, pinnedVersion, mutateRecord, _internal_btql) {;_class9.prototype.__init37.call(this);
+  constructor(objectType, pinnedVersion, mutateRecord, _internal_btql) {;_class9.prototype.__init38.call(this);
     this.objectType = objectType;
     this.pinnedVersion = pinnedVersion;
     this.mutateRecord = mutateRecord;
     this._internal_btql = _internal_btql;
   }
-  __init37() {this._fetchedData = void 0}
+  __init38() {this._fetchedData = void 0}
   get id() {
     throw new Error("ObjectFetcher subclasses must have an 'id' attribute");
   }
@@ -5917,7 +6000,7 @@ var ObjectFetcher = (_class9 = class {
           throw new Error("Too many BTQL iterations");
         }
       }
-      this._fetchedData = this.mutateRecord ? _optionalChain([data, 'optionalAccess', _55 => _55.map, 'call', _56 => _56(this.mutateRecord)]) : data;
+      this._fetchedData = this.mutateRecord ? _optionalChain([data, 'optionalAccess', _59 => _59.map, 'call', _60 => _60(this.mutateRecord)]) : data;
     }
     return this._fetchedData || [];
   }
@@ -5948,9 +6031,9 @@ var Experiment2 = (_class10 = class extends ObjectFetcher {
   
   
   // For type identification.
-  __init38() {this.kind = "experiment"}
+  __init39() {this.kind = "experiment"}
   constructor(state, lazyMetadata, dataset) {
-    super("experiment", void 0, (r) => enrichAttachments(r, state));_class10.prototype.__init38.call(this);;
+    super("experiment", void 0, (r) => enrichAttachments(r, state));_class10.prototype.__init39.call(this);;
     this.lazyMetadata = lazyMetadata;
     this.dataset = dataset;
     this.lastStartTime = getCurrentUnixTimestamp();
@@ -5998,7 +6081,7 @@ var Experiment2 = (_class10 = class extends ObjectFetcher {
    * @returns The `id` of the logged event.
    */
   log(event, options) {
-    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _57 => _57.allowConcurrentWithSpans])) {
+    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _61 => _61.allowConcurrentWithSpans])) {
       throw new Error(
         "Cannot run toplevel `log` method while using spans. To log to the span, call `experiment.traced` and then log with `span.log`"
       );
@@ -6051,12 +6134,12 @@ var Experiment2 = (_class10 = class extends ObjectFetcher {
       state: this.state,
       ...startSpanParentArgs({
         state: this.state,
-        parent: _optionalChain([args, 'optionalAccess', _58 => _58.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _62 => _62.parent]),
         parentObjectType: this.parentObjectType(),
         parentObjectId: this.lazyId,
         parentComputeObjectMetadataArgs: void 0,
         parentSpanIds: void 0,
-        propagatedEvent: _optionalChain([args, 'optionalAccess', _59 => _59.propagatedEvent])
+        propagatedEvent: _optionalChain([args, 'optionalAccess', _63 => _63.propagatedEvent])
       }),
       defaultRootType: "eval" /* EVAL */
     });
@@ -6257,8 +6340,8 @@ var SpanImpl = (_class11 = class _SpanImpl {
   
   
   
-  __init39() {this.kind = "span"}
-  constructor(args) {;_class11.prototype.__init39.call(this);
+  __init40() {this.kind = "span"}
+  constructor(args) {;_class11.prototype.__init40.call(this);
     this._state = args.state;
     const spanAttributes = _nullishCoalesce(args.spanAttributes, () => ( {}));
     const rawEvent = _nullishCoalesce(args.event, () => ( {}));
@@ -6298,13 +6381,17 @@ var SpanImpl = (_class11 = class _SpanImpl {
       },
       created: (/* @__PURE__ */ new Date()).toISOString()
     };
-    this._id = _nullishCoalesce(eventId, () => ( _uuid.v4.call(void 0, )));
-    this._spanId = _nullishCoalesce(args.spanId, () => ( _uuid.v4.call(void 0, )));
+    this._id = _nullishCoalesce(eventId, () => ( this._state.idGenerator.getSpanId()));
+    this._spanId = _nullishCoalesce(args.spanId, () => ( this._state.idGenerator.getSpanId()));
     if (args.parentSpanIds) {
       this._rootSpanId = args.parentSpanIds.rootSpanId;
       this._spanParents = "parentSpanIds" in args.parentSpanIds ? args.parentSpanIds.parentSpanIds : [args.parentSpanIds.spanId];
     } else {
-      this._rootSpanId = this._spanId;
+      if (this._state.idGenerator.shareRootSpanId()) {
+        this._rootSpanId = this._spanId;
+      } else {
+        this._rootSpanId = this._state.idGenerator.getTraceId();
+      }
       this._spanParents = void 0;
     }
     this.isMerge = false;
@@ -6348,10 +6435,10 @@ var SpanImpl = (_class11 = class _SpanImpl {
       ...serializableInternalData,
       [IS_MERGE_FIELD]: this.isMerge
     });
-    if (_optionalChain([partialRecord, 'access', _60 => _60.metrics, 'optionalAccess', _61 => _61.end])) {
-      this.loggedEndTime = _optionalChain([partialRecord, 'access', _62 => _62.metrics, 'optionalAccess', _63 => _63.end]);
+    if (_optionalChain([partialRecord, 'access', _64 => _64.metrics, 'optionalAccess', _65 => _65.end])) {
+      this.loggedEndTime = _optionalChain([partialRecord, 'access', _66 => _66.metrics, 'optionalAccess', _67 => _67.end]);
     }
-    if ((_nullishCoalesce(partialRecord.tags, () => ( []))).length > 0 && _optionalChain([this, 'access', _64 => _64._spanParents, 'optionalAccess', _65 => _65.length])) {
+    if ((_nullishCoalesce(partialRecord.tags, () => ( []))).length > 0 && _optionalChain([this, 'access', _68 => _68._spanParents, 'optionalAccess', _69 => _69.length])) {
       throw new Error("Tags can only be logged to the root span");
     }
     const computeRecord = async () => ({
@@ -6396,18 +6483,18 @@ var SpanImpl = (_class11 = class _SpanImpl {
     );
   }
   startSpan(args) {
-    const parentSpanIds = _optionalChain([args, 'optionalAccess', _66 => _66.parent]) ? void 0 : { spanId: this._spanId, rootSpanId: this._rootSpanId };
+    const parentSpanIds = _optionalChain([args, 'optionalAccess', _70 => _70.parent]) ? void 0 : { spanId: this._spanId, rootSpanId: this._rootSpanId };
     return new _SpanImpl({
       state: this._state,
       ...args,
       ...startSpanParentArgs({
         state: this._state,
-        parent: _optionalChain([args, 'optionalAccess', _67 => _67.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _71 => _71.parent]),
         parentObjectType: this.parentObjectType,
         parentObjectId: this.parentObjectId,
         parentComputeObjectMetadataArgs: this.parentComputeObjectMetadataArgs,
         parentSpanIds,
-        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _68 => _68.propagatedEvent]), () => ( this.propagatedEvent))
+        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _72 => _72.propagatedEvent]), () => ( this.propagatedEvent))
       })
     });
   }
@@ -6421,12 +6508,12 @@ var SpanImpl = (_class11 = class _SpanImpl {
       ...args,
       ...startSpanParentArgs({
         state: this._state,
-        parent: _optionalChain([args, 'optionalAccess', _69 => _69.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _73 => _73.parent]),
         parentObjectType: this.parentObjectType,
         parentObjectId: this.parentObjectId,
         parentComputeObjectMetadataArgs: this.parentComputeObjectMetadataArgs,
         parentSpanIds,
-        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _70 => _70.propagatedEvent]), () => ( this.propagatedEvent))
+        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _74 => _74.propagatedEvent]), () => ( this.propagatedEvent))
       }),
       spanId
     });
@@ -6435,7 +6522,7 @@ var SpanImpl = (_class11 = class _SpanImpl {
     let endTime;
     let internalData = {};
     if (!this.loggedEndTime) {
-      endTime = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _71 => _71.endTime]), () => ( getCurrentUnixTimestamp()));
+      endTime = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _75 => _75.endTime]), () => ( getCurrentUnixTimestamp()));
       internalData = { metrics: { end: endTime } };
     } else {
       endTime = this.loggedEndTime;
@@ -6478,8 +6565,8 @@ var SpanImpl = (_class11 = class _SpanImpl {
     const args = this.parentComputeObjectMetadataArgs;
     switch (this.parentObjectType) {
       case 2 /* PROJECT_LOGS */: {
-        const projectID = _optionalChain([args, 'optionalAccess', _72 => _72.project_id]) || this.parentObjectId.getSync().value;
-        const projectName = _optionalChain([args, 'optionalAccess', _73 => _73.project_name]);
+        const projectID = _optionalChain([args, 'optionalAccess', _76 => _76.project_id]) || this.parentObjectId.getSync().value;
+        const projectName = _optionalChain([args, 'optionalAccess', _77 => _77.project_name]);
         if (projectID) {
           return `${baseUrl}/object?object_type=project_logs&object_id=${projectID}&id=${this._id}`;
         } else if (projectName) {
@@ -6489,7 +6576,7 @@ var SpanImpl = (_class11 = class _SpanImpl {
         }
       }
       case 1 /* EXPERIMENT */: {
-        const expID = _optionalChain([args, 'optionalAccess', _74 => _74.experiment_id]) || _optionalChain([this, 'access', _75 => _75.parentObjectId, 'optionalAccess', _76 => _76.getSync, 'call', _77 => _77(), 'optionalAccess', _78 => _78.value]);
+        const expID = _optionalChain([args, 'optionalAccess', _78 => _78.experiment_id]) || _optionalChain([this, 'access', _79 => _79.parentObjectId, 'optionalAccess', _80 => _80.getSync, 'call', _81 => _81(), 'optionalAccess', _82 => _82.value]);
         if (!expID) {
           return getErrPermlink("provide-experiment-id");
         } else {
@@ -6579,13 +6666,13 @@ var Dataset2 = (_class12 = class extends ObjectFetcher {
         )
       ),
       _internal_btql
-    );_class12.prototype.__init40.call(this);_class12.prototype.__init41.call(this);;
+    );_class12.prototype.__init41.call(this);_class12.prototype.__init42.call(this);;
     this.state = state;
     this.lazyMetadata = lazyMetadata;
   }
   
-  __init40() {this.__braintrust_dataset_marker = true}
-  __init41() {this.newRecords = 0}
+  __init41() {this.__braintrust_dataset_marker = true}
+  __init42() {this.newRecords = 0}
   get id() {
     return (async () => {
       return (await this.lazyMetadata.get()).dataset.id;
@@ -6908,14 +6995,14 @@ function renderPromptParams(params, args, options) {
   return params;
 }
 var Prompt2 = (_class13 = class _Prompt {
-  constructor(metadata, defaults, noTrace) {;_class13.prototype.__init42.call(this);_class13.prototype.__init43.call(this);
+  constructor(metadata, defaults, noTrace) {;_class13.prototype.__init43.call(this);_class13.prototype.__init44.call(this);
     this.metadata = metadata;
     this.defaults = defaults;
     this.noTrace = noTrace;
   }
   
-  __init42() {this.hasParsedPromptData = false}
-  __init43() {this.__braintrust_prompt_marker = true}
+  __init43() {this.hasParsedPromptData = false}
+  __init44() {this.__braintrust_prompt_marker = true}
   get id() {
     return this.metadata.id;
   }
@@ -6929,13 +7016,13 @@ var Prompt2 = (_class13 = class _Prompt {
     return "slug" in this.metadata ? this.metadata.slug : this.metadata.id;
   }
   get prompt() {
-    return _optionalChain([this, 'access', _79 => _79.getParsedPromptData, 'call', _80 => _80(), 'optionalAccess', _81 => _81.prompt]);
+    return _optionalChain([this, 'access', _83 => _83.getParsedPromptData, 'call', _84 => _84(), 'optionalAccess', _85 => _85.prompt]);
   }
   get version() {
     return this.metadata[TRANSACTION_ID_FIELD];
   }
   get options() {
-    return _optionalChain([this, 'access', _82 => _82.getParsedPromptData, 'call', _83 => _83(), 'optionalAccess', _84 => _84.options]) || {};
+    return _optionalChain([this, 'access', _86 => _86.getParsedPromptData, 'call', _87 => _87(), 'optionalAccess', _88 => _88.options]) || {};
   }
   get promptData() {
     return this.getParsedPromptData();
@@ -7086,7 +7173,7 @@ var Prompt2 = (_class13 = class _Prompt {
       return {
         type: "chat",
         messages,
-        ..._optionalChain([prompt, 'access', _85 => _85.tools, 'optionalAccess', _86 => _86.trim, 'call', _87 => _87()]) ? {
+        ..._optionalChain([prompt, 'access', _89 => _89.tools, 'optionalAccess', _90 => _90.trim, 'call', _91 => _91()]) ? {
           tools: render(prompt.tools)
         } : void 0
       };
@@ -7196,9 +7283,15 @@ async function getPromptVersions(projectId, promptId) {
     );
   }
   const result = await response.json();
-  return _optionalChain([result, 'access', _88 => _88.data, 'optionalAccess', _89 => _89.filter, 'call', _90 => _90(
-    (entry) => ["upsert", "merge"].includes(_optionalChain([entry, 'access', _91 => _91.audit_data, 'optionalAccess', _92 => _92.action]))
-  ), 'access', _93 => _93.map, 'call', _94 => _94((entry) => prettifyXact(entry._xact_id))]) || [];
+  return _optionalChain([result, 'access', _92 => _92.data, 'optionalAccess', _93 => _93.filter, 'call', _94 => _94(
+    (entry) => ["upsert", "merge"].includes(_optionalChain([entry, 'access', _95 => _95.audit_data, 'optionalAccess', _96 => _96.action]))
+  ), 'access', _97 => _97.map, 'call', _98 => _98((entry) => prettifyXact(entry._xact_id))]) || [];
+}
+function resetIdGenStateForTests() {
+  const state = _internalGetGlobalState();
+  if (state) {
+    state.resetIdGenState();
+  }
 }
 var _exportsForTestingOnly = {
   extractAttachments,
@@ -7210,7 +7303,8 @@ var _exportsForTestingOnly = {
   setInitialTestState,
   initTestExperiment,
   isGeneratorFunction,
-  isAsyncGeneratorFunction
+  isAsyncGeneratorFunction,
+  resetIdGenStateForTests
 };
 
 // src/browser-config.ts
@@ -7457,7 +7551,7 @@ function parseSpanFromResponseCreateParams(params) {
 }
 function parseEventFromResponseCreateResult(result) {
   const data = {};
-  if (_optionalChain([result, 'optionalAccess', _95 => _95.output]) !== void 0) {
+  if (_optionalChain([result, 'optionalAccess', _99 => _99.output]) !== void 0) {
     data.output = processImagesInOutput(result.output);
   }
   if (result) {
@@ -7466,7 +7560,7 @@ function parseEventFromResponseCreateResult(result) {
       data.metadata = metadata;
     }
   }
-  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _96 => _96.usage]));
+  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _100 => _100.usage]));
   return data;
 }
 function processImagesInOutput(output) {
@@ -7520,7 +7614,7 @@ function parseSpanFromResponseParseParams(params) {
 }
 function parseEventFromResponseParseResult(result) {
   const data = {};
-  if (_optionalChain([result, 'optionalAccess', _97 => _97.output]) !== void 0) {
+  if (_optionalChain([result, 'optionalAccess', _101 => _101.output]) !== void 0) {
     data.output = processImagesInOutput(result.output);
   }
   if (result) {
@@ -7529,7 +7623,7 @@ function parseEventFromResponseParseResult(result) {
       data.metadata = metadata;
     }
   }
-  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _98 => _98.usage]));
+  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _102 => _102.usage]));
   return data;
 }
 function traceResponseCreateStream(stream, timedSpan) {
@@ -7546,7 +7640,7 @@ function traceResponseCreateStream(stream, timedSpan) {
       return result;
     }
     const item = result.value;
-    if (!item || !_optionalChain([item, 'optionalAccess', _99 => _99.type]) || !_optionalChain([item, 'optionalAccess', _100 => _100.response])) {
+    if (!item || !_optionalChain([item, 'optionalAccess', _103 => _103.type]) || !_optionalChain([item, 'optionalAccess', _104 => _104.response])) {
       return result;
     }
     const event = parseLogFromItem(item);
@@ -7557,14 +7651,14 @@ function traceResponseCreateStream(stream, timedSpan) {
   };
 }
 function parseLogFromItem(item) {
-  if (!item || !_optionalChain([item, 'optionalAccess', _101 => _101.type]) || !_optionalChain([item, 'optionalAccess', _102 => _102.response])) {
+  if (!item || !_optionalChain([item, 'optionalAccess', _105 => _105.type]) || !_optionalChain([item, 'optionalAccess', _106 => _106.response])) {
     return {};
   }
   const response = item.response;
   switch (item.type) {
     case "response.completed":
       const data = {};
-      if (_optionalChain([response, 'optionalAccess', _103 => _103.output]) !== void 0) {
+      if (_optionalChain([response, 'optionalAccess', _107 => _107.output]) !== void 0) {
         data.output = processImagesInOutput(response.output);
       }
       if (response) {
@@ -7573,7 +7667,7 @@ function parseLogFromItem(item) {
           data.metadata = metadata;
         }
       }
-      data.metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _104 => _104.usage]));
+      data.metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _108 => _108.usage]));
       return data;
     default:
       return {};
@@ -7757,8 +7851,8 @@ function wrapOpenAIv4(openai) {
   const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
   const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy;
-  if (_optionalChain([openai, 'access', _105 => _105.beta, 'optionalAccess', _106 => _106.chat, 'optionalAccess', _107 => _107.completions, 'optionalAccess', _108 => _108.stream])) {
-    const betaChatCompletionProxy = new Proxy(_optionalChain([openai, 'optionalAccess', _109 => _109.beta, 'optionalAccess', _110 => _110.chat, 'access', _111 => _111.completions]), {
+  if (_optionalChain([openai, 'access', _109 => _109.beta, 'optionalAccess', _110 => _110.chat, 'optionalAccess', _111 => _111.completions, 'optionalAccess', _112 => _112.stream])) {
+    const betaChatCompletionProxy = new Proxy(_optionalChain([openai, 'optionalAccess', _113 => _113.beta, 'optionalAccess', _114 => _114.chat, 'access', _115 => _115.completions]), {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -7806,7 +7900,7 @@ function wrapOpenAIv4(openai) {
   });
 }
 function logCompletionResponse(startTime, response, span) {
-  const metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _112 => _112.usage]));
+  const metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _116 => _116.usage]));
   metrics.time_to_first_token = getCurrentUnixTimestamp() - startTime;
   span.log({
     output: response.choices,
@@ -8036,7 +8130,7 @@ function parseChatCompletionParams(params) {
 function processEmbeddingResponse(result, span) {
   span.log({
     output: { embedding_length: result.data[0].embedding.length },
-    metrics: parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _113 => _113.usage]))
+    metrics: parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _117 => _117.usage]))
   });
 }
 function processModerationResponse(result, span) {
@@ -8066,10 +8160,10 @@ function postprocessStreamingResults(allResults) {
     if (result.usage) {
       metrics = {
         ...metrics,
-        ...parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _114 => _114.usage]))
+        ...parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _118 => _118.usage]))
       };
     }
-    const delta = _optionalChain([result, 'access', _115 => _115.choices, 'optionalAccess', _116 => _116[0], 'optionalAccess', _117 => _117.delta]);
+    const delta = _optionalChain([result, 'access', _119 => _119.choices, 'optionalAccess', _120 => _120[0], 'optionalAccess', _121 => _121.delta]);
     if (!delta) {
       continue;
     }