@bitfab/sdk 0.44.0 → 0.45.0

package/dist/{http-QWUWDTFN.js → http-E7RYN3BX.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-6JTVCM2L.js";
+} from "./chunk-TCOD5O2B.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BitfabError,
@@ -17,4 +17,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-QWUWDTFN.js.map
+//# sourceMappingURL=http-E7RYN3BX.js.map

package/dist/{http-Q4TSO32V.js → http-QGTV3X7R.js}

@@ -6,7 +6,7 @@ import {
   flushTraces,
   parseRetryAfterMs,
   serializePayloadBody
-} from "./chunk-AX6EZPGH.js";
+} from "./chunk-SILI6DDS.js";
 export {
   BitfabError,
   HttpClient,
@@ -16,4 +16,4 @@ export {
   parseRetryAfterMs,
   serializePayloadBody
 };
-//# sourceMappingURL=http-Q4TSO32V.js.map
+//# sourceMappingURL=http-QGTV3X7R.js.map

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.44.0";
+    __version__ = "0.45.0";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -2719,7 +2719,9 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
         tokens: null,
         model: serverItem.originalModel ?? serverItem.model ?? null,
         dbSnapshotRef: dbSnapshotRef ?? null,
-        dbBranchTimings: dbBranchTimings ?? null
+        dbBranchTimings: dbBranchTimings ?? null,
+        traceOutline: null,
+        originalTraceOutline: null
       };
     }
     try {
@@ -2803,7 +2805,9 @@ async function processItem(httpClient, serverItem, fn, testRunId, mockStrategy,
     tokens: null,
     model: originalModel,
     dbSnapshotRef: dbSnapshotRef ?? null,
-    dbBranchTimings: dbBranchTimings ?? null
+    dbBranchTimings: dbBranchTimings ?? null,
+    traceOutline: null,
+    originalTraceOutline: null
   };
 }
 async function waitForReplayPersistence(httpClient, testRunId, replayedTraceIds) {
@@ -3058,7 +3062,9 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
             tokens: item.tokens,
             model: item.model,
             dbSnapshotRef: item.dbSnapshotRef,
-            dbBranchTimings: item.dbBranchTimings
+            dbBranchTimings: item.dbBranchTimings,
+            traceOutline: item.traceOutline,
+            originalTraceOutline: item.originalTraceOutline
           }
         });
       } catch {
@@ -3138,7 +3144,9 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
       }
       if (mapped !== void 0) {
         item.tokens = replayTokens?.[mapped] ?? null;
+        item.traceOutline = completeResult.traceOutlines?.[mapped] ?? null;
       }
+      item.originalTraceOutline = completeResult.originalTraceOutlines?.[item.originalTraceId] ?? null;
     }
     if (completedCount > 0 && missing.length === completedCount) {
       const serverCount = completeResult.traceCount !== void 0 ? ` The server persisted ${completeResult.traceCount} trace(s) for this run.` : "";
@@ -3261,10 +3269,12 @@ __export(index_exports, {
   DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
   HttpClient: () => HttpClient,
+  LabelsClient: () => LabelsClient,
   MixedTracingError: () => MixedTracingError,
   NO_MOCK_OVERRIDE: () => NO_MOCK_OVERRIDE,
   ReplayError: () => ReplayError,
   SUPPORTED_PROVIDERS: () => SUPPORTED_PROVIDERS,
+  TracesClient: () => TracesClient,
   __version__: () => __version__,
   defineReplayRegistry: () => defineReplayRegistry,
   finalizers: () => finalizers,
@@ -3278,6 +3288,33 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/assertions.ts
+function expectationsPath(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(expectationsPath(traceId));
+  }
+  async saveAssertions(params) {
+    const response = await this.httpClient.request(expectationsPath(params.traceId), {
+      assertions: params.assertions,
+      source: params.source ?? "agent"
+    });
+    return response.assertions;
+  }
+  async archiveAssertions(params) {
+    const response = await this.httpClient.request(
+      expectationsPath(params.traceId, "/archive"),
+      { assertionIds: params.assertionIds }
+    );
+    return response.archived;
+  }
+};
+
 // src/claudeAgentSdk.ts
 init_constants();
 init_http();
@@ -4528,6 +4565,28 @@ function buildSnapshotRef(config, sdkWallClockBeforeFn) {
 init_errors();
 init_http();
 
+// src/labels.ts
+var LABELS_PATH = "/api/sdk/traces/labels";
+var LabelsClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  async save(params) {
+    const response = await this.httpClient.request(
+      LABELS_PATH,
+      {
+        labels: params.labels,
+        ...params.testRunId === void 0 ? {} : { testRunId: params.testRunId }
+      }
+    );
+    return response.labels;
+  }
+  async saveOne(update, testRunId) {
+    const [outcome] = await this.save({ labels: [update], testRunId });
+    return outcome;
+  }
+};
+
 // src/langgraph.ts
 init_constants();
 init_http();
@@ -6301,6 +6360,8 @@ var Bitfab = class {
       timeout: this.timeout
     });
     this.datasets = new DatasetsClient(this.httpClient);
+    this.traces = new TracesClient(this.httpClient);
+    this.labels = new LabelsClient(this.httpClient);
   }
   /**
    * Decorate a class method as an automatically expanded trace root.
@@ -8202,10 +8263,12 @@ function resolveTraceFunctionKey(registration) {
   DatasetsClient,
   DbBranchReplayError,
   HttpClient,
+  LabelsClient,
   MixedTracingError,
   NO_MOCK_OVERRIDE,
   ReplayError,
   SUPPORTED_PROVIDERS,
+  TracesClient,
   __version__,
   defineReplayRegistry,
   finalizers,