bitfab 0.64.30 → 0.65.1

package/dist/{chunk-KDNS4BVZ.js → chunk-HW4ZKFYV.js}

@@ -3,10 +3,11 @@ import {
   clearAutoTraceNodeConfiguration,
   currentAutoTraceEntries,
   invokeThroughAutoTraceEntries,
+  runAsAutoTraceRootFunction,
   runWithAutoTraceEntry,
   runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext
-} from "./chunk-YWXQM6J3.js";
+} from "./chunk-QRNHRY7F.js";
 import {
   NO_MOCK_OVERRIDE,
   createIntegrationFixtureContext,
@@ -17,7 +18,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-FY5WZ6M4.js";
+} from "./chunk-3FTI76L6.js";
 import {
   BitfabError,
   CONTENT_OFF_KEY,
@@ -37,7 +38,7 @@ import {
   readEnv,
   recordCallerTraceMetadata,
   retireTraceMetadata
-} from "./chunk-GUPAIDK4.js";
+} from "./chunk-GS7FUQ6U.js";
 import {
   __privateAdd,
   __privateGet,
@@ -49,9 +50,9 @@ import {
   warnOnce
 } from "./chunk-MT36GJMW.js";
 
-// src/assertionCategories.ts
-var CATEGORIES_PATH = "/api/sdk/assertionCategories";
-var AssertionCategoriesClient = class {
+// src/categories.ts
+var CATEGORIES_PATH = "/api/sdk/categories";
+var CategoriesClient = class {
   constructor(httpClient) {
     this.httpClient = httpClient;
   }
@@ -78,6 +79,7 @@ var AssertionCategoriesClient = class {
     return response.category;
   }
 };
+var AssertionCategoriesClient = CategoriesClient;
 
 // src/processorPayload.ts
 var SERIALIZATION_DEGRADED_STEP = "serialization_degraded";
@@ -3444,6 +3446,27 @@ function draftsInGenerationOrder(assertions, assertionIds) {
     return assertion === void 0 ? [] : [assertion];
   });
 }
+function withCategoryId(assertion) {
+  const categoryId = assertion.category_id ?? assertion.category_assertion_id ?? null;
+  return {
+    ...assertion,
+    category_id: categoryId,
+    category_assertion_id: categoryId
+  };
+}
+function withCategoryIds(assertions) {
+  return assertions.map(withCategoryId);
+}
+function toWireAssertion({
+  category_assertion_id,
+  ...assertion
+}) {
+  const categoryId = assertion.category_id === void 0 ? category_assertion_id : assertion.category_id;
+  if (categoryId === void 0) {
+    return assertion;
+  }
+  return { ...assertion, category_id: categoryId };
+}
 function traceAssertionsPath(traceId, suffix = "") {
   return `/api/sdk/traces/${encodeURIComponent(traceId)}/assertions${suffix}`;
 }
@@ -3458,9 +3481,10 @@ var TracesClient = class {
     return this.httpClient.request(SEARCH_PATH, params);
   }
   async getAssertions(traceId) {
-    return this.httpClient.get(
+    const result = await this.httpClient.get(
       traceAssertionsPath(traceId)
     );
+    return { ...result, assertions: withCategoryIds(result.assertions) };
   }
   async saveAssertions(params) {
     return this.saveAssertionsAll({
@@ -3473,10 +3497,13 @@ var TracesClient = class {
       return [];
     }
     const response = await this.httpClient.request(ASSERTIONS_PATH, {
-      updates: params.updates,
+      updates: params.updates.map((update) => ({
+        ...update,
+        assertions: update.assertions.map(toWireAssertion)
+      })),
       source: params.source ?? "agent"
     });
-    return response.assertions;
+    return withCategoryIds(response.assertions);
   }
   async generateAssertions(traceId, options = {}) {
     const started = await this.httpClient.request(
@@ -3488,7 +3515,7 @@ var TracesClient = class {
       }
     );
     if (isGenerationFinished(traceId, started.generation)) {
-      return started.assertions;
+      return withCategoryIds(started.assertions);
     }
     const timeoutMs = options.timeoutMs ?? DEFAULT_GENERATION_TIMEOUT_MS;
     const deadline = Date.now() + timeoutMs;
@@ -4431,6 +4458,11 @@ function readBooleanEnv(name) {
   );
   return void 0;
 }
+var AUTO_TRACE_NODE_FUNCTION_NAMES = /* @__PURE__ */ new WeakMap();
+function registerAutoTraceNode(configured, functionName) {
+  AUTO_TRACE_NODE_FUNCTION_NAMES.set(configured, functionName);
+  return configured;
+}
 var DEFAULT_AUTO_TRACE_MAX_DEPTH = 30;
 var DEFAULT_AUTO_TRACE_MAX_CAPTURED_SUBTREE_SPANS = 500;
 function spanOwnerName(receiver) {
@@ -4506,7 +4538,8 @@ var Bitfab = class {
       this.simulationPlan.refresh();
     }
     this.datasets = new DatasetsClient(this.httpClient);
-    this.assertionCategories = new AssertionCategoriesClient(this.httpClient);
+    this.categories = new CategoriesClient(this.httpClient);
+    this.assertionCategories = this.categories;
     this.organizationMembers = new OrganizationMembersClient(this.httpClient);
     this.traces = new TracesClient(this.httpClient);
     this.spans = new SpansClient(this.httpClient);
@@ -4565,7 +4598,13 @@ var Bitfab = class {
       throw new BitfabError("bitfab.withTrace requires a function");
     }
     const name = fn.name !== "" ? fn.name : traceFunctionKey;
-    return this.createAutoTraceRoot(traceFunctionKey, name, options, fn);
+    return this.createAutoTraceRoot(
+      traceFunctionKey,
+      name,
+      options,
+      fn,
+      AUTO_TRACE_NODE_FUNCTION_NAMES.get(fn) ?? fn.name.replace(/^bound /, "")
+    );
   }
   /**
    * Configure a transformed class method when it is discovered beneath a
@@ -4646,7 +4685,7 @@ var Bitfab = class {
   }
   createAutoTraceNode(configuration, fn, functionName) {
     const nodeConfiguration = { ...configuration, functionName };
-    return function(...args) {
+    return registerAutoTraceNode(function(...args) {
       if (!__bitfabAutoTraceActive()) {
         if (enclosingSurface() === "opt-in") {
           throw mixedTracingError("node()", "opt-out", "opt-in");
@@ -4657,11 +4696,15 @@ var Bitfab = class {
         nodeConfiguration,
         () => fn.apply(this, args)
       );
-    };
+    }, functionName);
   }
-  createAutoTraceRoot(traceFunctionKey, name, options, fn) {
+  createAutoTraceRoot(traceFunctionKey, name, options, fn, rootFunctionName = "") {
     const self = this;
     self.simulationPlan.refresh();
+    const runRootFunction = (receiver, args) => runAsAutoTraceRootFunction(
+      rootFunctionName,
+      () => fn.apply(receiver, args)
+    );
     const maxDepth = autoTraceLimit(
       options.maxDepth,
       DEFAULT_AUTO_TRACE_MAX_DEPTH
@@ -4824,7 +4867,7 @@ var Bitfab = class {
         return runWithAutoTraceRootContext(
           autoTraceContext,
           getSpanStack(),
-          () => fn.apply(this, args)
+          () => runRootFunction(this, args)
         );
       }
     );
@@ -4857,7 +4900,7 @@ var Bitfab = class {
         ),
         () => {
           if (link === void 0) {
-            return fn.apply(this, args);
+            return runRootFunction(this, args);
           }
           const innermost = enclosing[enclosing.length - 1];
           const innermostNow = currentAutoTraceEntries().find(
@@ -6376,7 +6419,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-NUZD6TOB.js");
+        const { flushTraces: flushTraces2 } = await import("./http-FN74JEHE.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -6435,7 +6478,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-7Y5ERBDN.js");
+    const { replay: doReplay } = await import("./replay-SVHAY4S6.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -6781,7 +6824,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-NUZD6TOB.js");
+  const { flushTraces: flushTraces2 } = await import("./http-FN74JEHE.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -6797,6 +6840,7 @@ function resolveTraceFunctionKey(registration) {
 }
 
 export {
+  CategoriesClient,
   AssertionCategoriesClient,
   BitfabClaudeAgentHandler,
   GraderRerunError,
@@ -6827,4 +6871,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-KDNS4BVZ.js.map
+//# sourceMappingURL=chunk-HW4ZKFYV.js.map