@bitfab/sdk 0.44.0 → 0.45.0

package/dist/index.d.cts

@@ -1,35 +1,3 @@
-/**
- * BAML execution utilities for the Bitfab TypeScript SDK.
- * This module provides functions to execute BAML prompts dynamically on the client side.
- */
-/**
- * Provider definition from the server.
- */
-interface ProviderDefinition {
-    provider: string;
-    apiKeyEnv: string;
-    models: Array<{
-        model: string;
-        description: string;
-    }>;
-}
-/**
- * Result of a BAML function execution with raw collector data.
- */
-interface BamlExecutionResult {
-    /** The parsed result of the function */
-    result: unknown;
-    /** Raw collector data for the server to parse */
-    rawCollector: Record<string, unknown> | null;
-}
-/**
- * Type for allowed environment variables.
- * Only OPENAI_API_KEY is currently supported.
- */
-type AllowedEnvVars = {
-    OPENAI_API_KEY?: string;
-};
-
 /**
  * Per-trace database snapshot ref capture.
  *
@@ -647,6 +615,32 @@ interface TokenUsage {
     cached: number | null;
     total: number | null;
 }
+interface TraceOutlineSpanError {
+    source: string;
+    error: string;
+    step?: string;
+}
+interface TraceOutlineSpan {
+    spanId: string;
+    name: string | null;
+    type: string;
+    traceFunctionKey: string | null;
+    durationMs: number | null;
+    tokens: TokenUsage | null;
+    model: string | null;
+    errors: TraceOutlineSpanError[] | null;
+    mocked: boolean;
+    children: TraceOutlineSpan[];
+}
+interface TraceOutline {
+    traceId: string;
+    name: string | null;
+    status: string;
+    traceFunctionKey: string | null;
+    durationMs: number | null;
+    spanCount: number;
+    spans: TraceOutlineSpan[];
+}
 /**
  * Describes a single file edited as part of a code change.
  *
@@ -746,6 +740,8 @@ interface CompleteReplayResponse {
      * `ReplayItem.tokens`. Absent on servers that predate this field.
      */
     tokens?: Record<string, TokenUsage | null>;
+    traceOutlines?: Record<string, TraceOutline>;
+    originalTraceOutlines?: Record<string, TraceOutline>;
     /**
      * Number of traces the server has persisted for this test run at
      * completion time. Lets the SDK distinguish "uploads failed" from
@@ -775,6 +771,86 @@ interface SpanTreeResponse {
     root: SpanTreeNode;
 }
 
+type TraceTargetOccurrence = "first" | "last" | number;
+type TraceTarget = {
+    kind: "output";
+} | {
+    kind: "span";
+    name: string;
+    occurrence?: TraceTargetOccurrence;
+};
+type TraceAssertionSource = "human" | "agent";
+type TraceAssertion = {
+    id: string;
+    traceId: string;
+    assertion: string;
+    passCriteria: string | null;
+    failCriteria: string | null;
+    targetOnEvaluatedTrace: TraceTarget | null;
+    source: TraceAssertionSource;
+    createdAt: string;
+    updatedAt: string;
+};
+type SaveAssertion = {
+    id?: string;
+    assertion: string;
+    passCriteria?: string | null;
+    failCriteria?: string | null;
+    targetOnEvaluatedTrace?: TraceTarget | null;
+};
+type TraceAssertionsResult = {
+    assertions: TraceAssertion[];
+    inheritedFrom: string | null;
+};
+type SaveAssertionsParams = {
+    traceId: string;
+    assertions: SaveAssertion[];
+    source?: TraceAssertionSource;
+};
+type ArchiveAssertionsParams = {
+    traceId: string;
+    assertionIds: string[];
+};
+declare class TracesClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    getAssertions(traceId: string): Promise<TraceAssertionsResult>;
+    saveAssertions(params: SaveAssertionsParams): Promise<TraceAssertion[]>;
+    archiveAssertions(params: ArchiveAssertionsParams): Promise<string[]>;
+}
+
+/**
+ * BAML execution utilities for the Bitfab TypeScript SDK.
+ * This module provides functions to execute BAML prompts dynamically on the client side.
+ */
+/**
+ * Provider definition from the server.
+ */
+interface ProviderDefinition {
+    provider: string;
+    apiKeyEnv: string;
+    models: Array<{
+        model: string;
+        description: string;
+    }>;
+}
+/**
+ * Result of a BAML function execution with raw collector data.
+ */
+interface BamlExecutionResult {
+    /** The parsed result of the function */
+    result: unknown;
+    /** Raw collector data for the server to parse */
+    rawCollector: Record<string, unknown> | null;
+}
+/**
+ * Type for allowed environment variables.
+ * Only OPENAI_API_KEY is currently supported.
+ */
+type AllowedEnvVars = {
+    OPENAI_API_KEY?: string;
+};
+
 /**
  * Claude Agent SDK handler for Bitfab tracing.
  *
@@ -1181,6 +1257,43 @@ declare class DatasetsClient {
     getGraderRerun(datasetId: string, runId?: string): Promise<GraderRerun | null>;
 }
 
+type LabelConfidence = "VeryLow" | "Low" | "Medium" | "High" | "VeryHigh";
+type LabelAction = "set" | "archived" | "no-active-label" | "skipped";
+type LabelTarget = {
+    traceId: string;
+    originalTraceId?: never;
+    attempt?: never;
+} | {
+    originalTraceId: string;
+    attempt?: number;
+    traceId?: never;
+};
+type LabelVerdict = {
+    label: boolean;
+    annotation: string;
+    confidence?: LabelConfidence;
+} | {
+    skip: true;
+} | {
+    archive: true;
+};
+type LabelUpdate = LabelTarget & LabelVerdict;
+type LabelOutcome = {
+    key: string;
+    traceId: string;
+    action: LabelAction;
+};
+type SaveLabelsParams = {
+    labels: LabelUpdate[];
+    testRunId?: string;
+};
+declare class LabelsClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    save(params: SaveLabelsParams): Promise<LabelOutcome[]>;
+    saveOne(update: LabelUpdate, testRunId?: string): Promise<LabelOutcome>;
+}
+
 /**
  * LangGraph/LangChain callback handler for Bitfab tracing.
  *
@@ -1693,6 +1806,8 @@ interface ReplayProgressItem {
     model?: string | null;
     dbSnapshotRef?: DbSnapshotRef | null;
     dbBranchTimings?: DbBranchTimings | null;
+    traceOutline?: TraceOutline | null;
+    originalTraceOutline?: TraceOutline | null;
 }
 /**
  * @deprecated Use {@link ReplayItemFinishProgress}. This legacy shape also
@@ -1844,6 +1959,8 @@ interface ReplayItem<T> {
      * no snapshot ref, or against a server that predates timings.
      */
     dbBranchTimings: DbBranchTimings | null;
+    traceOutline: TraceOutline | null;
+    originalTraceOutline: TraceOutline | null;
 }
 
 interface ReplayResult<T> {
@@ -2437,6 +2554,8 @@ declare class Bitfab {
     private readonly httpClient;
     /** Dataset operations for the authenticated organization. */
     readonly datasets: DatasetsClient;
+    readonly traces: TracesClient;
+    readonly labels: LabelsClient;
     private readonly bamlClient;
     private readonly dbSnapshot;
     private readonly autoTracePolicyRefreshes;
@@ -3082,7 +3201,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.44.0";
+declare const __version__ = "0.45.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -3212,4 +3331,4 @@ declare function seedFromRegistry(registry: ReplayRegistry, pipeline: string, ca
     run?: boolean;
 }): Promise<SeedResult>;
 
-export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type Dataset, type DatasetGraderRef, type DatasetTraceIds, DatasetsClient, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, type GraderRerun, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, HttpClient, type LangGraphIntegrationOptions, type ListDatasetsParams, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, SUPPORTED_PROVIDERS, type SaveDatasetParams, type SaveDatasetResult, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, type TokenUsage, type TraceIngestionType, type TraceResponse, type TracingProcessor, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, seedFromRegistry, serializeReplayResult };
+export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, type ArchiveAssertionsParams, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type Dataset, type DatasetGraderRef, type DatasetTraceIds, DatasetsClient, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, type GraderRerun, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, HttpClient, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionsParams, type SaveDatasetParams, type SaveDatasetResult, type SaveLabelsParams, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceIngestionType, type TraceOutline, type TraceOutlineSpan, type TraceOutlineSpanError, type TraceResponse, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, seedFromRegistry, serializeReplayResult };

package/dist/index.d.ts

@@ -1,35 +1,3 @@
-/**
- * BAML execution utilities for the Bitfab TypeScript SDK.
- * This module provides functions to execute BAML prompts dynamically on the client side.
- */
-/**
- * Provider definition from the server.
- */
-interface ProviderDefinition {
-    provider: string;
-    apiKeyEnv: string;
-    models: Array<{
-        model: string;
-        description: string;
-    }>;
-}
-/**
- * Result of a BAML function execution with raw collector data.
- */
-interface BamlExecutionResult {
-    /** The parsed result of the function */
-    result: unknown;
-    /** Raw collector data for the server to parse */
-    rawCollector: Record<string, unknown> | null;
-}
-/**
- * Type for allowed environment variables.
- * Only OPENAI_API_KEY is currently supported.
- */
-type AllowedEnvVars = {
-    OPENAI_API_KEY?: string;
-};
-
 /**
  * Per-trace database snapshot ref capture.
  *
@@ -647,6 +615,32 @@ interface TokenUsage {
     cached: number | null;
     total: number | null;
 }
+interface TraceOutlineSpanError {
+    source: string;
+    error: string;
+    step?: string;
+}
+interface TraceOutlineSpan {
+    spanId: string;
+    name: string | null;
+    type: string;
+    traceFunctionKey: string | null;
+    durationMs: number | null;
+    tokens: TokenUsage | null;
+    model: string | null;
+    errors: TraceOutlineSpanError[] | null;
+    mocked: boolean;
+    children: TraceOutlineSpan[];
+}
+interface TraceOutline {
+    traceId: string;
+    name: string | null;
+    status: string;
+    traceFunctionKey: string | null;
+    durationMs: number | null;
+    spanCount: number;
+    spans: TraceOutlineSpan[];
+}
 /**
  * Describes a single file edited as part of a code change.
  *
@@ -746,6 +740,8 @@ interface CompleteReplayResponse {
      * `ReplayItem.tokens`. Absent on servers that predate this field.
      */
     tokens?: Record<string, TokenUsage | null>;
+    traceOutlines?: Record<string, TraceOutline>;
+    originalTraceOutlines?: Record<string, TraceOutline>;
     /**
      * Number of traces the server has persisted for this test run at
      * completion time. Lets the SDK distinguish "uploads failed" from
@@ -775,6 +771,86 @@ interface SpanTreeResponse {
     root: SpanTreeNode;
 }
 
+type TraceTargetOccurrence = "first" | "last" | number;
+type TraceTarget = {
+    kind: "output";
+} | {
+    kind: "span";
+    name: string;
+    occurrence?: TraceTargetOccurrence;
+};
+type TraceAssertionSource = "human" | "agent";
+type TraceAssertion = {
+    id: string;
+    traceId: string;
+    assertion: string;
+    passCriteria: string | null;
+    failCriteria: string | null;
+    targetOnEvaluatedTrace: TraceTarget | null;
+    source: TraceAssertionSource;
+    createdAt: string;
+    updatedAt: string;
+};
+type SaveAssertion = {
+    id?: string;
+    assertion: string;
+    passCriteria?: string | null;
+    failCriteria?: string | null;
+    targetOnEvaluatedTrace?: TraceTarget | null;
+};
+type TraceAssertionsResult = {
+    assertions: TraceAssertion[];
+    inheritedFrom: string | null;
+};
+type SaveAssertionsParams = {
+    traceId: string;
+    assertions: SaveAssertion[];
+    source?: TraceAssertionSource;
+};
+type ArchiveAssertionsParams = {
+    traceId: string;
+    assertionIds: string[];
+};
+declare class TracesClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    getAssertions(traceId: string): Promise<TraceAssertionsResult>;
+    saveAssertions(params: SaveAssertionsParams): Promise<TraceAssertion[]>;
+    archiveAssertions(params: ArchiveAssertionsParams): Promise<string[]>;
+}
+
+/**
+ * BAML execution utilities for the Bitfab TypeScript SDK.
+ * This module provides functions to execute BAML prompts dynamically on the client side.
+ */
+/**
+ * Provider definition from the server.
+ */
+interface ProviderDefinition {
+    provider: string;
+    apiKeyEnv: string;
+    models: Array<{
+        model: string;
+        description: string;
+    }>;
+}
+/**
+ * Result of a BAML function execution with raw collector data.
+ */
+interface BamlExecutionResult {
+    /** The parsed result of the function */
+    result: unknown;
+    /** Raw collector data for the server to parse */
+    rawCollector: Record<string, unknown> | null;
+}
+/**
+ * Type for allowed environment variables.
+ * Only OPENAI_API_KEY is currently supported.
+ */
+type AllowedEnvVars = {
+    OPENAI_API_KEY?: string;
+};
+
 /**
  * Claude Agent SDK handler for Bitfab tracing.
  *
@@ -1181,6 +1257,43 @@ declare class DatasetsClient {
     getGraderRerun(datasetId: string, runId?: string): Promise<GraderRerun | null>;
 }
 
+type LabelConfidence = "VeryLow" | "Low" | "Medium" | "High" | "VeryHigh";
+type LabelAction = "set" | "archived" | "no-active-label" | "skipped";
+type LabelTarget = {
+    traceId: string;
+    originalTraceId?: never;
+    attempt?: never;
+} | {
+    originalTraceId: string;
+    attempt?: number;
+    traceId?: never;
+};
+type LabelVerdict = {
+    label: boolean;
+    annotation: string;
+    confidence?: LabelConfidence;
+} | {
+    skip: true;
+} | {
+    archive: true;
+};
+type LabelUpdate = LabelTarget & LabelVerdict;
+type LabelOutcome = {
+    key: string;
+    traceId: string;
+    action: LabelAction;
+};
+type SaveLabelsParams = {
+    labels: LabelUpdate[];
+    testRunId?: string;
+};
+declare class LabelsClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    save(params: SaveLabelsParams): Promise<LabelOutcome[]>;
+    saveOne(update: LabelUpdate, testRunId?: string): Promise<LabelOutcome>;
+}
+
 /**
  * LangGraph/LangChain callback handler for Bitfab tracing.
  *
@@ -1693,6 +1806,8 @@ interface ReplayProgressItem {
     model?: string | null;
     dbSnapshotRef?: DbSnapshotRef | null;
     dbBranchTimings?: DbBranchTimings | null;
+    traceOutline?: TraceOutline | null;
+    originalTraceOutline?: TraceOutline | null;
 }
 /**
  * @deprecated Use {@link ReplayItemFinishProgress}. This legacy shape also
@@ -1844,6 +1959,8 @@ interface ReplayItem<T> {
      * no snapshot ref, or against a server that predates timings.
      */
     dbBranchTimings: DbBranchTimings | null;
+    traceOutline: TraceOutline | null;
+    originalTraceOutline: TraceOutline | null;
 }
 
 interface ReplayResult<T> {
@@ -2437,6 +2554,8 @@ declare class Bitfab {
     private readonly httpClient;
     /** Dataset operations for the authenticated organization. */
     readonly datasets: DatasetsClient;
+    readonly traces: TracesClient;
+    readonly labels: LabelsClient;
     private readonly bamlClient;
     private readonly dbSnapshot;
     private readonly autoTracePolicyRefreshes;
@@ -3082,7 +3201,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.44.0";
+declare const __version__ = "0.45.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -3212,4 +3331,4 @@ declare function seedFromRegistry(registry: ReplayRegistry, pipeline: string, ca
     run?: boolean;
 }): Promise<SeedResult>;
 
-export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type Dataset, type DatasetGraderRef, type DatasetTraceIds, DatasetsClient, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, type GraderRerun, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, HttpClient, type LangGraphIntegrationOptions, type ListDatasetsParams, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, SUPPORTED_PROVIDERS, type SaveDatasetParams, type SaveDatasetResult, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, type TokenUsage, type TraceIngestionType, type TraceResponse, type TracingProcessor, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, seedFromRegistry, serializeReplayResult };
+export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, type ArchiveAssertionsParams, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type Dataset, type DatasetGraderRef, type DatasetTraceIds, DatasetsClient, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, type GraderRerun, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, HttpClient, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionsParams, type SaveDatasetParams, type SaveDatasetResult, type SaveLabelsParams, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceIngestionType, type TraceOutline, type TraceOutlineSpan, type TraceOutlineSpanError, type TraceResponse, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, seedFromRegistry, serializeReplayResult };

package/dist/index.js

@@ -8,14 +8,16 @@ import {
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   DatasetsClient,
+  LabelsClient,
   SUPPORTED_PROVIDERS,
+  TracesClient,
   defineReplayRegistry,
   finalizers,
   getCurrentReplayBranch,
   getCurrentSpan,
   getCurrentTrace,
   seedFromRegistry
-} from "./chunk-XYMG2FH7.js";
+} from "./chunk-GN6E2IBU.js";
 import "./chunk-ZUD7OFYB.js";
 import {
   BITFAB_PROGRESS_PREFIX,
@@ -24,7 +26,7 @@ import {
   ReplayError,
   reportReplayProgress,
   serializeReplayResult
-} from "./chunk-3WIN3VU3.js";
+} from "./chunk-H4WJUUZH.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -32,7 +34,7 @@ import {
   MixedTracingError,
   __version__,
   flushTraces
-} from "./chunk-6JTVCM2L.js";
+} from "./chunk-TCOD5O2B.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BITFAB_PROGRESS_PREFIX,
@@ -50,10 +52,12 @@ export {
   DatasetsClient,
   DbBranchReplayError,
   HttpClient,
+  LabelsClient,
   MixedTracingError,
   NO_MOCK_OVERRIDE,
   ReplayError,
   SUPPORTED_PROVIDERS,
+  TracesClient,
   __version__,
   defineReplayRegistry,
   finalizers,

package/dist/node.cjs

@@ -88,7 +88,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";
   }
 });
@@ -2726,7 +2726,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 {
@@ -2810,7 +2812,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) {
@@ -3065,7 +3069,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 {
@@ -3145,7 +3151,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.` : "";
@@ -3268,10 +3276,12 @@ __export(node_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,
@@ -3292,6 +3302,33 @@ registerAsyncLocalStorageClass(
   import_node_async_hooks.AsyncLocalStorage
 );
 
+// 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();
@@ -4542,6 +4579,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();
@@ -6315,6 +6374,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.
@@ -8220,10 +8281,12 @@ assertAsyncStorageRegistered();
   DatasetsClient,
   DbBranchReplayError,
   HttpClient,
+  LabelsClient,
   MixedTracingError,
   NO_MOCK_OVERRIDE,
   ReplayError,
   SUPPORTED_PROVIDERS,
+  TracesClient,
   __version__,
   defineReplayRegistry,
   finalizers,