bitfab 0.54.3 → 0.54.5

package/dist/{chunk-PELZFWPC.js → chunk-NATC7L2W.js}

@@ -18,7 +18,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-DL4KHJON.js";
+} from "./chunk-3O5MS4XJ.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -35,7 +35,7 @@ import {
   recordCallerTraceMetadata,
   retireTraceMetadata,
   warnOnce
-} from "./chunk-ZO4NEEUA.js";
+} from "./chunk-GVK7AES5.js";
 import {
   __privateAdd,
   __privateGet,
@@ -76,45 +76,6 @@ var AssertionCategoriesClient = class {
   }
 };
 
-// src/assertions.ts
-var ASSERTIONS_PATH = "/api/sdk/traces/assertions";
-function traceAssertionsPath(traceId, suffix = "") {
-  return `/api/sdk/traces/${encodeURIComponent(traceId)}/assertions${suffix}`;
-}
-var TracesClient = class {
-  constructor(httpClient) {
-    this.httpClient = httpClient;
-  }
-  async getAssertions(traceId) {
-    return this.httpClient.get(
-      traceAssertionsPath(traceId)
-    );
-  }
-  async saveAssertions(params) {
-    return this.saveAssertionsAll({
-      updates: [{ traceId: params.traceId, assertions: params.assertions }],
-      source: params.source
-    });
-  }
-  async saveAssertionsAll(params) {
-    if (params.updates.length === 0) {
-      return [];
-    }
-    const response = await this.httpClient.request(ASSERTIONS_PATH, {
-      updates: params.updates,
-      source: params.source ?? "agent"
-    });
-    return response.assertions;
-  }
-  async archiveAssertions(params) {
-    const response = await this.httpClient.request(
-      traceAssertionsPath(params.traceId, "/archive"),
-      { assertionIds: params.assertionIds }
-    );
-    return response.archived;
-  }
-};
-
 // src/processorPayload.ts
 var SERIALIZATION_DEGRADED_STEP = "serialization_degraded";
 function degradedError(dropped) {
@@ -331,6 +292,7 @@ var BitfabClaudeAgentHandler = class {
       spanInfo.dropped = [...inputDropped];
     }
     this.runToSpan.set(spanId, spanInfo);
+    this.httpClient.traceCompletion.start(spanInfo.traceId, spanInfo.spanId);
     return spanInfo;
   }
   completeSpan(spanId, output, error, extraContexts) {
@@ -352,6 +314,7 @@ var BitfabClaudeAgentHandler = class {
       spanInfo.contexts.push(extraContexts);
     }
     this.sendSpan(spanInfo);
+    this.httpClient.traceCompletion.end(spanInfo.traceId, spanInfo.spanId);
   }
   sendSpan(spanInfo) {
     const spanData = {
@@ -1824,6 +1787,9 @@ var BitfabLangGraphCallbackHandler = class {
       spanInfo.hidden = true;
     }
     this.runToSpan.set(runId, spanInfo);
+    if (spanInfo.hidden !== true) {
+      this.httpClient.traceCompletion.start(spanInfo.traceId, spanInfo.spanId);
+    }
     if (isRootInvocation) {
       this.sendTraceStart(spanInfo);
     }
@@ -1855,6 +1821,7 @@ var BitfabLangGraphCallbackHandler = class {
       this.sendTraceCompletion(spanInfo, invocation?.activeContext ?? null);
       this.invocations.delete(runId);
     }
+    this.httpClient.traceCompletion.end(spanInfo.traceId, spanInfo.spanId);
   }
   sendSpan(spanInfo) {
     const spanData = {
@@ -2539,6 +2506,55 @@ var StreamFinalizationError = class extends Error {
   }
 };
 
+// src/traces.ts
+var ASSERTIONS_PATH = "/api/sdk/traces/assertions";
+var SEARCH_PATH = "/api/sdk/traces/search";
+function traceAssertionsPath(traceId, suffix = "") {
+  return `/api/sdk/traces/${encodeURIComponent(traceId)}/assertions${suffix}`;
+}
+var TracesClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  /**
+   * Search traces by exact caller metadata.
+   *
+   * Keys and values are matched case-insensitively. Up to 10 supplied pairs
+   * must match. Use `nextCursor` from the result to fetch another page.
+   */
+  async search(params) {
+    return this.httpClient.request(SEARCH_PATH, params);
+  }
+  async getAssertions(traceId) {
+    return this.httpClient.get(
+      traceAssertionsPath(traceId)
+    );
+  }
+  async saveAssertions(params) {
+    return this.saveAssertionsAll({
+      updates: [{ traceId: params.traceId, assertions: params.assertions }],
+      source: params.source
+    });
+  }
+  async saveAssertionsAll(params) {
+    if (params.updates.length === 0) {
+      return [];
+    }
+    const response = await this.httpClient.request(ASSERTIONS_PATH, {
+      updates: params.updates,
+      source: params.source ?? "agent"
+    });
+    return response.assertions;
+  }
+  async archiveAssertions(params) {
+    const response = await this.httpClient.request(
+      traceAssertionsPath(params.traceId, "/archive"),
+      { assertionIds: params.assertionIds }
+    );
+    return response.archived;
+  }
+};
+
 // src/tracing.ts
 var BitfabOpenAITracingProcessor = class {
   /**
@@ -2550,6 +2566,8 @@ var BitfabOpenAITracingProcessor = class {
     this.activeTraces = {};
     this.activeSpanMappings = {};
     this.canonicalTraceIds = {};
+    this.startedSpans = /* @__PURE__ */ new Map();
+    this.endedTraces = /* @__PURE__ */ new Set();
     this.ownsHttpClient = config._httpClient === void 0;
     this.httpClient = config._httpClient ?? new HttpClient({
       apiKey: config.apiKey,
@@ -2588,6 +2606,10 @@ var BitfabOpenAITracingProcessor = class {
     }
     const canonicalTraceId = activeContext?.traceId ?? this.getCanonicalTraceId(trace.traceId);
     this.canonicalTraceIds[trace.traceId] = canonicalTraceId;
+    this.httpClient.traceCompletion.hold(
+      activeContext?.traceId ?? trace.traceId,
+      `openai:${trace.traceId}`
+    );
     this.sendTrace(trace, {
       id: canonicalTraceId,
       sourceTraceId: activeContext?.traceId
@@ -2605,16 +2627,33 @@ var BitfabOpenAITracingProcessor = class {
       id: mapping?.traceId ?? this.getCanonicalTraceId(trace.traceId),
       sourceTraceId: mapping?.traceId
     });
-    delete this.activeSpanMappings[trace.traceId];
-    delete this.canonicalTraceIds[trace.traceId];
-    delete this.activeTraces[trace.traceId];
+    this.endedTraces.add(trace.traceId);
+    this.httpClient.traceCompletion.end(
+      mapping?.traceId ?? trace.traceId,
+      `openai:${trace.traceId}`
+    );
+    this.cleanupTrace(trace.traceId);
+  }
+  cleanupTrace(traceId) {
+    if (!this.endedTraces.has(traceId) || [...this.startedSpans.values()].includes(traceId)) {
+      return;
+    }
+    this.endedTraces.delete(traceId);
+    delete this.activeSpanMappings[traceId];
+    delete this.canonicalTraceIds[traceId];
+    delete this.activeTraces[traceId];
   }
   /**
    * Called when a span is started. Span payloads are authoritative completed
    * snapshots, so start notifications do not cross the transport boundary.
    */
   // biome-ignore lint/suspicious/noExplicitAny: OpenAI Agents SDK uses any for span data
-  async onSpanStart(_span) {
+  async onSpanStart(span) {
+    if (span.spanId !== void 0 && span.traceId !== null && span.traceId !== void 0) {
+      this.startedSpans.set(span.spanId, span.traceId);
+      const traceId = this.activeSpanMappings[span.traceId]?.traceId ?? span.traceId;
+      this.httpClient.traceCompletion.start(traceId, span.spanId);
+    }
   }
   /**
    * Called when a span is ended.
@@ -2623,7 +2662,17 @@ var BitfabOpenAITracingProcessor = class {
    */
   // biome-ignore lint/suspicious/noExplicitAny: OpenAI Agents SDK uses any for span data
   async onSpanEnd(span) {
-    this.sendSpan(span);
+    const originalTraceId = span.traceId ?? "";
+    const traceId = this.activeSpanMappings[originalTraceId]?.traceId ?? originalTraceId;
+    try {
+      this.sendSpan(span);
+    } finally {
+      if (span.spanId !== void 0) {
+        this.httpClient.traceCompletion.end(traceId, span.spanId);
+        this.startedSpans.delete(span.spanId);
+      }
+      this.cleanupTrace(originalTraceId);
+    }
   }
   /**
    * Called when a trace is being flushed.
@@ -4328,6 +4377,7 @@ var Bitfab = class {
       let newStack;
       let executeWithContext;
       let registeredTraceId;
+      let startedSpan;
       try {
         const currentStack = getSpanStack();
         const parentContext = currentStack[currentStack.length - 1];
@@ -4379,6 +4429,8 @@ var Bitfab = class {
           });
           registeredTraceId = traceId;
         }
+        self.httpClient.traceCompletion.start(traceId, spanId);
+        startedSpan = { traceId, spanId };
         const functionName = fn.name !== "" ? fn.name : void 0;
         const rootTraceFunctionKey = activeTraceStates.get(traceId)?.traceFunctionKey ?? traceFunctionKey;
         const baseSpanParams = {
@@ -4453,6 +4505,8 @@ var Bitfab = class {
               retireTraceMetadata(traceId);
             }
           } catch {
+          } finally {
+            self.httpClient.traceCompletion.end(traceId, spanId);
           }
         };
         const recordSpan = (result) => {
@@ -4469,6 +4523,14 @@ var Bitfab = class {
             void sendSpan({ result });
           }
         };
+        const resolveMockOrAbort = async (resolve) => {
+          try {
+            return await resolve();
+          } catch (error) {
+            self.httpClient.traceCompletion.abort(traceId, spanId);
+            throw error;
+          }
+        };
         executeWithContext = () => {
           let result;
           try {
@@ -4525,7 +4587,7 @@ var Bitfab = class {
               );
             }
             return (async () => {
-              const output = await pending;
+              const output = await resolveMockOrAbort(() => pending);
               void sendSpan({
                 result: output,
                 mocked: true,
@@ -4595,7 +4657,7 @@ var Bitfab = class {
                 );
               }
               return runWithSpanStack(newStack, async () => {
-                const resolved2 = await resolution;
+                const resolved2 = await resolveMockOrAbort(() => resolution);
                 if (resolved2.matched) {
                   void sendSpan({
                     result: resolved2.output,
@@ -4611,7 +4673,9 @@ var Bitfab = class {
                   );
                 }
                 if (shouldMockWithBaseStrategy) {
-                  const output = await resolveRecordedOutput();
+                  const output = await resolveMockOrAbort(
+                    () => resolveRecordedOutput()
+                  );
                   void sendSpan({
                     result: output,
                     mocked: true,
@@ -4641,6 +4705,12 @@ var Bitfab = class {
           }
         }
       } catch (setupError) {
+        if (startedSpan) {
+          self.httpClient.traceCompletion.abort(
+            startedSpan.traceId,
+            startedSpan.spanId
+          );
+        }
         if (registeredTraceId) {
           activeTraceStates.delete(registeredTraceId);
           retireTraceMetadata(registeredTraceId);
@@ -5107,7 +5177,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-A5BXPX5R.js");
+        const { flushTraces: flushTraces2 } = await import("./http-RW2JO6TU.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -5166,7 +5236,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-FOVT4PNE.js");
+    const { replay: doReplay } = await import("./replay-MHYULXCH.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -5450,7 +5520,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-A5BXPX5R.js");
+  const { flushTraces: flushTraces2 } = await import("./http-RW2JO6TU.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -5467,7 +5537,6 @@ function resolveTraceFunctionKey(registration) {
 
 export {
   AssertionCategoriesClient,
-  TracesClient,
   BitfabClaudeAgentHandler,
   DatasetsClient,
   SUPPORTED_PROVIDERS,
@@ -5476,6 +5545,7 @@ export {
   BitfabLangGraphCallbackHandler,
   BitfabLangGraphIntegration,
   BitfabOpenAIAgentHandler,
+  TracesClient,
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   getCurrentSpan,
@@ -5488,4 +5558,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-PELZFWPC.js.map
+//# sourceMappingURL=chunk-NATC7L2W.js.map