@bitfab/sdk 0.64.29 → 0.65.0

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.64.29";
+    __version__ = "0.65.0";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -5367,6 +5367,7 @@ __export(index_exports, {
   BitfabOpenAITracingProcessor: () => BitfabOpenAITracingProcessor,
   BitfabRivetIntegration: () => BitfabRivetIntegration,
   BitfabVercelAiHandler: () => BitfabVercelAiHandler,
+  CategoriesClient: () => CategoriesClient,
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
   DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
@@ -5397,9 +5398,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;
   }
@@ -5426,6 +5427,7 @@ var AssertionCategoriesClient = class {
     return response.category;
   }
 };
+var AssertionCategoriesClient = CategoriesClient;
 
 // src/claudeAgentSdk.ts
 init_constants();
@@ -9025,6 +9027,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}`;
 }
@@ -9039,9 +9062,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({
@@ -9054,10 +9078,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(
@@ -9069,7 +9096,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;
@@ -10092,7 +10119,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);
@@ -12411,6 +12439,7 @@ function resolveTraceFunctionKey(registration) {
   BitfabOpenAITracingProcessor,
   BitfabRivetIntegration,
   BitfabVercelAiHandler,
+  CategoriesClient,
   DEFAULT_SERVICE_URL,
   DatasetsClient,
   DbBranchReplayError,