braintrust 0.4.0 → 0.4.1

package/dev/dist/index.d.mts

@@ -7302,6 +7302,10 @@ declare class LazyValue<T> {
     get hasSucceeded(): boolean;
 }
 
+interface ContextParentSpanIds {
+    rootSpanId: string;
+    spanParents: string[];
+}
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -7452,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.
  */
@@ -7549,10 +7558,12 @@ declare class BraintrustState {
     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

@@ -7302,6 +7302,10 @@ declare class LazyValue<T> {
     get hasSucceeded(): boolean;
 }
 
+interface ContextParentSpanIds {
+    rootSpanId: string;
+    spanParents: string[];
+}
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -7452,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.
  */
@@ -7549,10 +7558,12 @@ declare class BraintrustState {
     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;