@bitfab/sdk 0.64.29 → 0.65.0

package/dist/index.d.cts

@@ -22,6 +22,38 @@ type ApprovalFields = {
     approvedAt: string | null;
 };
 
+/**
+ * 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;
+};
+
 /** A node is identified by its workflow key and span name, never its display label. */
 interface ReplayNodeIdentity {
     traceFunctionKey: string;
@@ -964,66 +996,50 @@ interface SpanTreeResponse {
     root: SpanTreeNode;
 }
 
-type AssertionCategorySummary = {
+type CategorySummary = {
     id: string;
     title: string;
     description: string;
 };
-type AssertionCategory = AssertionCategorySummary & {
+type Category = CategorySummary & {
     organizationId: string;
     createdAt: string;
     updatedAt: string;
 };
-type SaveAssertionCategoryParams = {
+type SaveCategoryParams = {
     id?: string;
     title: string;
     description?: string;
 };
-/** Organization-scoped groupings for assertions. */
-declare class AssertionCategoriesClient {
+/** @deprecated Use {@link CategorySummary}. Same shape. */
+type AssertionCategorySummary = CategorySummary;
+/** @deprecated Use {@link Category}. Same shape. */
+type AssertionCategory = Category;
+/** @deprecated Use {@link SaveCategoryParams}. Same shape. */
+type SaveAssertionCategoryParams = SaveCategoryParams;
+/** Organization-scoped groupings for assertions and scenarios. */
+declare class CategoriesClient {
     private readonly httpClient;
     constructor(httpClient: HttpClient);
     /** Create a category, or update it by ID. Omitted descriptions are preserved. */
-    save(params: SaveAssertionCategoryParams): Promise<AssertionCategory>;
+    save(params: SaveCategoryParams): Promise<Category>;
     /** Read one category in the API key's organization. */
-    get(id: string): Promise<AssertionCategory>;
+    get(id: string): Promise<Category>;
     /** List the organization's categories ordered by title. */
-    list(): Promise<AssertionCategory[]>;
+    list(): Promise<Category[]>;
     /** Delete a category and clear its assignments, preserving assertions and verdicts. */
-    delete(id: string): Promise<AssertionCategory>;
-}
-
-/**
- * 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;
-    }>;
+    delete(id: string): Promise<Category>;
 }
 /**
- * Result of a BAML function execution with raw collector data.
+ * @deprecated Use {@link CategoriesClient}. Same class.
+ * @alias
  */
-interface BamlExecutionResult {
-    /** The parsed result of the function */
-    result: unknown;
-    /** Raw collector data for the server to parse */
-    rawCollector: Record<string, unknown> | null;
-}
+declare const AssertionCategoriesClient: typeof CategoriesClient;
 /**
- * Type for allowed environment variables.
- * Only OPENAI_API_KEY is currently supported.
+ * @deprecated Use {@link CategoriesClient}. Same class.
+ * @alias
  */
-type AllowedEnvVars = {
-    OPENAI_API_KEY?: string;
-};
+type AssertionCategoriesClient = CategoriesClient;
 
 /**
  * Claude Agent SDK handler for Bitfab tracing.
@@ -1528,7 +1544,7 @@ type ExperimentTotals = {
     skipped: number;
 };
 type ExperimentCategoryTally = {
-    category: AssertionCategorySummary;
+    category: CategorySummary;
     labels: ExperimentTally;
 };
 type ExperimentJitter = {
@@ -1610,8 +1626,10 @@ type TraceAssertion = {
     traceId: string;
     assertion: string;
     humanNote: string | null;
+    category_id: string | null;
+    /** @deprecated Use {@link category_id}. Same value. */
     category_assertion_id: string | null;
-    category: AssertionCategorySummary | null;
+    category: CategorySummary | null;
     passCriteria: string | null;
     failCriteria: string | null;
     targetOnEvaluatedTrace: TraceTarget | null;
@@ -1640,6 +1658,8 @@ type AssertionLabelEvidence = {
 };
 type SaveAssertion = {
     /** Omit to preserve the category, or pass null to clear it. */
+    category_id?: string | null;
+    /** @deprecated Use {@link category_id}. Ignored when `category_id` is also set. */
     category_assertion_id?: string | null;
     id?: string;
     assertion: string;
@@ -1653,7 +1673,7 @@ type SaveAssertion = {
      */
     justification?: Justification | null;
     /**
-     * Why `category_assertion_id` is the right category. Changing the category
+     * Why `category_id` is the right category. Changing the category
      * clears it. Omit on an edit to preserve it.
      */
     categoryJustification?: Justification | null;
@@ -1757,6 +1777,8 @@ type Scenario = {
     dataSummary: string;
     /** What the run actually did, or null when none has been saved. */
     actualOutcome: ScenarioActualOutcome | null;
+    /** Every category the scenario is filed under, sorted by title. */
+    categories: CategorySummary[];
 };
 type SaveScenarioHistoryEntry = Omit<ScenarioHistoryEntry, "source"> & {
     source?: ScenarioSource | null;
@@ -1786,6 +1808,12 @@ type SaveScenarioParams = {
      * Sending only this field, null included, is a valid save.
      */
     actualOutcome?: SaveScenarioActualOutcome | null;
+    /**
+     * Categories to file the scenario under. Replaces the whole set. Leave it
+     * out to keep the current set, or pass an empty array to clear it. An id
+     * outside the organization fails the save.
+     */
+    categoryIds?: string[];
 } & ({
     data?: SaveScenarioData;
     appendData?: never;
@@ -1804,6 +1832,8 @@ type SaveScenarioParams = {
     appendData: ScenarioDataAppend;
 } | {
     actualOutcome: SaveScenarioActualOutcome | null;
+} | {
+    categoryIds: string[];
 });
 type AssertionGenerationStatus = "running" | "completed" | "failed";
 type AssertionGeneration = {
@@ -3537,7 +3567,10 @@ declare class Bitfab {
     private readonly httpClient;
     /** Dataset operations for the authenticated organization. */
     readonly datasets: DatasetsClient;
-    readonly assertionCategories: AssertionCategoriesClient;
+    /** Organization-scoped groupings for assertions and scenarios. */
+    readonly categories: CategoriesClient;
+    /** @deprecated Use {@link categories}. Same client. */
+    readonly assertionCategories: CategoriesClient;
     readonly organizationMembers: OrganizationMembersClient;
     readonly traces: TracesClient;
     readonly spans: SpansClient;
@@ -4251,7 +4284,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.64.29";
+declare const __version__ = "0.65.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -4390,4 +4423,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, type ApprovalFields, type ApprovalState, type Approver, type ArchiveAssertionsParams, type ArchiveLabelTarget, AssertionCategoriesClient, type AssertionCategory, type AssertionCategorySummary, type AssertionEvidenceParameter, type AssertionGeneration, AssertionGenerationError, type AssertionGenerationStatus, AssertionGenerationTimeoutError, type AssertionLabelEvidence, type AssertionVerdict, type Assignee, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabRivetIntegration, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CommitRef, 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 Experiment, type ExperimentCategoryTally, type ExperimentCodeChangeTotals, type ExperimentJitter, type ExperimentPage, type ExperimentRollup, type ExperimentRollupResult, type ExperimentRunBy, type ExperimentStatus, type ExperimentTally, type ExperimentTotals, ExperimentsClient, type GenerateAssertionsOptions, type GetGraderLabelsParams, type GraderLabel, type GraderLabelOutcome, type GraderLabelSource, type GraderLabelUpdate, type GraderRerun, GraderRerunError, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, GraderRerunTimeoutError, GradersClient, HttpClient, type HumanLabelOutcome, type HumanLabelUpdate, type Justification, type JustificationSpan, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelSource, type LabelStatus, type LabelTarget, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, type ListExperimentsParams, type ListedSpan, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type OrganizationMember, OrganizationMembersClient, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayExperiment, type ReplayInterrupt, type ReplayInterruptSignal, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayNodeIdentity, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, type ReseedOptions, type ReseedRegistryResult, type ReseedResult, type RivetContextLike, type RivetEffectType, type RivetIntegrationOptions, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionCategoryParams, type SaveAssertionsAllParams, type SaveAssertionsParams, type SaveDatasetParams, type SaveDatasetResult, type SaveScenarioActualOutcome, type SaveScenarioData, type SaveScenarioHistoryEntry, type SaveScenarioParams, type Scenario, type ScenarioAction, type ScenarioActualOutcome, type ScenarioData, type ScenarioDataAppend, type ScenarioDataOrigin, type ScenarioHistoryEntry, type ScenarioSource, type ScenarioTrigger, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SelectiveReplayDecision, type SelectiveReplayOptions, type SelectiveReplayReport, type SpanListField, type SpanListParams, type SpanListResult, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, SpansClient, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceAssertionsUpdate, type TraceIngestionType, type TraceLabels, type TraceResponse, type TraceSearchFilters, type TraceSearchParams, type TraceSearchResult, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type UpdateDatasetParams, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type Verdict, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, reseedFromRegistry, seedFromRegistry, serializeReplayResult };
+export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, type ApprovalFields, type ApprovalState, type Approver, type ArchiveAssertionsParams, type ArchiveLabelTarget, AssertionCategoriesClient, type AssertionCategory, type AssertionCategorySummary, type AssertionEvidenceParameter, type AssertionGeneration, AssertionGenerationError, type AssertionGenerationStatus, AssertionGenerationTimeoutError, type AssertionLabelEvidence, type AssertionVerdict, type Assignee, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabRivetIntegration, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, CategoriesClient, type Category, type CategorySummary, type CodeChangeFile, type CommitRef, 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 Experiment, type ExperimentCategoryTally, type ExperimentCodeChangeTotals, type ExperimentJitter, type ExperimentPage, type ExperimentRollup, type ExperimentRollupResult, type ExperimentRunBy, type ExperimentStatus, type ExperimentTally, type ExperimentTotals, ExperimentsClient, type GenerateAssertionsOptions, type GetGraderLabelsParams, type GraderLabel, type GraderLabelOutcome, type GraderLabelSource, type GraderLabelUpdate, type GraderRerun, GraderRerunError, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, GraderRerunTimeoutError, GradersClient, HttpClient, type HumanLabelOutcome, type HumanLabelUpdate, type Justification, type JustificationSpan, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelSource, type LabelStatus, type LabelTarget, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, type ListExperimentsParams, type ListedSpan, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type OrganizationMember, OrganizationMembersClient, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayExperiment, type ReplayInterrupt, type ReplayInterruptSignal, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayNodeIdentity, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, type ReseedOptions, type ReseedRegistryResult, type ReseedResult, type RivetContextLike, type RivetEffectType, type RivetIntegrationOptions, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionCategoryParams, type SaveAssertionsAllParams, type SaveAssertionsParams, type SaveCategoryParams, type SaveDatasetParams, type SaveDatasetResult, type SaveScenarioActualOutcome, type SaveScenarioData, type SaveScenarioHistoryEntry, type SaveScenarioParams, type Scenario, type ScenarioAction, type ScenarioActualOutcome, type ScenarioData, type ScenarioDataAppend, type ScenarioDataOrigin, type ScenarioHistoryEntry, type ScenarioSource, type ScenarioTrigger, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SelectiveReplayDecision, type SelectiveReplayOptions, type SelectiveReplayReport, type SpanListField, type SpanListParams, type SpanListResult, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, SpansClient, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceAssertionsUpdate, type TraceIngestionType, type TraceLabels, type TraceResponse, type TraceSearchFilters, type TraceSearchParams, type TraceSearchResult, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type UpdateDatasetParams, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type Verdict, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, reseedFromRegistry, seedFromRegistry, serializeReplayResult };

package/dist/index.d.ts

@@ -22,6 +22,38 @@ type ApprovalFields = {
     approvedAt: string | null;
 };
 
+/**
+ * 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;
+};
+
 /** A node is identified by its workflow key and span name, never its display label. */
 interface ReplayNodeIdentity {
     traceFunctionKey: string;
@@ -964,66 +996,50 @@ interface SpanTreeResponse {
     root: SpanTreeNode;
 }
 
-type AssertionCategorySummary = {
+type CategorySummary = {
     id: string;
     title: string;
     description: string;
 };
-type AssertionCategory = AssertionCategorySummary & {
+type Category = CategorySummary & {
     organizationId: string;
     createdAt: string;
     updatedAt: string;
 };
-type SaveAssertionCategoryParams = {
+type SaveCategoryParams = {
     id?: string;
     title: string;
     description?: string;
 };
-/** Organization-scoped groupings for assertions. */
-declare class AssertionCategoriesClient {
+/** @deprecated Use {@link CategorySummary}. Same shape. */
+type AssertionCategorySummary = CategorySummary;
+/** @deprecated Use {@link Category}. Same shape. */
+type AssertionCategory = Category;
+/** @deprecated Use {@link SaveCategoryParams}. Same shape. */
+type SaveAssertionCategoryParams = SaveCategoryParams;
+/** Organization-scoped groupings for assertions and scenarios. */
+declare class CategoriesClient {
     private readonly httpClient;
     constructor(httpClient: HttpClient);
     /** Create a category, or update it by ID. Omitted descriptions are preserved. */
-    save(params: SaveAssertionCategoryParams): Promise<AssertionCategory>;
+    save(params: SaveCategoryParams): Promise<Category>;
     /** Read one category in the API key's organization. */
-    get(id: string): Promise<AssertionCategory>;
+    get(id: string): Promise<Category>;
     /** List the organization's categories ordered by title. */
-    list(): Promise<AssertionCategory[]>;
+    list(): Promise<Category[]>;
     /** Delete a category and clear its assignments, preserving assertions and verdicts. */
-    delete(id: string): Promise<AssertionCategory>;
-}
-
-/**
- * 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;
-    }>;
+    delete(id: string): Promise<Category>;
 }
 /**
- * Result of a BAML function execution with raw collector data.
+ * @deprecated Use {@link CategoriesClient}. Same class.
+ * @alias
  */
-interface BamlExecutionResult {
-    /** The parsed result of the function */
-    result: unknown;
-    /** Raw collector data for the server to parse */
-    rawCollector: Record<string, unknown> | null;
-}
+declare const AssertionCategoriesClient: typeof CategoriesClient;
 /**
- * Type for allowed environment variables.
- * Only OPENAI_API_KEY is currently supported.
+ * @deprecated Use {@link CategoriesClient}. Same class.
+ * @alias
  */
-type AllowedEnvVars = {
-    OPENAI_API_KEY?: string;
-};
+type AssertionCategoriesClient = CategoriesClient;
 
 /**
  * Claude Agent SDK handler for Bitfab tracing.
@@ -1528,7 +1544,7 @@ type ExperimentTotals = {
     skipped: number;
 };
 type ExperimentCategoryTally = {
-    category: AssertionCategorySummary;
+    category: CategorySummary;
     labels: ExperimentTally;
 };
 type ExperimentJitter = {
@@ -1610,8 +1626,10 @@ type TraceAssertion = {
     traceId: string;
     assertion: string;
     humanNote: string | null;
+    category_id: string | null;
+    /** @deprecated Use {@link category_id}. Same value. */
     category_assertion_id: string | null;
-    category: AssertionCategorySummary | null;
+    category: CategorySummary | null;
     passCriteria: string | null;
     failCriteria: string | null;
     targetOnEvaluatedTrace: TraceTarget | null;
@@ -1640,6 +1658,8 @@ type AssertionLabelEvidence = {
 };
 type SaveAssertion = {
     /** Omit to preserve the category, or pass null to clear it. */
+    category_id?: string | null;
+    /** @deprecated Use {@link category_id}. Ignored when `category_id` is also set. */
     category_assertion_id?: string | null;
     id?: string;
     assertion: string;
@@ -1653,7 +1673,7 @@ type SaveAssertion = {
      */
     justification?: Justification | null;
     /**
-     * Why `category_assertion_id` is the right category. Changing the category
+     * Why `category_id` is the right category. Changing the category
      * clears it. Omit on an edit to preserve it.
      */
     categoryJustification?: Justification | null;
@@ -1757,6 +1777,8 @@ type Scenario = {
     dataSummary: string;
     /** What the run actually did, or null when none has been saved. */
     actualOutcome: ScenarioActualOutcome | null;
+    /** Every category the scenario is filed under, sorted by title. */
+    categories: CategorySummary[];
 };
 type SaveScenarioHistoryEntry = Omit<ScenarioHistoryEntry, "source"> & {
     source?: ScenarioSource | null;
@@ -1786,6 +1808,12 @@ type SaveScenarioParams = {
      * Sending only this field, null included, is a valid save.
      */
     actualOutcome?: SaveScenarioActualOutcome | null;
+    /**
+     * Categories to file the scenario under. Replaces the whole set. Leave it
+     * out to keep the current set, or pass an empty array to clear it. An id
+     * outside the organization fails the save.
+     */
+    categoryIds?: string[];
 } & ({
     data?: SaveScenarioData;
     appendData?: never;
@@ -1804,6 +1832,8 @@ type SaveScenarioParams = {
     appendData: ScenarioDataAppend;
 } | {
     actualOutcome: SaveScenarioActualOutcome | null;
+} | {
+    categoryIds: string[];
 });
 type AssertionGenerationStatus = "running" | "completed" | "failed";
 type AssertionGeneration = {
@@ -3537,7 +3567,10 @@ declare class Bitfab {
     private readonly httpClient;
     /** Dataset operations for the authenticated organization. */
     readonly datasets: DatasetsClient;
-    readonly assertionCategories: AssertionCategoriesClient;
+    /** Organization-scoped groupings for assertions and scenarios. */
+    readonly categories: CategoriesClient;
+    /** @deprecated Use {@link categories}. Same client. */
+    readonly assertionCategories: CategoriesClient;
     readonly organizationMembers: OrganizationMembersClient;
     readonly traces: TracesClient;
     readonly spans: SpansClient;
@@ -4251,7 +4284,7 @@ interface CommitRef {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.64.29";
+declare const __version__ = "0.65.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -4390,4 +4423,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, type ApprovalFields, type ApprovalState, type Approver, type ArchiveAssertionsParams, type ArchiveLabelTarget, AssertionCategoriesClient, type AssertionCategory, type AssertionCategorySummary, type AssertionEvidenceParameter, type AssertionGeneration, AssertionGenerationError, type AssertionGenerationStatus, AssertionGenerationTimeoutError, type AssertionLabelEvidence, type AssertionVerdict, type Assignee, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabRivetIntegration, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CommitRef, 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 Experiment, type ExperimentCategoryTally, type ExperimentCodeChangeTotals, type ExperimentJitter, type ExperimentPage, type ExperimentRollup, type ExperimentRollupResult, type ExperimentRunBy, type ExperimentStatus, type ExperimentTally, type ExperimentTotals, ExperimentsClient, type GenerateAssertionsOptions, type GetGraderLabelsParams, type GraderLabel, type GraderLabelOutcome, type GraderLabelSource, type GraderLabelUpdate, type GraderRerun, GraderRerunError, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, GraderRerunTimeoutError, GradersClient, HttpClient, type HumanLabelOutcome, type HumanLabelUpdate, type Justification, type JustificationSpan, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelSource, type LabelStatus, type LabelTarget, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, type ListExperimentsParams, type ListedSpan, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type OrganizationMember, OrganizationMembersClient, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayExperiment, type ReplayInterrupt, type ReplayInterruptSignal, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayNodeIdentity, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, type ReseedOptions, type ReseedRegistryResult, type ReseedResult, type RivetContextLike, type RivetEffectType, type RivetIntegrationOptions, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionCategoryParams, type SaveAssertionsAllParams, type SaveAssertionsParams, type SaveDatasetParams, type SaveDatasetResult, type SaveScenarioActualOutcome, type SaveScenarioData, type SaveScenarioHistoryEntry, type SaveScenarioParams, type Scenario, type ScenarioAction, type ScenarioActualOutcome, type ScenarioData, type ScenarioDataAppend, type ScenarioDataOrigin, type ScenarioHistoryEntry, type ScenarioSource, type ScenarioTrigger, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SelectiveReplayDecision, type SelectiveReplayOptions, type SelectiveReplayReport, type SpanListField, type SpanListParams, type SpanListResult, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, SpansClient, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceAssertionsUpdate, type TraceIngestionType, type TraceLabels, type TraceResponse, type TraceSearchFilters, type TraceSearchParams, type TraceSearchResult, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type UpdateDatasetParams, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type Verdict, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, reseedFromRegistry, seedFromRegistry, serializeReplayResult };
+export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AddDatasetGradersResult, type AddDatasetTracesResult, type AllowedEnvVars, type ApprovalFields, type ApprovalState, type Approver, type ArchiveAssertionsParams, type ArchiveLabelTarget, AssertionCategoriesClient, type AssertionCategory, type AssertionCategorySummary, type AssertionEvidenceParameter, type AssertionGeneration, AssertionGenerationError, type AssertionGenerationStatus, AssertionGenerationTimeoutError, type AssertionLabelEvidence, type AssertionVerdict, type Assignee, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabRivetIntegration, BitfabVercelAiHandler, type CaptureSurface, type CaptureWhen, type CapturedSpan, CategoriesClient, type Category, type CategorySummary, type CodeChangeFile, type CommitRef, 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 Experiment, type ExperimentCategoryTally, type ExperimentCodeChangeTotals, type ExperimentJitter, type ExperimentPage, type ExperimentRollup, type ExperimentRollupResult, type ExperimentRunBy, type ExperimentStatus, type ExperimentTally, type ExperimentTotals, ExperimentsClient, type GenerateAssertionsOptions, type GetGraderLabelsParams, type GraderLabel, type GraderLabelOutcome, type GraderLabelSource, type GraderLabelUpdate, type GraderRerun, GraderRerunError, type GraderRerunProgress, type GraderRerunResult, type GraderRerunStatus, GraderRerunTimeoutError, GradersClient, HttpClient, type HumanLabelOutcome, type HumanLabelUpdate, type Justification, type JustificationSpan, type LabelAction, type LabelConfidence, type LabelOutcome, type LabelSource, type LabelStatus, type LabelTarget, type LabelUpdate, LabelsClient, type LangGraphIntegrationOptions, type ListDatasetsParams, type ListExperimentsParams, type ListedSpan, MixedTracingError, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type OrganizationMember, OrganizationMembersClient, type ProviderDefinition, type RemoveDatasetGradersResult, type RemoveDatasetTracesResult, ReplayBranch, ReplayError, type ReplayExperiment, type ReplayInterrupt, type ReplayInterruptSignal, type ReplayItem, type ReplayItemFinishProgress, type ReplayItemStartProgress, type ReplayNodeIdentity, type ReplayOptions, type ReplayOptionsFactory, type ReplayProgress, type ReplayProgressItem, type ReplayRegistration, type ReplayRegistry, type ReplayRegistryContext, type ReplayRegistryOptions, type ReplayResult, type RerunGradersOptions, type RerunGradersResult, type ReseedOptions, type ReseedRegistryResult, type ReseedResult, type RivetContextLike, type RivetEffectType, type RivetIntegrationOptions, SUPPORTED_PROVIDERS, type SaveAssertion, type SaveAssertionCategoryParams, type SaveAssertionsAllParams, type SaveAssertionsParams, type SaveCategoryParams, type SaveDatasetParams, type SaveDatasetResult, type SaveScenarioActualOutcome, type SaveScenarioData, type SaveScenarioHistoryEntry, type SaveScenarioParams, type Scenario, type ScenarioAction, type ScenarioActualOutcome, type ScenarioData, type ScenarioDataAppend, type ScenarioDataOrigin, type ScenarioHistoryEntry, type ScenarioSource, type ScenarioTrigger, type SeedCase, type SeedCaseOptions, type SeedResult, type SeedRunOptions, type SelectiveReplayDecision, type SelectiveReplayOptions, type SelectiveReplayReport, type SpanListField, type SpanListParams, type SpanListResult, type SpanLookup, type SpanMethodDecorator, type SpanMethodDecoratorContext, type SpanNodeMeta, type SpanOccurrence, type SpanOptions, type SpanType, SpansClient, type TokenUsage, type TraceAssertion, type TraceAssertionSource, type TraceAssertionsResult, type TraceAssertionsUpdate, type TraceIngestionType, type TraceLabels, type TraceResponse, type TraceSearchFilters, type TraceSearchParams, type TraceSearchResult, type TraceTarget, type TraceTargetOccurrence, TracesClient, type TracingProcessor, type UpdateDatasetParams, type VercelCallParams, type VercelGenerateResult, type VercelStreamResult, type Verdict, type WrapBAMLOptions, type WrappedBamlFn, __version__, defineReplayRegistry, finalizers, flushTraces, getCurrentReplayBranch, getCurrentSpan, getCurrentTrace, reportReplayProgress, reseedFromRegistry, seedFromRegistry, serializeReplayResult };

package/dist/index.js

@@ -11,6 +11,7 @@ import {
   BitfabOpenAITracingProcessor,
   BitfabRivetIntegration,
   BitfabVercelAiHandler,
+  CategoriesClient,
   DatasetsClient,
   ExperimentsClient,
   GraderRerunError,
@@ -28,7 +29,7 @@ import {
   getCurrentTrace,
   reseedFromRegistry,
   seedFromRegistry
-} from "./chunk-BDPDIXLA.js";
+} from "./chunk-SUXEIG4U.js";
 import "./chunk-YWXQM6J3.js";
 import {
   BITFAB_PROGRESS_PREFIX,
@@ -37,7 +38,7 @@ import {
   ReplayError,
   reportReplayProgress,
   serializeReplayResult
-} from "./chunk-UIDS5FUO.js";
+} from "./chunk-JIGT4NC5.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -45,7 +46,7 @@ import {
   MixedTracingError,
   __version__,
   flushTraces
-} from "./chunk-CBF3YILJ.js";
+} from "./chunk-TNIJXUGC.js";
 import "./chunk-MT36GJMW.js";
 export {
   AssertionCategoriesClient,
@@ -63,6 +64,7 @@ export {
   BitfabOpenAITracingProcessor,
   BitfabRivetIntegration,
   BitfabVercelAiHandler,
+  CategoriesClient,
   DEFAULT_SERVICE_URL,
   DatasetsClient,
   DbBranchReplayError,

package/dist/node.cjs

@@ -88,7 +88,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";
   }
 });
@@ -5374,6 +5374,7 @@ __export(node_exports, {
   BitfabOpenAITracingProcessor: () => BitfabOpenAITracingProcessor,
   BitfabRivetIntegration: () => BitfabRivetIntegration,
   BitfabVercelAiHandler: () => BitfabVercelAiHandler,
+  CategoriesClient: () => CategoriesClient,
   DEFAULT_SERVICE_URL: () => DEFAULT_SERVICE_URL,
   DatasetsClient: () => DatasetsClient,
   DbBranchReplayError: () => DbBranchReplayError,
@@ -5411,9 +5412,9 @@ registerAsyncLocalStorageClass(
   import_node_async_hooks.AsyncLocalStorage
 );
 
-// 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;
   }
@@ -5440,6 +5441,7 @@ var AssertionCategoriesClient = class {
     return response.category;
   }
 };
+var AssertionCategoriesClient = CategoriesClient;
 
 // src/claudeAgentSdk.ts
 init_constants();
@@ -9039,6 +9041,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}`;
 }
@@ -9053,9 +9076,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({
@@ -9068,10 +9092,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(
@@ -9083,7 +9110,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;
@@ -10106,7 +10133,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);
@@ -12429,6 +12457,7 @@ assertAsyncStorageRegistered();
   BitfabOpenAITracingProcessor,
   BitfabRivetIntegration,
   BitfabVercelAiHandler,
+  CategoriesClient,
   DEFAULT_SERVICE_URL,
   DatasetsClient,
   DbBranchReplayError,