@bitfab/sdk 0.54.10 → 0.54.12

package/dist/index.d.cts

@@ -2214,18 +2214,18 @@ declare const RIVET_GROUP_EFFECT_METHODS: {
     readonly queue: readonly ["send", "next", "nextBatch", "waitForNames", "waitForAvailable", "complete", "enqueueAndWait", "tryNext", "tryNextBatch"];
     readonly run: readonly ["setWakeAt"];
 };
-declare const RIVET_TOP_LEVEL_EFFECTS: readonly ["broadcast", "saveState", "sleep", "destroy"];
+declare const RIVET_TOP_LEVEL_EFFECTS: readonly ["broadcast", "saveState", "sleep", "destroy", "step", "tryStep", "try", "loop", "sleepUntil", "rollbackCheckpoint", "join", "race", "removed", "getVersion"];
 type RivetGroup = keyof typeof RIVET_GROUP_EFFECT_METHODS;
 type RivetGroupEffectType = {
     [TGroup in RivetGroup]: `${TGroup}.${(typeof RIVET_GROUP_EFFECT_METHODS)[TGroup][number]}`;
 }[RivetGroup];
-/** Finite Rivet context operations supported as replay effects. */
+/** Rivet context operations supported as replay effects. */
 type RivetEffectType = RivetGroupEffectType | (typeof RIVET_TOP_LEVEL_EFFECTS)[number] | "db.transaction.execute" | "queue.message.complete";
 /** Configuration for {@link BitfabRivetIntegration}. */
 interface RivetIntegrationOptions {
     /**
      * Context operations whose recorded outputs replay should serve instead of
-     * executing against a live actor. Nothing is marked by default.
+     * executing against a live actor or workflow. Nothing is marked by default.
      */
     mockOnReplay?: readonly RivetEffectType[];
 }
@@ -2246,8 +2246,8 @@ interface RivetIntegrationRuntime {
     recordRootFixture(context: Record<string, unknown>, recordedRoots: Set<string>): void;
 }
 /**
- * Binds a live Rivet actor context to an existing traced function and exposes
- * replay-aware proxies for supported context operations.
+ * Binds a live Rivet actor or workflow context to an existing traced function
+ * and exposes replay-aware proxies for supported context operations.
  *
  * The integration never creates a trace root. Use it with the same
  * `BitfabFunction` handle whose `withTrace()` wrapper is the production and
@@ -2267,13 +2267,15 @@ declare class BitfabRivetIntegration<TContext extends object = RivetContextLike>
     withContext<TReturn>(context: TContext, fn: () => TReturn): TReturn;
     /**
      * Adapt a traced function into a Rivet handler whose first argument is the
-     * live actor context. The context is not added to the traced root's inputs.
+     * live actor or workflow context. The context is not added to the traced
+     * root's inputs.
      */
     bind<This, TArgs extends unknown[], TReturn>(fn: (this: This, ...args: TArgs) => TReturn): (this: This, context: TContext, ...args: TArgs) => TReturn;
     /**
      * Return the current Rivet context proxy. In production it delegates to the
      * bound live context. During replay it hydrates recorded actor state and
-     * intercepts supported effects without requiring a live Rivet actor.
+     * intercepts supported effects without requiring a live Rivet actor or
+     * workflow.
      */
     getContext(): TContext;
     private createContextProxy;
@@ -3123,9 +3125,9 @@ declare class Bitfab {
      */
     getLangGraphIntegration(traceFunctionKey: string, options?: LangGraphIntegrationOptions): BitfabLangGraphIntegration;
     /**
-     * Get the first-class Rivet actor-context integration for an existing trace
-     * root. The integration binds context and records effect spans; it never
-     * creates another root.
+     * Get the first-class Rivet actor/workflow-context integration for an
+     * existing trace root. The integration binds context and records effect
+     * spans; it never creates another root.
      *
      * @param traceFunctionKey - Key used by the enclosing `withTrace()` root
      * @param options - Explicit per-context-path replay mocking policy
@@ -3602,7 +3604,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.54.10";
+declare const __version__ = "0.54.12";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -2214,18 +2214,18 @@ declare const RIVET_GROUP_EFFECT_METHODS: {
     readonly queue: readonly ["send", "next", "nextBatch", "waitForNames", "waitForAvailable", "complete", "enqueueAndWait", "tryNext", "tryNextBatch"];
     readonly run: readonly ["setWakeAt"];
 };
-declare const RIVET_TOP_LEVEL_EFFECTS: readonly ["broadcast", "saveState", "sleep", "destroy"];
+declare const RIVET_TOP_LEVEL_EFFECTS: readonly ["broadcast", "saveState", "sleep", "destroy", "step", "tryStep", "try", "loop", "sleepUntil", "rollbackCheckpoint", "join", "race", "removed", "getVersion"];
 type RivetGroup = keyof typeof RIVET_GROUP_EFFECT_METHODS;
 type RivetGroupEffectType = {
     [TGroup in RivetGroup]: `${TGroup}.${(typeof RIVET_GROUP_EFFECT_METHODS)[TGroup][number]}`;
 }[RivetGroup];
-/** Finite Rivet context operations supported as replay effects. */
+/** Rivet context operations supported as replay effects. */
 type RivetEffectType = RivetGroupEffectType | (typeof RIVET_TOP_LEVEL_EFFECTS)[number] | "db.transaction.execute" | "queue.message.complete";
 /** Configuration for {@link BitfabRivetIntegration}. */
 interface RivetIntegrationOptions {
     /**
      * Context operations whose recorded outputs replay should serve instead of
-     * executing against a live actor. Nothing is marked by default.
+     * executing against a live actor or workflow. Nothing is marked by default.
      */
     mockOnReplay?: readonly RivetEffectType[];
 }
@@ -2246,8 +2246,8 @@ interface RivetIntegrationRuntime {
     recordRootFixture(context: Record<string, unknown>, recordedRoots: Set<string>): void;
 }
 /**
- * Binds a live Rivet actor context to an existing traced function and exposes
- * replay-aware proxies for supported context operations.
+ * Binds a live Rivet actor or workflow context to an existing traced function
+ * and exposes replay-aware proxies for supported context operations.
  *
  * The integration never creates a trace root. Use it with the same
  * `BitfabFunction` handle whose `withTrace()` wrapper is the production and
@@ -2267,13 +2267,15 @@ declare class BitfabRivetIntegration<TContext extends object = RivetContextLike>
     withContext<TReturn>(context: TContext, fn: () => TReturn): TReturn;
     /**
      * Adapt a traced function into a Rivet handler whose first argument is the
-     * live actor context. The context is not added to the traced root's inputs.
+     * live actor or workflow context. The context is not added to the traced
+     * root's inputs.
      */
     bind<This, TArgs extends unknown[], TReturn>(fn: (this: This, ...args: TArgs) => TReturn): (this: This, context: TContext, ...args: TArgs) => TReturn;
     /**
      * Return the current Rivet context proxy. In production it delegates to the
      * bound live context. During replay it hydrates recorded actor state and
-     * intercepts supported effects without requiring a live Rivet actor.
+     * intercepts supported effects without requiring a live Rivet actor or
+     * workflow.
      */
     getContext(): TContext;
     private createContextProxy;
@@ -3123,9 +3125,9 @@ declare class Bitfab {
      */
     getLangGraphIntegration(traceFunctionKey: string, options?: LangGraphIntegrationOptions): BitfabLangGraphIntegration;
     /**
-     * Get the first-class Rivet actor-context integration for an existing trace
-     * root. The integration binds context and records effect spans; it never
-     * creates another root.
+     * Get the first-class Rivet actor/workflow-context integration for an
+     * existing trace root. The integration binds context and records effect
+     * spans; it never creates another root.
      *
      * @param traceFunctionKey - Key used by the enclosing `withTrace()` root
      * @param options - Explicit per-context-path replay mocking policy
@@ -3602,7 +3604,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.54.10";
+declare const __version__ = "0.54.12";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -21,7 +21,7 @@ import {
   getCurrentTrace,
   reseedFromRegistry,
   seedFromRegistry
-} from "./chunk-3F7SICKD.js";
+} from "./chunk-JEFU6FTD.js";
 import "./chunk-EGWJYCNT.js";
 import {
   BITFAB_PROGRESS_PREFIX,
@@ -30,7 +30,7 @@ import {
   ReplayError,
   reportReplayProgress,
   serializeReplayResult
-} from "./chunk-VJOLQ7UY.js";
+} from "./chunk-XTUINMAL.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -38,7 +38,7 @@ import {
   MixedTracingError,
   __version__,
   flushTraces
-} from "./chunk-XREH4NJT.js";
+} from "./chunk-I5NJU64I.js";
 import "./chunk-VCRFFCLY.js";
 export {
   AssertionCategoriesClient,

package/dist/node.cjs

@@ -88,7 +88,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.54.10";
+    __version__ = "0.54.12";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -114,20 +114,20 @@ var init_traceCompletion = __esm({
         this.closed = /* @__PURE__ */ new Map();
       }
       state(traceId) {
-        let state = this.traces.get(traceId);
-        if (state === void 0) {
-          state = { spanIds: /* @__PURE__ */ new Set(), active: /* @__PURE__ */ new Set() };
-          this.traces.set(traceId, state);
+        let state2 = this.traces.get(traceId);
+        if (state2 === void 0) {
+          state2 = { spanIds: /* @__PURE__ */ new Set(), active: /* @__PURE__ */ new Set() };
+          this.traces.set(traceId, state2);
         }
-        return state;
+        return state2;
       }
       start(traceId, spanId) {
         if (this.closed.has(traceId)) {
           return;
         }
-        const state = this.state(traceId);
-        state.spanIds.add(spanId);
-        state.active.add(spanId);
+        const state2 = this.state(traceId);
+        state2.spanIds.add(spanId);
+        state2.active.add(spanId);
       }
       open(traceId) {
         if (!this.closed.has(traceId)) {
@@ -143,24 +143,24 @@ var init_traceCompletion = __esm({
         }
       }
       end(traceId, spanId) {
-        const state = this.traces.get(traceId);
-        if (state !== void 0) {
-          state.active.delete(spanId);
-          this.drain(traceId, state);
+        const state2 = this.traces.get(traceId);
+        if (state2 !== void 0) {
+          state2.active.delete(spanId);
+          this.drain(traceId, state2);
         }
       }
       abort(traceId, spanId) {
-        const state = this.traces.get(traceId);
-        if (state === void 0) {
+        const state2 = this.traces.get(traceId);
+        if (state2 === void 0) {
           return;
         }
-        state.active.delete(spanId);
-        state.spanIds.delete(spanId);
-        if (state.active.size === 0 && state.spanIds.size === 0 && state.complete === void 0) {
+        state2.active.delete(spanId);
+        state2.spanIds.delete(spanId);
+        if (state2.active.size === 0 && state2.spanIds.size === 0 && state2.complete === void 0) {
           this.traces.delete(traceId);
           return;
         }
-        this.drain(traceId, state);
+        this.drain(traceId, state2);
       }
       close(traceId, complete, dropped = false) {
         const count = this.closed.get(traceId);
@@ -168,29 +168,29 @@ var init_traceCompletion = __esm({
           complete(count);
           return;
         }
-        const state = this.traces.get(traceId);
-        if (state === void 0) {
+        const state2 = this.traces.get(traceId);
+        if (state2 === void 0) {
           return;
         }
-        state.complete = complete;
+        state2.complete = complete;
         if (dropped) {
-          state.active.clear();
+          state2.active.clear();
         }
-        this.drain(traceId, state);
+        this.drain(traceId, state2);
       }
-      drain(traceId, state) {
-        if (state.active.size !== 0 || state.complete === void 0) {
+      drain(traceId, state2) {
+        if (state2.active.size !== 0 || state2.complete === void 0) {
           return;
         }
         this.traces.delete(traceId);
-        this.closed.set(traceId, state.spanIds.size);
+        this.closed.set(traceId, state2.spanIds.size);
         if (this.closed.size > 1024) {
           const oldest = this.closed.keys().next().value;
           if (oldest !== void 0) {
             this.closed.delete(oldest);
           }
         }
-        state.complete(state.spanIds.size);
+        state2.complete(state2.spanIds.size);
       }
     };
   }
@@ -441,8 +441,8 @@ function text(value) {
   const trimmed = value?.trim();
   return trimmed ? trimmed : null;
 }
-function isEmptyGitState(state) {
-  return Object.values(state).every((value) => value === null);
+function isEmptyGitState(state2) {
+  return Object.values(state2).every((value) => value === null);
 }
 async function resolveGitState(cwd) {
   const git = await gitRunner({
@@ -463,14 +463,14 @@ async function resolveGitState(cwd) {
   if (!root) {
     return null;
   }
-  const state = {
+  const state2 = {
     githubEmail: text(email),
     branch: text(branch),
     commitSha: sha(commitSha),
     baseSha: sha(baseSha),
     experimentSha: await resolveWorkingTreeSha(run, root.trim())
   };
-  return isEmptyGitState(state) ? null : state;
+  return isEmptyGitState(state2) ? null : state2;
 }
 async function resolveWorkingTreeSha(run, root) {
   let indexPath;
@@ -6985,6 +6985,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/client.ts
 init_policyRefresh();
 init_randomUuid();
@@ -7134,18 +7211,42 @@ var RIVET_TOP_LEVEL_EFFECTS = [
   "broadcast",
   "saveState",
   "sleep",
-  "destroy"
+  "destroy",
+  "step",
+  "tryStep",
+  "try",
+  "loop",
+  "sleepUntil",
+  "rollbackCheckpoint",
+  "join",
+  "race",
+  "removed",
+  "getVersion"
 ];
 var TOP_LEVEL_UNAVAILABLE_CAPABILITIES = /* @__PURE__ */ new Set([
   "conn",
   "conns",
-  "log",
   "request",
   "actorRuntimeSocket",
   "keepAwake",
   "waitUntil",
   "client"
 ]);
+var replayLog = () => void 0;
+var replayLogger = {
+  level: "silent",
+  msgPrefix: void 0,
+  fatal: replayLog,
+  trace: replayLog,
+  silent: replayLog,
+  debug: replayLog,
+  info: replayLog,
+  warn: replayLog,
+  error: replayLog,
+  child: () => replayLogger,
+  bindings: () => ({}),
+  isLevelEnabled: () => false
+};
 var SYNC_VOID_EFFECTS = /* @__PURE__ */ new Set([
   "broadcast",
   "sleep",
@@ -7175,16 +7276,28 @@ function snapshotContext(context) {
     "state",
     "vars",
     "actorId",
+    "workflowId",
     "name",
     "key",
-    "region",
-    "aborted"
+    "region"
   ]) {
     const value = readProperty(source, property);
     if (value !== void 0) {
       snapshot[property] = value;
     }
   }
+  const aborted = readProperty(source, "aborted");
+  if (typeof aborted === "boolean") {
+    snapshot.aborted = aborted;
+  } else {
+    const abortSignal = readProperty(source, "abortSignal");
+    if (isRecord5(abortSignal)) {
+      const signalAborted = readProperty(abortSignal, "aborted");
+      if (typeof signalAborted === "boolean") {
+        snapshot.aborted = signalAborted;
+      }
+    }
+  }
   return snapshot;
 }
 function unavailableCapability(path) {
@@ -7276,7 +7389,8 @@ var BitfabRivetIntegration = class {
   }
   /**
    * Adapt a traced function into a Rivet handler whose first argument is the
-   * live actor context. The context is not added to the traced root's inputs.
+   * live actor or workflow context. The context is not added to the traced
+   * root's inputs.
    */
   bind(fn) {
     const integration = this;
@@ -7287,7 +7401,8 @@ var BitfabRivetIntegration = class {
   /**
    * Return the current Rivet context proxy. In production it delegates to the
    * bound live context. During replay it hydrates recorded actor state and
-   * intercepts supported effects without requiring a live Rivet actor.
+   * intercepts supported effects without requiring a live Rivet actor or
+   * workflow.
    */
   getContext() {
     const replay2 = getReplayContext();
@@ -7371,6 +7486,12 @@ var BitfabRivetIntegration = class {
         if (property === "setPreventSleep" && replay2) {
           return (_preventSleep) => void 0;
         }
+        if (property === "isEvicted" && replay2) {
+          return () => replayTarget.abortSignal?.aborted === true;
+        }
+        if (property === "log" && replay2) {
+          return replayLogger;
+        }
         if (replay2 && typeof property === "string" && TOP_LEVEL_UNAVAILABLE_CAPABILITIES.has(property)) {
           throw unavailableCapability(property);
         }
@@ -7544,7 +7665,7 @@ var BitfabRivetIntegration = class {
     return { ...value, complete };
   }
   queueCallIsCompletable(effect, args) {
-    const options = effect === "queue.waitForNames" ? args[1] : args[0];
+    const options = effect === "queue.waitForNames" || (effect === "queue.next" || effect === "queue.nextBatch") && typeof args[0] === "string" ? args[1] : args[0];
     return isRecord5(options) && options.completable === true;
   }
   createQueueIterator(queue, options) {
@@ -7708,6 +7829,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;
@@ -7745,6 +7869,7 @@ var BitfabOpenAITracingProcessor = class {
     }
     this.endedTraces.delete(traceId);
     delete this.activeSpanMappings[traceId];
+    unmapOpenAiAgentsTrace(traceId);
     delete this.canonicalTraceIds[traceId];
     delete this.activeTraces[traceId];
   }
@@ -7769,6 +7894,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 {
@@ -7789,6 +7917,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 = {};
@@ -8946,16 +9077,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(
@@ -8966,10 +9097,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);
       }
     );
   }
@@ -9275,9 +9406,9 @@ var Bitfab = class {
     });
   }
   /**
-   * Get the first-class Rivet actor-context integration for an existing trace
-   * root. The integration binds context and records effect spans; it never
-   * creates another root.
+   * Get the first-class Rivet actor/workflow-context integration for an
+   * existing trace root. The integration binds context and records effect
+   * spans; it never creates another root.
    *
    * @param traceFunctionKey - Key used by the enclosing `withTrace()` root
    * @param options - Explicit per-context-path replay mocking policy
@@ -9580,7 +9711,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;
@@ -9594,11 +9729,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;