braintrust 3.26.0 → 3.28.0

package/dist/auto-instrumentations/loader/cjs-patch.cjs

@@ -52,8 +52,10 @@ var GLOBAL_INSTRUMENTATION_HOOKS_KEY = "__braintrust_instrumentation_hooks";
 var GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION = 1;
 var GLOBAL_INSTRUMENTATION_HOOKS_REGISTRY_BRAND = "braintrust.global-instrumentation-hooks.registry";
 var GLOBAL_INSTRUMENTATION_HOOK_BRAND = "braintrust.global-instrumentation-hooks.hook";
+var GLOBAL_INVOCATION_HOOK_BRAND = "braintrust.global-instrumentation-hooks.invocation-hook";
 var registryBrand = Symbol.for(GLOBAL_INSTRUMENTATION_HOOKS_REGISTRY_BRAND);
 var hookBrand = Symbol.for(GLOBAL_INSTRUMENTATION_HOOK_BRAND);
+var invocationHookBrand = Symbol.for(GLOBAL_INVOCATION_HOOK_BRAND);
 var errorReporter;
 function reportError(error) {
   try {
@@ -189,12 +191,69 @@ var traceEvents = [
   "asyncEnd",
   "error"
 ];
+function traceInvocation(hook, operator, target, thisArg, args, additional, callbackIndex = -1) {
+  const context = {
+    ...additional,
+    arguments: args,
+    self: thisArg
+  };
+  const invoke = () => hook.invoke(target, thisArg, args, additional);
+  if (operator === "traceCallback") {
+    return hook.traceCallback(invoke, callbackIndex, context);
+  }
+  if (operator === "tracePromise") {
+    return hook.tracePromise(invoke, context);
+  }
+  return hook.traceSync(invoke, context);
+}
+var InvocationHook = class {
+  interceptors = [];
+  get hasInterceptors() {
+    return this.interceptors.length > 0;
+  }
+  intercept(interceptor) {
+    if (typeof interceptor !== "function") {
+      throw new TypeError("interceptor must be a function");
+    }
+    this.interceptors = [...this.interceptors, interceptor];
+    let active = true;
+    return () => {
+      if (!active) {
+        return;
+      }
+      active = false;
+      const index = this.interceptors.indexOf(interceptor);
+      if (index !== -1) {
+        this.interceptors = [
+          ...this.interceptors.slice(0, index),
+          ...this.interceptors.slice(index + 1)
+        ];
+      }
+    };
+  }
+  invoke(target, thisArg, args, additional) {
+    const interceptors = this.interceptors;
+    if (interceptors.length === 0) {
+      return Reflect.apply(target, thisArg, args);
+    }
+    let next = target;
+    for (let index = interceptors.length - 1; index >= 0; index -= 1) {
+      const interceptor = interceptors[index];
+      const downstream = next;
+      next = function(...nextArgs) {
+        return interceptor(downstream, this, nextArgs, additional);
+      };
+    }
+    return Reflect.apply(next, thisArg, args);
+  }
+};
 var TracingHook = class {
   start;
   end;
   asyncStart;
   asyncEnd;
   error;
+  invocationHook = new InvocationHook();
   constructor(nameOrChannels) {
     Object.defineProperty(this, hookBrand, {
       configurable: false,
@@ -202,6 +261,12 @@ var TracingHook = class {
       value: GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION,
       writable: false
     });
+    Object.defineProperty(this, invocationHookBrand, {
+      configurable: false,
+      enumerable: false,
+      value: GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION,
+      writable: false
+    });
     if (typeof nameOrChannels === "string") {
       this.start = new HookChannel(`tracing:${nameOrChannels}:start`);
       this.end = new HookChannel(`tracing:${nameOrChannels}:end`);
@@ -219,6 +284,26 @@ var TracingHook = class {
   get hasSubscribers() {
     return this.start.hasSubscribers || this.end.hasSubscribers || this.asyncStart.hasSubscribers || this.asyncEnd.hasSubscribers || this.error.hasSubscribers;
   }
+  get hasInterceptors() {
+    return this.invocationHook.hasInterceptors;
+  }
+  intercept(interceptor) {
+    return this.invocationHook.intercept(interceptor);
+  }
+  invoke(target, thisArg, args, additional) {
+    return this.invocationHook.invoke(target, thisArg, args, additional);
+  }
+  traceInvocation(operator, target, thisArg, args, additional, callbackIndex = -1) {
+    return traceInvocation(
+      this,
+      operator,
+      target,
+      thisArg,
+      args,
+      additional,
+      callbackIndex
+    );
+  }
   subscribe(handlers) {
     for (const eventName of traceEvents) {
       const handler = handlers[eventName];
@@ -432,8 +517,7 @@ var transforms = {
   tracingHookDeclaration(state, node) {
     const {
       channelName,
-      module: { name },
-      operator
+      module: { name }
     } = state;
     const channelVariable = formatChannelVariable(channelName);
     const channelGetter = formatChannelGetter(channelName);
@@ -446,7 +530,7 @@ var transforms = {
       return;
     }
     const sharedHookLookup = hasSharedHookLookup ? "" : `
-      const ${SHARED_HOOK_LOOKUP} = (__bt$hookName, __bt$operator) => {
+      const ${SHARED_HOOK_LOOKUP} = (__bt$hookName) => {
         try {
           const __bt$hooks = globalThis[${JSON.stringify(
       GLOBAL_INSTRUMENTATION_HOOKS_KEY
@@ -468,8 +552,7 @@ var transforms = {
             __bt$hook[Symbol.for(${JSON.stringify(
       GLOBAL_INSTRUMENTATION_HOOK_BRAND
     )})] !== ${GLOBAL_INSTRUMENTATION_HOOKS_PROTOCOL_VERSION} ||
-            typeof __bt$hook.hasSubscribers !== "boolean" ||
-            typeof __bt$hook[__bt$operator] !== "function"
+            typeof __bt$hook.traceInvocation !== "function"
           ) return undefined;
           return __bt$hook;
         } catch {
@@ -482,8 +565,7 @@ var transforms = {
       let ${channelVariable};
       const ${channelGetter} = () =>
         ${channelVariable} ??= ${SHARED_HOOK_LOOKUP}(
-          ${JSON.stringify(`orchestrion:${name}:${channelName}`)},
-          ${JSON.stringify(operator)}
+          ${JSON.stringify(`orchestrion:${name}:${channelName}`)}
         );
     `;
     const sharedHookLookupIndex = node.body.findIndex(
@@ -558,28 +640,18 @@ function traceInstanceMethod(state, node, program) {
   ctor.value.body.body.push(...ctorBody);
 }
 function wrap(state, node) {
-  const { operator, moduleVersion } = state;
-  const wrapper = operator === "traceCallback" ? wrapCallback(state) : operator === "tracePromise" ? wrapPromise(state) : wrapSync(state);
+  const wrapper = wrapInvocation(state);
   const block = wrapper.body[0].body;
   const common = (0, import_meriyah.parse)(
     node.type === "ArrowFunctionExpression" ? `
-    const __bt$ctx = {
-      arguments,
-      moduleVersion: ${JSON.stringify(moduleVersion)}
-    };
-    const __bt$traced = () => {
+    const __bt$target = (...__bt$args) => {
       const __bt$wrapped = () => {};
-      return __bt$wrapped(...arguments);
+      return __bt$wrapped(...__bt$args);
     };
   ` : `
-    const __bt$ctx = {
-      arguments,
-      self: this,
-      moduleVersion: ${JSON.stringify(moduleVersion)}
-    };
-    const __bt$traced = () => {
+    const __bt$target = function (...__bt$args) {
       const __bt$wrapped = () => {};
-      return __bt$wrapped.apply(this, arguments);
+      return __bt$wrapped.apply(this, __bt$args);
     };
   `
   ).body;
@@ -621,49 +693,25 @@ function wrapSuper(node) {
     node.body.body.unshift((0, import_meriyah.parse)("const __bt$super = {}").body[0]);
   }
 }
-function wrapCallback(state) {
-  const {
-    channelName,
-    functionQuery: { callbackIndex = -1 }
-  } = state;
+function wrapInvocation(state) {
+  const { channelName, moduleVersion, operator, functionQuery } = state;
   const channelGetter = formatChannelGetter(channelName);
+  const callbackIndex = functionQuery.callbackIndex ?? -1;
   return (0, import_meriyah.parse)(`
     function wrapper () {
       const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.traceCallback(
-        __bt$traced,
-        ${callbackIndex},
-        __bt$ctx
+      if (!__bt$hook) return __bt$target.apply(this, arguments);
+      return __bt$hook.traceInvocation(
+        ${JSON.stringify(operator)},
+        __bt$target,
+        this,
+        arguments,
+        { moduleVersion: ${JSON.stringify(moduleVersion)} },
+        ${callbackIndex}
       );
     }
   `);
 }
-function wrapPromise(state) {
-  const { channelName } = state;
-  const channelGetter = formatChannelGetter(channelName);
-  return (0, import_meriyah.parse)(`
-    function wrapper () {
-      const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.tracePromise(__bt$traced, __bt$ctx);
-    }
-  `);
-}
-function wrapSync(state) {
-  const { channelName } = state;
-  const channelGetter = formatChannelGetter(channelName);
-  return (0, import_meriyah.parse)(`
-    function wrapper () {
-      const __bt$hook = ${channelGetter}();
-      if (!__bt$hook?.hasSubscribers) return __bt$traced();
-      __bt$ctx.self ??= this;
-      return __bt$hook.traceSync(__bt$traced, __bt$ctx);
-    }
-  `);
-}
 
 // src/auto-instrumentations/orchestrion-js/transformer.ts
 var Transformer = class {

package/dist/auto-instrumentations/loader/cjs-patch.mjs

@@ -1,8 +1,8 @@
 import {
   applySpecialCasePatch,
   create
-} from "../chunk-XYN63IG5.mjs";
-import "../chunk-NRE4QUQR.mjs";
+} from "../chunk-I55G56ZL.mjs";
+import "../chunk-JPUFNW7X.mjs";
 import {
   getPackageName,
   getPackageVersion

package/dist/auto-instrumentations/loader/esm-hook.mjs

@@ -4,7 +4,7 @@ import {
   getPackageName,
   getPackageVersion,
   isSpecialCaseTarget
-} from "../chunk-AKKPLXTP.mjs";
+} from "../chunk-2AMDGD65.mjs";
 
 // src/auto-instrumentations/loader/esm-hook.mts
 import { readFile } from "fs/promises";