@bitfab/sdk 0.39.0 → 0.41.0

package/dist/index.d.cts

@@ -540,7 +540,11 @@ declare class HttpClient {
     lookupFunction<T>(name: string): Promise<T>;
     getAutoTracePolicy<T>(traceFunctionKey: string, protocol: string): Promise<T>;
     getTraceSpan(traceId: string, lookup: SpanLookup): Promise<CapturedSpan | null>;
-    private get;
+    /**
+     * GET a JSON endpoint on the service with the client's API key. Throws a
+     * `BitfabError` carrying the status text for any non-2xx response.
+     */
+    get<T>(endpoint: string): Promise<T>;
     /**
      * Queue an internal trace (from local BAML execution via `call()`) onto this
      * client's batching transport. `functionId` moves into the payload because
@@ -574,6 +578,7 @@ declare class HttpClient {
         appendContexts?: Record<string, unknown>[];
         mergeMetadata?: Record<string, unknown>;
         setSessionId?: string;
+        setName?: string;
     }): Promise<void>;
     /**
      * Start a replay session by fetching historical traces.
@@ -1034,6 +1039,141 @@ declare class BitfabVercelAiHandler {
     get middleware(): BitfabLanguageModelMiddleware;
 }
 
+interface DatasetGraderRef {
+    id: string;
+    name: string | null;
+}
+interface Dataset {
+    id: string;
+    traceFunctionKey: string;
+    name: string;
+    description: string | null;
+    traceCount: number;
+    graders: DatasetGraderRef[];
+    createdAt: string;
+    updatedAt: string;
+}
+interface SaveDatasetParams {
+    traceFunctionKey: string;
+    name: string;
+    description?: string;
+}
+interface SaveDatasetResult {
+    dataset: Dataset;
+    created: boolean;
+}
+interface ListDatasetsParams {
+    traceFunctionKey?: string;
+}
+interface DatasetTraceIds {
+    datasetId: string;
+    traceIds: string[];
+}
+interface AddDatasetTracesResult {
+    dataset: Dataset;
+    addedTraceIds: string[];
+    alreadyPresentTraceIds: string[];
+    skippedTraceIds: string[];
+}
+interface RemoveDatasetTracesResult {
+    dataset: Dataset;
+    removedTraceIds: string[];
+    notPresentTraceIds: string[];
+}
+interface AddDatasetGradersResult {
+    dataset: Dataset;
+    addedGraderIds: string[];
+    alreadyAssignedGraderIds: string[];
+    skippedGraderIds: string[];
+}
+interface RemoveDatasetGradersResult {
+    dataset: Dataset;
+    removedGraderIds: string[];
+    notAssignedGraderIds: string[];
+}
+type GraderRerunStatus = "pending" | "running" | "completed" | "errored";
+interface GraderRerunProgress {
+    completedTraces: number;
+    totalTraces: number;
+    graderCount: number;
+}
+interface GraderRerunResult {
+    tracesGraded: number;
+    gradersRun: number;
+}
+interface GraderRerun {
+    id: string;
+    status: GraderRerunStatus;
+    graderIds: string[];
+    progress: GraderRerunProgress | null;
+    result: GraderRerunResult | null;
+    error: string | null;
+    createdAt: string;
+    updatedAt: string;
+}
+interface RerunGradersOptions {
+    graderIds?: string[];
+    wait?: boolean;
+    timeoutMs?: number;
+    pollIntervalMs?: number;
+}
+interface RerunGradersResult {
+    run: GraderRerun;
+    joinedExisting: boolean;
+}
+/**
+ * Dataset operations for the authenticated organization, reached as
+ * `client.datasets`. A dataset is a named bucket of traces scoped to one trace
+ * function. Experiments replay against it and its graders score its members.
+ */
+declare class DatasetsClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    /**
+     * Create a dataset, or update the one already named this way under the same
+     * trace function. `created` reports which happened. An omitted description
+     * leaves an existing one untouched.
+     */
+    save(params: SaveDatasetParams): Promise<SaveDatasetResult>;
+    /**
+     * List datasets, scoped to one trace function when `traceFunctionKey` is
+     * given and organization-wide otherwise.
+     */
+    list(params?: ListDatasetsParams): Promise<Dataset[]>;
+    /** Fetch one dataset by id. Rejects with a 404 `BitfabError` when it is not in this organization. */
+    get(datasetId: string): Promise<Dataset>;
+    /** The ids of every trace in the dataset, the same membership a replay with `datasetId` selects. */
+    listTraces(datasetId: string): Promise<DatasetTraceIds>;
+    /**
+     * Add traces to the dataset (1 to 100 ids per call). Traces outside the
+     * organization or under another trace function are reported in
+     * `skippedTraceIds` rather than failing the call.
+     */
+    addTraces(datasetId: string, traceIds: string[]): Promise<AddDatasetTracesResult>;
+    /** Remove traces from the dataset. The traces themselves are never deleted. */
+    removeTraces(datasetId: string, traceIds: string[]): Promise<RemoveDatasetTracesResult>;
+    /**
+     * Assign graders to the dataset (1 to 100 ids per call). Graders outside the
+     * organization or under another trace function are reported in
+     * `skippedGraderIds` rather than failing the call.
+     */
+    addGraders(datasetId: string, graderIds: string[]): Promise<AddDatasetGradersResult>;
+    /** Unassign graders from the dataset. */
+    removeGraders(datasetId: string, graderIds: string[]): Promise<RemoveDatasetGradersResult>;
+    /**
+     * Re-run graders over every trace in the dataset. Defaults to every assigned
+     * grader; an unassigned id is rejected. Waits for the run to finish (up to
+     * `timeoutMs`, default 90s) unless `wait` is `false`, and returns the last
+     * run state seen either way. A request matching an in-flight run joins it.
+     */
+    rerunGraders(datasetId: string, options?: RerunGradersOptions): Promise<RerunGradersResult>;
+    /**
+     * The dataset's active grader re-run, or the run named by `runId`. Returns
+     * `null` when nothing is active or the run does not belong to this dataset.
+     */
+    getGraderRerun(datasetId: string, runId?: string): Promise<GraderRerun | null>;
+}
+
 /**
  * LangGraph/LangChain callback handler for Bitfab tracing.
  *
@@ -2021,6 +2161,7 @@ interface DetachedTrace {
      * Rejects if the server refused the update.
      */
     setSessionId(sessionId: string): Promise<void>;
+    setName(name: string): Promise<void>;
 }
 /**
  * A handle to the current active trace, allowing trace-level context to be set.
@@ -2030,6 +2171,7 @@ interface CurrentTrace {
      * Set the session ID for this trace. Stored in the database session_id column.
      */
     setSessionId(sessionId: string): void;
+    setName(name: string): void;
     /**
      * Set metadata for this trace. Stored in rawData.metadata.
      * Subsequent calls merge with existing metadata, with later values taking precedence.
@@ -2272,6 +2414,8 @@ declare class Bitfab {
     private readonly explicitlyEnabled;
     private readonly strict;
     private readonly httpClient;
+    /** Dataset operations for the authenticated organization. */
+    readonly datasets: DatasetsClient;
     private readonly bamlClient;
     private readonly dbSnapshot;
     private readonly autoTracePolicyRefreshes;
@@ -2777,6 +2921,7 @@ declare class Bitfab {
          */
         metadata?: Record<string, unknown>;
         sessionId?: string;
+        name?: string;
         spanName?: string;
         spanType?: SpanType;
     }): string;
@@ -2955,7 +3100,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.39.0";
+declare const __version__ = "0.41.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -3083,4 +3228,4 @@ interface SeedResult {
  */
 declare function seedFromRegistry(registry: ReplayRegistry, pipeline: string, cases: readonly SeedCase[]): Promise<SeedResult>;
 
-export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AllowedEnvVars, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, HttpClient, type LangGraphIntegrationOptions, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, 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, SUPPORTED_PROVIDERS, type SeedCase, type SeedResult, 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, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, 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, 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 SeedResult, 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 };

package/dist/index.d.ts

@@ -540,7 +540,11 @@ declare class HttpClient {
     lookupFunction<T>(name: string): Promise<T>;
     getAutoTracePolicy<T>(traceFunctionKey: string, protocol: string): Promise<T>;
     getTraceSpan(traceId: string, lookup: SpanLookup): Promise<CapturedSpan | null>;
-    private get;
+    /**
+     * GET a JSON endpoint on the service with the client's API key. Throws a
+     * `BitfabError` carrying the status text for any non-2xx response.
+     */
+    get<T>(endpoint: string): Promise<T>;
     /**
      * Queue an internal trace (from local BAML execution via `call()`) onto this
      * client's batching transport. `functionId` moves into the payload because
@@ -574,6 +578,7 @@ declare class HttpClient {
         appendContexts?: Record<string, unknown>[];
         mergeMetadata?: Record<string, unknown>;
         setSessionId?: string;
+        setName?: string;
     }): Promise<void>;
     /**
      * Start a replay session by fetching historical traces.
@@ -1034,6 +1039,141 @@ declare class BitfabVercelAiHandler {
     get middleware(): BitfabLanguageModelMiddleware;
 }
 
+interface DatasetGraderRef {
+    id: string;
+    name: string | null;
+}
+interface Dataset {
+    id: string;
+    traceFunctionKey: string;
+    name: string;
+    description: string | null;
+    traceCount: number;
+    graders: DatasetGraderRef[];
+    createdAt: string;
+    updatedAt: string;
+}
+interface SaveDatasetParams {
+    traceFunctionKey: string;
+    name: string;
+    description?: string;
+}
+interface SaveDatasetResult {
+    dataset: Dataset;
+    created: boolean;
+}
+interface ListDatasetsParams {
+    traceFunctionKey?: string;
+}
+interface DatasetTraceIds {
+    datasetId: string;
+    traceIds: string[];
+}
+interface AddDatasetTracesResult {
+    dataset: Dataset;
+    addedTraceIds: string[];
+    alreadyPresentTraceIds: string[];
+    skippedTraceIds: string[];
+}
+interface RemoveDatasetTracesResult {
+    dataset: Dataset;
+    removedTraceIds: string[];
+    notPresentTraceIds: string[];
+}
+interface AddDatasetGradersResult {
+    dataset: Dataset;
+    addedGraderIds: string[];
+    alreadyAssignedGraderIds: string[];
+    skippedGraderIds: string[];
+}
+interface RemoveDatasetGradersResult {
+    dataset: Dataset;
+    removedGraderIds: string[];
+    notAssignedGraderIds: string[];
+}
+type GraderRerunStatus = "pending" | "running" | "completed" | "errored";
+interface GraderRerunProgress {
+    completedTraces: number;
+    totalTraces: number;
+    graderCount: number;
+}
+interface GraderRerunResult {
+    tracesGraded: number;
+    gradersRun: number;
+}
+interface GraderRerun {
+    id: string;
+    status: GraderRerunStatus;
+    graderIds: string[];
+    progress: GraderRerunProgress | null;
+    result: GraderRerunResult | null;
+    error: string | null;
+    createdAt: string;
+    updatedAt: string;
+}
+interface RerunGradersOptions {
+    graderIds?: string[];
+    wait?: boolean;
+    timeoutMs?: number;
+    pollIntervalMs?: number;
+}
+interface RerunGradersResult {
+    run: GraderRerun;
+    joinedExisting: boolean;
+}
+/**
+ * Dataset operations for the authenticated organization, reached as
+ * `client.datasets`. A dataset is a named bucket of traces scoped to one trace
+ * function. Experiments replay against it and its graders score its members.
+ */
+declare class DatasetsClient {
+    private readonly httpClient;
+    constructor(httpClient: HttpClient);
+    /**
+     * Create a dataset, or update the one already named this way under the same
+     * trace function. `created` reports which happened. An omitted description
+     * leaves an existing one untouched.
+     */
+    save(params: SaveDatasetParams): Promise<SaveDatasetResult>;
+    /**
+     * List datasets, scoped to one trace function when `traceFunctionKey` is
+     * given and organization-wide otherwise.
+     */
+    list(params?: ListDatasetsParams): Promise<Dataset[]>;
+    /** Fetch one dataset by id. Rejects with a 404 `BitfabError` when it is not in this organization. */
+    get(datasetId: string): Promise<Dataset>;
+    /** The ids of every trace in the dataset, the same membership a replay with `datasetId` selects. */
+    listTraces(datasetId: string): Promise<DatasetTraceIds>;
+    /**
+     * Add traces to the dataset (1 to 100 ids per call). Traces outside the
+     * organization or under another trace function are reported in
+     * `skippedTraceIds` rather than failing the call.
+     */
+    addTraces(datasetId: string, traceIds: string[]): Promise<AddDatasetTracesResult>;
+    /** Remove traces from the dataset. The traces themselves are never deleted. */
+    removeTraces(datasetId: string, traceIds: string[]): Promise<RemoveDatasetTracesResult>;
+    /**
+     * Assign graders to the dataset (1 to 100 ids per call). Graders outside the
+     * organization or under another trace function are reported in
+     * `skippedGraderIds` rather than failing the call.
+     */
+    addGraders(datasetId: string, graderIds: string[]): Promise<AddDatasetGradersResult>;
+    /** Unassign graders from the dataset. */
+    removeGraders(datasetId: string, graderIds: string[]): Promise<RemoveDatasetGradersResult>;
+    /**
+     * Re-run graders over every trace in the dataset. Defaults to every assigned
+     * grader; an unassigned id is rejected. Waits for the run to finish (up to
+     * `timeoutMs`, default 90s) unless `wait` is `false`, and returns the last
+     * run state seen either way. A request matching an in-flight run joins it.
+     */
+    rerunGraders(datasetId: string, options?: RerunGradersOptions): Promise<RerunGradersResult>;
+    /**
+     * The dataset's active grader re-run, or the run named by `runId`. Returns
+     * `null` when nothing is active or the run does not belong to this dataset.
+     */
+    getGraderRerun(datasetId: string, runId?: string): Promise<GraderRerun | null>;
+}
+
 /**
  * LangGraph/LangChain callback handler for Bitfab tracing.
  *
@@ -2021,6 +2161,7 @@ interface DetachedTrace {
      * Rejects if the server refused the update.
      */
     setSessionId(sessionId: string): Promise<void>;
+    setName(name: string): Promise<void>;
 }
 /**
  * A handle to the current active trace, allowing trace-level context to be set.
@@ -2030,6 +2171,7 @@ interface CurrentTrace {
      * Set the session ID for this trace. Stored in the database session_id column.
      */
     setSessionId(sessionId: string): void;
+    setName(name: string): void;
     /**
      * Set metadata for this trace. Stored in rawData.metadata.
      * Subsequent calls merge with existing metadata, with later values taking precedence.
@@ -2272,6 +2414,8 @@ declare class Bitfab {
     private readonly explicitlyEnabled;
     private readonly strict;
     private readonly httpClient;
+    /** Dataset operations for the authenticated organization. */
+    readonly datasets: DatasetsClient;
     private readonly bamlClient;
     private readonly dbSnapshot;
     private readonly autoTracePolicyRefreshes;
@@ -2777,6 +2921,7 @@ declare class Bitfab {
          */
         metadata?: Record<string, unknown>;
         sessionId?: string;
+        name?: string;
         spanName?: string;
         spanType?: SpanType;
     }): string;
@@ -2955,7 +3100,7 @@ declare class BitfabFunction {
 /**
  * SDK version from package.json (injected at build time)
  */
-declare const __version__ = "0.39.0";
+declare const __version__ = "0.41.0";
 
 /**
  * Constants for the Bitfab SDK.
@@ -3083,4 +3228,4 @@ interface SeedResult {
  */
 declare function seedFromRegistry(registry: ReplayRegistry, pipeline: string, cases: readonly SeedCase[]): Promise<SeedResult>;
 
-export { type ActiveSpanContext, type AdaptContext, type AdaptInputsFn, type AllowedEnvVars, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, type CaptureWhen, type CapturedSpan, type CodeChangeFile, type CurrentSpan, type CurrentTrace, DEFAULT_SERVICE_URL, type DbBranchOptions, DbBranchReplayError, type DbBranchTimings, type DbSnapshotConfig, type DbSnapshotProvider, type DbSnapshotRef, type DetachedTrace, HttpClient, type LangGraphIntegrationOptions, type MockOverride, type MockOverrideCtx, type MockOverrideInput, type MockOverrideResolver, type MockStrategy, type MockValue, NO_MOCK_OVERRIDE, type NodeMatcher, type NodeMethodDecorator, type NodeOptions, type ProviderDefinition, 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, SUPPORTED_PROVIDERS, type SeedCase, type SeedResult, 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, BITFAB_PROGRESS_PREFIX, type BamlExecutionResult, Bitfab, BitfabClaudeAgentHandler, type BitfabConfig, BitfabError, BitfabFunction, BitfabLangGraphCallbackHandler as BitfabLangChainCallbackHandler, BitfabLangGraphCallbackHandler, BitfabLangGraphIntegration, type BitfabLanguageModelMiddleware, BitfabOpenAIAgentHandler, BitfabOpenAITracingProcessor, BitfabVercelAiHandler, 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, 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 SeedResult, 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 };

package/dist/index.js

@@ -7,6 +7,7 @@ import {
   BitfabOpenAIAgentHandler,
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
+  DatasetsClient,
   SUPPORTED_PROVIDERS,
   defineReplayRegistry,
   finalizers,
@@ -14,7 +15,7 @@ import {
   getCurrentSpan,
   getCurrentTrace,
   seedFromRegistry
-} from "./chunk-SHJ3ABUG.js";
+} from "./chunk-U3AKIFW3.js";
 import "./chunk-ZUD7OFYB.js";
 import {
   BITFAB_PROGRESS_PREFIX,
@@ -23,14 +24,14 @@ import {
   ReplayError,
   reportReplayProgress,
   serializeReplayResult
-} from "./chunk-FW7PZ3EP.js";
+} from "./chunk-RWZYZRYI.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
   HttpClient,
   __version__,
   flushTraces
-} from "./chunk-2JQSYJJR.js";
+} from "./chunk-IFYX3KW6.js";
 import "./chunk-H6LZRFMN.js";
 export {
   BITFAB_PROGRESS_PREFIX,
@@ -45,6 +46,7 @@ export {
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
   DEFAULT_SERVICE_URL,
+  DatasetsClient,
   DbBranchReplayError,
   HttpClient,
   NO_MOCK_OVERRIDE,