bitfab 0.64.30 → 0.65.1

package/dist/index.cjs

@@ -51,7 +51,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.64.30";
+    __version__ = "0.65.1";
     __packageName__ = "bitfab";
   }
 });
@@ -5376,6 +5376,7 @@ __export(index_exports, {
   BitfabOpenAITracingProcessor: () => BitfabOpenAITracingProcessor,
   BitfabRivetIntegration: () => BitfabRivetIntegration,
   BitfabVercelAiHandler: () => BitfabVercelAiHandler,
+  CategoriesClient: () => CategoriesClient,
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
   DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
@@ -5406,9 +5407,9 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
-// 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;
   }
@@ -5435,6 +5436,7 @@ var AssertionCategoriesClient = class {
     return response.category;
   }
 };
+var AssertionCategoriesClient = CategoriesClient;
 
 // src/claudeAgentSdk.ts
 init_constants();
@@ -6143,6 +6145,21 @@ function runWithAutoTraceNodeConfiguration(nodeConfiguration, fn) {
   }
   return result;
 }
+function runAsAutoTraceRootFunction(functionName, fn) {
+  const scope = currentAutoTraceScope();
+  if (!scope || scope.entries.length === 0 || functionName === "") {
+    return fn();
+  }
+  const rootFunction = {
+    name: functionName,
+    entering: true
+  };
+  try {
+    return runWithAutoTraceScope({ ...scope, rootFunction }, fn);
+  } finally {
+    rootFunction.entering = false;
+  }
+}
 var SUBTREE_TRACING_DOCS_URL = "https://docs.bitfab.ai/typescript-sdk#experimental-subtree-tracing";
 function transformInstalled() {
   return autoTraceGlobal.__bitfab_auto_trace_transformed_v1__ === true;
@@ -9034,6 +9051,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}`;
 }
@@ -9048,9 +9086,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({
@@ -9063,10 +9102,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(
@@ -9078,7 +9120,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;
@@ -10026,6 +10068,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) {
@@ -10101,7 +10148,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);
@@ -10160,7 +10208,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
@@ -10241,7 +10295,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");
@@ -10252,11 +10306,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
@@ -10419,7 +10477,7 @@ var Bitfab = class {
         return runWithAutoTraceRootContext(
           autoTraceContext,
           getSpanStack(),
-          () => fn.apply(this, args)
+          () => runRootFunction(this, args)
         );
       }
     );
@@ -10452,7 +10510,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(
@@ -12420,6 +12478,7 @@ function resolveTraceFunctionKey(registration) {
   BitfabOpenAITracingProcessor,
   BitfabRivetIntegration,
   BitfabVercelAiHandler,
+  CategoriesClient,
   DEFAULT_SERVICE_URL,
   DatasetsClient,
   DbBranchReplayError,