braintrust 0.3.8 → 0.4.1

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;
@@ -7284,8 +7302,10 @@ declare class LazyValue<T> {
     get hasSucceeded(): boolean;
 }
 
-/// <reference lib="dom" />
-
+interface ContextParentSpanIds {
+    rootSpanId: string;
+    spanParents: string[];
+}
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -7436,6 +7456,11 @@ interface Span extends Exportable {
     state(): BraintrustState;
     kind: "span";
 }
+declare abstract class ContextManager {
+    abstract getParentSpanIds(): ContextParentSpanIds | undefined;
+    abstract runInContext<R>(span: Span, callback: () => R): R;
+    abstract getCurrentSpan(): Span | undefined;
+}
 /**
  * A fake implementation of the Span API which does nothing. This can be used as the default span.
  */
@@ -7532,8 +7557,13 @@ declare class BraintrustState {
     private _apiConn;
     private _proxyConn;
     promptCache: PromptCache;
+    private _idGenerator;
+    private _contextManager;
     constructor(loginParams: LoginOptions);
     resetLoginInfo(): void;
+    resetIdGenState(): void;
+    get idGenerator(): IDGenerator;
+    get contextManager(): ContextManager;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
     static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;

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;
@@ -7284,8 +7302,10 @@ declare class LazyValue<T> {
     get hasSucceeded(): boolean;
 }
 
-/// <reference lib="dom" />
-
+interface ContextParentSpanIds {
+    rootSpanId: string;
+    spanParents: string[];
+}
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -7436,6 +7456,11 @@ interface Span extends Exportable {
     state(): BraintrustState;
     kind: "span";
 }
+declare abstract class ContextManager {
+    abstract getParentSpanIds(): ContextParentSpanIds | undefined;
+    abstract runInContext<R>(span: Span, callback: () => R): R;
+    abstract getCurrentSpan(): Span | undefined;
+}
 /**
  * A fake implementation of the Span API which does nothing. This can be used as the default span.
  */
@@ -7532,8 +7557,13 @@ declare class BraintrustState {
     private _apiConn;
     private _proxyConn;
     promptCache: PromptCache;
+    private _idGenerator;
+    private _contextManager;
     constructor(loginParams: LoginOptions);
     resetLoginInfo(): void;
+    resetIdGenState(): void;
+    get idGenerator(): IDGenerator;
+    get contextManager(): ContextManager;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
     static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;