braintrust 0.4.0 → 0.4.2

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;
 };
@@ -7450,8 +7454,19 @@ interface Span extends Exportable {
      */
     startSpanWithParents(spanId: string, spanParents: string[], args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    /**
+     * Internal method to get the OTEL parent string for this span.
+     * This is used by OtelContextManager to set the braintrust.parent attribute.
+     * @returns A string like "project_id:X" or "experiment_id:X", or undefined if no parent
+     */
+    _getOtelParent(): string | undefined;
     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.
  */
@@ -7475,6 +7490,7 @@ declare class NoopSpan implements Span {
     setAttributes(_args: Omit<StartSpanArgs, "event">): void;
     startSpanWithParents(_spanId: string, _spanParents: string[], _args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    _getOtelParent(): string | undefined;
     toString(): string;
 }
 declare const NOOP_SPAN: NoopSpan;
@@ -7549,10 +7565,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;
@@ -7850,6 +7868,12 @@ declare class Experiment extends ObjectFetcher<ExperimentEvent> implements Expor
     kind: "experiment";
     constructor(state: BraintrustState, lazyMetadata: LazyValue<ProjectExperimentMetadata>, dataset?: AnyDataset);
     get id(): Promise<string>;
+    /**
+     * Wait for the experiment ID to be resolved. This is useful for ensuring the ID
+     * is available synchronously in child spans (for OTEL parent attributes).
+     * @internal
+     */
+    _waitForId(): Promise<void>;
     get name(): Promise<string>;
     get project(): Promise<ObjectMetadata>;
     private parentObjectType;

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;
 };
@@ -7450,8 +7454,19 @@ interface Span extends Exportable {
      */
     startSpanWithParents(spanId: string, spanParents: string[], args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    /**
+     * Internal method to get the OTEL parent string for this span.
+     * This is used by OtelContextManager to set the braintrust.parent attribute.
+     * @returns A string like "project_id:X" or "experiment_id:X", or undefined if no parent
+     */
+    _getOtelParent(): string | undefined;
     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.
  */
@@ -7475,6 +7490,7 @@ declare class NoopSpan implements Span {
     setAttributes(_args: Omit<StartSpanArgs, "event">): void;
     startSpanWithParents(_spanId: string, _spanParents: string[], _args?: StartSpanArgs): Span;
     state(): BraintrustState;
+    _getOtelParent(): string | undefined;
     toString(): string;
 }
 declare const NOOP_SPAN: NoopSpan;
@@ -7549,10 +7565,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;
@@ -7850,6 +7868,12 @@ declare class Experiment extends ObjectFetcher<ExperimentEvent> implements Expor
     kind: "experiment";
     constructor(state: BraintrustState, lazyMetadata: LazyValue<ProjectExperimentMetadata>, dataset?: AnyDataset);
     get id(): Promise<string>;
+    /**
+     * Wait for the experiment ID to be resolved. This is useful for ensuring the ID
+     * is available synchronously in child spans (for OTEL parent attributes).
+     * @internal
+     */
+    _waitForId(): Promise<void>;
     get name(): Promise<string>;
     get project(): Promise<ObjectMetadata>;
     private parentObjectType;