bitfab 0.54.10 → 0.54.11

package/dist/{chunk-PRXW4B5X.js → chunk-G4LWALQ7.js}

@@ -6,7 +6,7 @@ import {
   replayContextReady,
   runWithReplayContext,
   warnOnce
-} from "./chunk-B2O33DBH.js";
+} from "./chunk-FMUJONTQ.js";
 import {
   runWithCaptureSuppressed
 } from "./chunk-VCRFFCLY.js";
@@ -1257,4 +1257,4 @@ export {
   sleepForReplayPersistence,
   replay
 };
-//# sourceMappingURL=chunk-PRXW4B5X.js.map
+//# sourceMappingURL=chunk-G4LWALQ7.js.map

package/dist/{chunk-A3OBQOLV.js → chunk-MEJOPEKV.js}

@@ -19,7 +19,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-PRXW4B5X.js";
+} from "./chunk-G4LWALQ7.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -36,7 +36,7 @@ import {
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-B2O33DBH.js";
+} from "./chunk-FMUJONTQ.js";
 import {
   __privateAdd,
   __privateGet,
@@ -2441,6 +2441,83 @@ var BitfabOpenAIAgentHandler = class {
   }
 };
 
+// src/openaiAgentsSpanContext.ts
+var OPENAI_AGENTS_CONTEXT_STORAGE = /* @__PURE__ */ Symbol.for(
+  "openai.agents.core.asyncLocalStorage"
+);
+var MAX_OPENAI_AGENTS_SPAN_CHAIN = 64;
+var spanGlobal = globalThis;
+var state = spanGlobal.__bitfabOpenAiAgentsSpansV1 ?? {
+  bitfabTraceIds: /* @__PURE__ */ new Map(),
+  endedSpanIds: /* @__PURE__ */ new Map()
+};
+spanGlobal.__bitfabOpenAiAgentsSpansV1 = state;
+function mapOpenAiAgentsTrace(agentsTraceId, bitfabTraceId) {
+  state.bitfabTraceIds.set(agentsTraceId, bitfabTraceId);
+}
+function unmapOpenAiAgentsTrace(agentsTraceId) {
+  state.bitfabTraceIds.delete(agentsTraceId);
+  state.endedSpanIds.delete(agentsTraceId);
+}
+function endOpenAiAgentsSpan(agentsTraceId, spanId) {
+  if (!state.bitfabTraceIds.has(agentsTraceId)) {
+    return;
+  }
+  const ended = state.endedSpanIds.get(agentsTraceId) ?? /* @__PURE__ */ new Set();
+  ended.add(spanId);
+  state.endedSpanIds.set(agentsTraceId, ended);
+}
+function isOpenAiAgentsContextStorage(value) {
+  return value !== null && typeof value === "object" && typeof value.getStore === "function";
+}
+function currentOpenAiAgentsSpan() {
+  const storage2 = spanGlobal[OPENAI_AGENTS_CONTEXT_STORAGE];
+  if (!isOpenAiAgentsContextStorage(storage2)) {
+    return void 0;
+  }
+  try {
+    const store = storage2.getStore();
+    return store?.active === true ? store.span ?? void 0 : void 0;
+  } catch {
+    return void 0;
+  }
+}
+function currentOpenAiAgentsSpans() {
+  const spans = [];
+  let span = currentOpenAiAgentsSpan();
+  while (span !== void 0 && spans.length < MAX_OPENAI_AGENTS_SPAN_CHAIN) {
+    spans.push(span);
+    span = span.previousSpan ?? void 0;
+  }
+  return spans;
+}
+function openAiAgentsSpanChain() {
+  if (state.bitfabTraceIds.size === 0) {
+    return [];
+  }
+  return currentOpenAiAgentsSpans().map((span) => span.spanId).filter((spanId) => typeof spanId === "string");
+}
+function openAiAgentsParentSpanId(bitfabTraceId, enclosingSpanIds) {
+  if (state.bitfabTraceIds.size === 0) {
+    return void 0;
+  }
+  for (const span of currentOpenAiAgentsSpans()) {
+    if (typeof span.spanId !== "string" || typeof span.traceId !== "string") {
+      return void 0;
+    }
+    if (state.bitfabTraceIds.get(span.traceId) !== bitfabTraceId) {
+      return void 0;
+    }
+    if (enclosingSpanIds?.includes(span.spanId)) {
+      return void 0;
+    }
+    if (!state.endedSpanIds.get(span.traceId)?.has(span.spanId)) {
+      return span.spanId;
+    }
+  }
+  return void 0;
+}
+
 // src/replayBranch.ts
 var _url, _context;
 var ReplayBranch = class {
@@ -3137,6 +3214,9 @@ var BitfabOpenAITracingProcessor = class {
     const activeContext = this.getActiveSpanContext?.();
     if (activeContext) {
       this.activeSpanMappings[trace.traceId] = activeContext;
+      if (this.shouldRecord()) {
+        mapOpenAiAgentsTrace(trace.traceId, activeContext.traceId);
+      }
     }
     const canonicalTraceId = activeContext?.traceId ?? this.getCanonicalTraceId(trace.traceId);
     this.canonicalTraceIds[trace.traceId] = canonicalTraceId;
@@ -3174,6 +3254,7 @@ var BitfabOpenAITracingProcessor = class {
     }
     this.endedTraces.delete(traceId);
     delete this.activeSpanMappings[traceId];
+    unmapOpenAiAgentsTrace(traceId);
     delete this.canonicalTraceIds[traceId];
     delete this.activeTraces[traceId];
   }
@@ -3198,6 +3279,9 @@ var BitfabOpenAITracingProcessor = class {
   async onSpanEnd(span) {
     const originalTraceId = span.traceId ?? "";
     const traceId = this.activeSpanMappings[originalTraceId]?.traceId ?? originalTraceId;
+    if (span.spanId !== void 0) {
+      endOpenAiAgentsSpan(originalTraceId, span.spanId);
+    }
     try {
       this.sendSpan(span);
     } finally {
@@ -3218,6 +3302,9 @@ var BitfabOpenAITracingProcessor = class {
    * Called when the trace processor is shutting down.
    */
   async shutdown(timeout) {
+    for (const traceId of Object.keys(this.activeSpanMappings)) {
+      unmapOpenAiAgentsTrace(traceId);
+    }
     this.activeTraces = {};
     this.activeSpanMappings = {};
     this.canonicalTraceIds = {};
@@ -4374,16 +4461,16 @@ var Bitfab = class {
     this.simulationPlan.send(payload, rootTraceFunctionKeyOf(payload), submit);
   }
   refreshAutoTraceCapturePolicy(traceFunctionKey) {
-    const state = this.autoTracePolicyRefreshes.get(traceFunctionKey) ?? new PolicyRefresh();
-    this.autoTracePolicyRefreshes.set(traceFunctionKey, state);
-    state.run(
+    const state2 = this.autoTracePolicyRefreshes.get(traceFunctionKey) ?? new PolicyRefresh();
+    this.autoTracePolicyRefreshes.set(traceFunctionKey, state2);
+    state2.run(
       () => this.httpClient.getAutoTracePolicy(
         traceFunctionKey,
         AUTO_TRACE_PROTOCOL
       ),
       (policy) => {
         if (policy.protocol !== AUTO_TRACE_PROTOCOL) {
-          state.hold(AUTO_TRACE_POLICY_RETRY_MS);
+          state2.hold(AUTO_TRACE_POLICY_RETRY_MS);
           return;
         }
         const functionIds = Array.isArray(policy.functionIds) ? policy.functionIds.filter(
@@ -4394,10 +4481,10 @@ var Bitfab = class {
           traceFunctionKey,
           policy.revision === null ? void 0 : functionIds
         );
-        state.hold(AUTO_TRACE_POLICY_REFRESH_MS);
+        state2.hold(AUTO_TRACE_POLICY_REFRESH_MS);
       },
       () => {
-        state.hold(AUTO_TRACE_POLICY_RETRY_MS);
+        state2.hold(AUTO_TRACE_POLICY_RETRY_MS);
       }
     );
   }
@@ -5008,7 +5095,11 @@ var Bitfab = class {
         const seedCtxForTraceId = parentContext ? null : getSeedContext();
         const traceId = parentContext?.traceId ?? replayCtxForTraceId?.traceId ?? seedCtxForTraceId?.traceId ?? randomUuid();
         const spanId = randomUuid();
-        const parentSpanId = parentContext?.spanId ?? null;
+        const openAiAgentsParent = parentContext !== void 0 && options.instrumentation === "trace" ? openAiAgentsParentSpanId(
+          parentContext.traceId,
+          parentContext.openAiAgentsSpanIds
+        ) : void 0;
+        const parentSpanId = openAiAgentsParent ?? parentContext?.spanId ?? null;
         const isRootSpan = parentSpanId === null;
         if (options.enclosingTrace !== void 0) {
           options.enclosingTrace.link.traceId = traceId;
@@ -5022,11 +5113,13 @@ var Bitfab = class {
           parentContext?.surface,
           traceFunctionKey
         );
+        const openAiAgentsSpanIds = openAiAgentsSpanChain();
         const newContext = {
           traceId,
           spanId,
           contexts: [],
-          ...surface !== void 0 && { surface }
+          ...surface !== void 0 && { surface },
+          ...openAiAgentsSpanIds.length > 0 && { openAiAgentsSpanIds }
         };
         newStack = [...currentStack, newContext];
         const inputs = args;
@@ -5801,7 +5894,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-AVWWAMFV.js");
+        const { flushTraces: flushTraces2 } = await import("./http-AXPT27DO.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -5860,7 +5953,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-BCWIDCKH.js");
+    const { replay: doReplay } = await import("./replay-DOG4KV7I.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -6180,7 +6273,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-AVWWAMFV.js");
+  const { flushTraces: flushTraces2 } = await import("./http-AXPT27DO.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -6219,4 +6312,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-A3OBQOLV.js.map
+//# sourceMappingURL=chunk-MEJOPEKV.js.map