braintrust 0.3.7 → 0.4.0

package/dev/dist/index.d.mts

@@ -7097,6 +7097,24 @@ type ToolFunctionDefinitionType = z.infer<typeof ToolFunctionDefinition>;
 
 type GenericFunction<Input, Output> = ((input: Input) => Output) | ((input: Input) => Promise<Output>);
 
+/**
+ * Abstract base class for ID generators
+ */
+declare abstract class IDGenerator {
+    /**
+     * Generate a span ID
+     */
+    abstract getSpanId(): string;
+    /**
+     * Generate a trace ID
+     */
+    abstract getTraceId(): string;
+    /**
+     * Return true if the generator should use span_id as root_span_id for backwards compatibility
+     */
+    abstract shareRootSpanId(): boolean;
+}
+
 interface IsoAsyncLocalStorage<T> {
     enterWith(store: T): void;
     run<R>(store: T | undefined, callback: () => R): R;
@@ -7457,6 +7475,7 @@ declare class NoopSpan implements Span {
     setAttributes(_args: Omit<StartSpanArgs, "event">): void;
     startSpanWithParents(_spanId: string, _spanParents: string[], _args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    toString(): string;
 }
 declare const NOOP_SPAN: NoopSpan;
 declare global {
@@ -7529,8 +7548,11 @@ declare class BraintrustState {
     private _apiConn;
     private _proxyConn;
     promptCache: PromptCache;
+    private _idGenerator;
     constructor(loginParams: LoginOptions);
     resetLoginInfo(): void;
+    resetIdGenState(): void;
+    get idGenerator(): IDGenerator;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
     static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;
@@ -7548,6 +7570,8 @@ declare class BraintrustState {
     loginReplaceApiConn(apiConn: HTTPConnection): void;
     disable(): void;
     enforceQueueSizeLimit(enforce: boolean): void;
+    toJSON(): Record<string, any>;
+    toString(): string;
 }
 declare class HTTPConnection {
     base_url: string;
@@ -7565,6 +7589,7 @@ declare class HTTPConnection {
     post(path: string, params?: Record<string, unknown> | string, config?: RequestInit): Promise<Response>;
     get_json(object_type: string, args?: Record<string, string | string[] | undefined> | undefined, retries?: number): Promise<any>;
     post_json(object_type: string, args?: Record<string, unknown> | string | undefined): Promise<any>;
+    toString(): string;
 }
 interface ObjectMetadata {
     id: string;

package/dev/dist/index.d.ts

@@ -7097,6 +7097,24 @@ type ToolFunctionDefinitionType = z.infer<typeof ToolFunctionDefinition>;
 
 type GenericFunction<Input, Output> = ((input: Input) => Output) | ((input: Input) => Promise<Output>);
 
+/**
+ * Abstract base class for ID generators
+ */
+declare abstract class IDGenerator {
+    /**
+     * Generate a span ID
+     */
+    abstract getSpanId(): string;
+    /**
+     * Generate a trace ID
+     */
+    abstract getTraceId(): string;
+    /**
+     * Return true if the generator should use span_id as root_span_id for backwards compatibility
+     */
+    abstract shareRootSpanId(): boolean;
+}
+
 interface IsoAsyncLocalStorage<T> {
     enterWith(store: T): void;
     run<R>(store: T | undefined, callback: () => R): R;
@@ -7457,6 +7475,7 @@ declare class NoopSpan implements Span {
     setAttributes(_args: Omit<StartSpanArgs, "event">): void;
     startSpanWithParents(_spanId: string, _spanParents: string[], _args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    toString(): string;
 }
 declare const NOOP_SPAN: NoopSpan;
 declare global {
@@ -7529,8 +7548,11 @@ declare class BraintrustState {
     private _apiConn;
     private _proxyConn;
     promptCache: PromptCache;
+    private _idGenerator;
     constructor(loginParams: LoginOptions);
     resetLoginInfo(): void;
+    resetIdGenState(): void;
+    get idGenerator(): IDGenerator;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
     static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;
@@ -7548,6 +7570,8 @@ declare class BraintrustState {
     loginReplaceApiConn(apiConn: HTTPConnection): void;
     disable(): void;
     enforceQueueSizeLimit(enforce: boolean): void;
+    toJSON(): Record<string, any>;
+    toString(): string;
 }
 declare class HTTPConnection {
     base_url: string;
@@ -7565,6 +7589,7 @@ declare class HTTPConnection {
     post(path: string, params?: Record<string, unknown> | string, config?: RequestInit): Promise<Response>;
     get_json(object_type: string, args?: Record<string, string | string[] | undefined> | undefined, retries?: number): Promise<any>;
     post_json(object_type: string, args?: Record<string, unknown> | string | undefined): Promise<any>;
+    toString(): string;
 }
 interface ObjectMetadata {
     id: string;