@bitfab/sdk 0.22.0 → 0.23.1

package/dist/index.d.cts

@@ -278,8 +278,8 @@ interface ActiveSpanContext$1 {
  */
 declare class BitfabLangGraphCallbackHandler {
     name: string;
-    ignoreRetriever: boolean;
     ignoreRetry: boolean;
+    ignoreRetriever: boolean;
     ignoreCustomEvent: boolean;
     private readonly httpClient;
     private readonly traceFunctionKey;
@@ -330,6 +330,14 @@ declare class BitfabLangGraphCallbackHandler {
  *      replayable with no hand-written `withSpan`. The processor's spans nest
  *      underneath it automatically (it remaps onto the active span context).
  *
+ * When `wrapRun` runs inside an enclosing Bitfab span (the replay auto-wrap, or
+ * a caller's own `withSpan`), that span is already the replayable root: the
+ * handler skips opening a second one and lets the processor nest the run's spans
+ * under the existing root. This mirrors the Claude Agent SDK and LangGraph
+ * handlers, which no-op their root span under an enclosing span, and keeps a
+ * replayed run's span tree identical to the original (no doubled root agent
+ * span).
+ *
  * Use both together: register the processor once at startup, then call
  * `handler.wrapRun(agent, input)` instead of `run(agent, input)`.
  */
@@ -339,6 +347,7 @@ type RootSpanOptions = {
     finalize: (result: unknown) => unknown | Promise<unknown>;
 };
 type WithSpanFn = <TArgs extends unknown[], TReturn>(traceFunctionKey: string, options: RootSpanOptions, fn: (...args: TArgs) => TReturn) => (...args: TArgs) => TReturn;
+type GetActiveSpanContextFn = () => unknown | null;
 type RunInput<TContext, TAgent extends Agent<any, any>> = string | AgentInputItem[] | RunState<TContext, TAgent>;
 /**
  * OpenAI Agents SDK handler that records a replayable root span around a run.
@@ -361,9 +370,11 @@ type RunInput<TContext, TAgent extends Agent<any, any>> = string | AgentInputIte
 declare class BitfabOpenAIAgentHandler {
     private readonly traceFunctionKey;
     private readonly withSpanFn;
+    private readonly getActiveSpanContext?;
     constructor(config: {
         traceFunctionKey: string;
         withSpan: WithSpanFn;
+        getActiveSpanContext?: GetActiveSpanContextFn;
     });
     /**
      * Drop-in replacement for the OpenAI Agents SDK's `run()` that records a
@@ -1210,7 +1221,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.22.0";
+declare const __version__ = "0.23.1";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.d.ts

@@ -278,8 +278,8 @@ interface ActiveSpanContext$1 {
  */
 declare class BitfabLangGraphCallbackHandler {
     name: string;
-    ignoreRetriever: boolean;
     ignoreRetry: boolean;
+    ignoreRetriever: boolean;
     ignoreCustomEvent: boolean;
     private readonly httpClient;
     private readonly traceFunctionKey;
@@ -330,6 +330,14 @@ declare class BitfabLangGraphCallbackHandler {
  *      replayable with no hand-written `withSpan`. The processor's spans nest
  *      underneath it automatically (it remaps onto the active span context).
  *
+ * When `wrapRun` runs inside an enclosing Bitfab span (the replay auto-wrap, or
+ * a caller's own `withSpan`), that span is already the replayable root: the
+ * handler skips opening a second one and lets the processor nest the run's spans
+ * under the existing root. This mirrors the Claude Agent SDK and LangGraph
+ * handlers, which no-op their root span under an enclosing span, and keeps a
+ * replayed run's span tree identical to the original (no doubled root agent
+ * span).
+ *
  * Use both together: register the processor once at startup, then call
  * `handler.wrapRun(agent, input)` instead of `run(agent, input)`.
  */
@@ -339,6 +347,7 @@ type RootSpanOptions = {
     finalize: (result: unknown) => unknown | Promise<unknown>;
 };
 type WithSpanFn = <TArgs extends unknown[], TReturn>(traceFunctionKey: string, options: RootSpanOptions, fn: (...args: TArgs) => TReturn) => (...args: TArgs) => TReturn;
+type GetActiveSpanContextFn = () => unknown | null;
 type RunInput<TContext, TAgent extends Agent<any, any>> = string | AgentInputItem[] | RunState<TContext, TAgent>;
 /**
  * OpenAI Agents SDK handler that records a replayable root span around a run.
@@ -361,9 +370,11 @@ type RunInput<TContext, TAgent extends Agent<any, any>> = string | AgentInputIte
 declare class BitfabOpenAIAgentHandler {
     private readonly traceFunctionKey;
     private readonly withSpanFn;
+    private readonly getActiveSpanContext?;
     constructor(config: {
         traceFunctionKey: string;
         withSpan: WithSpanFn;
+        getActiveSpanContext?: GetActiveSpanContextFn;
     });
     /**
      * Drop-in replacement for the OpenAI Agents SDK's `run()` that records a
@@ -1210,7 +1221,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.22.0";
+declare const __version__ = "0.23.1";
 
 /**
  * Constants for the Bitfab SDK.

package/dist/index.js

@@ -13,7 +13,7 @@ import {
   flushTraces,
   getCurrentSpan,
   getCurrentTrace
-} from "./chunk-4SLFC266.js";
+} from "./chunk-2EDITKO2.js";
 import {
   BitfabError
 } from "./chunk-EQI6ZJC3.js";

package/dist/node.cjs

@@ -505,7 +505,7 @@ registerAsyncLocalStorageClass(
 );
 
 // src/version.generated.ts
-var __version__ = "0.22.0";
+var __version__ = "0.23.1";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1956,8 +1956,9 @@ function extractLangGraphMetadata(metadata) {
 var BitfabLangGraphCallbackHandler = class {
   constructor(config) {
     this.name = "BitfabLangGraphCallbackHandler";
-    this.ignoreRetriever = true;
     this.ignoreRetry = true;
+    // Retriever callbacks ARE captured (retriever queries -> function spans).
+    this.ignoreRetriever = false;
     this.ignoreCustomEvent = true;
     this.runToSpan = /* @__PURE__ */ new Map();
     this.invocations = /* @__PURE__ */ new Map();
@@ -2292,9 +2293,17 @@ var BitfabOpenAIAgentHandler = class {
   constructor(config) {
     this.traceFunctionKey = config.traceFunctionKey;
     this.withSpanFn = config.withSpan;
+    this.getActiveSpanContext = config.getActiveSpanContext;
   }
   async wrapRun(agent, input, options) {
     const { run } = await import("@openai/agents");
+    if (this.getActiveSpanContext?.() != null) {
+      return run(
+        agent,
+        input,
+        options
+      );
+    }
     const isStreaming = options?.stream === true;
     const finalize = async (result) => {
       const res = result;
@@ -3038,7 +3047,11 @@ var Bitfab = class {
   getOpenAiAgentHandler(traceFunctionKey) {
     return new BitfabOpenAIAgentHandler({
       traceFunctionKey,
-      withSpan: this.withSpan.bind(this)
+      withSpan: this.withSpan.bind(this),
+      getActiveSpanContext: () => {
+        const stack = getSpanStack();
+        return stack[stack.length - 1] ?? null;
+      }
     });
   }
   /**
@@ -3663,7 +3676,7 @@ var Bitfab = class {
     if (wrappedKey === void 0) {
       replayFn = this.withSpan(
         traceFunctionKey,
-        { name: fn.name || "Replay", type: "agent" },
+        { name: traceFunctionKey, type: "agent" },
         fn
       );
     } else if (wrappedKey !== traceFunctionKey) {