braintrust 3.1.0 → 3.2.0

package/dist/auto-instrumentations/hook.mjs

@@ -0,0 +1,378 @@
+import {
+  getPackageVersion
+} from "./chunk-ZEC7BCL4.mjs";
+
+// src/auto-instrumentations/hook.mts
+import { register } from "node:module";
+
+// src/auto-instrumentations/configs/openai.ts
+var openaiConfigs = [
+  // Chat Completions
+  {
+    channelName: "chat.completions.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Embeddings
+  {
+    channelName: "embeddings.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/embeddings.mjs"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Chat Completions Parse
+  {
+    channelName: "beta.chat.completions.parse",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/beta/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "parse",
+      kind: "Async"
+    }
+  },
+  // Moderations
+  {
+    channelName: "moderations.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/moderations.mjs"
+    },
+    functionQuery: {
+      className: "Moderations",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Chat Completions Stream
+  {
+    channelName: "beta.chat.completions.stream",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/beta/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  // Responses API (v4.87.0+)
+  {
+    channelName: "responses.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: "responses.stream",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: "responses.parse",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "parse",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/anthropic.ts
+var anthropicConfigs = [
+  // Messages API - create (supports streaming via stream=true parameter)
+  {
+    channelName: "messages.create",
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.60.0",
+      filePath: "resources/messages.mjs"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Messages API - create (supports streaming via stream=true parameter)
+  {
+    channelName: "beta.messages.create",
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.60.0",
+      filePath: "resources/beta/messages/messages.mjs"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/ai-sdk.ts
+var aiSDKConfigs = [
+  // generateText - async function
+  {
+    channelName: "generateText",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "generateText",
+      kind: "Async"
+    }
+  },
+  // streamText - async function
+  {
+    channelName: "streamText",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamText",
+      kind: "Async"
+    }
+  },
+  // generateObject - async function
+  {
+    channelName: "generateObject",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "generateObject",
+      kind: "Async"
+    }
+  },
+  // streamObject - async function
+  {
+    channelName: "streamObject",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamObject",
+      kind: "Async"
+    }
+  },
+  // Agent.generate - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: "Agent.generate",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0 <6.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  // Agent.stream - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: "Agent.stream",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0 <6.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "stream",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/claude-agent-sdk.ts
+var claudeAgentSDKConfigs = [
+  // query - Main entry point for agent interactions (top-level exported async generator function)
+  {
+    channelName: "query",
+    module: {
+      name: "@anthropic-ai/claude-agent-sdk",
+      versionRange: ">=0.1.0",
+      filePath: "sdk.mjs"
+    },
+    functionQuery: {
+      functionName: "query",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/google-genai.ts
+var googleGenAIConfigs = [
+  // Models.generateContentInternal - The actual class method (Node.js entry point)
+  // Note: generateContent is an arrow function property that calls this internal method
+  {
+    channelName: "models.generateContent",
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.0.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "generateContentInternal",
+      kind: "Async"
+    }
+  },
+  // Models.generateContentStreamInternal - The actual class method (Node.js entry point)
+  // Note: generateContentStream is an arrow function property that calls this internal method
+  {
+    channelName: "models.generateContentStream",
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.0.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "generateContentStreamInternal",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/loader/cjs-patch.ts
+import {
+  create
+} from "@apm-js-collab/code-transformer";
+import * as Module from "node:module";
+import { sep } from "node:path";
+import moduleDetailsFromPath from "module-details-from-path";
+var ModulePatch = class {
+  packages;
+  instrumentator;
+  originalCompile;
+  constructor({
+    instrumentations = []
+  } = {}) {
+    this.packages = new Set(instrumentations.map((i) => i.module.name));
+    this.instrumentator = create(instrumentations);
+    this.originalCompile = Module.prototype._compile;
+  }
+  /**
+   * Patches the Node.js module class method that is responsible for compiling code.
+   * If a module is found that has an instrumentator, it will transform the code before compiling it
+   * with tracing channel methods.
+   */
+  patch() {
+    const self = this;
+    Module.prototype._compile = function wrappedCompile(...args) {
+      const [content, filename] = args;
+      const normalizedForPlatform = filename.split("/").join(sep);
+      const resolvedModule = moduleDetailsFromPath(normalizedForPlatform);
+      if (resolvedModule && self.packages.has(resolvedModule.name)) {
+        const version = getPackageVersion(resolvedModule.basedir);
+        const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        const transformer = self.instrumentator.getTransformer(
+          resolvedModule.name,
+          version,
+          normalizedModulePath
+        );
+        if (transformer) {
+          try {
+            const transformedCode = transformer.transform(content, "unknown");
+            args[0] = transformedCode?.code;
+          } catch (error) {
+            console.warn(`Error transforming module ${filename}:`, error);
+          } finally {
+            transformer.free();
+          }
+        }
+      }
+      return self.originalCompile.apply(this, args);
+    };
+  }
+  /**
+   * Restores the original Module.prototype._compile method
+   * **Note**: This is intended to be used in testing only.
+   */
+  unpatch() {
+    Module.prototype._compile = this.originalCompile;
+  }
+};
+
+// src/auto-instrumentations/hook.mts
+var allConfigs = [
+  ...openaiConfigs,
+  ...anthropicConfigs,
+  ...aiSDKConfigs,
+  ...claudeAgentSDKConfigs,
+  ...googleGenAIConfigs
+];
+register("./loader/esm-hook.mjs", {
+  parentURL: import.meta.url,
+  data: { instrumentations: allConfigs }
+});
+try {
+  const patch = new ModulePatch({ instrumentations: allConfigs });
+  patch.patch();
+  if (process.env.DEBUG === "@braintrust*" || process.env.DEBUG === "*") {
+    console.log(
+      "[Braintrust] Auto-instrumentation active (ESM + CJS) for:",
+      allConfigs.map((c) => c.channelName).join(", ")
+    );
+  }
+} catch (err) {
+  if (process.env.DEBUG === "@braintrust*" || process.env.DEBUG === "*") {
+    console.log(
+      "[Braintrust] Auto-instrumentation active (ESM only) for:",
+      allConfigs.map((c) => c.channelName).join(", ")
+    );
+    console.error("[Braintrust] CJS patch failed:", err);
+  }
+}

package/dist/auto-instrumentations/index.cjs

@@ -0,0 +1,318 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/auto-instrumentations/index.ts
+var index_exports = {};
+__export(index_exports, {
+  aiSDKConfigs: () => aiSDKConfigs,
+  anthropicConfigs: () => anthropicConfigs,
+  claudeAgentSDKConfigs: () => claudeAgentSDKConfigs,
+  googleGenAIConfigs: () => googleGenAIConfigs,
+  openaiConfigs: () => openaiConfigs
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/auto-instrumentations/configs/openai.ts
+var openaiConfigs = [
+  // Chat Completions
+  {
+    channelName: "chat.completions.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Embeddings
+  {
+    channelName: "embeddings.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/embeddings.mjs"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Chat Completions Parse
+  {
+    channelName: "beta.chat.completions.parse",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/beta/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "parse",
+      kind: "Async"
+    }
+  },
+  // Moderations
+  {
+    channelName: "moderations.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/moderations.mjs"
+    },
+    functionQuery: {
+      className: "Moderations",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Chat Completions Stream
+  {
+    channelName: "beta.chat.completions.stream",
+    module: {
+      name: "openai",
+      versionRange: ">=4.0.0",
+      filePath: "resources/beta/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  // Responses API (v4.87.0+)
+  {
+    channelName: "responses.create",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: "responses.stream",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "stream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: "responses.parse",
+    module: {
+      name: "openai",
+      versionRange: ">=4.87.0",
+      filePath: "resources/responses/responses.mjs"
+    },
+    functionQuery: {
+      className: "Responses",
+      methodName: "parse",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/anthropic.ts
+var anthropicConfigs = [
+  // Messages API - create (supports streaming via stream=true parameter)
+  {
+    channelName: "messages.create",
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.60.0",
+      filePath: "resources/messages.mjs"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  // Beta Messages API - create (supports streaming via stream=true parameter)
+  {
+    channelName: "beta.messages.create",
+    module: {
+      name: "@anthropic-ai/sdk",
+      versionRange: ">=0.60.0",
+      filePath: "resources/beta/messages/messages.mjs"
+    },
+    functionQuery: {
+      className: "Messages",
+      methodName: "create",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/ai-sdk.ts
+var aiSDKConfigs = [
+  // generateText - async function
+  {
+    channelName: "generateText",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "generateText",
+      kind: "Async"
+    }
+  },
+  // streamText - async function
+  {
+    channelName: "streamText",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamText",
+      kind: "Async"
+    }
+  },
+  // generateObject - async function
+  {
+    channelName: "generateObject",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "generateObject",
+      kind: "Async"
+    }
+  },
+  // streamObject - async function
+  {
+    channelName: "streamObject",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      functionName: "streamObject",
+      kind: "Async"
+    }
+  },
+  // Agent.generate - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: "Agent.generate",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0 <6.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  // Agent.stream - async method (v3-v5 only, Agent structure changed in v6)
+  {
+    channelName: "Agent.stream",
+    module: {
+      name: "ai",
+      versionRange: ">=3.0.0 <6.0.0",
+      filePath: "dist/index.mjs"
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "stream",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/claude-agent-sdk.ts
+var claudeAgentSDKConfigs = [
+  // query - Main entry point for agent interactions (top-level exported async generator function)
+  {
+    channelName: "query",
+    module: {
+      name: "@anthropic-ai/claude-agent-sdk",
+      versionRange: ">=0.1.0",
+      filePath: "sdk.mjs"
+    },
+    functionQuery: {
+      functionName: "query",
+      kind: "Async"
+    }
+  }
+];
+
+// src/auto-instrumentations/configs/google-genai.ts
+var googleGenAIConfigs = [
+  // Models.generateContentInternal - The actual class method (Node.js entry point)
+  // Note: generateContent is an arrow function property that calls this internal method
+  {
+    channelName: "models.generateContent",
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.0.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "generateContentInternal",
+      kind: "Async"
+    }
+  },
+  // Models.generateContentStreamInternal - The actual class method (Node.js entry point)
+  // Note: generateContentStream is an arrow function property that calls this internal method
+  {
+    channelName: "models.generateContentStream",
+    module: {
+      name: "@google/genai",
+      versionRange: ">=1.0.0",
+      filePath: "dist/node/index.mjs"
+    },
+    functionQuery: {
+      className: "Models",
+      methodName: "generateContentStreamInternal",
+      kind: "Async"
+    }
+  }
+];
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  aiSDKConfigs,
+  anthropicConfigs,
+  claudeAgentSDKConfigs,
+  googleGenAIConfigs,
+  openaiConfigs
+});