bitfab 0.22.0 → 0.23.1

package/dist/{chunk-4SLFC266.js → chunk-2EDITKO2.js}

@@ -10,7 +10,7 @@ import {
 } from "./chunk-EQI6ZJC3.js";
 
 // src/version.generated.ts
-var __version__ = "0.22.0";
+var __version__ = "0.23.1";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1454,8 +1454,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();
@@ -1790,9 +1791,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;
@@ -2529,7 +2538,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;
+      }
     });
   }
   /**
@@ -3154,7 +3167,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) {
@@ -3291,4 +3304,4 @@ export {
   BitfabFunction,
   finalizers
 };
-//# sourceMappingURL=chunk-4SLFC266.js.map
+//# sourceMappingURL=chunk-2EDITKO2.js.map