braintrust 3.3.0 → 3.5.0

package/dist/instrumentation/index.d.mts

@@ -1,4 +1,10 @@
-import { ChannelHandlers as ChannelHandlers$1 } from 'dc-browser';
+interface IsoChannelHandlers<M = any> {
+    start?: (context: M, name: string) => void;
+    end?: (context: M, name: string) => void;
+    asyncStart?: (context: M, name: string) => void;
+    asyncEnd?: (context: M, name: string) => void;
+    error?: (context: M, name: string) => void;
+}
 
 /**
  * Base class for creating instrumentation plugins.
@@ -33,7 +39,7 @@ declare abstract class BasePlugin {
      * @param channelName - The channel name to subscribe to
      * @param handlers - Event handlers
      */
-    protected subscribe(channelName: string, handlers: ChannelHandlers$1): void;
+    protected subscribe(channelName: string, handlers: IsoChannelHandlers): void;
     /**
      * Subscribe to a channel for async methods (non-streaming).
      * Creates a span and logs input/output/metrics.
@@ -45,8 +51,9 @@ declare abstract class BasePlugin {
             input: any;
             metadata: any;
         };
-        extractOutput: (result: any) => any;
-        extractMetrics: (result: any, startTime?: number) => Record<string, number>;
+        extractOutput: (result: any, endEvent?: any) => any;
+        extractMetadata?: (result: any, endEvent?: any) => any;
+        extractMetrics: (result: any, startTime?: number, endEvent?: any) => Record<string, number>;
     }): void;
     /**
      * Subscribe to a channel for async methods that may return streams.
@@ -59,11 +66,13 @@ declare abstract class BasePlugin {
             input: any;
             metadata: any;
         };
-        extractOutput: (result: any) => any;
-        extractMetrics: (result: any, startTime?: number) => Record<string, number>;
-        aggregateChunks?: (chunks: any[]) => {
+        extractOutput: (result: any, endEvent?: any) => any;
+        extractMetadata?: (result: any, endEvent?: any) => any;
+        extractMetrics: (result: any, startTime?: number, endEvent?: any) => Record<string, number>;
+        aggregateChunks?: (chunks: any[], result?: any, endEvent?: any) => {
             output: any;
             metrics: Record<string, number>;
+            metadata?: any;
         };
     }): void;
     /**
@@ -122,17 +131,6 @@ declare function parseChannelName(channelName: string): {
  */
 declare function isValidChannelName(channelName: string): boolean;
 
-/**
- * Standard event types for diagnostics_channel-based instrumentation.
- *
- * These types follow the TracingChannel pattern from Node.js.
- * For async functions (tracePromise):
- * - start: Called before the synchronous portion executes
- * - end: Called after the synchronous portion completes (promise returned)
- * - asyncStart: Called when the promise begins to settle
- * - asyncEnd: Called when the promise finishes settling (before user code continues)
- * - error: Called if the function throws or the promise rejects
- */
 /**
  * Base context object shared across all events in a trace.
  */
@@ -186,10 +184,6 @@ interface ErrorEvent extends BaseContext {
      */
     arguments?: unknown[];
 }
-/**
- * Event emitted when a promise begins to settle.
- * This fires after the synchronous portion and when the async continuation starts.
- */
 interface AsyncStartEvent<TInput = unknown> extends StartEvent<TInput> {
 }
 /**

package/dist/instrumentation/index.d.ts

@@ -1,4 +1,10 @@
-import { ChannelHandlers as ChannelHandlers$1 } from 'dc-browser';
+interface IsoChannelHandlers<M = any> {
+    start?: (context: M, name: string) => void;
+    end?: (context: M, name: string) => void;
+    asyncStart?: (context: M, name: string) => void;
+    asyncEnd?: (context: M, name: string) => void;
+    error?: (context: M, name: string) => void;
+}
 
 /**
  * Base class for creating instrumentation plugins.
@@ -33,7 +39,7 @@ declare abstract class BasePlugin {
      * @param channelName - The channel name to subscribe to
      * @param handlers - Event handlers
      */
-    protected subscribe(channelName: string, handlers: ChannelHandlers$1): void;
+    protected subscribe(channelName: string, handlers: IsoChannelHandlers): void;
     /**
      * Subscribe to a channel for async methods (non-streaming).
      * Creates a span and logs input/output/metrics.
@@ -45,8 +51,9 @@ declare abstract class BasePlugin {
             input: any;
             metadata: any;
         };
-        extractOutput: (result: any) => any;
-        extractMetrics: (result: any, startTime?: number) => Record<string, number>;
+        extractOutput: (result: any, endEvent?: any) => any;
+        extractMetadata?: (result: any, endEvent?: any) => any;
+        extractMetrics: (result: any, startTime?: number, endEvent?: any) => Record<string, number>;
     }): void;
     /**
      * Subscribe to a channel for async methods that may return streams.
@@ -59,11 +66,13 @@ declare abstract class BasePlugin {
             input: any;
             metadata: any;
         };
-        extractOutput: (result: any) => any;
-        extractMetrics: (result: any, startTime?: number) => Record<string, number>;
-        aggregateChunks?: (chunks: any[]) => {
+        extractOutput: (result: any, endEvent?: any) => any;
+        extractMetadata?: (result: any, endEvent?: any) => any;
+        extractMetrics: (result: any, startTime?: number, endEvent?: any) => Record<string, number>;
+        aggregateChunks?: (chunks: any[], result?: any, endEvent?: any) => {
             output: any;
             metrics: Record<string, number>;
+            metadata?: any;
         };
     }): void;
     /**
@@ -122,17 +131,6 @@ declare function parseChannelName(channelName: string): {
  */
 declare function isValidChannelName(channelName: string): boolean;
 
-/**
- * Standard event types for diagnostics_channel-based instrumentation.
- *
- * These types follow the TracingChannel pattern from Node.js.
- * For async functions (tracePromise):
- * - start: Called before the synchronous portion executes
- * - end: Called after the synchronous portion completes (promise returned)
- * - asyncStart: Called when the promise begins to settle
- * - asyncEnd: Called when the promise finishes settling (before user code continues)
- * - error: Called if the function throws or the promise rejects
- */
 /**
  * Base context object shared across all events in a trace.
  */
@@ -186,10 +184,6 @@ interface ErrorEvent extends BaseContext {
      */
     arguments?: unknown[];
 }
-/**
- * Event emitted when a promise begins to settle.
- * This fires after the synchronous portion and when the async continuation starts.
- */
 interface AsyncStartEvent<TInput = unknown> extends StartEvent<TInput> {
 }
 /**