braintrust 3.12.0 → 3.14.0

package/dist/instrumentation/index.d.mts

@@ -3487,6 +3487,26 @@ declare const GitMetadataSettings: z.ZodObject<{
     fields?: ("dirty" | "tag" | "commit" | "branch" | "author_name" | "author_email" | "commit_message" | "commit_time" | "git_diff")[] | undefined;
 }>;
 type GitMetadataSettingsType = z.infer<typeof GitMetadataSettings>;
+declare const ObjectReference$1: z.ZodObject<{
+    object_type: z.ZodEnum<["project_logs", "experiment", "dataset", "prompt", "function", "prompt_session"]>;
+    object_id: z.ZodString;
+    id: z.ZodString;
+    _xact_id: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNull]>>;
+    created: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNull]>>;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    object_type: "function" | "experiment" | "dataset" | "prompt" | "prompt_session" | "project_logs";
+    object_id: string;
+    created?: string | null | undefined;
+    _xact_id?: string | null | undefined;
+}, {
+    id: string;
+    object_type: "function" | "experiment" | "dataset" | "prompt" | "prompt_session" | "project_logs";
+    object_id: string;
+    created?: string | null | undefined;
+    _xact_id?: string | null | undefined;
+}>;
+type ObjectReferenceType$1 = z.infer<typeof ObjectReference$1>;
 declare const Prompt$1: z.ZodObject<{
     id: z.ZodString;
     _xact_id: z.ZodString;
@@ -6117,6 +6137,7 @@ interface InstrumentationIntegrationsConfig {
     cursor?: boolean;
     cursorSDK?: boolean;
     flue?: boolean;
+    mastra?: boolean;
     openAIAgents?: boolean;
     openrouter?: boolean;
     openrouterAgent?: boolean;
@@ -6322,6 +6343,7 @@ declare const ObjectReference: z.ZodObject<{
     created?: string | null | undefined;
     _xact_id?: string | null | undefined;
 }>;
+type ObjectReferenceType = z.infer<typeof ObjectReference>;
 
 type IdField = {
     id: string;
@@ -6342,7 +6364,7 @@ type OtherExperimentLogFields = {
     metadata: Record<string, unknown>;
     metrics: Record<string, unknown>;
     datasetRecordId: string;
-    origin: z.infer<typeof ObjectReference>;
+    origin: ObjectReferenceType;
     span_attributes: Record<string, unknown>;
     [ASYNC_SCORING_CONTROL_FIELD]: AsyncScoringControlType;
     [MERGE_PATHS_FIELD]: string[][];
@@ -6384,6 +6406,7 @@ type DatasetEvent = {
     tags?: string[];
     metadata?: unknown;
     created?: string;
+    origin?: ObjectReferenceType;
     id: string;
     dataset_id: string;
 } & ({
@@ -6715,17 +6738,15 @@ interface PromptKey {
     id?: string;
 }
 /**
- * A two-layer cache for Braintrust prompts with both in-memory and filesystem storage.
+ * A configurable cache for Braintrust prompts with optional in-memory and filesystem storage.
  *
- * This cache implements either a one or two-layer caching strategy:
- * 1. A fast in-memory LRU cache for frequently accessed prompts.
- * 2. An optional persistent filesystem-based cache that serves as a backing store.
+ * This cache can use either layer independently, both layers together, or no layers.
  */
 declare class PromptCache {
-    private readonly memoryCache;
+    private readonly memoryCache?;
     private readonly diskCache?;
     constructor(options: {
-        memoryCache: LRUCache<string, Prompt>;
+        memoryCache?: LRUCache<string, Prompt>;
         diskCache?: DiskCache<Prompt>;
     });
     /**
@@ -6752,10 +6773,10 @@ interface ParametersKey {
     id?: string;
 }
 declare class ParametersCache {
-    private readonly memoryCache;
+    private readonly memoryCache?;
     private readonly diskCache?;
     constructor(options: {
-        memoryCache: LRUCache<string, RemoteEvalParameters>;
+        memoryCache?: LRUCache<string, RemoteEvalParameters>;
         diskCache?: DiskCache<RemoteEvalParameters>;
     });
     get(key: ParametersKey): Promise<RemoteEvalParameters | undefined>;
@@ -6775,9 +6796,15 @@ declare class ParametersCache {
 interface CachedSpan {
     input?: unknown;
     output?: unknown;
+    expected?: unknown;
+    error?: unknown;
+    scores?: Record<string, unknown>;
+    metrics?: Record<string, unknown>;
     metadata?: Record<string, unknown>;
+    tags?: string[];
     span_id: string;
     span_parents?: string[];
+    is_root?: boolean | null;
     span_attributes?: {
         name?: string;
         type?: string;
@@ -7310,6 +7337,13 @@ interface LogOptions<IsAsyncFlush> {
     linkArgs?: LinkArgs;
 }
 type PromiseUnless<B, R> = B extends true ? R : Promise<Awaited<R>>;
+type DatasetPipelineDeferredJSONAttachmentHook = (data: unknown, options?: {
+    filename?: string;
+    pretty?: boolean;
+}) => object;
+declare global {
+    var __BT_DATASET_PIPELINE_DEFER_JSON_ATTACHMENT__: DatasetPipelineDeferredJSONAttachmentHook | undefined;
+}
 interface ParentSpanIds {
     spanId: string;
     rootSpanId: string;
@@ -7480,7 +7514,7 @@ interface LoginOptions {
      */
     appUrl?: string;
     /**
-     * The API key to use. If the parameter is not specified, will try to use the `BRAINTRUST_API_KEY` environment variable.
+     * The API key to use. If the parameter is not specified, will try to use the `BRAINTRUST_API_KEY` environment variable. In Node.js, if that is unset, will try the nearest `.env.braintrust` file in the current working directory or parent directories.
      */
     apiKey?: string;
     /**
@@ -7719,17 +7753,19 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
      * about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the
      * `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any
      * JSON-serializable type, but its keys must be strings.
+     * @param event.origin (Optional) a reference to the source object this dataset record was derived from.
      * @param event.id (Optional) a unique identifier for the event. If you don't provide one, Braintrust will generate one for you.
      * @param event.output: (Deprecated) The output of your application. Use `expected` instead.
      * @returns The `id` of the logged record.
      */
-    insert({ input, expected, metadata, tags, id, output, }: {
+    insert({ input, expected, metadata, tags, id, output, origin, }: {
         readonly input?: unknown;
         readonly expected?: unknown;
         readonly tags?: string[];
         readonly metadata?: Record<string, unknown>;
         readonly id?: string;
         readonly output?: unknown;
+        readonly origin?: ObjectReferenceType$1;
     }): string;
     /**
      * Update fields of a single record in the dataset. The updated fields will be batched and uploaded behind the scenes.

package/dist/instrumentation/index.d.ts

@@ -3487,6 +3487,26 @@ declare const GitMetadataSettings: z.ZodObject<{
     fields?: ("dirty" | "tag" | "commit" | "branch" | "author_name" | "author_email" | "commit_message" | "commit_time" | "git_diff")[] | undefined;
 }>;
 type GitMetadataSettingsType = z.infer<typeof GitMetadataSettings>;
+declare const ObjectReference$1: z.ZodObject<{
+    object_type: z.ZodEnum<["project_logs", "experiment", "dataset", "prompt", "function", "prompt_session"]>;
+    object_id: z.ZodString;
+    id: z.ZodString;
+    _xact_id: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNull]>>;
+    created: z.ZodOptional<z.ZodUnion<[z.ZodString, z.ZodNull]>>;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    object_type: "function" | "experiment" | "dataset" | "prompt" | "prompt_session" | "project_logs";
+    object_id: string;
+    created?: string | null | undefined;
+    _xact_id?: string | null | undefined;
+}, {
+    id: string;
+    object_type: "function" | "experiment" | "dataset" | "prompt" | "prompt_session" | "project_logs";
+    object_id: string;
+    created?: string | null | undefined;
+    _xact_id?: string | null | undefined;
+}>;
+type ObjectReferenceType$1 = z.infer<typeof ObjectReference$1>;
 declare const Prompt$1: z.ZodObject<{
     id: z.ZodString;
     _xact_id: z.ZodString;
@@ -6117,6 +6137,7 @@ interface InstrumentationIntegrationsConfig {
     cursor?: boolean;
     cursorSDK?: boolean;
     flue?: boolean;
+    mastra?: boolean;
     openAIAgents?: boolean;
     openrouter?: boolean;
     openrouterAgent?: boolean;
@@ -6322,6 +6343,7 @@ declare const ObjectReference: z.ZodObject<{
     created?: string | null | undefined;
     _xact_id?: string | null | undefined;
 }>;
+type ObjectReferenceType = z.infer<typeof ObjectReference>;
 
 type IdField = {
     id: string;
@@ -6342,7 +6364,7 @@ type OtherExperimentLogFields = {
     metadata: Record<string, unknown>;
     metrics: Record<string, unknown>;
     datasetRecordId: string;
-    origin: z.infer<typeof ObjectReference>;
+    origin: ObjectReferenceType;
     span_attributes: Record<string, unknown>;
     [ASYNC_SCORING_CONTROL_FIELD]: AsyncScoringControlType;
     [MERGE_PATHS_FIELD]: string[][];
@@ -6384,6 +6406,7 @@ type DatasetEvent = {
     tags?: string[];
     metadata?: unknown;
     created?: string;
+    origin?: ObjectReferenceType;
     id: string;
     dataset_id: string;
 } & ({
@@ -6715,17 +6738,15 @@ interface PromptKey {
     id?: string;
 }
 /**
- * A two-layer cache for Braintrust prompts with both in-memory and filesystem storage.
+ * A configurable cache for Braintrust prompts with optional in-memory and filesystem storage.
  *
- * This cache implements either a one or two-layer caching strategy:
- * 1. A fast in-memory LRU cache for frequently accessed prompts.
- * 2. An optional persistent filesystem-based cache that serves as a backing store.
+ * This cache can use either layer independently, both layers together, or no layers.
  */
 declare class PromptCache {
-    private readonly memoryCache;
+    private readonly memoryCache?;
     private readonly diskCache?;
     constructor(options: {
-        memoryCache: LRUCache<string, Prompt>;
+        memoryCache?: LRUCache<string, Prompt>;
         diskCache?: DiskCache<Prompt>;
     });
     /**
@@ -6752,10 +6773,10 @@ interface ParametersKey {
     id?: string;
 }
 declare class ParametersCache {
-    private readonly memoryCache;
+    private readonly memoryCache?;
     private readonly diskCache?;
     constructor(options: {
-        memoryCache: LRUCache<string, RemoteEvalParameters>;
+        memoryCache?: LRUCache<string, RemoteEvalParameters>;
         diskCache?: DiskCache<RemoteEvalParameters>;
     });
     get(key: ParametersKey): Promise<RemoteEvalParameters | undefined>;
@@ -6775,9 +6796,15 @@ declare class ParametersCache {
 interface CachedSpan {
     input?: unknown;
     output?: unknown;
+    expected?: unknown;
+    error?: unknown;
+    scores?: Record<string, unknown>;
+    metrics?: Record<string, unknown>;
     metadata?: Record<string, unknown>;
+    tags?: string[];
     span_id: string;
     span_parents?: string[];
+    is_root?: boolean | null;
     span_attributes?: {
         name?: string;
         type?: string;
@@ -7310,6 +7337,13 @@ interface LogOptions<IsAsyncFlush> {
     linkArgs?: LinkArgs;
 }
 type PromiseUnless<B, R> = B extends true ? R : Promise<Awaited<R>>;
+type DatasetPipelineDeferredJSONAttachmentHook = (data: unknown, options?: {
+    filename?: string;
+    pretty?: boolean;
+}) => object;
+declare global {
+    var __BT_DATASET_PIPELINE_DEFER_JSON_ATTACHMENT__: DatasetPipelineDeferredJSONAttachmentHook | undefined;
+}
 interface ParentSpanIds {
     spanId: string;
     rootSpanId: string;
@@ -7480,7 +7514,7 @@ interface LoginOptions {
      */
     appUrl?: string;
     /**
-     * The API key to use. If the parameter is not specified, will try to use the `BRAINTRUST_API_KEY` environment variable.
+     * The API key to use. If the parameter is not specified, will try to use the `BRAINTRUST_API_KEY` environment variable. In Node.js, if that is unset, will try the nearest `.env.braintrust` file in the current working directory or parent directories.
      */
     apiKey?: string;
     /**
@@ -7719,17 +7753,19 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
      * about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the
      * `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any
      * JSON-serializable type, but its keys must be strings.
+     * @param event.origin (Optional) a reference to the source object this dataset record was derived from.
      * @param event.id (Optional) a unique identifier for the event. If you don't provide one, Braintrust will generate one for you.
      * @param event.output: (Deprecated) The output of your application. Use `expected` instead.
      * @returns The `id` of the logged record.
      */
-    insert({ input, expected, metadata, tags, id, output, }: {
+    insert({ input, expected, metadata, tags, id, output, origin, }: {
         readonly input?: unknown;
         readonly expected?: unknown;
         readonly tags?: string[];
         readonly metadata?: Record<string, unknown>;
         readonly id?: string;
         readonly output?: unknown;
+        readonly origin?: ObjectReferenceType$1;
     }): string;
     /**
      * Update fields of a single record in the dataset. The updated fields will be batched and uploaded behind the scenes.

package/dist/instrumentation/index.js

@@ -125,6 +125,7 @@ var iso = {
   getRepoInfo: async (_settings) => void 0,
   getPastNAncestors: async () => [],
   getEnv: (_name) => void 0,
+  getBraintrustApiKey: async () => void 0,
   getCallerLocation: () => void 0,
   newAsyncLocalStorage: () => new DefaultAsyncLocalStorage(),
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -4018,6 +4019,76 @@ var LRUCache = class {
   }
 };
 
+// src/prompt-cache/cache-config.ts
+var CACHE_LOCATION_ENV_VAR = "BRAINTRUST_CACHE_LOCATION";
+var DEFAULT_CACHE_MEMORY_MAX = 1 << 10;
+var DEFAULT_CACHE_DISK_MAX = 1 << 20;
+var warnedInvalidCacheModeEnvValue = false;
+var warnedUnavailableDiskCacheMode = false;
+function warnInvalidCacheMode(value) {
+  if (warnedInvalidCacheModeEnvValue) {
+    return;
+  }
+  warnedInvalidCacheModeEnvValue = true;
+  debugLogger.warn(
+    `Invalid ${CACHE_LOCATION_ENV_VAR} value "${value}". Expected "mixed", "memory", "disk", or "none". Falling back to "mixed".`
+  );
+}
+function warnUnavailableDiskCache() {
+  if (warnedUnavailableDiskCacheMode) {
+    return;
+  }
+  warnedUnavailableDiskCacheMode = true;
+  debugLogger.warn(
+    `Disk cache is not supported on this platform, so ${CACHE_LOCATION_ENV_VAR}="disk" disables prompt and parameters caching.`
+  );
+}
+function parseCacheMode() {
+  const value = isomorph_default.getEnv(CACHE_LOCATION_ENV_VAR);
+  const normalized = value?.trim().toLowerCase();
+  if (!normalized) {
+    return "mixed";
+  }
+  if (normalized === "mixed" || normalized === "memory" || normalized === "disk" || normalized === "none") {
+    return normalized;
+  }
+  warnInvalidCacheMode(value ?? "");
+  return "mixed";
+}
+function parsePositiveIntegerEnv(envVar, defaultValue) {
+  const value = Number(isomorph_default.getEnv(envVar));
+  return Number.isInteger(value) && value > 0 ? value : defaultValue;
+}
+function createCacheLayers({
+  memoryMaxEnvVar,
+  diskCacheDirEnvVar,
+  diskMaxEnvVar,
+  getDefaultDiskCacheDir
+}) {
+  const mode = parseCacheMode();
+  const memoryCache = mode === "mixed" || mode === "memory" ? new LRUCache({
+    max: parsePositiveIntegerEnv(
+      memoryMaxEnvVar,
+      DEFAULT_CACHE_MEMORY_MAX
+    )
+  }) : void 0;
+  let diskCache;
+  if (mode === "mixed" || mode === "disk") {
+    if (canUseDiskCache()) {
+      diskCache = new DiskCache({
+        cacheDir: isomorph_default.getEnv(diskCacheDirEnvVar) ?? getDefaultDiskCacheDir(),
+        max: parsePositiveIntegerEnv(diskMaxEnvVar, DEFAULT_CACHE_DISK_MAX)
+      });
+    } else if (mode === "disk") {
+      warnUnavailableDiskCache();
+    }
+  }
+  if (diskCache) {
+    return { memoryCache, diskCache };
+  }
+  return { memoryCache };
+}
+
 // src/prompt-cache/prompt-cache.ts
 function createCacheKey(key) {
   if (key.id) {
@@ -4045,16 +4116,18 @@ var PromptCache = class {
    */
   async get(key) {
     const cacheKey = createCacheKey(key);
-    const memoryPrompt = this.memoryCache.get(cacheKey);
-    if (memoryPrompt !== void 0) {
-      return memoryPrompt;
+    if (this.memoryCache) {
+      const memoryPrompt = this.memoryCache.get(cacheKey);
+      if (memoryPrompt !== void 0) {
+        return memoryPrompt;
+      }
     }
     if (this.diskCache) {
       const diskPrompt = await this.diskCache.get(cacheKey);
       if (!diskPrompt) {
         return void 0;
       }
-      this.memoryCache.set(cacheKey, diskPrompt);
+      this.memoryCache?.set(cacheKey, diskPrompt);
       return diskPrompt;
     }
     return void 0;
@@ -4069,7 +4142,7 @@ var PromptCache = class {
    */
   async set(key, value) {
     const cacheKey = createCacheKey(key);
-    this.memoryCache.set(cacheKey, value);
+    this.memoryCache?.set(cacheKey, value);
     if (this.diskCache) {
       await this.diskCache.set(cacheKey, value);
     }
@@ -4099,23 +4172,25 @@ var ParametersCache = class {
   }
   async get(key) {
     const cacheKey = createCacheKey2(key);
-    const memoryParams = this.memoryCache.get(cacheKey);
-    if (memoryParams !== void 0) {
-      return memoryParams;
+    if (this.memoryCache) {
+      const memoryParams = this.memoryCache.get(cacheKey);
+      if (memoryParams !== void 0) {
+        return memoryParams;
+      }
     }
     if (this.diskCache) {
       const diskParams = await this.diskCache.get(cacheKey);
       if (!diskParams) {
         return void 0;
       }
-      this.memoryCache.set(cacheKey, diskParams);
+      this.memoryCache?.set(cacheKey, diskParams);
       return diskParams;
     }
     return void 0;
   }
   async set(key, value) {
     const cacheKey = createCacheKey2(key);
-    this.memoryCache.set(cacheKey, value);
+    this.memoryCache?.set(cacheKey, value);
     if (this.diskCache) {
       await this.diskCache.set(cacheKey, value);
     }
@@ -4647,21 +4722,22 @@ var BraintrustState = class _BraintrustState {
       setGlobalDebugLogLevel(void 0);
     }
     this.resetLoginInfo();
-    const memoryCache = new LRUCache({
-      max: Number(isomorph_default.getEnv("BRAINTRUST_PROMPT_CACHE_MEMORY_MAX")) ?? 1 << 10
+    const { memoryCache, diskCache } = createCacheLayers({
+      memoryMaxEnvVar: "BRAINTRUST_PROMPT_CACHE_MEMORY_MAX",
+      diskCacheDirEnvVar: "BRAINTRUST_PROMPT_CACHE_DIR",
+      diskMaxEnvVar: "BRAINTRUST_PROMPT_CACHE_DISK_MAX",
+      getDefaultDiskCacheDir: () => `${isomorph_default.getEnv("HOME") ?? isomorph_default.homedir()}/.braintrust/prompt_cache`
     });
-    const diskCache = canUseDiskCache() ? new DiskCache({
-      cacheDir: isomorph_default.getEnv("BRAINTRUST_PROMPT_CACHE_DIR") ?? `${isomorph_default.getEnv("HOME") ?? isomorph_default.homedir()}/.braintrust/prompt_cache`,
-      max: Number(isomorph_default.getEnv("BRAINTRUST_PROMPT_CACHE_DISK_MAX")) ?? 1 << 20
-    }) : void 0;
     this.promptCache = new PromptCache({ memoryCache, diskCache });
-    const parametersMemoryCache = new LRUCache({
-      max: Number(isomorph_default.getEnv("BRAINTRUST_PARAMETERS_CACHE_MEMORY_MAX")) ?? 1 << 10
+    const {
+      memoryCache: parametersMemoryCache,
+      diskCache: parametersDiskCache
+    } = createCacheLayers({
+      memoryMaxEnvVar: "BRAINTRUST_PARAMETERS_CACHE_MEMORY_MAX",
+      diskCacheDirEnvVar: "BRAINTRUST_PARAMETERS_CACHE_DIR",
+      diskMaxEnvVar: "BRAINTRUST_PARAMETERS_CACHE_DISK_MAX",
+      getDefaultDiskCacheDir: () => `${isomorph_default.getEnv("HOME") ?? isomorph_default.homedir()}/.braintrust/parameters_cache`
     });
-    const parametersDiskCache = canUseDiskCache() ? new DiskCache({
-      cacheDir: isomorph_default.getEnv("BRAINTRUST_PARAMETERS_CACHE_DIR") ?? `${isomorph_default.getEnv("HOME") ?? isomorph_default.homedir()}/.braintrust/parameters_cache`,
-      max: Number(isomorph_default.getEnv("BRAINTRUST_PARAMETERS_CACHE_DISK_MAX")) ?? 1 << 20
-    }) : void 0;
     this.parametersCache = new ParametersCache({
       memoryCache: parametersMemoryCache,
       diskCache: parametersDiskCache
@@ -6548,10 +6624,11 @@ function initLogger(options = {}) {
 async function loginToState(options = {}) {
   const {
     appUrl = isomorph_default.getEnv("BRAINTRUST_APP_URL") || "https://www.braintrust.dev",
-    apiKey = isomorph_default.getEnv("BRAINTRUST_API_KEY"),
+    apiKey: apiKeyArg,
     orgName = isomorph_default.getEnv("BRAINTRUST_ORG_NAME"),
     fetch: fetch2 = globalThis.fetch
   } = options || {};
+  const apiKey = apiKeyArg !== void 0 ? apiKeyArg : await isomorph_default.getBraintrustApiKey();
   const appPublicUrl = isomorph_default.getEnv("BRAINTRUST_APP_PUBLIC_URL") || appUrl;
   const state = new BraintrustState(options);
   state.resetLoginInfo();
@@ -7482,9 +7559,15 @@ var SpanImpl = class _SpanImpl {
       const cachedSpan = {
         input: partialRecord.input,
         output: partialRecord.output,
+        expected: partialRecord.expected,
+        error: partialRecord.error,
+        scores: partialRecord.scores,
+        metrics: partialRecord.metrics,
         metadata: partialRecord.metadata,
+        tags: partialRecord.tags,
         span_id: this._spanId,
         span_parents: this._spanParents,
+        is_root: this._spanId === this._rootSpanId,
         span_attributes: partialRecord.span_attributes
       };
       this._state.spanCache.queueWrite(
@@ -7820,6 +7903,7 @@ var Dataset2 = class extends ObjectFetcher {
     metadata,
     tags,
     output,
+    origin,
     isMerge
   }) {
     return new LazyValue(async () => {
@@ -7834,6 +7918,7 @@ var Dataset2 = class extends ObjectFetcher {
         created: !isMerge ? (/* @__PURE__ */ new Date()).toISOString() : void 0,
         //if we're merging/updating an event we will not add this ts
         metadata,
+        origin,
         ...!!isMerge ? {
           [IS_MERGE_FIELD]: true
         } : {}
@@ -7853,6 +7938,7 @@ var Dataset2 = class extends ObjectFetcher {
    * about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the
    * `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any
    * JSON-serializable type, but its keys must be strings.
+   * @param event.origin (Optional) a reference to the source object this dataset record was derived from.
    * @param event.id (Optional) a unique identifier for the event. If you don't provide one, Braintrust will generate one for you.
    * @param event.output: (Deprecated) The output of your application. Use `expected` instead.
    * @returns The `id` of the logged record.
@@ -7863,7 +7949,8 @@ var Dataset2 = class extends ObjectFetcher {
     metadata,
     tags,
     id,
-    output
+    output,
+    origin
   }) {
     this.validateEvent({ metadata, expected, output, tags });
     const rowId = id || (0, import_uuid2.v4)();
@@ -7875,6 +7962,7 @@ var Dataset2 = class extends ObjectFetcher {
         metadata,
         tags,
         output,
+        origin,
         isMerge: false
       })
     );
@@ -23621,7 +23709,7 @@ var BraintrustPlugin = class extends BasePlugin {
     this.config = config;
   }
   onEnable() {
-    const integrations = this.config.integrations || {};
+    const integrations = this.config.integrations ?? {};
     if (integrations.openai !== false) {
       this.openaiPlugin = new OpenAIPlugin();
       this.openaiPlugin.enable();
@@ -23686,7 +23774,7 @@ var BraintrustPlugin = class extends BasePlugin {
       this.genkitPlugin = new GenkitPlugin();
       this.genkitPlugin.enable();
     }
-    if (getIntegrationConfig(integrations, "gitHubCopilot") !== false) {
+    if (integrations.gitHubCopilot !== false) {
       this.gitHubCopilotPlugin = new GitHubCopilotPlugin();
       this.gitHubCopilotPlugin.enable();
     }
@@ -23799,6 +23887,7 @@ var envIntegrationAliases = {
   cursorsdk: "cursorSDK",
   flue: "flue",
   "flue-runtime": "flue",
+  mastra: "mastra",
   "openai-agents": "openAIAgents",
   openaiagents: "openAIAgents",
   "openai-agents-core": "openAIAgents",
@@ -23841,6 +23930,7 @@ function getDefaultInstrumentationIntegrations() {
     cursor: true,
     cursorSDK: true,
     flue: true,
+    mastra: true,
     openAIAgents: true,
     openrouter: true,
     openrouterAgent: true,