braintrust 3.10.0 → 3.11.0

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

@@ -93,6 +93,27 @@ function getPackageName(baseDir) {
   return void 0;
 }
 
+// src/auto-instrumentations/loader/openai-api-promise-patch.ts
+var OPENAI_API_PROMISE_PATCH = `
+;(function __btPatchAPIPromise() {
+  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
+  APIPromise.prototype.__btParsePatched = true;
+  var _origThen = APIPromise.prototype.then;
+  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
+    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
+      this.__btParseWrapped = true;
+      var _origParse = this.parseResponse;
+      var _cached;
+      this.parseResponse = function() {
+        if (!_cached) _cached = _origParse.apply(this, arguments);
+        return _cached;
+      };
+    }
+    return _origThen.call(this, onfulfilled, onrejected);
+  };
+})();
+`;
+
 // src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
@@ -124,8 +145,12 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
+          args[0] = content + OPENAI_API_PROMISE_PATCH;
+          return self.originalCompile.apply(this, args);
+        }
+        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,

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

@@ -10,6 +10,29 @@ import {
 import * as NodeModule from "module";
 import { sep } from "path";
 import moduleDetailsFromPath from "module-details-from-path";
+
+// src/auto-instrumentations/loader/openai-api-promise-patch.ts
+var OPENAI_API_PROMISE_PATCH = `
+;(function __btPatchAPIPromise() {
+  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
+  APIPromise.prototype.__btParsePatched = true;
+  var _origThen = APIPromise.prototype.then;
+  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
+    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
+      this.__btParseWrapped = true;
+      var _origParse = this.parseResponse;
+      var _cached;
+      this.parseResponse = function() {
+        if (!_cached) _cached = _origParse.apply(this, arguments);
+        return _cached;
+      };
+    }
+    return _origThen.call(this, onfulfilled, onrejected);
+  };
+})();
+`;
+
+// src/auto-instrumentations/loader/cjs-patch.ts
 var ModulePatch = class {
   packages;
   instrumentator;
@@ -40,8 +63,12 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
+          args[0] = content + OPENAI_API_PROMISE_PATCH;
+          return self.originalCompile.apply(this, args);
+        }
+        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,

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

@@ -1,7 +1,8 @@
 import {
+  OPENAI_API_PROMISE_PATCH,
   getPackageName,
   getPackageVersion
-} from "../chunk-ITP7RAUY.mjs";
+} from "../chunk-MWZXZQUO.mjs";
 
 // src/auto-instrumentations/loader/esm-hook.mts
 import { readFile } from "fs/promises";
@@ -42,8 +43,22 @@ async function resolve(specifier, context, nextResolve) {
   }
   return url;
 }
+function isOpenAIApiPromise(url) {
+  return url.includes("/openai") && url.includes("api-promise");
+}
 async function load(url, context, nextLoad) {
   const result = await nextLoad(url, context);
+  if (isOpenAIApiPromise(url)) {
+    if (result.format === "commonjs") {
+      const parsedUrl = new URL(result.responseURL ?? url);
+      result.source ??= await readFile(parsedUrl);
+    }
+    if (result.source) {
+      result.source = result.source.toString("utf8") + OPENAI_API_PROMISE_PATCH;
+      result.shortCircuit = true;
+    }
+    return result;
+  }
   if (!transformers.has(url)) {
     return result;
   }