braintrust 3.27.0 → 3.29.0

package/dist/{chunk-YKD22IMR.mjs → chunk-M6XPNJC4.mjs}

@@ -10,11 +10,13 @@ import {
   cloudflareThinkChannels,
   cohereChannels,
   cursorSDKChannels,
+  debugLogger,
   detectSpanOriginEnvironment,
   flueChannels,
   genkitChannels,
   genkitCoreChannels,
   getDefaultInstrumentationIntegrations,
+  getEnvDebugLogLevel,
   getHuggingFaceTransformersPipelineInfo,
   getSpanInstrumentationName,
   gitHubCopilotChannels,
@@ -32,6 +34,7 @@ import {
   langSmithChannels,
   mergeSpanOriginContext,
   mistralChannels,
+  normalizeDebugLogLevelOption,
   ollamaChannels,
   openAIAgentsCoreChannels,
   openAIChannels,
@@ -41,129 +44,18 @@ import {
   piCodingAgentChannels,
   readDisabledInstrumentationEnvConfig,
   registerHuggingFaceTransformersPipeline,
-  setGlobalHookErrorReporter,
+  resetDebugLoggerForTests,
+  setDebugLogStateResolver,
+  setGlobalDebugLogLevel,
   smithyClientChannels,
   smithyCoreChannels,
   strandsAgentSDKChannels,
+  voyageAIChannels,
   withSpanInstrumentationName
-} from "./chunk-CZM5JIQL.mjs";
+} from "./chunk-7FA6VP2S.mjs";
 
 // src/id-gen.ts
 import { v4 as uuidv4 } from "uuid";
-
-// src/debug-logger.ts
-var PREFIX = "[braintrust]";
-var DEBUG_LOG_LEVEL_SYMBOL = /* @__PURE__ */ Symbol.for("braintrust-debug-log-level");
-var LOG_LEVEL_PRIORITY = {
-  error: 0,
-  warn: 1,
-  info: 2,
-  debug: 3
-};
-var hasWarnedAboutInvalidEnvValue = false;
-var debugLogStateResolver = void 0;
-function warnInvalidEnvValue(value) {
-  if (hasWarnedAboutInvalidEnvValue) {
-    return;
-  }
-  hasWarnedAboutInvalidEnvValue = true;
-  console.warn(
-    PREFIX,
-    `Invalid BRAINTRUST_DEBUG_LOG_LEVEL value "${value}". Expected "error", "warn", "info", or "debug".`
-  );
-}
-function normalizeDebugLogLevelOption(option) {
-  if (option === false) {
-    return void 0;
-  }
-  if (option === "error" || option === "warn" || option === "info" || option === "debug") {
-    return option;
-  }
-  throw new Error(
-    `Invalid debugLogLevel value "${option}". Expected false, "error", "warn", "info", or "debug".`
-  );
-}
-function parseDebugLogLevelEnv(value) {
-  if (!value) {
-    return void 0;
-  }
-  if (value === "error" || value === "warn" || value === "info" || value === "debug") {
-    return value;
-  }
-  warnInvalidEnvValue(value);
-  return void 0;
-}
-function getEnvDebugLogLevel() {
-  return parseDebugLogLevelEnv(isomorph_default.getEnv("BRAINTRUST_DEBUG_LOG_LEVEL"));
-}
-function setGlobalDebugLogLevel(level) {
-  globalThis[DEBUG_LOG_LEVEL_SYMBOL] = level;
-}
-function resetDebugLoggerForTests() {
-  hasWarnedAboutInvalidEnvValue = false;
-  setGlobalDebugLogLevel(void 0);
-}
-function setDebugLogStateResolver(resolver) {
-  debugLogStateResolver = resolver;
-}
-function resolveDebugLogLevel(state) {
-  const stateLevel = state?.getDebugLogLevel?.();
-  const hasStateOverride = state?.hasDebugLogLevelOverride?.() ?? false;
-  if (hasStateOverride) {
-    return stateLevel;
-  }
-  const globalLevel = (
-    // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-    globalThis[DEBUG_LOG_LEVEL_SYMBOL]
-  );
-  if (globalLevel !== void 0) {
-    return globalLevel === false ? void 0 : globalLevel;
-  }
-  return getEnvDebugLogLevel();
-}
-function emit(method, state, args) {
-  const level = resolveDebugLogLevel(state);
-  if (!level || LOG_LEVEL_PRIORITY[method] > LOG_LEVEL_PRIORITY[level]) {
-    return;
-  }
-  if (method === "info") {
-    console.log(PREFIX, ...args);
-  } else if (method === "debug") {
-    console.debug(PREFIX, ...args);
-  } else if (method === "warn") {
-    console.warn(PREFIX, ...args);
-  } else {
-    console.error(PREFIX, ...args);
-  }
-}
-function createDebugLogger(state) {
-  const resolveState = () => state ?? debugLogStateResolver?.();
-  return {
-    info(...args) {
-      emit("info", resolveState(), args);
-    },
-    debug(...args) {
-      emit("debug", resolveState(), args);
-    },
-    warn(...args) {
-      emit("warn", resolveState(), args);
-    },
-    error(...args) {
-      emit("error", resolveState(), args);
-    }
-  };
-}
-var debugLogger = {
-  ...createDebugLogger(),
-  forState(state) {
-    return createDebugLogger(state);
-  }
-};
-setGlobalHookErrorReporter((error) => {
-  debugLogger.error("Global instrumentation hook error:", error);
-});
-
-// src/id-gen.ts
 var IDGenerator = class {
 };
 var UUIDGenerator = class extends IDGenerator {
@@ -1121,15 +1013,15 @@ function mergeDictsWithPaths({
 function mergeDictsWithPathsHelper({
   mergeInto,
   mergeFrom,
-  path: path2,
+  path: path3,
   mergePaths
 }) {
   Object.entries(mergeFrom).forEach(([k, mergeFromV]) => {
     if (FORBIDDEN_MERGE_KEYS.has(k)) return;
-    const fullPath = path2.concat([k]);
+    const fullPath = path3.concat([k]);
     const fullPathSerialized = JSON.stringify(fullPath);
     const mergeIntoV = recordFind(mergeInto, k);
-    const isSetUnionField = path2.length === 0 && SET_UNION_FIELDS.has(k) && !mergePaths.has(fullPathSerialized);
+    const isSetUnionField = path3.length === 0 && SET_UNION_FIELDS.has(k) && !mergePaths.has(fullPathSerialized);
     if (isSetUnionField && isArray(mergeIntoV) && isArray(mergeFromV)) {
       const seen = /* @__PURE__ */ new Set();
       const combined = [];
@@ -1162,9 +1054,9 @@ function mergeDicts(mergeInto, mergeFrom) {
 function recordFind(m, k) {
   return m[k];
 }
-function getObjValueByPath(row, path2) {
+function getObjValueByPath(row, path3) {
   let curr = row;
-  for (const p of path2) {
+  for (const p of path3) {
     if (!isObjectOrArray(curr)) {
       return null;
     }
@@ -1780,7 +1672,10 @@ var AclObjectType = z6.union([
     "org_member",
     "project_log",
     "org_project",
-    "org_audit_logs"
+    "org_audit_logs",
+    "project_group",
+    "ai_secret",
+    "org_ai_secret"
   ]),
   z6.null()
 ]);
@@ -1909,7 +1804,8 @@ var AsyncScoringState = z6.union([
     token: z6.string(),
     function_ids: z6.array(z6.unknown()),
     skip_logging: z6.union([z6.boolean(), z6.null()]).optional(),
-    triggered_functions: z6.union([z6.record(TriggeredFunctionState), z6.null()]).optional()
+    triggered_functions: z6.union([z6.record(TriggeredFunctionState), z6.null()]).optional(),
+    last_triggered_xact_id: z6.union([z6.string(), z6.number(), z6.null()]).optional()
   }),
   z6.object({ status: z6.literal("disabled") }),
   z6.null(),
@@ -1978,7 +1874,7 @@ var FunctionTypeEnum = z6.enum([
   "parameters",
   "sandbox"
 ]);
-var NullableSavedFunctionId = z6.union([
+var FacetPreprocessorId = z6.union([
   z6.object({
     type: z6.literal("function"),
     id: z6.string(),
@@ -1989,10 +1885,23 @@ var NullableSavedFunctionId = z6.union([
     name: z6.string(),
     function_type: FunctionTypeEnum.optional().default("scorer")
   }),
+  z6.object({ type: z6.literal("inline"), code: z6.string().min(1) }),
   z6.null()
 ]);
+var SavedFunctionId = z6.union([
+  z6.object({
+    type: z6.literal("function"),
+    id: z6.string(),
+    version: z6.string().optional()
+  }),
+  z6.object({
+    type: z6.literal("global"),
+    name: z6.string(),
+    function_type: FunctionTypeEnum.optional().default("scorer")
+  })
+]);
 var TopicMapGenerationSettings = z6.object({
-  algorithm: z6.enum(["hdbscan", "kmeans"]),
+  algorithm: z6.enum(["hdbscan", "kmeans", "community"]),
   dimension_reduction: z6.enum(["umap", "pca", "none"]),
   sample_size: z6.number().int().gt(0).optional(),
   n_clusters: z6.number().int().gt(0).optional(),
@@ -2004,6 +1913,7 @@ var TopicMapGenerationSettings = z6.object({
 var TopicMapData = z6.object({
   type: z6.literal("topic_map"),
   source_facet: z6.string(),
+  source_facet_function: SavedFunctionId.and(z6.unknown()).optional(),
   embedding_model: z6.string(),
   bundle_key: z6.string().optional(),
   report_key: z6.string().optional(),
@@ -2017,7 +1927,7 @@ var TopicMapData = z6.object({
 });
 var BatchedFacetData = z6.object({
   type: z6.literal("batched_facet"),
-  preprocessor: NullableSavedFunctionId.and(z6.unknown()).optional(),
+  preprocessor: FacetPreprocessorId.optional(),
   facets: z6.array(
     z6.object({
       name: z6.string(),
@@ -2278,18 +2188,6 @@ var ObjectReferenceNullish = z6.union([
   }),
   z6.null()
 ]);
-var SavedFunctionId = z6.union([
-  z6.object({
-    type: z6.literal("function"),
-    id: z6.string(),
-    version: z6.string().optional()
-  }),
-  z6.object({
-    type: z6.literal("global"),
-    name: z6.string(),
-    function_type: FunctionTypeEnum.optional().default("scorer")
-  })
-]);
 var DatasetEvent = z6.object({
   id: z6.string(),
   _xact_id: z6.string(),
@@ -2511,7 +2409,7 @@ var ExtendedSavedFunctionId = z6.union([
 ]);
 var FacetData = z6.object({
   type: z6.literal("facet"),
-  preprocessor: NullableSavedFunctionId.and(z6.unknown()).optional(),
+  preprocessor: FacetPreprocessorId.optional(),
   prompt: z6.string(),
   model: z6.string().optional(),
   embedding_model: z6.string().optional(),
@@ -2610,7 +2508,7 @@ var PromptParserNullish = z6.union([
   }),
   z6.null()
 ]);
-var PreprocessorSavedFunctionId = z6.union([
+var PreprocessorId = z6.union([
   z6.object({
     type: z6.literal("function"),
     id: z6.string(),
@@ -2621,6 +2519,7 @@ var PreprocessorSavedFunctionId = z6.union([
     name: z6.string(),
     function_type: z6.literal("preprocessor").optional().default("preprocessor")
   }),
+  z6.object({ type: z6.literal("inline"), code: z6.string().min(1) }),
   z6.null()
 ]);
 var PromptDataNullish = z6.union([
@@ -2628,7 +2527,7 @@ var PromptDataNullish = z6.union([
     prompt: PromptBlockDataNullish,
     options: PromptOptionsNullish,
     parser: PromptParserNullish,
-    preprocessor: PreprocessorSavedFunctionId,
+    preprocessor: PreprocessorId,
     tool_functions: z6.union([z6.array(SavedFunctionId), z6.null()]),
     template_format: z6.union([
       z6.enum(["mustache", "nunjucks", "none"]),
@@ -2830,7 +2729,7 @@ var PromptData = z6.object({
   prompt: PromptBlockDataNullish,
   options: PromptOptionsNullish,
   parser: PromptParserNullish,
-  preprocessor: PreprocessorSavedFunctionId,
+  preprocessor: PreprocessorId,
   tool_functions: z6.union([z6.array(SavedFunctionId), z6.null()]),
   template_format: z6.union([
     z6.enum(["mustache", "nunjucks", "none"]),
@@ -3016,6 +2915,19 @@ var MessageRole = z6.enum([
   "model",
   "developer"
 ]);
+var NullableSavedFunctionId = z6.union([
+  z6.object({
+    type: z6.literal("function"),
+    id: z6.string(),
+    version: z6.string().optional()
+  }),
+  z6.object({
+    type: z6.literal("global"),
+    name: z6.string(),
+    function_type: FunctionTypeEnum.optional().default("scorer")
+  }),
+  z6.null()
+]);
 var ObjectReference = z6.object({
   object_type: z6.enum([
     "project_logs",
@@ -3037,6 +2949,7 @@ var TraceScope = z6.object({
 });
 var OnlineScoreConfig = z6.union([
   z6.object({
+    status: AutomationStatus.optional(),
     sampling_rate: z6.number().gte(0).lte(1),
     scorers: z6.array(SavedFunctionId),
     btql_filter: z6.union([z6.string(), z6.null()]).optional(),
@@ -3117,6 +3030,72 @@ var Project = z6.object({
   user_id: z6.union([z6.string(), z6.null()]).optional(),
   settings: ProjectSettings.optional()
 });
+var WindowedAutomationConfig = z6.object({
+  event_type: z6.literal("windowed"),
+  product_origin: z6.union([z6.literal("patterns"), z6.null()]).optional(),
+  status: AutomationStatus.optional(),
+  threshold: z6.object({
+    calculation: z6.object({
+      type: z6.literal("btql"),
+      btql_query: z6.string().min(1),
+      output: z6.object({
+        type: z6.literal("scalar"),
+        value_column: z6.string().min(1)
+      })
+    }),
+    policy: z6.object({
+      condition: z6.object({
+        type: z6.literal("threshold"),
+        operator: z6.enum(["lt", "lte", "gt", "gte", "eq", "neq"]),
+        threshold: z6.number()
+      }),
+      pending_seconds: z6.number().int().gte(0).lte(2592e3),
+      no_data_behavior: z6.enum(["keep_last", "resolve", "alert"]),
+      renotify_interval_seconds: z6.union([z6.number(), z6.null()]).optional(),
+      notify_on_recovery: z6.boolean().optional().default(true)
+    })
+  }).optional(),
+  window: z6.object({
+    window_seconds: z6.number().int().gte(1).lte(2592e3),
+    schedule: z6.union([
+      z6.object({
+        type: z6.literal("interval"),
+        evaluation_interval_seconds: z6.number().int().gte(1).lte(2592e3)
+      }),
+      z6.object({
+        type: z6.literal("cron"),
+        cron_expression: z6.string().min(1),
+        timezone: z6.union([z6.string(), z6.null()]).optional()
+      })
+    ]),
+    evaluation_delay_seconds: z6.number().int().gte(0).lte(2592e3)
+  }),
+  loop: z6.object({
+    prompt: z6.string().min(1).max(1e4),
+    include_trigger_input: z6.boolean().optional().default(false),
+    agent_slug: z6.string().min(1),
+    auto_approve_tools: z6.array(z6.string().min(1)).optional().default([]),
+    harness: z6.enum(["native", "codex", "claude-code"]).optional(),
+    model: z6.string().min(1).optional(),
+    reasoning_effort: z6.enum(["none", "minimal", "low", "medium", "high", "xhigh", "max"]).optional()
+  }).optional(),
+  actions: z6.array(
+    z6.union([
+      z6.object({
+        type: z6.literal("webhook"),
+        url: z6.string(),
+        formatting_prompt: z6.string().min(1).max(1e4).optional()
+      }),
+      z6.object({
+        type: z6.literal("slack"),
+        workspace_id: z6.string(),
+        channel: z6.string(),
+        message_template: z6.string().optional(),
+        formatting_prompt: z6.string().min(1).max(1e4).optional()
+      })
+    ])
+  ).max(20).optional().default([])
+});
 var TopicAutomationFacetModel = z6.union([
   z6.enum(["brain-facet-latest", "brain-facet-1", "brain-facet-2"]),
   z6.null()
@@ -3158,7 +3137,8 @@ var TopicDigestAutomationConfig = z6.object({
     type: z6.literal("slack"),
     workspace_id: z6.string(),
     channel: z6.string(),
-    message_template: z6.string().optional()
+    message_template: z6.string().optional(),
+    formatting_prompt: z6.string().min(1).max(1e4).optional()
   }),
   topic_map_function_ids: z6.array(z6.string()).max(10).optional()
 });
@@ -3172,15 +3152,21 @@ var ProjectAutomation = z6.object({
   config: z6.union([
     z6.object({
       event_type: z6.literal("logs"),
+      status: AutomationStatus.optional(),
       btql_filter: z6.string(),
       interval_seconds: z6.number().gte(1).lte(2592e3),
       action: z6.union([
-        z6.object({ type: z6.literal("webhook"), url: z6.string() }),
+        z6.object({
+          type: z6.literal("webhook"),
+          url: z6.string(),
+          formatting_prompt: z6.string().min(1).max(1e4).optional()
+        }),
         z6.object({
           type: z6.literal("slack"),
           workspace_id: z6.string(),
           channel: z6.string(),
-          message_template: z6.string().optional()
+          message_template: z6.string().optional(),
+          formatting_prompt: z6.string().min(1).max(1e4).optional()
         })
       ])
     }),
@@ -3231,21 +3217,38 @@ var ProjectAutomation = z6.object({
     }),
     z6.object({
       event_type: z6.literal("environment_update"),
+      status: AutomationStatus.optional(),
       environment_filter: z6.array(z6.string()).optional(),
       action: z6.union([
-        z6.object({ type: z6.literal("webhook"), url: z6.string() }),
+        z6.object({
+          type: z6.literal("webhook"),
+          url: z6.string(),
+          formatting_prompt: z6.string().min(1).max(1e4).optional()
+        }),
         z6.object({
           type: z6.literal("slack"),
           workspace_id: z6.string(),
           channel: z6.string(),
-          message_template: z6.string().optional()
+          message_template: z6.string().optional(),
+          formatting_prompt: z6.string().min(1).max(1e4).optional()
         })
       ])
     }),
+    WindowedAutomationConfig,
     TopicAutomationConfig,
     TopicDigestAutomationConfig
   ])
 });
+var ProjectGroup = z6.object({
+  id: z6.string().uuid(),
+  org_id: z6.string().uuid(),
+  user_id: z6.union([z6.string(), z6.null()]).optional(),
+  created: z6.union([z6.string(), z6.null()]).optional(),
+  name: z6.string(),
+  description: z6.union([z6.string(), z6.null()]).optional(),
+  deleted_at: z6.union([z6.string(), z6.null()]).optional(),
+  member_projects: z6.array(z6.string().uuid()).max(1e4)
+});
 var ProjectLogsEvent = z6.object({
   id: z6.string(),
   _xact_id: z6.string(),
@@ -3463,7 +3466,8 @@ var RunEval = z6.object({
       dataset_environment: z6.union([z6.string(), z6.null()]).optional(),
       _internal_btql: z6.union([z6.object({}).partial().passthrough(), z6.null()]).optional()
     }),
-    z6.object({ data: z6.array(z6.unknown()) })
+    z6.object({ data: z6.array(z6.unknown()) }),
+    z6.object({ experiment_name: z6.string() })
   ]),
   name: z6.string().optional(),
   parameters: z6.object({}).partial().passthrough().optional(),
@@ -3668,7 +3672,10 @@ var View = z6.object({
     "for_review_datasets"
   ]),
   name: z6.string(),
+  description: z6.union([z6.string(), z6.null()]).optional(),
+  starred: z6.boolean().optional(),
   created: z6.union([z6.string(), z6.null()]).optional(),
+  updated_at: z6.union([z6.string(), z6.null()]).optional(),
   view_data: ViewData.optional(),
   options: ViewOptions.optional(),
   user_id: z6.union([z6.string(), z6.null()]).optional(),
@@ -4252,10 +4259,10 @@ var DiskCache = class {
       return;
     }
     const stats = await Promise.all(
-      paths.map(async (path2) => {
-        const stat2 = await isomorph_default.stat(path2);
+      paths.map(async (path3) => {
+        const stat2 = await isomorph_default.stat(path3);
         return {
-          path: path2,
+          path: path3,
           mtime: stat2.mtime.getTime()
         };
       })
@@ -4427,44 +4434,72 @@ function createCacheLayers({
 }
 
 // src/prompt-cache/prompt-cache.ts
-function createCacheKey(key) {
+function createCacheKey(key, namespace) {
+  let cacheKey;
   if (key.id) {
-    return `id:${key.id}`;
-  }
-  const prefix = key.projectId ?? key.projectName;
-  if (!prefix) {
-    throw new Error("Either projectId or projectName must be provided");
-  }
-  if (!key.slug) {
-    throw new Error("Slug must be provided when not using ID");
+    cacheKey = `id:${key.id}`;
+  } else {
+    const prefix = key.projectId ?? key.projectName;
+    if (!prefix) {
+      throw new Error("Either projectId or projectName must be provided");
+    }
+    if (!key.slug) {
+      throw new Error("Slug must be provided when not using ID");
+    }
+    cacheKey = `${prefix}:${key.slug}:${key.version ?? "latest"}`;
   }
-  return `${prefix}:${key.slug}:${key.version ?? "latest"}`;
+  return namespace === void 0 ? cacheKey : `${namespace.length}:${namespace}:${cacheKey}`;
 }
-var PromptCache = class {
+var PromptCache = class _PromptCache {
   memoryCache;
   diskCache;
+  namespace;
+  expectedResolvedOrgIdentity;
   constructor(options) {
     this.memoryCache = options.memoryCache;
     this.diskCache = options.diskCache;
+    this.namespace = options.namespace;
+    this.expectedResolvedOrgIdentity = options.expectedResolvedOrgIdentity;
+  }
+  /**
+   * Returns a cache view that shares the same storage layers but isolates all
+   * entries under the provided namespace.
+   */
+  withNamespace(namespace, expectedResolvedOrgIdentity) {
+    return new _PromptCache({
+      memoryCache: this.memoryCache,
+      diskCache: this.diskCache,
+      namespace,
+      expectedResolvedOrgIdentity
+    });
   }
   /**
    * Retrieves a prompt from the cache.
    * First checks the in-memory LRU cache, then falls back to checking the disk cache if available.
    */
   async get(key) {
-    const cacheKey = createCacheKey(key);
+    const cacheKey = createCacheKey(key, this.namespace);
     if (this.memoryCache) {
-      const memoryPrompt = this.memoryCache.get(cacheKey);
-      if (memoryPrompt !== void 0) {
-        return memoryPrompt;
+      const memoryEntry = this.memoryCache.get(cacheKey);
+      if (memoryEntry !== void 0 && (this.expectedResolvedOrgIdentity === void 0 || memoryEntry.resolvedOrgIdentity === this.expectedResolvedOrgIdentity)) {
+        return memoryEntry.value;
       }
     }
     if (this.diskCache) {
-      const diskPrompt = await this.diskCache.get(cacheKey);
-      if (!diskPrompt) {
+      const diskEntry = await this.diskCache.get(cacheKey);
+      if (!diskEntry || this.expectedResolvedOrgIdentity !== void 0 && diskEntry.resolvedOrgIdentity !== this.expectedResolvedOrgIdentity) {
         return void 0;
       }
-      this.memoryCache?.set(cacheKey, diskPrompt);
+      const serializedPrompt = diskEntry.value;
+      const diskPrompt = new Prompt2(
+        serializedPrompt.metadata,
+        serializedPrompt.defaults,
+        serializedPrompt.noTrace
+      );
+      this.memoryCache?.set(cacheKey, {
+        value: diskPrompt,
+        resolvedOrgIdentity: diskEntry.resolvedOrgIdentity
+      });
       return diskPrompt;
     }
     return void 0;
@@ -4478,58 +4513,91 @@ var PromptCache = class {
    * @throws If there is an error writing to the disk cache.
    */
   async set(key, value) {
-    const cacheKey = createCacheKey(key);
-    this.memoryCache?.set(cacheKey, value);
+    const cacheKey = createCacheKey(key, this.namespace);
+    const memoryEntry = {
+      value,
+      resolvedOrgIdentity: this.expectedResolvedOrgIdentity
+    };
+    this.memoryCache?.set(cacheKey, memoryEntry);
     if (this.diskCache) {
-      await this.diskCache.set(cacheKey, value);
+      await this.diskCache.set(cacheKey, {
+        value: value._internalSerializeForCache(),
+        resolvedOrgIdentity: this.expectedResolvedOrgIdentity
+      });
     }
   }
 };
 
 // src/prompt-cache/parameters-cache.ts
-function createCacheKey2(key) {
+function createCacheKey2(key, namespace) {
+  let cacheKey;
   if (key.id) {
-    return `parameters:id:${key.id}`;
-  }
-  const prefix = key.projectId ?? key.projectName;
-  if (!prefix) {
-    throw new Error("Either projectId or projectName must be provided");
-  }
-  if (!key.slug) {
-    throw new Error("Slug must be provided when not using ID");
+    cacheKey = `parameters:id:${key.id}`;
+  } else {
+    const prefix = key.projectId ?? key.projectName;
+    if (!prefix) {
+      throw new Error("Either projectId or projectName must be provided");
+    }
+    if (!key.slug) {
+      throw new Error("Slug must be provided when not using ID");
+    }
+    cacheKey = `parameters:${prefix}:${key.slug}:${key.version ?? "latest"}`;
   }
-  return `parameters:${prefix}:${key.slug}:${key.version ?? "latest"}`;
+  return namespace === void 0 ? cacheKey : `${namespace.length}:${namespace}:${cacheKey}`;
 }
-var ParametersCache = class {
+var ParametersCache = class _ParametersCache {
   memoryCache;
   diskCache;
+  namespace;
+  expectedResolvedOrgIdentity;
   constructor(options) {
     this.memoryCache = options.memoryCache;
     this.diskCache = options.diskCache;
+    this.namespace = options.namespace;
+    this.expectedResolvedOrgIdentity = options.expectedResolvedOrgIdentity;
+  }
+  withNamespace(namespace, expectedResolvedOrgIdentity) {
+    return new _ParametersCache({
+      memoryCache: this.memoryCache,
+      diskCache: this.diskCache,
+      namespace,
+      expectedResolvedOrgIdentity
+    });
   }
   async get(key) {
-    const cacheKey = createCacheKey2(key);
+    const cacheKey = createCacheKey2(key, this.namespace);
     if (this.memoryCache) {
-      const memoryParams = this.memoryCache.get(cacheKey);
-      if (memoryParams !== void 0) {
-        return memoryParams;
+      const memoryEntry = this.memoryCache.get(cacheKey);
+      if (memoryEntry !== void 0 && (this.expectedResolvedOrgIdentity === void 0 || memoryEntry.resolvedOrgIdentity === this.expectedResolvedOrgIdentity)) {
+        return memoryEntry.value;
       }
     }
     if (this.diskCache) {
-      const diskParams = await this.diskCache.get(cacheKey);
-      if (!diskParams) {
+      const diskEntry = await this.diskCache.get(cacheKey);
+      if (!diskEntry || this.expectedResolvedOrgIdentity !== void 0 && diskEntry.resolvedOrgIdentity !== this.expectedResolvedOrgIdentity) {
         return void 0;
       }
-      this.memoryCache?.set(cacheKey, diskParams);
-      return diskParams;
+      const diskParameters = new RemoteEvalParameters(diskEntry.value.metadata);
+      this.memoryCache?.set(cacheKey, {
+        value: diskParameters,
+        resolvedOrgIdentity: diskEntry.resolvedOrgIdentity
+      });
+      return diskParameters;
     }
     return void 0;
   }
   async set(key, value) {
-    const cacheKey = createCacheKey2(key);
-    this.memoryCache?.set(cacheKey, value);
+    const cacheKey = createCacheKey2(key, this.namespace);
+    const memoryEntry = {
+      value,
+      resolvedOrgIdentity: this.expectedResolvedOrgIdentity
+    };
+    this.memoryCache?.set(cacheKey, memoryEntry);
     if (this.diskCache) {
-      await this.diskCache.set(cacheKey, value);
+      await this.diskCache.set(cacheKey, {
+        value: value._internalSerializeForCache(),
+        resolvedOrgIdentity: this.expectedResolvedOrgIdentity
+      });
     }
   }
 };
@@ -4834,6 +4902,16 @@ var datasetSnapshotRegisterResponseSchema = z8.object({
   dataset_snapshot: DatasetSnapshot,
   found_existing: z8.boolean().optional()
 });
+var datasetObjectInfoSchema = z8.object({
+  object_id: z8.string(),
+  object_name: z8.string(),
+  parent_cols: z8.object({
+    project: z8.object({
+      id: z8.string(),
+      name: z8.string()
+    })
+  })
+});
 var datasetRestorePreviewResultSchema = z8.object({
   rows_to_restore: z8.number(),
   rows_to_delete: z8.number()
@@ -4910,6 +4988,9 @@ function applyMaskingToField(maskingFunction, data, fieldName) {
 var INITIAL_SPAN_WRITE_AS_MERGE = /* @__PURE__ */ Symbol(
   "braintrust.initial-span-write-as-merge"
 );
+var RESUME_SPAN_WITHOUT_INITIAL_WRITE = /* @__PURE__ */ Symbol(
+  "braintrust.resume-span-without-initial-write"
+);
 var INTERNAL_SPAN_CONTEXT = /* @__PURE__ */ Symbol("braintrust.internal-span-context");
 var BRAINTRUST_CURRENT_SPAN_STORE = /* @__PURE__ */ Symbol.for(
   "braintrust.currentSpanStore"
@@ -5044,12 +5125,53 @@ var loginSchema = z8.strictObject({
 });
 var stateNonce = 0;
 var V1_PROXY_SUFFIX = "/v1/proxy";
+var LOADER_LOGIN_CACHE_MAX = 16;
 function normalizeProxyConnUrl(proxyUrl) {
   return proxyUrl.endsWith(V1_PROXY_SUFFIX) ? proxyUrl.slice(0, proxyUrl.length - V1_PROXY_SUFFIX.length) : proxyUrl;
 }
 var BraintrustState = class _BraintrustState {
+  id;
+  currentExperiment;
+  // Note: the value of IsAsyncFlush doesn't really matter here, since we
+  // (safely) dynamically cast it whenever retrieving the logger.
+  currentLogger;
+  currentParent;
+  currentSpan;
+  // Any time we re-log in, we directly update the apiConn inside the logger.
+  // This is preferable to replacing the whole logger, which would create the
+  // possibility of multiple loggers floating around, which may not log in a
+  // deterministic order.
+  _bgLogger;
+  _overrideBgLogger = null;
+  appUrl = null;
+  appPublicUrl = null;
+  loginToken = null;
+  orgId = null;
+  orgName = null;
+  apiUrl = null;
+  proxyUrl = null;
+  loggedIn = false;
+  gitMetadataSettings;
+  debugLogLevel;
+  debugLogLevelConfigured = false;
+  fetch = globalThis.fetch;
+  _appConn = null;
+  _apiConn = null;
+  _proxyConn = null;
+  promptCache;
+  parametersCache;
+  spanCache;
+  _idGenerator = null;
+  _contextManager = null;
+  _otelFlushCallback = null;
+  spanOriginEnvironment;
+  traceContextSigningSecret;
+  loaderLoginCache = /* @__PURE__ */ new WeakMap();
+  loginParams;
+  activeLoginOrgNameSelector;
   constructor(loginParams) {
-    this.loginParams = loginParams;
+    this.loginParams = { ...loginParams };
+    this.activeLoginOrgNameSelector = loginParams.orgName ?? isomorph_default.getEnv("BRAINTRUST_ORG_NAME");
     this.id = `${(/* @__PURE__ */ new Date()).toLocaleString()}-${stateNonce++}`;
     this.currentExperiment = void 0;
     this.currentLogger = void 0;
@@ -5086,12 +5208,14 @@ var BraintrustState = class _BraintrustState {
     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`
-    });
+    } = 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`
+      }
+    );
     this.parametersCache = new ParametersCache({
       memoryCache: parametersMemoryCache,
       diskCache: parametersDiskCache
@@ -5100,43 +5224,6 @@ var BraintrustState = class _BraintrustState {
     this.spanOriginEnvironment = detectSpanOriginEnvironment();
     this._internalSetTraceContextSigningSecret(loginParams.apiKey);
   }
-  loginParams;
-  id;
-  currentExperiment;
-  // Note: the value of IsAsyncFlush doesn't really matter here, since we
-  // (safely) dynamically cast it whenever retrieving the logger.
-  currentLogger;
-  currentParent;
-  currentSpan;
-  // Any time we re-log in, we directly update the apiConn inside the logger.
-  // This is preferable to replacing the whole logger, which would create the
-  // possibility of multiple loggers floating around, which may not log in a
-  // deterministic order.
-  _bgLogger;
-  _overrideBgLogger = null;
-  appUrl = null;
-  appPublicUrl = null;
-  loginToken = null;
-  orgId = null;
-  orgName = null;
-  apiUrl = null;
-  proxyUrl = null;
-  loggedIn = false;
-  gitMetadataSettings;
-  debugLogLevel;
-  debugLogLevelConfigured = false;
-  fetch = globalThis.fetch;
-  _appConn = null;
-  _apiConn = null;
-  _proxyConn = null;
-  promptCache;
-  parametersCache;
-  spanCache;
-  _idGenerator = null;
-  _contextManager = null;
-  _otelFlushCallback = null;
-  spanOriginEnvironment;
-  traceContextSigningSecret;
   /** @internal */
   _internalSetTraceContextSigningSecret(secret) {
     const normalizedSecret = secret?.trim();
@@ -5161,6 +5248,101 @@ var BraintrustState = class _BraintrustState {
     this._appConn = null;
     this._apiConn = null;
     this._proxyConn = null;
+    this.loaderLoginCache = /* @__PURE__ */ new WeakMap();
+  }
+  /** @internal */
+  async _internalResolveLoaderLoginOptions({
+    apiKey,
+    appUrl,
+    orgName,
+    fetch: fetch2,
+    forceLogin
+  }) {
+    const resolvedAppUrl = appUrl ?? (this.loggedIn ? this.appUrl ?? void 0 : void 0) ?? this.loginParams.appUrl ?? isomorph_default.getEnv("BRAINTRUST_APP_URL") ?? "https://www.braintrust.dev";
+    const resolvedApiKey = apiKey ?? (this.loggedIn ? this.loginToken ?? void 0 : void 0) ?? this.loginParams.apiKey ?? await isomorph_default.getBraintrustApiKey();
+    if (!resolvedApiKey) {
+      throw new Error(
+        "Please specify an api key (e.g. by setting BRAINTRUST_API_KEY)."
+      );
+    }
+    const normalizedApiKey = HTTPConnection.sanitize_token(resolvedApiKey);
+    const usesActiveCredential = this.loggedIn && normalizedApiKey === this.loginToken;
+    const requestedOrgName = orgName ?? (usesActiveCredential ? this.activeLoginOrgNameSelector : void 0) ?? this.loginParams.orgName ?? isomorph_default.getEnv("BRAINTRUST_ORG_NAME");
+    const resolvedOrgName = orgName ?? (usesActiveCredential ? this.orgName ?? void 0 : void 0) ?? requestedOrgName;
+    const resolvedFetch = fetch2 ?? (this.loggedIn ? this.fetch : void 0) ?? this.loginParams.fetch ?? globalThis.fetch;
+    const credentialCacheNamespace = JSON.stringify([
+      "loader-credential",
+      resolvedAppUrl,
+      requestedOrgName,
+      normalizedApiKey
+    ]);
+    return {
+      apiKey: normalizedApiKey,
+      appUrl: resolvedAppUrl,
+      orgName: resolvedOrgName,
+      fetch: resolvedFetch,
+      forceLogin,
+      credentialCacheNamespace,
+      existingState: !forceLogin && usesActiveCredential && resolvedAppUrl === this.appUrl && resolvedOrgName === this.orgName && resolvedFetch === this.fetch ? this : void 0
+    };
+  }
+  /** @internal */
+  _internalGetLoaderCacheViews(loginOptions, requestState) {
+    const expectedResolvedOrgIdentity = requestState?.orgId && requestState.appUrl ? JSON.stringify([
+      "loader-org",
+      requestState.appUrl,
+      requestState.orgId
+    ]) : void 0;
+    return {
+      promptCache: this.promptCache.withNamespace(
+        loginOptions.credentialCacheNamespace,
+        expectedResolvedOrgIdentity
+      ),
+      parametersCache: this.parametersCache.withNamespace(
+        loginOptions.credentialCacheNamespace,
+        expectedResolvedOrgIdentity
+      )
+    };
+  }
+  /** @internal */
+  async _internalGetLoaderState({
+    apiKey,
+    appUrl,
+    orgName,
+    fetch: fetch2,
+    forceLogin,
+    existingState
+  }) {
+    if (existingState) {
+      return existingState;
+    }
+    let cache = this.loaderLoginCache.get(fetch2);
+    if (!cache) {
+      cache = new LRUCache({ max: LOADER_LOGIN_CACHE_MAX });
+      this.loaderLoginCache.set(fetch2, cache);
+    }
+    const cacheKey = JSON.stringify([appUrl, orgName, apiKey]);
+    if (!forceLogin) {
+      const cachedState = cache.get(cacheKey);
+      if (cachedState) {
+        return cachedState;
+      }
+    }
+    const statePromise = loginToLoaderRequestState({
+      orgName,
+      apiKey,
+      appUrl,
+      fetch: fetch2
+    });
+    cache.set(cacheKey, statePromise);
+    try {
+      return await statePromise;
+    } catch (error) {
+      if (cache.get(cacheKey) === statePromise) {
+        cache.delete(cacheKey);
+      }
+      throw error;
+    }
   }
   resetIdGenState() {
     this._idGenerator = null;
@@ -5209,6 +5391,8 @@ var BraintrustState = class _BraintrustState {
     this.debugLogLevel = other.debugLogLevel;
     this.debugLogLevelConfigured = other.debugLogLevelConfigured;
     this.traceContextSigningSecret = other.traceContextSigningSecret;
+    this.fetch = other.fetch;
+    this.activeLoginOrgNameSelector = other.activeLoginOrgNameSelector;
     setGlobalDebugLogLevel(
       this.debugLogLevelConfigured ? this.debugLogLevel ?? false : void 0
     );
@@ -5448,36 +5632,85 @@ var FailedHTTPResponse = class extends Error {
   status;
   text;
   data;
-  constructor(status, text, data) {
+  cause;
+  constructor(status, text, data, cause) {
     super(`${status}: ${text} (${data})`);
     this.status = status;
     this.text = text;
     this.data = data;
+    this.cause = cause;
   }
 };
+var HTTPTransportError = class extends Error {
+  cause;
+  constructor(cause) {
+    super(cause instanceof Error ? cause.message : String(cause));
+    this.name = "HTTPTransportError";
+    this.cause = cause;
+  }
+};
+var httpTransportErrorCauses = /* @__PURE__ */ new WeakSet();
+function recordHTTPTransportError(error) {
+  if (typeof error === "object" && error !== null || typeof error === "function") {
+    httpTransportErrorCauses.add(error);
+  }
+}
+function rethrowHTTPTransportError(error, classifyTransportErrors) {
+  if (classifyTransportErrors) {
+    throw new HTTPTransportError(error);
+  }
+  recordHTTPTransportError(error);
+  throw error;
+}
+function isLoaderCacheFallbackError(error) {
+  if (error instanceof FailedHTTPResponse) {
+    return error.status === 408 || error.status === 429 || error.status >= 500;
+  }
+  if (error instanceof HTTPTransportError) {
+    return true;
+  }
+  return (typeof error === "object" && error !== null || typeof error === "function") && httpTransportErrorCauses.has(error);
+}
+async function readJSONResponse(response, classifyTransportErrors = false) {
+  let data;
+  try {
+    data = await response.text();
+  } catch (error) {
+    rethrowHTTPTransportError(error, classifyTransportErrors);
+  }
+  return JSON.parse(data);
+}
 async function checkResponse(resp) {
   if (resp.ok) {
     return resp;
-  } else {
+  }
+  let data;
+  try {
+    data = await resp.text();
+  } catch (error) {
     throw new FailedHTTPResponse(
       resp.status,
       resp.statusText,
-      await resp.text()
+      "Unable to read response body",
+      error
     );
   }
+  throw new FailedHTTPResponse(resp.status, resp.statusText, data);
 }
 var HTTPConnection = class _HTTPConnection {
-  base_url;
-  token;
-  headers;
-  fetch;
-  constructor(base_url, fetch2) {
+  constructor(base_url, fetch2, classifyTransportErrors = false) {
+    this.classifyTransportErrors = classifyTransportErrors;
     this.base_url = base_url;
     this.token = null;
     this.headers = {};
     this._reset();
     this.fetch = fetch2;
   }
+  classifyTransportErrors;
+  base_url;
+  token;
+  headers;
+  fetch;
   setFetch(fetch2) {
     this.fetch = fetch2;
   }
@@ -5507,9 +5740,9 @@ var HTTPConnection = class _HTTPConnection {
       this.headers["Authorization"] = `Bearer ${this.token}`;
     }
   }
-  async get(path2, params = void 0, config) {
+  async get(path3, params = void 0, config) {
     const { headers, ...rest } = config || {};
-    const url = new URL(_urljoin(this.base_url, path2));
+    const url = new URL(_urljoin(this.base_url, path3));
     url.search = new URLSearchParams(
       params ? Object.entries(params).filter(([_, v]) => v !== void 0).flatMap(
         ([k, v]) => v !== void 0 ? typeof v === "string" ? [[k, v]] : v.map((x) => [k, x]) : []
@@ -5517,9 +5750,9 @@ var HTTPConnection = class _HTTPConnection {
     ).toString();
     const this_fetch = this.fetch;
     const this_headers = this.headers;
-    return await checkResponse(
-      // Using toString() here makes it work with isomorphic fetch
-      await this_fetch(url.toString(), {
+    let response;
+    try {
+      response = await this_fetch(url.toString(), {
         headers: {
           Accept: "application/json",
           ...this_headers,
@@ -5527,10 +5760,16 @@ var HTTPConnection = class _HTTPConnection {
         },
         keepalive: true,
         ...rest
-      })
-    );
+      });
+    } catch (error) {
+      if (config?.signal?.aborted) {
+        throw getAbortReason(config.signal);
+      }
+      rethrowHTTPTransportError(error, this.classifyTransportErrors);
+    }
+    return await checkResponse(response);
   }
-  async post(path2, params, config, retries = 0) {
+  async post(path3, params, config, retries = 0) {
     const { headers, ...rest } = config || {};
     const this_fetch = this.fetch;
     const this_base_url = this.base_url;
@@ -5538,8 +5777,9 @@ var HTTPConnection = class _HTTPConnection {
     const tries = retries + 1;
     for (let i = 0; i < tries; i++) {
       try {
-        return await checkResponse(
-          await this_fetch(_urljoin(this_base_url, path2), {
+        let response;
+        try {
+          response = await this_fetch(_urljoin(this_base_url, path3), {
             method: "POST",
             headers: {
               Accept: "application/json",
@@ -5550,8 +5790,14 @@ var HTTPConnection = class _HTTPConnection {
             body: typeof params === "string" ? params : params ? JSON.stringify(params) : void 0,
             keepalive: true,
             ...rest
-          })
-        );
+          });
+        } catch (error) {
+          if (config?.signal?.aborted) {
+            throw getAbortReason(config.signal);
+          }
+          rethrowHTTPTransportError(error, this.classifyTransportErrors);
+        }
+        return await checkResponse(response);
       } catch (error) {
         if (config?.signal?.aborted) {
           throw getAbortReason(config.signal);
@@ -5560,7 +5806,7 @@ var HTTPConnection = class _HTTPConnection {
           throw error;
         }
         debugLogger.debug(
-          `Retrying API request ${path2} after ${formatHTTPError(error)}`
+          `Retrying API request ${path3} after ${formatHTTPError(error)}`
         );
         const sleepTimeMs = HTTP_RETRY_BASE_SLEEP_TIME_S * 1e3 * 2 ** i + Math.random() * HTTP_RETRY_JITTER_MS;
         debugLogger.info(
@@ -5576,7 +5822,7 @@ var HTTPConnection = class _HTTPConnection {
     for (let i = 0; i < tries; i++) {
       try {
         const resp = await this.get(`${object_type}`, args);
-        return await resp.json();
+        return await readJSONResponse(resp, this.classifyTransportErrors);
       } catch (e) {
         if (i < tries - 1) {
           debugLogger.debug(
@@ -5599,7 +5845,7 @@ var HTTPConnection = class _HTTPConnection {
     const resp = await this.post(`${object_type}`, args, {
       headers: { "Content-Type": "application/json" }
     });
-    return await resp.json();
+    return await readJSONResponse(resp, this.classifyTransportErrors);
   }
   // Custom inspect for Node.js console.log
   [/* @__PURE__ */ Symbol.for("nodejs.util.inspect.custom")]() {
@@ -6133,6 +6379,30 @@ function updateSpan({
     event
   });
 }
+function _internalResumeSpan({
+  exported,
+  state
+}) {
+  const resolvedState = state ?? _globalState;
+  const components = SpanComponentsV4.fromStr(exported);
+  const { row_id, root_span_id, span_id } = components.data;
+  if (!row_id || !root_span_id || !span_id) {
+    throw new Error("Only exported root spans can be resumed");
+  }
+  return new SpanImpl({
+    state: resolvedState,
+    parentObjectType: components.data.object_type,
+    parentObjectId: new LazyValue(
+      spanComponentsToObjectIdLambda(resolvedState, components)
+    ),
+    parentComputeObjectMetadataArgs: void 0,
+    parentSpanIds: { parentSpanIds: [], rootSpanId: root_span_id },
+    spanId: span_id,
+    event: { id: row_id },
+    propagatedEvent: components.data.propagated_event ?? void 0,
+    [RESUME_SPAN_WITHOUT_INITIAL_WRITE]: true
+  });
+}
 function spanComponentsToObjectIdLambda(state, components) {
   if (components.data.object_id) {
     const ret = components.data.object_id;
@@ -7623,6 +7893,7 @@ function initDataset(projectOrOptions, optionalOptions) {
   const {
     project,
     dataset,
+    datasetId,
     description,
     version,
     snapshotName,
@@ -7638,6 +7909,11 @@ function initDataset(projectOrOptions, optionalOptions) {
     state: stateArg,
     _internal_btql
   } = options;
+  if (datasetId !== void 0 && (description !== void 0 || metadata !== void 0)) {
+    throw new Error(
+      "Cannot specify description or metadata when datasetId is provided"
+    );
+  }
   const selection = normalizeDatasetSelection({
     version,
     environment,
@@ -7658,6 +7934,35 @@ function initDataset(projectOrOptions, optionalOptions) {
         fetch: fetch2,
         forceLogin
       });
+      if (datasetId !== void 0) {
+        const objectInfo = datasetObjectInfoSchema.array().parse(
+          await state.appConn().post_json("api/self/get_object_info", {
+            object_type: "dataset",
+            object_ids: [datasetId]
+          })
+        );
+        if (objectInfo.length === 0) {
+          throw new Error(`Dataset with ID ${datasetId} not found`);
+        }
+        if (objectInfo.length !== 1) {
+          throw new Error(
+            `Expected exactly one dataset with ID ${datasetId}, but found ${objectInfo.length}`
+          );
+        }
+        const datasetInfo = objectInfo[0];
+        return {
+          project: {
+            id: datasetInfo.parent_cols.project.id,
+            name: datasetInfo.parent_cols.project.name,
+            fullInfo: datasetInfo.parent_cols.project
+          },
+          dataset: {
+            id: datasetInfo.object_id,
+            name: datasetInfo.object_name,
+            fullInfo: datasetInfo
+          }
+        };
+      }
       const args = {
         org_id: state.orgId,
         project_name: project,
@@ -7811,6 +8116,24 @@ function initLogger(options = {}) {
   }
   return ret;
 }
+async function runCredentialScopedLoaderRequest(state, loginOptions, request) {
+  const resolvedLoginOptions = await state._internalResolveLoaderLoginOptions(loginOptions);
+  let cacheViews = state._internalGetLoaderCacheViews(resolvedLoginOptions);
+  try {
+    const requestState = await state._internalGetLoaderState(resolvedLoginOptions);
+    cacheViews = state._internalGetLoaderCacheViews(
+      resolvedLoginOptions,
+      requestState
+    );
+    return {
+      ok: true,
+      response: await request(requestState),
+      ...cacheViews
+    };
+  } catch (error) {
+    return { ok: false, error, ...cacheViews };
+  }
+}
 async function loadPrompt({
   projectName,
   projectId,
@@ -7835,43 +8158,47 @@ async function loadPrompt({
     throw new Error("Must specify slug");
   }
   const state = stateArg ?? _globalState;
-  let response;
-  try {
-    await state.login({
-      orgName,
-      apiKey,
-      appUrl,
-      fetch: fetch2,
-      forceLogin
-    });
-    if (id) {
-      response = await state.apiConn().get_json(`v1/prompt/${id}`, versionOrEnvironment);
-      if (response) {
-        response = { objects: [response] };
+  const result = await runCredentialScopedLoaderRequest(
+    state,
+    { orgName, apiKey, appUrl, fetch: fetch2, forceLogin },
+    async (requestState) => {
+      let response2;
+      if (id) {
+        response2 = await requestState.apiConn().get_json(`v1/prompt/${id}`, versionOrEnvironment);
+        if (response2) {
+          response2 = { objects: [response2] };
+        }
+      } else {
+        response2 = await requestState.apiConn().get_json("v1/prompt", {
+          project_name: projectName,
+          project_id: projectId,
+          slug,
+          ...versionOrEnvironment
+        });
       }
-    } else {
-      response = await state.apiConn().get_json("v1/prompt", {
-        project_name: projectName,
-        project_id: projectId,
-        slug,
-        ...versionOrEnvironment
-      });
+      return response2;
+    }
+  );
+  const { promptCache } = result;
+  if (!result.ok) {
+    const e = result.error;
+    if (!isLoaderCacheFallbackError(e)) {
+      throw e;
     }
-  } catch (e) {
     if (version || environment) {
       throw new Error(`Prompt not found with specified parameters: ${e}`);
     }
     debugLogger.forState(state).warn("Failed to load prompt, attempting to fall back to cache:", e);
     let prompt2;
     if (id) {
-      prompt2 = await state.promptCache.get({ id });
+      prompt2 = await promptCache.get({ id });
       if (!prompt2) {
         throw new Error(
           `Prompt with id ${id} not found (not found on server or in local cache): ${e}`
         );
       }
     } else {
-      prompt2 = await state.promptCache.get({
+      prompt2 = await promptCache.get({
         slug,
         projectId,
         projectName,
@@ -7887,6 +8214,7 @@ async function loadPrompt({
     }
     return prompt2;
   }
+  const { response } = result;
   if (!("objects" in response) || response.objects.length === 0) {
     if (id) {
       throw new Error(`Prompt with id ${id} not found.`);
@@ -7910,9 +8238,9 @@ async function loadPrompt({
   const prompt = new Prompt2(metadata, defaults || {}, noTrace);
   try {
     if (id) {
-      await state.promptCache.set({ id }, prompt);
+      await promptCache.set({ id }, prompt);
     } else if (slug) {
-      await state.promptCache.set(
+      await promptCache.set(
         { slug, projectId, projectName, version: version ?? "latest" },
         prompt
       );
@@ -7944,46 +8272,50 @@ async function loadParameters({
     throw new Error("Must specify slug");
   }
   const state = stateArg ?? _globalState;
-  let response;
-  try {
-    await state.login({
-      orgName,
-      apiKey,
-      appUrl,
-      fetch: fetch2,
-      forceLogin
-    });
-    if (id) {
-      response = await state.apiConn().get_json(`v1/function/${id}`, {
-        ...versionOrEnvironment
-      });
-      if (response) {
-        response = { objects: [response] };
+  const result = await runCredentialScopedLoaderRequest(
+    state,
+    { orgName, apiKey, appUrl, fetch: fetch2, forceLogin },
+    async (requestState) => {
+      let response2;
+      if (id) {
+        response2 = await requestState.apiConn().get_json(`v1/function/${id}`, {
+          ...versionOrEnvironment
+        });
+        if (response2) {
+          response2 = { objects: [response2] };
+        }
+      } else {
+        response2 = await requestState.apiConn().get_json("v1/function", {
+          project_name: projectName,
+          project_id: projectId,
+          slug,
+          function_type: "parameters",
+          ...versionOrEnvironment
+        });
       }
-    } else {
-      response = await state.apiConn().get_json("v1/function", {
-        project_name: projectName,
-        project_id: projectId,
-        slug,
-        function_type: "parameters",
-        ...versionOrEnvironment
-      });
+      return response2;
+    }
+  );
+  const { parametersCache } = result;
+  if (!result.ok) {
+    const e = result.error;
+    if (!isLoaderCacheFallbackError(e)) {
+      throw e;
     }
-  } catch (e) {
     if (version || environment) {
       throw new Error(`Parameters not found with specified parameters: ${e}`);
     }
     debugLogger.forState(state).warn("Failed to load parameters, attempting to fall back to cache:", e);
     let parameters2;
     if (id) {
-      parameters2 = await state.parametersCache.get({ id });
+      parameters2 = await parametersCache.get({ id });
       if (!parameters2) {
         throw new Error(
           `Parameters with id ${id} not found (not found on server or in local cache): ${e}`
         );
       }
     } else {
-      parameters2 = await state.parametersCache.get({
+      parameters2 = await parametersCache.get({
         slug,
         projectId,
         projectName,
@@ -7999,6 +8331,7 @@ async function loadParameters({
     }
     return parameters2;
   }
+  const { response } = result;
   if (!("objects" in response) || response.objects.length === 0) {
     if (id) {
       throw new Error(`Parameters with id ${id} not found.`);
@@ -8022,9 +8355,9 @@ async function loadParameters({
   const parameters = new RemoteEvalParameters(metadata);
   try {
     if (id) {
-      await state.parametersCache.set({ id }, parameters);
+      await parametersCache.set({ id }, parameters);
     } else if (slug) {
-      await state.parametersCache.set(
+      await parametersCache.set(
         { slug, projectId, projectName, version: version ?? "latest" },
         parameters
       );
@@ -8065,6 +8398,52 @@ async function login(options = {}) {
   await state.login(options);
   return state;
 }
+async function loginToLoaderRequestState({
+  appUrl,
+  apiKey,
+  orgName,
+  fetch: fetch2
+}) {
+  let orgId;
+  let apiUrl;
+  if (apiKey === TEST_API_KEY) {
+    orgId = "test-org-id";
+    apiUrl = "https://braintrust.dev/fake-api-url";
+  } else {
+    let loginResponse;
+    try {
+      loginResponse = await fetch2(_urljoin(appUrl, `/api/apikey/login`), {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          Authorization: `Bearer ${apiKey}`
+        }
+      });
+    } catch (error) {
+      throw new HTTPTransportError(error);
+    }
+    const info = await readJSONResponse(
+      await checkResponse(loginResponse),
+      true
+    );
+    const org = selectLoginOrg(info.org_info, orgName);
+    orgId = org.id;
+    apiUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
+    if (!apiUrl) {
+      throw new Error(
+        orgName ? `Unable to log into organization '${orgName}'. Are you sure this credential is scoped to the organization?` : "Unable to log into any organization with the provided credential."
+      );
+    }
+  }
+  const apiConnection = new HTTPConnection(apiUrl, fetch2, true);
+  apiConnection.set_token(apiKey);
+  apiConnection.make_long_lived();
+  return {
+    appUrl,
+    orgId,
+    apiConn: () => apiConnection
+  };
+}
 async function loginToState(options = {}) {
   const {
     appUrl = isomorph_default.getEnv("BRAINTRUST_APP_URL") || "https://www.braintrust.dev",
@@ -8096,16 +8475,18 @@ async function loginToState(options = {}) {
     _saveOrgInfo(state, testOrgInfo, testOrgInfo[0].name);
     return state;
   } else {
-    const resp = await checkResponse(
-      await fetch2(_urljoin(state.appUrl, `/api/apikey/login`), {
+    const loginResponse = await fetch2(
+      _urljoin(state.appUrl, `/api/apikey/login`),
+      {
         method: "POST",
         headers: {
           "Content-Type": "application/json",
           Authorization: `Bearer ${apiKey}`
         }
-      })
+      }
     );
-    const info = await resp.json();
+    const resp = await checkResponse(loginResponse);
+    const info = await readJSONResponse(resp);
     _saveOrgInfo(state, info.org_info, orgName);
     if (!state.apiUrl) {
       if (orgName) {
@@ -8608,11 +8989,11 @@ function wrapTraced(fn, args) {
   }
   if (args?.asyncFlush) {
     return ((...fnArgs) => traced((span) => {
-      if (!hasExplicitInput) {
+      if (!args?.noTraceIO && !hasExplicitInput) {
         span.log({ input: fnArgs });
       }
       const output = fn(...fnArgs);
-      if (!hasExplicitOutput) {
+      if (!args?.noTraceIO && !hasExplicitOutput) {
         if (output instanceof Promise) {
           return (async () => {
             const result = await output;
@@ -8627,12 +9008,12 @@ function wrapTraced(fn, args) {
     }, spanArgs));
   } else {
     return ((...fnArgs) => traced(async (span) => {
-      if (!hasExplicitInput) {
+      if (!args?.noTraceIO && !hasExplicitInput) {
         span.log({ input: fnArgs });
       }
       const outputResult = fn(...fnArgs);
       const output = await outputResult;
-      if (!hasExplicitOutput) {
+      if (!args?.noTraceIO && !hasExplicitOutput) {
         span.log({ output });
       }
       return output;
@@ -8758,27 +9139,28 @@ async function* asyncGeneratorWithCurrent(span, gen, state = void 0) {
 function withParent(parent, callback, state = void 0) {
   return (state ?? _globalState).currentParent.run(parent, () => callback());
 }
-function _saveOrgInfo(state, org_info, org_name) {
-  if (org_info.length === 0) {
+function _saveOrgInfo(state, orgInfo, orgName) {
+  const org = selectLoginOrg(orgInfo, orgName);
+  state.orgId = org.id;
+  state.orgName = org.name;
+  state.apiUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
+  state.proxyUrl = isomorph_default.getEnv("BRAINTRUST_PROXY_URL") ?? org.proxy_url;
+  state.gitMetadataSettings = org.git_metadata || void 0;
+}
+function selectLoginOrg(orgInfo, orgName) {
+  if (orgInfo.length === 0) {
     throw new LoginInvalidOrgError(
       "This user is not part of any organizations."
     );
   }
-  for (const org of org_info) {
-    if (org_name === void 0 || org.name === org_name) {
-      state.orgId = org.id;
-      state.orgName = org.name;
-      state.apiUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
-      state.proxyUrl = isomorph_default.getEnv("BRAINTRUST_PROXY_URL") ?? org.proxy_url;
-      state.gitMetadataSettings = org.git_metadata || void 0;
-      break;
+  for (const org of orgInfo) {
+    if (orgName === void 0 || org.name === orgName) {
+      return org;
     }
   }
-  if (state.orgId === void 0) {
-    throw new LoginInvalidOrgError(
-      `Organization ${org_name} not found. Must be one of ${org_info.map((x) => x.name).join(", ")}`
-    );
-  }
+  throw new LoginInvalidOrgError(
+    `Organization ${orgName} not found. Must be one of ${orgInfo.map((org) => org.name).join(", ")}`
+  );
 }
 function validateTags(tags) {
   const seen = /* @__PURE__ */ new Set();
@@ -9013,7 +9395,7 @@ var ObjectFetcher = class {
     const objectId = await this.id;
     const batchLimit = batchSize ?? DEFAULT_FETCH_BATCH_SIZE;
     const internalLimit = getInternalBtqlLimit(this._internal_btql);
-    const limit = batchSize !== void 0 ? batchSize : internalLimit ?? batchLimit;
+    let remainingLimit = internalLimit;
     const internalBtqlWithoutReservedQueryKeys = Object.fromEntries(
       Object.entries(this._internal_btql ?? {}).filter(
         ([key]) => key !== "cursor" && key !== "limit" && key !== "select" && key !== "from"
@@ -9022,6 +9404,10 @@ var ObjectFetcher = class {
     let cursor = void 0;
     let iterations = 0;
     while (true) {
+      if (remainingLimit !== void 0 && remainingLimit <= 0) {
+        return;
+      }
+      const limit = remainingLimit === void 0 ? batchLimit : Math.min(batchLimit, remainingLimit);
       const resp = await state.apiConn().post(
         `btql`,
         {
@@ -9061,7 +9447,14 @@ var ObjectFetcher = class {
       const respJson = await resp.json();
       const mutate = this.mutateRecord;
       for (const record of respJson.data ?? []) {
-        yield mutate ? mutate(record) : record;
+        if (remainingLimit !== void 0 && remainingLimit <= 0) {
+          return;
+        }
+        const mutatedRecord = mutate ? mutate(record) : record;
+        if (remainingLimit !== void 0) {
+          remainingLimit--;
+        }
+        yield mutatedRecord;
       }
       if (!respJson.cursor) {
         break;
@@ -9581,7 +9974,9 @@ var SpanImpl = class _SpanImpl {
     this._rootSpanId = resolvedIds.rootSpanId;
     this._spanParents = resolvedIds.spanParents;
     this.isMerge = args[INITIAL_SPAN_WRITE_AS_MERGE] === true;
-    this.logInternal({ event, internalData });
+    if (!args[RESUME_SPAN_WITHOUT_INITIAL_WRITE]) {
+      this.logInternal({ event, internalData });
+    }
     this.isMerge = true;
   }
   getParentInfo() {
@@ -10671,6 +11066,14 @@ var Prompt2 = class _Prompt {
   static isPrompt(data) {
     return typeof data === "object" && data !== null && "__braintrust_prompt_marker" in data;
   }
+  /** @internal */
+  _internalSerializeForCache() {
+    return {
+      metadata: this.metadata,
+      defaults: this.defaults,
+      noTrace: this.noTrace
+    };
+  }
   static fromPromptData(name, promptData) {
     return new _Prompt(
       {
@@ -10711,6 +11114,10 @@ var RemoteEvalParameters = class {
   get data() {
     return this.metadata.function_data.data ?? {};
   }
+  /** @internal */
+  _internalSerializeForCache() {
+    return { metadata: this.metadata };
+  }
   validate(data) {
     if (typeof data !== "object" || data === null) {
       return false;
@@ -11571,56 +11978,14 @@ function suppressionStore() {
   autoInstrumentationSuppressionStore ??= isomorph_default.newAsyncLocalStorage();
   return autoInstrumentationSuppressionStore;
 }
-function currentFrames() {
-  return suppressionStore().getStore()?.frames ?? [];
-}
 function isAutoInstrumentationSuppressed() {
-  const frames = currentFrames();
-  return frames[frames.length - 1]?.mode === "suppress";
+  return suppressionStore().getStore() === true;
 }
 function runWithAutoInstrumentationSuppressed(callback) {
-  const frame = {
-    id: /* @__PURE__ */ Symbol("braintrust.auto-instrumentation-suppress"),
-    mode: "suppress"
-  };
-  return suppressionStore().run(
-    { frames: [...currentFrames(), frame] },
-    callback
-  );
-}
-function bindAutoInstrumentationSuppressionToStart(tracingChannel) {
-  const startChannel = tracingChannel.start;
-  if (!startChannel) {
-    return void 0;
-  }
-  const store = suppressionStore();
-  startChannel.bindStore(store, () => ({
-    frames: [
-      ...currentFrames(),
-      {
-        id: /* @__PURE__ */ Symbol("braintrust.auto-instrumentation-suppress"),
-        mode: "suppress"
-      }
-    ]
-  }));
-  return () => {
-    startChannel.unbindStore(store);
-  };
+  return suppressionStore().run(true, callback);
 }
-function enterAutoInstrumentationAllowed() {
-  const frame = {
-    id: /* @__PURE__ */ Symbol("braintrust.auto-instrumentation-allow"),
-    mode: "allow"
-  };
-  suppressionStore().enterWith({
-    frames: [...currentFrames(), frame]
-  });
-  return () => {
-    const frames = currentFrames().filter(
-      (candidate) => candidate.id !== frame.id
-    );
-    suppressionStore().enterWith(frames.length > 0 ? { frames } : void 0);
-  };
+function runWithAutoInstrumentationAllowed(callback) {
+  return suppressionStore().run(void 0, callback);
 }
 
 // src/instrumentation/core/channel-tracing.ts
@@ -12311,6 +12676,22 @@ function processInputAttachments(input) {
         };
       }
     }
+    const voyageBase64Key = node.type === "image_base64" ? Object.hasOwn(node, "imageBase64") ? "imageBase64" : "image_base64" : node.type === "video_base64" ? Object.hasOwn(node, "videoBase64") ? "videoBase64" : "video_base64" : void 0;
+    const voyageBase64Value = voyageBase64Key ? node[voyageBase64Key] : void 0;
+    if (voyageBase64Key && typeof voyageBase64Value === "string" && voyageBase64Value.startsWith("data:")) {
+      const mediaType = inferMediaTypeFromDataUrl(
+        voyageBase64Value,
+        node.type === "video_base64" ? "video/mp4" : "image/png"
+      );
+      const filename = `${node.type === "video_base64" ? "video" : "image"}.${getExtensionFromMediaType(mediaType)}`;
+      const attachment = toAttachment(voyageBase64Value, mediaType, filename);
+      if (attachment) {
+        return {
+          ...node,
+          [voyageBase64Key]: attachment
+        };
+      }
+    }
     if (node.type === "file" && node.file && typeof node.file === "object" && typeof node.file.file_data === "string" && node.file.file_data.startsWith("data:")) {
       const mediaType = inferMediaTypeFromDataUrl(
         node.file.file_data,
@@ -12857,7 +13238,7 @@ var AISDKPlugin = class extends BasePlugin {
   }
   subscribeToAISDK() {
     const denyOutputPaths = this.config.denyOutputPaths || DEFAULT_DENY_OUTPUT_PATHS;
-    this.unsubscribers.push(subscribeToAISDKV7TelemetryDispatcher());
+    this.unsubscribers.push(interceptAISDKV7TelemetryDispatcher());
     this.unsubscribers.push(subscribeToHarnessAgentCreateSession());
     this.unsubscribers.push(
       subscribeToHarnessContinuation(
@@ -12885,6 +13266,18 @@ var AISDKPlugin = class extends BasePlugin {
         aggregateChunks: aggregateAISDKChunks
       })
     );
+    this.unsubscribers.push(
+      traceAsyncChannel(aiSDKChannels.generateImage, {
+        name: "generateImage",
+        type: "llm" /* LLM */,
+        extractInput: ([params], event) => prepareAISDKGenerateImageInput(params, event.self),
+        extractOutput: (result, endEvent) => processAISDKGenerateImageOutput(
+          result,
+          resolveDenyOutputPaths(endEvent, denyOutputPaths)
+        ),
+        extractMetrics: (result) => extractTokenMetrics(result)
+      })
+    );
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.streamText, {
         name: "streamText",
@@ -13374,26 +13767,29 @@ function subscribeToHarnessContinuation(continuationChannel, defaultDenyOutputPa
     channel.unsubscribe(handlers);
   };
 }
-function subscribeToAISDKV7TelemetryDispatcher() {
-  const channel = aiSDKChannels.v7CreateTelemetryDispatcher.tracingChannel();
+function interceptAISDKV7TelemetryDispatcher() {
   const telemetry = braintrustAISDKTelemetry();
-  const handlers = {
-    end: (event) => {
-      const telemetryOptions = event.arguments?.[0]?.telemetry;
-      if (telemetryOptions?.isEnabled === false) {
-        return;
+  return aiSDKChannels.v7CreateTelemetryDispatcher.intercept(
+    (target, thisArg, args) => {
+      const dispatcher = Reflect.apply(target, thisArg, args);
+      const telemetryOptions = args[0]?.telemetry;
+      if (telemetryOptions?.isEnabled !== false) {
+        try {
+          patchAISDKV7TelemetryDispatcher(
+            dispatcher,
+            telemetry,
+            telemetryOptions
+          );
+        } catch (error) {
+          debugLogger.error(
+            "Error instrumenting AI SDK v7 telemetry dispatcher:",
+            error
+          );
+        }
       }
-      patchAISDKV7TelemetryDispatcher(
-        event.result,
-        telemetry,
-        telemetryOptions
-      );
+      return dispatcher;
     }
-  };
-  channel.subscribe(handlers);
-  return () => {
-    channel.unsubscribe(handlers);
-  };
+  );
 }
 function patchAISDKV7TelemetryDispatcher(dispatcher, telemetry, telemetryOptions) {
   if (!isObject(dispatcher)) {
@@ -13493,7 +13889,7 @@ function resolveDenyOutputPaths(event, defaultDenyOutputPaths) {
     return defaultDenyOutputPaths;
   }
   const runtimeDenyOutputPaths = firstArgument2[RUNTIME_DENY_OUTPUT_PATHS];
-  if (Array.isArray(runtimeDenyOutputPaths) && runtimeDenyOutputPaths.every((path2) => typeof path2 === "string")) {
+  if (Array.isArray(runtimeDenyOutputPaths) && runtimeDenyOutputPaths.every((path3) => typeof path3 === "string")) {
     return runtimeDenyOutputPaths;
   }
   return defaultDenyOutputPaths;
@@ -13718,16 +14114,10 @@ var convertImageToAttachment = (image, explicitMimeType) => {
       }
     }
     if (explicitMimeType) {
-      if (image instanceof Uint8Array) {
-        return new Attachment({
-          data: new Blob([image], { type: explicitMimeType }),
-          filename: `image.${getExtensionFromMediaType(explicitMimeType)}`,
-          contentType: explicitMimeType
-        });
-      }
-      if (typeof Buffer !== "undefined" && Buffer.isBuffer(image)) {
+      const blob = convertDataToBlob(image, explicitMimeType);
+      if (blob) {
         return new Attachment({
-          data: new Blob([image], { type: explicitMimeType }),
+          data: blob,
           filename: `image.${getExtensionFromMediaType(explicitMimeType)}`,
           contentType: explicitMimeType
         });
@@ -13781,6 +14171,25 @@ var convertDataToAttachment = (data, mimeType, filename) => {
 function processAISDKCallInput(params) {
   return processInputAttachmentsSync(params);
 }
+function processAISDKGenerateImageInput(params) {
+  const prompt = params.prompt;
+  if (!isObject(prompt) || Array.isArray(prompt)) {
+    return processAISDKCallInput(params);
+  }
+  const processedPrompt = { ...prompt };
+  if (Array.isArray(prompt.images)) {
+    processedPrompt.images = prompt.images.map(
+      (image) => convertImageToAttachment(image, "image/png") ?? image
+    );
+  }
+  if (prompt.mask !== void 0) {
+    processedPrompt.mask = convertImageToAttachment(prompt.mask, "image/png") ?? prompt.mask;
+  }
+  return processAISDKCallInput({
+    ...params,
+    prompt: processedPrompt
+  });
+}
 function processAISDKWorkflowAgentCallInput(params) {
   const processed = processAISDKCallInput(params);
   return {
@@ -13919,6 +14328,12 @@ function prepareAISDKEmbedInput(params, self) {
     metadata: extractMetadataFromEmbedParams(params, self)
   };
 }
+function prepareAISDKGenerateImageInput(params, self) {
+  return {
+    input: processAISDKGenerateImageInput(params).input,
+    metadata: extractMetadataFromCallParams(params, self)
+  };
+}
 function prepareAISDKRerankInput(params, self) {
   const { documents, query } = params;
   return {
@@ -15255,11 +15670,11 @@ function processAISDKOutput(output, denyOutputPaths) {
   if (!output) return output;
   const merged = extractSerializableOutputFields(output);
   const deleteOutputPaths = denyOutputPaths.filter(
-    (path2) => path2.toLowerCase().endsWith("headers")
+    (path3) => path3.toLowerCase().endsWith("headers")
   );
   const sanitized = omit(merged, denyOutputPaths, deleteOutputPaths);
-  for (const path2 of TRANSPORT_PAYLOAD_ROOT_PATHS) {
-    const stack = [{ obj: sanitized, keys: parsePath(path2) }];
+  for (const path3 of TRANSPORT_PAYLOAD_ROOT_PATHS) {
+    const stack = [{ obj: sanitized, keys: parsePath(path3) }];
     while (stack.length > 0) {
       const entry = stack.pop();
       if (!entry || entry.keys.length === 0) {
@@ -15294,6 +15709,57 @@ function processAISDKOutput(output, denyOutputPaths) {
   }
   return normalizeAISDKLoggedOutput(sanitized);
 }
+function processAISDKGenerateImageOutput(output, denyOutputPaths) {
+  if (!output || typeof output !== "object") {
+    return output;
+  }
+  const summarized = {};
+  for (const field of [
+    "usage",
+    "warnings",
+    "providerMetadata",
+    "experimental_providerMetadata",
+    "responses"
+  ]) {
+    const value = safeSerializableFieldRead(output, field);
+    if (value !== void 0 && isSerializableOutputValue(value)) {
+      summarized[field] = value;
+    }
+  }
+  const images = safeSerializableFieldRead(output, "images");
+  const image = safeSerializableFieldRead(output, "image");
+  const generatedFiles = Array.isArray(images) && images.length > 0 ? images : image !== void 0 ? [image] : [];
+  const loggedOutput = normalizeAISDKLoggedOutput(
+    omit(summarized, denyOutputPaths)
+  );
+  if (generatedFiles.length > 0) {
+    loggedOutput.images = generatedFiles.map(
+      (file, index) => convertAISDKGeneratedFileToAttachment(file, index)
+    );
+  }
+  return loggedOutput;
+}
+function convertAISDKGeneratedFileToAttachment(file, index) {
+  if (!file || typeof file !== "object") {
+    return file;
+  }
+  const generatedFile = file;
+  const generatedMediaType = safeSerializableFieldRead(
+    generatedFile,
+    "mediaType"
+  );
+  const mediaType = typeof generatedMediaType === "string" ? generatedMediaType : "application/octet-stream";
+  const data = safeSerializableFieldRead(generatedFile, "base64") ?? safeSerializableFieldRead(generatedFile, "uint8Array");
+  const blob = convertDataToBlob(data, mediaType);
+  if (blob) {
+    return new Attachment({
+      data: blob,
+      filename: `generated_image_${index}.${getExtensionFromMediaType(mediaType)}`,
+      contentType: mediaType
+    });
+  }
+  return file;
+}
 function processAISDKEmbeddingOutput(output, denyOutputPaths) {
   if (!output || typeof output !== "object") {
     return output;
@@ -15698,11 +16164,11 @@ function firstNumber(...values) {
 function deepCopy(obj) {
   return JSON.parse(JSON.stringify(obj));
 }
-function parsePath(path2) {
+function parsePath(path3) {
   const keys = [];
   let current = "";
-  for (let i = 0; i < path2.length; i++) {
-    const char = path2[i];
+  for (let i = 0; i < path3.length; i++) {
+    const char = path3[i];
     if (char === ".") {
       if (current) {
         keys.push(current);
@@ -15715,8 +16181,8 @@ function parsePath(path2) {
       }
       let bracketContent = "";
       i++;
-      while (i < path2.length && path2[i] !== "]") {
-        bracketContent += path2[i];
+      while (i < path3.length && path3[i] !== "]") {
+        bracketContent += path3[i];
         i++;
       }
       if (bracketContent === "") {
@@ -15771,9 +16237,9 @@ function omitAtPath(obj, keys, deleteLeaf = false) {
 function omit(obj, paths, deletePaths = []) {
   const result = deepCopy(obj);
   const deletePathSet = new Set(deletePaths);
-  for (const path2 of paths) {
-    const keys = parsePath(path2);
-    omitAtPath(result, keys, deletePathSet.has(path2));
+  for (const path3 of paths) {
+    const keys = parsePath(path3);
+    omitAtPath(result, keys, deletePathSet.has(path3));
   }
   return result;
 }
@@ -15782,7 +16248,6 @@ function omit(obj, paths, deletePaths = []) {
 function braintrustAISDKTelemetry() {
   const operations = /* @__PURE__ */ new Map();
   const operationKeysByCallId = /* @__PURE__ */ new Map();
-  const workflowOperationKeyStore = isomorph_default.newAsyncLocalStorage();
   const modelSpans = /* @__PURE__ */ new Map();
   const objectSpans = /* @__PURE__ */ new Map();
   const embedSpans = /* @__PURE__ */ new Map();
@@ -15825,9 +16290,6 @@ function braintrustAISDKTelemetry() {
       return;
     }
     operations.delete(operationKey);
-    if (workflowOperationKeyStore.getStore() === operationKey) {
-      workflowOperationKeyStore.enterWith(void 0);
-    }
     const keys = operationKeysByCallId.get(state.callId);
     if (!keys) {
       return;
@@ -15875,14 +16337,7 @@ function braintrustAISDKTelemetry() {
         return key;
       }
     }
-    const workflowOperationKey = workflowOperationKeyStore.getStore();
-    if (workflowOperationKey && keys.includes(workflowOperationKey)) {
-      return workflowOperationKey;
-    }
-    if (callId === "workflow-agent") {
-      return void 0;
-    }
-    return mode === "finish" ? keys[0] : keys[keys.length - 1];
+    return callId === "workflow-agent" || mode === "active" ? keys[keys.length - 1] : keys[0];
   };
   const operationKeyFromEvent = (event, mode = "active") => {
     const explicit = explicitOperationKey(event);
@@ -15896,17 +16351,13 @@ function braintrustAISDKTelemetry() {
         if (operationKey) {
           return operationKey;
         }
-        const workflowOperationKey2 = workflowOperationKeyStore.getStore();
-        if (workflowOperationKey2 && operations.has(workflowOperationKey2)) {
-          return workflowOperationKey2;
+        const workflowAgentKeys2 = operationKeysByCallId.get("workflow-agent");
+        if (workflowAgentKeys2?.length) {
+          return workflowAgentKeys2[workflowAgentKeys2.length - 1];
         }
         return callId === "workflow-agent" ? void 0 : callId;
       }
     }
-    const workflowOperationKey = workflowOperationKeyStore.getStore();
-    if (workflowOperationKey && operations.has(workflowOperationKey)) {
-      return workflowOperationKey;
-    }
     const wrapperSpan = currentWorkflowAgentWrapperSpan();
     if (wrapperSpan?.spanId) {
       for (const [operationKey, state] of operations) {
@@ -15916,8 +16367,8 @@ function braintrustAISDKTelemetry() {
       }
     }
     const workflowAgentKeys = operationKeysByCallId.get("workflow-agent");
-    if (workflowAgentKeys?.length === 1) {
-      return workflowAgentKeys[0];
+    if (workflowAgentKeys?.length) {
+      return workflowAgentKeys[workflowAgentKeys.length - 1];
     }
     if (operations.size === 1) {
       return operations.keys().next().value;
@@ -16120,9 +16571,6 @@ function braintrustAISDKTelemetry() {
         if (!ownsSpan) {
           return;
         }
-        if (workflowAgent) {
-          workflowOperationKeyStore.enterWith(operationKey);
-        }
         let metadata = metadataFromEvent(event);
         const logPayload = { metadata };
         const workflowAgentCallInput = workflowAgent ? operationInput(event, operationName) : void 0;
@@ -16999,17 +17447,20 @@ var FlueObserveBridge = class {
       return;
     }
     const metadata = {
+      ...event.runId ? this.runsById.get(event.runId)?.metadata : {},
       ...extractEventMetadata(event),
       "flue.operation": event.operationKind,
       provider: "flue"
     };
     const parent = this.parentSpanForEvent(event);
-    const span = startFlueSpan(parent, {
+    const args = {
       name: `flue.${event.operationKind}`,
       spanAttributes: { type: "task" /* TASK */ },
       startTime: eventTime(event.timestamp),
       event: { metadata }
-    });
+    };
+    const runSpan = event.runId ? this.runsById.get(event.runId)?.span : void 0;
+    const span = event.operationKind === "prompt" && (!parent || parent === runSpan) ? startFlueRootSpan(args) : startFlueSpan(parent, args);
     this.operationsById.set(event.operationId, { metadata, span });
   }
   handleOperation(event) {
@@ -17024,6 +17475,11 @@ var FlueObserveBridge = class {
       ...event.isError !== void 0 ? { "flue.is_error": event.isError } : {},
       ...event.usage ? { "flue.usage": event.usage } : {}
     };
+    const input = flueOperationInput(event);
+    if (!state.loggedInput && input !== void 0) {
+      safeLog(state.span, { input });
+      state.loggedInput = true;
+    }
     this.finishPendingChildrenForOperation(event, output);
     safeLog(state.span, {
       ...event.isError ? { error: toLoggedError(event.errorInfo ?? event.error) } : {},
@@ -17040,6 +17496,8 @@ var FlueObserveBridge = class {
       return;
     }
     const input = flueTurnRequestInput(event);
+    const operation = event.operationId ? this.operationsById.get(event.operationId) : void 0;
+    const turnInput = prepareFlueTurnInput(event, input, operation);
     const model = flueTurnRequestModel(event);
     const provider = flueTurnRequestProvider(event);
     const api = flueTurnRequestApi(event);
@@ -17052,8 +17510,7 @@ var FlueObserveBridge = class {
       ...provider ? { "flue.provider": provider } : {},
       ...event.purpose ? { "flue.turn_purpose": event.purpose } : {},
       ...reasoning ? { reasoning } : {},
-      ...input?.systemPrompt ? { "flue.system_prompt": input.systemPrompt } : {},
-      ...input?.tools ? { tools: input.tools } : {}
+      ...turnInput.metadata
     };
     const parent = this.parentSpanForTurn(event);
     const span = startFlueSpan(parent, {
@@ -17061,11 +17518,14 @@ var FlueObserveBridge = class {
       spanAttributes: { type: "llm" /* LLM */ },
       startTime: eventTime(event.timestamp),
       event: {
-        input: input?.messages,
+        input: turnInput.messages,
         metadata
       }
     });
-    this.logOperationInput(event.operationId, input?.messages ?? input);
+    this.logOperationInput(
+      event.operationId,
+      latestUserMessageInput(input?.messages)
+    );
     this.turnsByKey.set(key, { metadata, span });
   }
   handleTurn(event) {
@@ -17312,16 +17772,20 @@ var FlueObserveBridge = class {
   }
   startSyntheticOperation(event) {
     const metadata = {
+      ...event.runId ? this.runsById.get(event.runId)?.metadata : {},
       ...extractEventMetadata(event),
       "flue.operation": event.operationKind,
       provider: "flue"
     };
-    const span = startFlueSpan(this.parentSpanForEvent(event), {
+    const args = {
       name: `flue.${event.operationKind}`,
       spanAttributes: { type: "task" /* TASK */ },
       startTime: eventTime(event.timestamp),
       event: { metadata }
-    });
+    };
+    const parent = this.parentSpanForEvent(event);
+    const runSpan = event.runId ? this.runsById.get(event.runId)?.span : void 0;
+    const span = event.operationKind === "prompt" && (!parent || parent === runSpan) ? startFlueRootSpan(args) : startFlueSpan(parent, args);
     return { metadata, span };
   }
   startSyntheticTurn(event) {
@@ -17500,6 +17964,71 @@ function flueRunInput(event) {
 function flueTurnRequestInput(event) {
   return event.request?.input ?? event.input;
 }
+function prepareFlueTurnInput(event, input, operation) {
+  const messages = input?.messages;
+  const tracksUserTurn = event.purpose === "agent" && operation?.metadata["flue.operation"] === "prompt" && Array.isArray(messages);
+  if (!tracksUserTurn) {
+    return {
+      messages,
+      metadata: {
+        ...input?.systemPrompt ? { "flue.system_prompt": input.systemPrompt } : {},
+        ...input?.tools ? { tools: input.tools } : {}
+      }
+    };
+  }
+  const previous = operation.turnInputState;
+  const previousMessageCount = previous?.messageCount ?? 0;
+  const boundaryFingerprint = messages.length > 0 ? fingerprintJsonValue(messages[messages.length - 1]) : void 0;
+  const continuesPreviousInput = previous !== void 0 && messages.length >= previousMessageCount && (previousMessageCount === 0 || previous.boundaryFingerprint !== void 0 && previous.boundaryFingerprint === fingerprintJsonValue(messages[previousMessageCount - 1]));
+  const inputMode = previous === void 0 ? "full" : continuesPreviousInput ? "delta" : "reset";
+  const systemPromptFingerprint = fingerprintJsonValue(input?.systemPrompt);
+  const toolsFingerprint = fingerprintJsonValue(input?.tools);
+  operation.turnInputState = {
+    ...boundaryFingerprint !== void 0 ? { boundaryFingerprint } : {},
+    messageCount: messages.length,
+    ...systemPromptFingerprint !== void 0 ? { systemPromptFingerprint } : {},
+    ...toolsFingerprint !== void 0 ? { toolsFingerprint } : {}
+  };
+  return {
+    messages: continuesPreviousInput ? messages.slice(previousMessageCount) : messages,
+    metadata: {
+      "flue.input_mode": inputMode,
+      ...continuesPreviousInput ? { "flue.input_message_offset": previousMessageCount } : {},
+      ...input?.systemPrompt && (!continuesPreviousInput || systemPromptFingerprint !== previous?.systemPromptFingerprint) ? { "flue.system_prompt": input.systemPrompt } : {},
+      ...input?.tools && (!continuesPreviousInput || toolsFingerprint !== previous?.toolsFingerprint) ? { tools: input.tools } : {}
+    }
+  };
+}
+function fingerprintJsonValue(value) {
+  try {
+    const serialized = JSON.stringify(value);
+    if (serialized === void 0) {
+      return void 0;
+    }
+    let hash = 2166136261;
+    for (let i = 0; i < serialized.length; i++) {
+      hash = Math.imul(hash ^ serialized.charCodeAt(i), 16777619);
+    }
+    return `${serialized.length}:${hash >>> 0}`;
+  } catch {
+    return void 0;
+  }
+}
+function latestUserMessageInput(messages) {
+  if (!messages) {
+    return void 0;
+  }
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i];
+    if (isObjectLike(message) && Reflect.get(message, "role") === "user") {
+      return [message];
+    }
+  }
+  return void 0;
+}
+function flueOperationInput(event) {
+  return typeof event.agentInput?.text === "string" ? [{ content: event.agentInput.text, role: "user" }] : void 0;
+}
 function flueTurnRequestModel(event) {
   return event.request?.requestedModel ?? event.request?.model ?? event.model;
 }
@@ -17657,6 +18186,21 @@ function startFlueSpan(parent, args) {
     withSpanInstrumentationName(args, INSTRUMENTATION_NAMES.FLUE)
   );
 }
+function startFlueRootSpan(args) {
+  const state = _internalGetGlobalState();
+  const spanId = state.idGenerator.getSpanId();
+  const rootSpanId = state.idGenerator.shareRootSpanId() ? spanId : state.idGenerator.getTraceId();
+  return withCurrent(
+    NOOP_SPAN,
+    () => startSpan({
+      ...withSpanInstrumentationName(args, INSTRUMENTATION_NAMES.FLUE),
+      parentSpanIds: { parentSpanIds: [], rootSpanId },
+      spanId,
+      state
+    }),
+    state
+  );
+}
 function runWithCurrentSpanStore(span, next) {
   const state = _internalGetGlobalState();
   const contextManager = state?.contextManager;
@@ -18020,12 +18564,14 @@ function getMetricsFromResponse(response) {
       continue;
     }
     const inputTokenDetails = usageMetadata.input_token_details;
+    const outputTokenDetails = usageMetadata.output_token_details;
     return normalizeTokenMetrics({
       total_tokens: usageMetadata.total_tokens,
       prompt_tokens: usageMetadata.input_tokens,
       completion_tokens: usageMetadata.output_tokens,
       prompt_cache_creation_tokens: isRecord(inputTokenDetails) ? inputTokenDetails.cache_creation : void 0,
-      prompt_cached_tokens: isRecord(inputTokenDetails) ? inputTokenDetails.cache_read : void 0
+      prompt_cached_tokens: isRecord(inputTokenDetails) ? inputTokenDetails.cache_read : void 0,
+      completion_reasoning_tokens: isRecord(outputTokenDetails) ? outputTokenDetails.reasoning : void 0
     });
   }
   const llmOutput = response.llmOutput || {};
@@ -18403,13 +18949,33 @@ function aggregateChatLogprobs(existing, incoming) {
   }
   return aggregated;
 }
+function createAggregatedChatChoice(index) {
+  return {
+    index,
+    role: void 0,
+    content: void 0,
+    refusal: void 0,
+    toolCallsByIndex: /* @__PURE__ */ new Map(),
+    logprobs: void 0,
+    finish_reason: void 0
+  };
+}
+function toChatChoice(choice) {
+  const toolCalls = Array.from(choice.toolCallsByIndex.entries()).sort(([left], [right]) => left - right).map(([, toolCall]) => toolCall);
+  return {
+    index: choice.index,
+    message: {
+      role: choice.role,
+      content: choice.content,
+      ...choice.refusal !== void 0 ? { refusal: choice.refusal } : {},
+      tool_calls: toolCalls.length > 0 ? toolCalls : void 0
+    },
+    logprobs: choice.logprobs ?? null,
+    finish_reason: choice.finish_reason
+  };
+}
 function aggregateChatCompletionChunks(chunks, streamResult, endEvent) {
-  let role = void 0;
-  let content = void 0;
-  let refusal = void 0;
-  let tool_calls = void 0;
-  let logprobs = void 0;
-  let finish_reason = void 0;
+  const choicesByIndex = /* @__PURE__ */ new Map();
   let metrics = {};
   for (const chunk of chunks) {
     if (chunk.usage) {
@@ -18418,62 +18984,75 @@ function aggregateChatCompletionChunks(chunks, streamResult, endEvent) {
         ...parseMetricsFromUsage(chunk.usage)
       };
     }
-    const choice = chunk.choices?.[0];
-    if (!choice) {
+    const choices = chunk.choices;
+    if (!choices?.length) {
       continue;
     }
-    if (choice.finish_reason) {
-      finish_reason = choice.finish_reason;
-    }
-    logprobs = aggregateChatLogprobs(logprobs, choice.logprobs);
-    const delta = choice.delta;
-    if (!delta) {
-      continue;
-    }
-    if (delta.finish_reason) {
-      finish_reason = delta.finish_reason;
-    }
-    if (!role && delta.role) {
-      role = delta.role;
-    }
-    if (delta.content) {
-      content = (content || "") + delta.content;
-    }
-    if (delta.refusal) {
-      refusal = (refusal || "") + delta.refusal;
-    }
-    if (delta.tool_calls) {
-      const toolDelta = delta.tool_calls[0];
-      if (!tool_calls || toolDelta.id && tool_calls[tool_calls.length - 1].id !== toolDelta.id) {
-        tool_calls = [
-          ...tool_calls || [],
-          {
-            id: toolDelta.id,
-            type: toolDelta.type,
-            function: toolDelta.function
+    for (const choice of choices) {
+      const choiceIndex = choice.index;
+      let aggregatedChoice = choicesByIndex.get(choiceIndex);
+      if (!aggregatedChoice) {
+        aggregatedChoice = createAggregatedChatChoice(choiceIndex);
+        choicesByIndex.set(choiceIndex, aggregatedChoice);
+      }
+      if (choice.finish_reason) {
+        aggregatedChoice.finish_reason = choice.finish_reason;
+      }
+      aggregatedChoice.logprobs = aggregateChatLogprobs(
+        aggregatedChoice.logprobs,
+        choice.logprobs
+      );
+      const delta = choice.delta;
+      if (!delta) {
+        continue;
+      }
+      if (delta.finish_reason) {
+        aggregatedChoice.finish_reason = delta.finish_reason;
+      }
+      if (!aggregatedChoice.role && delta.role) {
+        aggregatedChoice.role = delta.role;
+      }
+      if (delta.content) {
+        aggregatedChoice.content = (aggregatedChoice.content || "") + delta.content;
+      }
+      if (delta.refusal) {
+        aggregatedChoice.refusal = (aggregatedChoice.refusal || "") + delta.refusal;
+      }
+      if (delta.tool_calls) {
+        for (const toolDelta of delta.tool_calls) {
+          let aggregatedToolCall = aggregatedChoice.toolCallsByIndex.get(
+            toolDelta.index
+          );
+          if (!aggregatedToolCall) {
+            aggregatedToolCall = {
+              function: { arguments: "" }
+            };
+            aggregatedChoice.toolCallsByIndex.set(
+              toolDelta.index,
+              aggregatedToolCall
+            );
           }
-        ];
-      } else {
-        tool_calls[tool_calls.length - 1].function.arguments += toolDelta.function.arguments;
+          if (toolDelta.id !== void 0) {
+            aggregatedToolCall.id = toolDelta.id;
+          }
+          if (toolDelta.type !== void 0) {
+            aggregatedToolCall.type = toolDelta.type;
+          }
+          if (toolDelta.function?.name !== void 0) {
+            aggregatedToolCall.function.name = toolDelta.function.name;
+          }
+          if (toolDelta.function?.arguments !== void 0) {
+            aggregatedToolCall.function.arguments += toolDelta.function.arguments;
+          }
+        }
       }
     }
   }
   metrics = withCachedMetric(metrics, streamResult, endEvent);
+  const output = Array.from(choicesByIndex.values()).sort((left, right) => left.index - right.index).map(toChatChoice);
   return {
     metrics,
-    output: [
-      {
-        index: 0,
-        message: {
-          role,
-          content,
-          ...refusal !== void 0 ? { refusal } : {},
-          tool_calls
-        },
-        logprobs: logprobs ?? null,
-        finish_reason
-      }
-    ]
+    output: output.length > 0 ? output : [toChatChoice(createAggregatedChatChoice(0))]
   };
 }
 function aggregateResponseStreamEvents(chunks, _streamResult, endEvent) {
@@ -19152,12 +19731,30 @@ function logInstrumentationError2(context, error) {
 
 // src/wrappers/anthropic-tokens-util.ts
 function finalizeAnthropicTokens(metrics) {
-  const prompt_tokens = (metrics.prompt_tokens || 0) + (metrics.prompt_cached_tokens || 0) + (metrics.prompt_cache_creation_tokens || 0);
-  return {
+  const hasSplitCacheCreationTokens = metrics.prompt_cache_creation_5m_tokens !== void 0 || metrics.prompt_cache_creation_1h_tokens !== void 0;
+  const splitCacheCreationTokens = (metrics.prompt_cache_creation_5m_tokens || 0) + (metrics.prompt_cache_creation_1h_tokens || 0);
+  const aggregateCacheCreationTokens = metrics.prompt_cache_creation_tokens || 0;
+  const effectiveCacheCreationTokens = Math.max(
+    aggregateCacheCreationTokens,
+    splitCacheCreationTokens
+  );
+  const prompt_tokens = (metrics.prompt_tokens || 0) + (metrics.prompt_cached_tokens || 0) + effectiveCacheCreationTokens;
+  const finalized = {
     ...metrics,
     prompt_tokens,
     tokens: prompt_tokens + (metrics.completion_tokens || 0)
   };
+  if (hasSplitCacheCreationTokens && splitCacheCreationTokens >= aggregateCacheCreationTokens) {
+    delete finalized.prompt_cache_creation_tokens;
+  }
+  return finalized;
+}
+function toNumericMetrics(metrics) {
+  return Object.fromEntries(
+    Object.entries(metrics).filter(
+      (entry) => entry[1] !== void 0
+    )
+  );
 }
 function extractAnthropicCacheTokens(cacheReadTokens = 0, cacheCreationTokens = 0) {
   const cacheTokens = {};
@@ -19716,6 +20313,24 @@ function parseMetricsFromUsage2(usage) {
   saveIfExistsTo("output_tokens", "completion_tokens");
   saveIfExistsTo("cache_read_input_tokens", "prompt_cached_tokens");
   saveIfExistsTo("cache_creation_input_tokens", "prompt_cache_creation_tokens");
+  if (isObject(usage.cache_creation)) {
+    const cacheCreation = usage.cache_creation;
+    for (const [source, target] of [
+      ["ephemeral_5m_input_tokens", "prompt_cache_creation_5m_tokens"],
+      ["ephemeral_1h_input_tokens", "prompt_cache_creation_1h_tokens"]
+    ]) {
+      const value = cacheCreation[source];
+      if (typeof value === "number") {
+        metrics[target] = value;
+      }
+    }
+  }
+  if (isObject(usage.output_tokens_details)) {
+    const thinkingTokens = usage.output_tokens_details.thinking_tokens;
+    if (typeof thinkingTokens === "number") {
+      metrics.completion_reasoning_tokens = thinkingTokens;
+    }
+  }
   if (isObject(usage.server_tool_use)) {
     for (const [name, value] of Object.entries(usage.server_tool_use)) {
       if (typeof value === "number") {
@@ -20281,118 +20896,22 @@ function coalesceInput(messages, system) {
 var CLAUDE_AGENT_SDK_SKIP_LOCAL_TOOL_HOOKS_OPTION = "__braintrust_skip_local_tool_hooks";
 
 // src/instrumentation/plugins/claude-agent-sdk-local-tool-context.ts
-var LOCAL_TOOL_CONTEXT_ASYNC_ITERATOR_PATCHED = /* @__PURE__ */ Symbol.for(
-  "braintrust.claude_agent_sdk.local_tool_context_async_iterator_patched"
-);
-function createLocalToolContextStore() {
-  const maybeIsoWithAsyncLocalStorage = isomorph_default;
-  if (typeof maybeIsoWithAsyncLocalStorage.newAsyncLocalStorage === "function") {
-    return maybeIsoWithAsyncLocalStorage.newAsyncLocalStorage();
-  }
-  let currentStore;
-  return {
-    enterWith(store) {
-      currentStore = store;
-    },
-    getStore() {
-      return currentStore;
-    },
-    run(store, callback) {
-      const previousStore = currentStore;
-      currentStore = store;
-      try {
-        return callback();
-      } finally {
-        currentStore = previousStore;
-      }
-    }
-  };
-}
-var localToolContextStore = createLocalToolContextStore();
-var fallbackLocalToolParentResolver;
-function createClaudeLocalToolContext() {
-  return {};
-}
-function runWithClaudeLocalToolContext(callback, context) {
-  return localToolContextStore.run(
-    context ?? createClaudeLocalToolContext(),
-    callback
-  );
-}
-function ensureClaudeLocalToolContext() {
-  const existing = localToolContextStore.getStore();
-  if (existing) {
-    return existing;
-  }
-  const created = {};
-  localToolContextStore.enterWith(created);
-  return created;
-}
-function setClaudeLocalToolParentResolver(resolver) {
-  fallbackLocalToolParentResolver = resolver;
-  const context = ensureClaudeLocalToolContext();
-  if (!context) {
-    return;
-  }
-  context.resolveLocalToolParent = resolver;
-}
-function getClaudeLocalToolParentResolver() {
-  return localToolContextStore.getStore()?.resolveLocalToolParent ?? fallbackLocalToolParentResolver;
+var localToolContextStore = isomorph_default.newAsyncLocalStorage();
+var localToolParentResolversByToolUseId = /* @__PURE__ */ new Map();
+function runWithClaudeLocalToolContext(callback, resolver) {
+  return localToolContextStore.run(resolver, callback);
 }
-function isAsyncIterable3(value) {
-  return value !== null && typeof value === "object" && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
+function registerClaudeLocalToolParentResolver(toolUseId, resolver) {
+  localToolParentResolversByToolUseId.set(toolUseId, resolver);
 }
-function bindClaudeLocalToolContextToAsyncIterable(result, localToolContext) {
-  if (!isAsyncIterable3(result) || Object.isFrozen(result) || Object.isSealed(result)) {
-    return result;
-  }
-  const stream = result;
-  const originalAsyncIterator = stream[Symbol.asyncIterator];
-  if (originalAsyncIterator[LOCAL_TOOL_CONTEXT_ASYNC_ITERATOR_PATCHED]) {
-    return result;
+function getClaudeLocalToolParentResolver(toolUseId) {
+  const currentResolver = localToolContextStore.getStore();
+  if (!toolUseId) {
+    return currentResolver;
   }
-  const patchedAsyncIterator = function() {
-    return runWithClaudeLocalToolContext(() => {
-      const iterator = Reflect.apply(originalAsyncIterator, this, []);
-      if (!iterator || typeof iterator !== "object") {
-        return iterator;
-      }
-      const patchMethod = (methodName) => {
-        const originalMethod = Reflect.get(iterator, methodName);
-        if (typeof originalMethod !== "function") {
-          return;
-        }
-        Reflect.set(
-          iterator,
-          methodName,
-          (...args) => runWithClaudeLocalToolContext(
-            () => Reflect.apply(
-              originalMethod,
-              iterator,
-              args
-            ),
-            localToolContext
-          )
-        );
-      };
-      patchMethod("next");
-      patchMethod("return");
-      patchMethod("throw");
-      return iterator;
-    }, localToolContext);
-  };
-  Object.defineProperty(
-    patchedAsyncIterator,
-    LOCAL_TOOL_CONTEXT_ASYNC_ITERATOR_PATCHED,
-    {
-      configurable: false,
-      enumerable: false,
-      value: true,
-      writable: false
-    }
-  );
-  Reflect.set(stream, Symbol.asyncIterator, patchedAsyncIterator);
-  return result;
+  const registeredResolver = localToolParentResolversByToolUseId.get(toolUseId);
+  localToolParentResolversByToolUseId.delete(toolUseId);
+  return currentResolver ?? registeredResolver;
 }
 
 // src/instrumentation/plugins/claude-agent-sdk-local-tool-spans.ts
@@ -20418,7 +20937,7 @@ function wrapLocalClaudeToolHandler(handler, getMetadata) {
     const metadata = getMetadata();
     const rawToolName = metadata.serverName ? `mcp__${metadata.serverName}__${metadata.toolName}` : metadata.toolName;
     const toolUseId = getToolUseIdFromExtra(handlerArgs[1]);
-    const localToolParentResolver = getClaudeLocalToolParentResolver();
+    const localToolParentResolver = getClaudeLocalToolParentResolver(toolUseId);
     const spanName = metadata.serverName ? `tool: ${metadata.serverName}/${metadata.toolName}` : `tool: ${metadata.toolName}`;
     const runWithResolvedParent = async () => {
       const parent = toolUseId && localToolParentResolver ? await localToolParentResolver(toolUseId).catch(() => void 0) : void 0;
@@ -20687,37 +21206,97 @@ function seedTaskToolUseIdMapping(taskIdToToolUseId, message) {
     taskIdToToolUseId.set(message.task_id, message.tool_use_id);
   }
 }
-function extractUsageFromMessage(message) {
-  const metrics = {};
-  let usage;
-  if (message.type === "assistant") {
-    usage = message.message?.usage;
-  } else if (message.type === "result") {
-    usage = message.usage;
-  }
+function tokenCount(value) {
+  return typeof value === "number" && Number.isFinite(value) && Number.isInteger(value) && value >= 0 ? value : void 0;
+}
+function copyUsage(usage) {
   if (!usage || typeof usage !== "object") {
-    return metrics;
+    return void 0;
+  }
+  const copy = {};
+  for (const key of [
+    "input_tokens",
+    "output_tokens",
+    "cache_read_input_tokens",
+    "cache_creation_input_tokens"
+  ]) {
+    const value = tokenCount(Reflect.get(usage, key));
+    if (value !== void 0) {
+      copy[key] = value;
+    }
+  }
+  const cacheCreation = Reflect.get(usage, "cache_creation");
+  if (cacheCreation && typeof cacheCreation === "object") {
+    const cacheCreationCopy = {};
+    for (const key of [
+      "ephemeral_5m_input_tokens",
+      "ephemeral_1h_input_tokens"
+    ]) {
+      const value = tokenCount(Reflect.get(cacheCreation, key));
+      if (value !== void 0) {
+        cacheCreationCopy[key] = value;
+      }
+    }
+    if (Object.keys(cacheCreationCopy).length > 0) {
+      copy.cache_creation = cacheCreationCopy;
+    }
+  }
+  return Object.keys(copy).length > 0 ? copy : void 0;
+}
+function mergeUsage(base, override) {
+  if (!base || !override) {
+    return override ?? base;
+  }
+  const cacheCreation = base.cache_creation || override.cache_creation ? { ...base.cache_creation, ...override.cache_creation } : void 0;
+  return {
+    ...base,
+    ...override,
+    ...cacheCreation && { cache_creation: cacheCreation }
+  };
+}
+function extractUsage(usage, includeOutput) {
+  const metrics = {};
+  if (!usage) {
+    return {};
   }
   const inputTokens = getNumberProperty(usage, "input_tokens");
   if (inputTokens !== void 0) {
     metrics.prompt_tokens = inputTokens;
   }
-  const outputTokens = getNumberProperty(usage, "output_tokens");
-  if (outputTokens !== void 0) {
-    metrics.completion_tokens = outputTokens;
+  if (includeOutput) {
+    const outputTokens = getNumberProperty(usage, "output_tokens");
+    if (outputTokens !== void 0) {
+      metrics.completion_tokens = outputTokens;
+    }
   }
   const cacheReadTokens = getNumberProperty(usage, "cache_read_input_tokens") || 0;
   const cacheCreationTokens = getNumberProperty(usage, "cache_creation_input_tokens") || 0;
-  if (cacheReadTokens > 0 || cacheCreationTokens > 0) {
-    Object.assign(
-      metrics,
-      extractAnthropicCacheTokens(cacheReadTokens, cacheCreationTokens)
-    );
+  Object.assign(
+    metrics,
+    extractAnthropicCacheTokens(cacheReadTokens, cacheCreationTokens)
+  );
+  const cacheCreation5mTokens = getNumberProperty(
+    usage.cache_creation,
+    "ephemeral_5m_input_tokens"
+  );
+  const cacheCreation1hTokens = getNumberProperty(
+    usage.cache_creation,
+    "ephemeral_1h_input_tokens"
+  );
+  if (cacheCreation5mTokens !== void 0) {
+    metrics.prompt_cache_creation_5m_tokens = cacheCreation5mTokens;
   }
-  if (Object.keys(metrics).length > 0) {
-    Object.assign(metrics, finalizeAnthropicTokens(metrics));
+  if (cacheCreation1hTokens !== void 0) {
+    metrics.prompt_cache_creation_1h_tokens = cacheCreation1hTokens;
   }
-  return metrics;
+  if (Object.keys(metrics).length === 0) {
+    return {};
+  }
+  const finalized = finalizeAnthropicTokens(metrics);
+  if (metrics.completion_tokens === void 0) {
+    delete finalized.tokens;
+  }
+  return toNumericMetrics(finalized);
 }
 function buildLLMInput(promptMessages, conversationHistory) {
   const inputParts = [...promptMessages, ...conversationHistory];
@@ -20751,16 +21330,16 @@ function buildRootPromptMessages(prompt, capturedPromptMessages) {
 function formatCapturedMessages(messages) {
   return messages.length > 0 ? messages : [];
 }
-async function createLLMSpanForMessages(messages, promptMessages, conversationHistory, options, startTime, parentSpan, existingSpan) {
+async function createLLMSpanForMessages(messages, promptMessages, conversationHistory, options, startTime, parentSpan, usage, hasFinalOutputUsage, existingSpan) {
   if (messages.length === 0) {
     return void 0;
   }
   const lastMessage = messages[messages.length - 1];
-  if (lastMessage.type !== "assistant" || !lastMessage.message?.usage) {
+  if (lastMessage.type !== "assistant") {
     return void 0;
   }
-  const model = lastMessage.message.model || options.model;
-  const usage = extractUsageFromMessage(lastMessage);
+  const model = lastMessage.message?.model || options.model;
+  const metrics = options.includePartialMessages ? extractUsage(usage, hasFinalOutputUsage) : {};
   const input = buildLLMInput(promptMessages, conversationHistory);
   const outputs = messages.map(
     (m) => m.message?.content && m.message?.role ? { content: m.message.content, role: m.message.role } : void 0
@@ -20782,8 +21361,8 @@ async function createLLMSpanForMessages(messages, promptMessages, conversationHi
   );
   span.log({
     input,
-    metadata: model ? { model } : void 0,
-    metrics: usage,
+    metadata: { ...model && { model }, provider: "anthropic" },
+    ...Object.keys(metrics).length > 0 ? { metrics } : {},
     output: outputs
   });
   const spanExport = await span.export();
@@ -20873,12 +21452,19 @@ function prepareLocalToolHandlersInMcpServers(mcpServers) {
   }
   return { hasLocalToolHandlers, localToolHookNames };
 }
-function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
+function createToolTracingHooks(resolveParentSpan, taskIdToToolUseId, toolUseToParent, activeToolSpans, mcpServers, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
   const preToolUse = async (input, toolUseID) => {
     if (input.hook_event_name !== "PreToolUse" || !toolUseID) {
       return {};
     }
+    if (!toolUseToParent.has(toolUseID) && input.agent_id) {
+      const parentToolUseId = taskIdToToolUseId.get(input.agent_id);
+      if (parentToolUseId) {
+        toolUseToParent.set(toolUseID, parentToolUseId);
+      }
+    }
     if (skipLocalToolHooks && (isLocalToolUse(input.tool_name, mcpServers) || localToolHookNames.has(input.tool_name))) {
+      registerClaudeLocalToolParentResolver(toolUseID, resolveParentSpan);
       return {};
     }
     const parsed = parseToolName(input.tool_name);
@@ -21073,9 +21659,6 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
     }
     const metadata = {
       ...subAgentDetailsToMetadata(details),
-      ...input.agent_transcript_path && {
-        "claude_agent_sdk.agent_transcript_path": input.agent_transcript_path
-      },
       "claude_agent_sdk.stop_hook_active": input.stop_hook_active
     };
     try {
@@ -21097,7 +21680,7 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
     subagentStop
   };
 }
-function injectTracingHooks(options, resolveParentSpan, activeToolSpans, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
+function injectTracingHooks(options, resolveParentSpan, taskIdToToolUseId, toolUseToParent, activeToolSpans, localToolHookNames, skipLocalToolHooks, subAgentDetailsByToolUseId, subAgentSpans, endedSubAgentSpans) {
   const {
     preToolUse,
     postToolUse,
@@ -21106,6 +21689,8 @@ function injectTracingHooks(options, resolveParentSpan, activeToolSpans, localTo
     subagentStop
   } = createToolTracingHooks(
     resolveParentSpan,
+    taskIdToToolUseId,
+    toolUseToParent,
     activeToolSpans,
     options.mcpServers,
     localToolHookNames,
@@ -21187,6 +21772,13 @@ async function finalizeCurrentMessageGroup(state) {
     }
   }
   const existingLlmSpan = state.activeLlmSpansByParentToolUse.get(parentKey);
+  const lastMessage = state.currentMessages[state.currentMessages.length - 1];
+  const messageId = lastMessage?.message?.id;
+  const usage = state.options.includePartialMessages ? mergeUsage(
+    copyUsage(lastMessage?.message?.usage),
+    messageId ? state.usageByMessageId.get(messageId) : void 0
+  ) : void 0;
+  const hasFinalOutputUsage = messageId !== void 0 && state.finalOutputUsageMessageIds.has(messageId);
   const llmSpanResult = await createLLMSpanForMessages(
     state.currentMessages,
     promptMessages,
@@ -21194,6 +21786,8 @@ async function finalizeCurrentMessageGroup(state) {
     state.options,
     state.currentMessageStartTime,
     parentSpan,
+    usage,
+    hasFinalOutputUsage,
     existingLlmSpan
   );
   if (llmSpanResult) {
@@ -21211,9 +21805,17 @@ async function finalizeCurrentMessageGroup(state) {
     }
   }
   state.activeLlmSpansByParentToolUse.delete(parentKey);
-  const lastMessage = state.currentMessages[state.currentMessages.length - 1];
-  if (lastMessage?.message?.usage) {
-    state.accumulatedOutputTokens += getNumberProperty(lastMessage.message.usage, "output_tokens") || 0;
+  if (messageId) {
+    state.usageByMessageId.delete(messageId);
+    state.finalOutputUsageMessageIds.delete(messageId);
+    for (const [
+      parent,
+      activeMessageId
+    ] of state.activePartialMessageIdByParentKey) {
+      if (activeMessageId === messageId) {
+        state.activePartialMessageIdByParentKey.delete(parent);
+      }
+    }
   }
   state.currentMessages.length = 0;
 }
@@ -21304,6 +21906,10 @@ async function ensureActiveLlmSpanForParentToolUse(rootSpan, activeLlmSpansByPar
     );
     llmParentSpan = await subAgentSpan.export();
   }
+  const racedLlmSpan = activeLlmSpansByParentToolUse.get(parentKey);
+  if (racedLlmSpan) {
+    return racedLlmSpan;
+  }
   const llmSpan = startSpan(
     withSpanInstrumentationName(
       {
@@ -21423,7 +22029,49 @@ async function maybeHandleTaskLifecycleMessage(state, message) {
   }
   return true;
 }
+function handlePartialUsageMessage(state, message) {
+  if (message.type !== "stream_event") {
+    return false;
+  }
+  const event = message.event;
+  if (!event || typeof event !== "object") {
+    return true;
+  }
+  const parentKey = llmParentKey(message.parent_tool_use_id ?? null);
+  if (event.type === "message_start") {
+    const messageId2 = event.message?.id;
+    const usage = copyUsage(event.message?.usage);
+    if (messageId2) {
+      state.activePartialMessageIdByParentKey.set(parentKey, messageId2);
+      if (usage) {
+        state.usageByMessageId.set(messageId2, usage);
+      }
+    }
+    return true;
+  }
+  const messageId = state.activePartialMessageIdByParentKey.get(parentKey);
+  if (!messageId) {
+    return true;
+  }
+  if (event.type === "message_delta") {
+    const update = copyUsage(event.usage);
+    if (update) {
+      const usage = state.usageByMessageId.get(messageId) ?? {};
+      Object.assign(usage, update);
+      state.usageByMessageId.set(messageId, usage);
+      if (update.output_tokens !== void 0) {
+        state.finalOutputUsageMessageIds.add(messageId);
+      }
+    }
+  } else if (event.type === "message_stop") {
+    state.activePartialMessageIdByParentKey.delete(parentKey);
+  }
+  return true;
+}
 async function handleStreamMessage(state, message) {
+  if (handlePartialUsageMessage(state, message)) {
+    return;
+  }
   maybeTrackToolUseContext(state, message);
   if (await maybeHandleTaskLifecycleMessage(state, message)) {
     return;
@@ -21470,36 +22118,9 @@ async function handleStreamMessage(state, message) {
     );
     state.currentMessages.push(message);
   }
-  if (message.type !== "result" || !message.usage) {
+  if (message.type !== "result") {
     return;
   }
-  const finalUsageMetrics = extractUsageFromMessage(message);
-  if (state.currentMessages.length > 0 && finalUsageMetrics.completion_tokens !== void 0) {
-    const lastMessage = state.currentMessages[state.currentMessages.length - 1];
-    if (lastMessage?.message?.usage) {
-      const adjustedTokens = finalUsageMetrics.completion_tokens - state.accumulatedOutputTokens;
-      if (adjustedTokens >= 0) {
-        lastMessage.message.usage.output_tokens = adjustedTokens;
-      }
-      const resultUsage = message.usage;
-      if (resultUsage && typeof resultUsage === "object") {
-        const cacheReadTokens = getNumberProperty(
-          resultUsage,
-          "cache_read_input_tokens"
-        );
-        if (cacheReadTokens !== void 0) {
-          lastMessage.message.usage.cache_read_input_tokens = cacheReadTokens;
-        }
-        const cacheCreationTokens = getNumberProperty(
-          resultUsage,
-          "cache_creation_input_tokens"
-        );
-        if (cacheCreationTokens !== void 0) {
-          lastMessage.message.usage.cache_creation_input_tokens = cacheCreationTokens;
-        }
-      }
-    }
-  }
   const metadata = {};
   if (message.num_turns !== void 0) {
     metadata.num_turns = message.num_turns;
@@ -21507,8 +22128,12 @@ async function handleStreamMessage(state, message) {
   if (message.session_id !== void 0) {
     metadata.session_id = message.session_id;
   }
-  if (Object.keys(metadata).length > 0) {
-    state.span.log({ metadata });
+  const metrics = state.options.includePartialMessages ? {} : extractUsage(copyUsage(message.usage), true);
+  if (Object.keys(metadata).length > 0 || Object.keys(metrics).length > 0) {
+    state.span.log({
+      ...Object.keys(metadata).length > 0 ? { metadata } : {},
+      ...Object.keys(metrics).length > 0 ? { metrics } : {}
+    });
   }
 }
 async function finalizeQuerySpan(state) {
@@ -21532,6 +22157,9 @@ async function finalizeQuerySpan(state) {
       llmSpan.end();
     }
     state.activeLlmSpansByParentToolUse.clear();
+    state.activePartialMessageIdByParentKey.clear();
+    state.finalOutputUsageMessageIds.clear();
+    state.usageByMessageId.clear();
     for (const toolSpan of state.activeToolSpans.values()) {
       toolSpan.end();
     }
@@ -21547,7 +22175,7 @@ async function finalizeQuerySpan(state) {
 }
 var ClaudeAgentSDKPlugin = class extends BasePlugin {
   onEnable() {
-    this.subscribeToQuery();
+    this.interceptQuery();
   }
   onDisable() {
     for (const unsubscribe of this.unsubscribers) {
@@ -21555,207 +22183,218 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
     }
     this.unsubscribers = [];
   }
-  subscribeToQuery() {
-    const channel = claudeAgentSDKChannels.query.tracingChannel();
-    const spans = /* @__PURE__ */ new WeakMap();
-    const handlers = {
-      start: (event) => {
-        const params = event.arguments[0] ?? {};
-        const originalPrompt = params.prompt;
-        const options = params.options ?? {};
-        const promptIsAsyncIterable = isAsyncIterable(originalPrompt);
-        let promptStarted = false;
-        let capturedPromptMessages;
-        let resolvePromptDone;
-        const promptDone = new Promise((resolve) => {
-          resolvePromptDone = resolve;
-        });
-        if (promptIsAsyncIterable) {
-          capturedPromptMessages = [];
-          const promptStream = originalPrompt;
-          params.prompt = (async function* () {
-            promptStarted = true;
-            try {
-              for await (const message of promptStream) {
-                capturedPromptMessages.push(message);
-                yield message;
-              }
-            } finally {
-              resolvePromptDone?.();
+  interceptQuery() {
+    const startQuery = (params) => {
+      const originalPrompt = params.prompt;
+      const options = params.options ?? {};
+      const promptIsAsyncIterable = isAsyncIterable(originalPrompt);
+      let promptStarted = false;
+      let capturedPromptMessages;
+      let resolvePromptDone;
+      const promptDone = new Promise((resolve) => {
+        resolvePromptDone = resolve;
+      });
+      if (promptIsAsyncIterable) {
+        capturedPromptMessages = [];
+        const promptStream = originalPrompt;
+        params.prompt = (async function* () {
+          promptStarted = true;
+          try {
+            for await (const message of promptStream) {
+              capturedPromptMessages.push(message);
+              yield message;
             }
-          })();
-        }
-        const span = startSpan(
-          withSpanInstrumentationName(
-            {
-              name: "Claude Agent",
-              spanAttributes: {
-                type: "task" /* TASK */
-              }
-            },
-            INSTRUMENTATION_NAMES.CLAUDE_AGENT_SDK
-          )
-        );
-        const startTime = getCurrentUnixTimestamp();
-        try {
-          span.log({
-            input: typeof originalPrompt === "string" ? originalPrompt : promptIsAsyncIterable ? void 0 : originalPrompt !== void 0 ? String(originalPrompt) : void 0,
-            metadata: filterSerializableOptions(options)
-          });
-        } catch (error) {
-          console.error("Error extracting input for Claude Agent SDK:", error);
-        }
-        const activeToolSpans = /* @__PURE__ */ new Map();
-        const activeLlmSpansByParentToolUse = /* @__PURE__ */ new Map();
-        const conversationHistoryByParentKey = /* @__PURE__ */ new Map();
-        const subAgentSpans = /* @__PURE__ */ new Map();
-        const endedSubAgentSpans = /* @__PURE__ */ new Set();
-        const toolUseToParent = /* @__PURE__ */ new Map();
-        const latestLlmParentBySubAgentToolUse = /* @__PURE__ */ new Map();
-        const latestRootLlmParentRef = {
-          value: void 0
-        };
-        const subAgentDetailsByToolUseId = /* @__PURE__ */ new Map();
-        const taskIdToToolUseId = /* @__PURE__ */ new Map();
-        const promptMessagesByParentKey = /* @__PURE__ */ new Map();
-        const promptSourcePriorityByParentKey = /* @__PURE__ */ new Map();
-        const localToolContext = createClaudeLocalToolContext();
-        const { hasLocalToolHandlers, localToolHookNames } = prepareLocalToolHandlersInMcpServers(options.mcpServers);
-        const skipLocalToolHooks = options[CLAUDE_AGENT_SDK_SKIP_LOCAL_TOOL_HOOKS_OPTION] === true || hasLocalToolHandlers;
-        const resolveToolUseParentSpan = async (toolUseID, context) => {
-          const trackedParentToolUseId = toolUseToParent.get(toolUseID);
-          const parentToolUseId = trackedParentToolUseId ?? (context?.agentId ? taskIdToToolUseId.get(context.agentId) ?? null : null);
-          const parentKey = llmParentKey(parentToolUseId);
-          const activeLlmSpan = activeLlmSpansByParentToolUse.get(parentKey);
-          const latestLlmParent = parentToolUseId ? latestLlmParentBySubAgentToolUse.get(parentToolUseId) : latestRootLlmParentRef.value;
-          if (!activeLlmSpan && !latestLlmParent) {
-            await ensureActiveLlmSpanForParentToolUse(
-              span,
-              activeLlmSpansByParentToolUse,
-              subAgentDetailsByToolUseId,
-              activeToolSpans,
-              subAgentSpans,
-              parentToolUseId,
-              getCurrentUnixTimestamp()
-            );
-          }
-          if (parentToolUseId) {
-            const subAgentSpan = await ensureSubAgentSpan(
-              subAgentDetailsByToolUseId,
-              span,
-              activeToolSpans,
-              subAgentSpans,
-              parentToolUseId
-            );
-            return subAgentSpan.export();
+          } finally {
+            resolvePromptDone?.();
           }
-          return span.export();
-        };
-        localToolContext.resolveLocalToolParent = resolveToolUseParentSpan;
-        setClaudeLocalToolParentResolver(resolveToolUseParentSpan);
-        const optionsWithHooks = injectTracingHooks(
-          options,
-          resolveToolUseParentSpan,
-          activeToolSpans,
-          localToolHookNames,
-          skipLocalToolHooks,
-          subAgentDetailsByToolUseId,
-          subAgentSpans,
-          endedSubAgentSpans
-        );
-        params.options = optionsWithHooks;
-        event.arguments[0] = params;
-        spans.set(event, {
-          accumulatedOutputTokens: 0,
-          activeLlmSpansByParentToolUse,
-          activeToolSpans,
-          conversationHistoryByParentKey,
-          capturedPromptMessages,
-          currentMessageId: void 0,
-          currentMessageStartTime: startTime,
-          currentMessages: [],
-          endedSubAgentSpans,
-          finalResults: [],
-          options: optionsWithHooks,
-          originalPrompt,
-          processing: Promise.resolve(),
-          promptDone,
-          promptMessagesByParentKey,
-          promptStarted: () => promptStarted,
-          promptSourcePriorityByParentKey,
-          span,
-          subAgentDetailsByToolUseId,
-          subAgentSpans,
-          taskIdToToolUseId,
-          latestLlmParentBySubAgentToolUse,
-          latestRootLlmParentRef,
-          toolUseToParent,
-          localToolContext
+        })();
+      }
+      const span = startSpan(
+        withSpanInstrumentationName(
+          {
+            name: "Claude Agent",
+            spanAttributes: {
+              type: "task" /* TASK */
+            }
+          },
+          INSTRUMENTATION_NAMES.CLAUDE_AGENT_SDK
+        )
+      );
+      const startTime = getCurrentUnixTimestamp();
+      try {
+        span.log({
+          input: typeof originalPrompt === "string" ? originalPrompt : promptIsAsyncIterable ? void 0 : originalPrompt !== void 0 ? String(originalPrompt) : void 0,
+          metadata: filterSerializableOptions(options)
         });
-      },
-      end: (event) => {
-        const state = spans.get(event);
-        if (!state) {
-          return;
-        }
-        const eventResult = bindClaudeLocalToolContextToAsyncIterable(
-          event.result,
-          state.localToolContext
-        );
-        if (eventResult === void 0) {
-          state.span.end();
-          spans.delete(event);
-          return;
-        }
-        if (isAsyncIterable(eventResult)) {
-          patchStreamIfNeeded(eventResult, {
-            onChunk: (message) => {
-              maybeTrackToolUseContext(state, message);
-              state.processing = state.processing.then(() => handleStreamMessage(state, message)).catch((error) => {
-                console.error(
-                  "Error processing Claude Agent SDK stream chunk:",
-                  error
-                );
-              });
-            },
-            onComplete: () => state.processing.then(() => finalizeQuerySpan(state)).finally(() => {
-              spans.delete(event);
-            }),
-            onError: (error) => state.processing.then(() => {
-              state.span.log({
-                error: error.message
-              });
-            }).then(() => finalizeQuerySpan(state)).finally(() => {
-              spans.delete(event);
-            })
-          });
-          return;
-        }
-        try {
-          state.span.log({ output: eventResult });
-        } catch (error) {
-          console.error("Error extracting output for Claude Agent SDK:", error);
-        } finally {
-          state.span.end();
-          spans.delete(event);
+      } catch (error) {
+        console.error("Error extracting input for Claude Agent SDK:", error);
+      }
+      const activeToolSpans = /* @__PURE__ */ new Map();
+      const activeLlmSpansByParentToolUse = /* @__PURE__ */ new Map();
+      const conversationHistoryByParentKey = /* @__PURE__ */ new Map();
+      const subAgentSpans = /* @__PURE__ */ new Map();
+      const endedSubAgentSpans = /* @__PURE__ */ new Set();
+      const toolUseToParent = /* @__PURE__ */ new Map();
+      const latestLlmParentBySubAgentToolUse = /* @__PURE__ */ new Map();
+      const latestRootLlmParentRef = {
+        value: void 0
+      };
+      const subAgentDetailsByToolUseId = /* @__PURE__ */ new Map();
+      const taskIdToToolUseId = /* @__PURE__ */ new Map();
+      const promptMessagesByParentKey = /* @__PURE__ */ new Map();
+      const promptSourcePriorityByParentKey = /* @__PURE__ */ new Map();
+      const { hasLocalToolHandlers, localToolHookNames } = prepareLocalToolHandlersInMcpServers(options.mcpServers);
+      const skipLocalToolHooks = options[CLAUDE_AGENT_SDK_SKIP_LOCAL_TOOL_HOOKS_OPTION] === true || hasLocalToolHandlers;
+      const resolveToolUseParentSpan = async (toolUseID, context) => {
+        const trackedParentToolUseId = toolUseToParent.get(toolUseID);
+        const parentToolUseId = trackedParentToolUseId ?? (context?.agentId ? taskIdToToolUseId.get(context.agentId) ?? null : null);
+        const parentKey = llmParentKey(parentToolUseId);
+        const activeLlmSpan = activeLlmSpansByParentToolUse.get(parentKey);
+        const latestLlmParent = parentToolUseId ? latestLlmParentBySubAgentToolUse.get(parentToolUseId) : latestRootLlmParentRef.value;
+        if (!activeLlmSpan && !latestLlmParent) {
+          await ensureActiveLlmSpanForParentToolUse(
+            span,
+            activeLlmSpansByParentToolUse,
+            subAgentDetailsByToolUseId,
+            activeToolSpans,
+            subAgentSpans,
+            parentToolUseId,
+            getCurrentUnixTimestamp()
+          );
         }
-      },
-      error: (event) => {
-        const state = spans.get(event);
-        if (!state || !event.error) {
-          return;
+        if (parentToolUseId) {
+          const subAgentSpan = await ensureSubAgentSpan(
+            subAgentDetailsByToolUseId,
+            span,
+            activeToolSpans,
+            subAgentSpans,
+            parentToolUseId
+          );
+          return subAgentSpan.export();
         }
-        state.span.log({
-          error: event.error.message
+        return span.export();
+      };
+      const optionsWithHooks = injectTracingHooks(
+        options,
+        resolveToolUseParentSpan,
+        taskIdToToolUseId,
+        toolUseToParent,
+        activeToolSpans,
+        localToolHookNames,
+        skipLocalToolHooks,
+        subAgentDetailsByToolUseId,
+        subAgentSpans,
+        endedSubAgentSpans
+      );
+      params.options = optionsWithHooks;
+      return {
+        activeLlmSpansByParentToolUse,
+        activePartialMessageIdByParentKey: /* @__PURE__ */ new Map(),
+        activeToolSpans,
+        conversationHistoryByParentKey,
+        capturedPromptMessages,
+        currentMessageId: void 0,
+        currentMessageStartTime: startTime,
+        currentMessages: [],
+        endedSubAgentSpans,
+        finalOutputUsageMessageIds: /* @__PURE__ */ new Set(),
+        finalResults: [],
+        options: optionsWithHooks,
+        originalPrompt,
+        processing: Promise.resolve(),
+        promptDone,
+        promptMessagesByParentKey,
+        promptStarted: () => promptStarted,
+        promptSourcePriorityByParentKey,
+        span,
+        subAgentDetailsByToolUseId,
+        subAgentSpans,
+        taskIdToToolUseId,
+        latestLlmParentBySubAgentToolUse,
+        latestRootLlmParentRef,
+        toolUseToParent,
+        usageByMessageId: /* @__PURE__ */ new Map(),
+        localToolParentResolver: resolveToolUseParentSpan
+      };
+    };
+    const finishQuery = (state, result) => {
+      if (isAsyncIterable(result)) {
+        patchStreamIfNeeded(result, {
+          aroundNext: (callback) => runWithClaudeLocalToolContext(
+            callback,
+            state.localToolParentResolver
+          ),
+          onChunk: (message) => {
+            maybeTrackToolUseContext(state, message);
+            state.processing = state.processing.then(() => handleStreamMessage(state, message)).catch((error) => {
+              console.error(
+                "Error processing Claude Agent SDK stream chunk:",
+                error
+              );
+            });
+          },
+          onComplete: () => state.processing.then(() => finalizeQuerySpan(state)),
+          onError: (error) => state.processing.then(() => {
+            state.span.log({ error: error.message });
+          }).then(() => finalizeQuerySpan(state))
         });
+        return;
+      }
+      try {
+        state.span.log({ output: result });
+      } catch (error) {
+        console.error("Error extracting output for Claude Agent SDK:", error);
+      } finally {
         state.span.end();
-        spans.delete(event);
       }
     };
-    channel.subscribe(handlers);
-    this.unsubscribers.push(() => {
-      channel.unsubscribe(handlers);
-    });
+    this.unsubscribers.push(
+      claudeAgentSDKChannels.query.intercept((target, thisArg, args) => {
+        let state;
+        try {
+          args[0] ??= {};
+          state = startQuery(args[0]);
+        } catch (error) {
+          debugLogger.error(
+            "Error starting Claude Agent SDK instrumentation:",
+            error
+          );
+        }
+        const invokeTarget = () => Reflect.apply(target, thisArg, args);
+        try {
+          const result = state ? runWithClaudeLocalToolContext(
+            invokeTarget,
+            state.localToolParentResolver
+          ) : invokeTarget();
+          if (state) {
+            try {
+              finishQuery(state, result);
+            } catch (error) {
+              debugLogger.error(
+                "Error finalizing Claude Agent SDK instrumentation:",
+                error
+              );
+            }
+          }
+          return result;
+        } catch (error) {
+          if (state) {
+            try {
+              state.span.log({
+                error: error instanceof Error ? error.message : String(error)
+              });
+              state.span.end();
+            } catch (instrumentationError) {
+              debugLogger.error(
+                "Error handling Claude Agent SDK instrumentation failure:",
+                instrumentationError
+              );
+            }
+          }
+          throw error;
+        }
+      })
+    );
   }
 };
 
@@ -24080,9 +24719,9 @@ function extractEmbedPromptTokenCount(response) {
   let sawAny = false;
   for (const embedding of embeddings) {
     const embeddingStats = tryToDict(tryToDict(embedding)?.statistics);
-    const tokenCount = embeddingStats?.tokenCount;
-    if (typeof tokenCount === "number" && Number.isFinite(tokenCount)) {
-      total += tokenCount;
+    const tokenCount2 = embeddingStats?.tokenCount;
+    if (typeof tokenCount2 === "number" && Number.isFinite(tokenCount2)) {
+      total += tokenCount2;
       sawAny = true;
     }
   }
@@ -25697,7 +26336,7 @@ function patchOpenRouterCallModelResult(args) {
         span,
         () => originalMethod.apply(resultLike, args2)
       );
-      if (!isAsyncIterable4(stream)) {
+      if (!isAsyncIterable3(stream)) {
         return stream;
       }
       return wrapAsyncIterableWithSpan({
@@ -25892,7 +26531,7 @@ function wrapAsyncIterableWithSpan(args) {
     }
   };
 }
-function isAsyncIterable4(value) {
+function isAsyncIterable3(value) {
   return !!value && (typeof value === "object" || typeof value === "function") && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
 }
 function normalizeError(error) {
@@ -26734,7 +27373,7 @@ function patchOpenRouterCallModelResult2(args) {
         span,
         () => originalMethod.apply(resultLike, args2)
       );
-      if (!isAsyncIterable5(stream)) {
+      if (!isAsyncIterable4(stream)) {
         return stream;
       }
       return wrapAsyncIterableWithSpan2({
@@ -26929,7 +27568,7 @@ function wrapAsyncIterableWithSpan2(args) {
     }
   };
 }
-function isAsyncIterable5(value) {
+function isAsyncIterable4(value) {
   return !!value && (typeof value === "object" || typeof value === "function") && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
 }
 function normalizeError2(error) {
@@ -30632,7 +31271,7 @@ function getStringProperty2(obj, key) {
   return typeof value === "string" ? value : void 0;
 }
 function extractMetricsFromUsage(usage) {
-  const metrics = {
+  const rawMetrics = {
     prompt_tokens: usage.inputTokens,
     completion_tokens: usage.outputTokens,
     ...extractAnthropicCacheTokens(
@@ -30641,10 +31280,10 @@ function extractMetricsFromUsage(usage) {
     )
   };
   if (usage.reasoningTokens !== void 0) {
-    metrics.completion_reasoning_tokens = usage.reasoningTokens;
-    metrics.reasoning_tokens = usage.reasoningTokens;
+    rawMetrics.completion_reasoning_tokens = usage.reasoningTokens;
+    rawMetrics.reasoning_tokens = usage.reasoningTokens;
   }
-  Object.assign(metrics, finalizeAnthropicTokens(metrics));
+  const metrics = finalizeAnthropicTokens(rawMetrics);
   const metadata = {
     model: usage.model
   };
@@ -31675,6 +32314,9 @@ function sanitizeLoggedValue(value, seen = /* @__PURE__ */ new WeakSet(), depth
 // src/instrumentation/plugins/pi-coding-agent-plugin.ts
 var piAgentPatchStates = /* @__PURE__ */ new WeakMap();
 var piAgentEventSubscriptions = /* @__PURE__ */ new WeakSet();
+var PI_TOOL_EXECUTE_WRAPPED = /* @__PURE__ */ Symbol.for(
+  "braintrust.pi_coding_agent.tool_execute_wrapped"
+);
 var piPromptContextStore;
 var PiCodingAgentPlugin = class extends BasePlugin {
   activePromptStates = /* @__PURE__ */ new Set();
@@ -31755,6 +32397,7 @@ function startPiPromptRun(event, onFinalize) {
     return void 0;
   }
   installPiAgentInstrumentation(agent);
+  wrapPiToolExecutors(agent.state?.tools);
   const metadata = {
     ...extractSessionMetadata(session),
     ...extractPromptOptionsMetadata(event.arguments[1]),
@@ -31851,6 +32494,7 @@ function makeInstrumentedStreamFn(agent, originalStreamFn) {
     if (!state || state.agent !== agent || state.finalized) {
       return invokeOriginal();
     }
+    wrapPiToolExecutors(context.tools);
     const llmState = await startPiLlmSpan(state, model, context, options);
     try {
       const stream = await runWithAutoInstrumentationSuppressed(invokeOriginal);
@@ -31861,6 +32505,33 @@ function makeInstrumentedStreamFn(agent, originalStreamFn) {
     }
   };
 }
+function wrapPiToolExecutors(tools) {
+  if (!tools) {
+    return;
+  }
+  for (const tool of tools) {
+    try {
+      const execute = tool.execute;
+      if (typeof execute !== "function" || execute[PI_TOOL_EXECUTE_WRAPPED]) {
+        continue;
+      }
+      const wrappedExecute = function(...args) {
+        return runWithAutoInstrumentationAllowed(
+          () => Reflect.apply(execute, this, args)
+        );
+      };
+      Object.defineProperty(wrappedExecute, PI_TOOL_EXECUTE_WRAPPED, {
+        configurable: false,
+        enumerable: false,
+        value: true,
+        writable: false
+      });
+      tool.execute = wrappedExecute;
+    } catch (error) {
+      logInstrumentationError4("Pi Coding Agent tool wrapping", error);
+    }
+  }
+}
 async function startPiLlmSpan(state, model, context, options) {
   const metadata = {
     ...extractModelMetadata2(model),
@@ -32032,35 +32703,26 @@ async function startPiToolSpan(state, event) {
   if (!event.toolCallId || state.activeToolSpans.has(event.toolCallId)) {
     return;
   }
-  const restoreAutoInstrumentation = enterAutoInstrumentationAllowed();
   const metadata = {
     "gen_ai.tool.call.id": event.toolCallId,
     "gen_ai.tool.name": event.toolName,
     "pi_coding_agent.tool.name": event.toolName
   };
-  try {
-    const span = startSpan(
-      withSpanInstrumentationName(
-        {
-          event: {
-            input: event.args,
-            metadata
-          },
-          name: event.toolName || "tool",
-          parent: await state.span.export(),
-          spanAttributes: { type: "tool" /* TOOL */ }
+  const span = startSpan(
+    withSpanInstrumentationName(
+      {
+        event: {
+          input: event.args,
+          metadata
         },
-        INSTRUMENTATION_NAMES.PI_CODING_AGENT
-      )
-    );
-    state.activeToolSpans.set(event.toolCallId, {
-      restoreAutoInstrumentation,
-      span
-    });
-  } catch (error) {
-    restoreAutoInstrumentation();
-    throw error;
-  }
+        name: event.toolName || "tool",
+        parent: await state.span.export(),
+        spanAttributes: { type: "tool" /* TOOL */ }
+      },
+      INSTRUMENTATION_NAMES.PI_CODING_AGENT
+    )
+  );
+  state.activeToolSpans.set(event.toolCallId, { span });
 }
 function finishPiToolSpan(state, event) {
   const toolState = state.activeToolSpans.get(event.toolCallId);
@@ -32081,11 +32743,7 @@ function finishPiToolSpan(state, event) {
       output: event.result
     });
   } finally {
-    try {
-      toolState.span.end();
-    } finally {
-      toolState.restoreAutoInstrumentation?.();
-    }
+    toolState.span.end();
   }
 }
 function finishPiPromptRun(state, error) {
@@ -32156,14 +32814,10 @@ function finishPiLlmSpan(promptState, llmState, message, error) {
 }
 function finishOpenToolSpans(state, error) {
   for (const [, toolState] of state.activeToolSpans) {
-    try {
-      safeLog4(toolState.span, {
-        error: error ? toLoggedError(error) : "Pi tool did not complete"
-      });
-      toolState.span.end();
-    } finally {
-      toolState.restoreAutoInstrumentation?.();
-    }
+    safeLog4(toolState.span, {
+      error: error ? toLoggedError(error) : "Pi tool did not complete"
+    });
+    toolState.span.end();
   }
   state.activeToolSpans.clear();
 }
@@ -32462,12 +33116,12 @@ var MAX_STRANDS_STRING_ATTACHMENT_CACHE_ENTRIES = 32;
 var StrandsAgentSDKPlugin = class extends BasePlugin {
   activeChildParents = /* @__PURE__ */ new WeakMap();
   onEnable() {
-    this.subscribeToAgentStream();
-    this.subscribeToMultiAgentStream(
+    this.interceptAgentStream();
+    this.interceptMultiAgentStream(
       strandsAgentSDKChannels.graphStream,
       "Graph.stream"
     );
-    this.subscribeToMultiAgentStream(
+    this.interceptMultiAgentStream(
       strandsAgentSDKChannels.swarmStream,
       "Swarm.stream"
     );
@@ -32478,143 +33132,109 @@ var StrandsAgentSDKPlugin = class extends BasePlugin {
     }
     this.unsubscribers = [];
   }
-  subscribeToAgentStream() {
-    const channel = strandsAgentSDKChannels.agentStream.tracingChannel();
-    const states = /* @__PURE__ */ new WeakMap();
-    const unbindAutoInstrumentationSuppression = bindAutoInstrumentationSuppressionToStart(channel);
-    const handlers = {
-      start: (event) => {
-        const state = startAgentStream(event, this.activeChildParents);
-        if (state) {
-          states.set(event, state);
-        }
-      },
-      end: (event) => {
-        const state = states.get(event);
-        if (!state) {
-          return;
-        }
-        const result = event.result;
-        if (isAsyncIterable(result)) {
-          patchStreamIfNeeded(result, {
-            aroundNext: (callback) => runWithAutoInstrumentationSuppressed(callback),
-            onChunk: (chunk) => handleAgentStreamEvent(state, chunk),
-            onComplete: () => {
-              finalizeAgentStream(state);
-              states.delete(event);
-            },
-            onError: (error) => {
-              finalizeAgentStream(state, error);
-              states.delete(event);
-            }
-          });
-          return;
-        }
-        finalizeAgentStream(state, void 0, result);
-        states.delete(event);
-      },
-      error: (event) => {
-        const state = states.get(event);
-        if (!state || !event.error) {
-          return;
-        }
-        finalizeAgentStream(state, event.error);
-        states.delete(event);
-      }
-    };
-    channel.subscribe(handlers);
-    this.unsubscribers.push(() => {
-      unbindAutoInstrumentationSuppression?.();
-      channel.unsubscribe(handlers);
-    });
+  interceptAgentStream() {
+    this.unsubscribers.push(
+      strandsAgentSDKChannels.agentStream.intercept(
+        (target, thisArg, args, additional) => instrumentStrandsStreamInvocation({
+          finalize: finalizeAgentStream,
+          handleChunk: handleAgentStreamEvent,
+          invoke: () => Reflect.apply(target, thisArg, args),
+          name: "Strands Agent SDK",
+          start: () => startAgentStream(
+            args[0],
+            extractAgent(additional.agent, thisArg),
+            this.activeChildParents
+          )
+        })
+      )
+    );
   }
-  subscribeToMultiAgentStream(channel, operation) {
-    const tracingChannel = channel.tracingChannel();
-    const states = /* @__PURE__ */ new WeakMap();
-    const unbindAutoInstrumentationSuppression = bindAutoInstrumentationSuppressionToStart(tracingChannel);
-    const handlers = {
-      start: (event) => {
-        const state = startMultiAgentStream(
-          event,
-          operation,
-          this.activeChildParents
-        );
-        if (state) {
-          states.set(event, state);
-        }
-      },
-      end: (event) => {
-        const state = states.get(event);
-        if (!state) {
-          return;
-        }
-        const result = event.result;
-        if (isAsyncIterable(result)) {
-          patchStreamIfNeeded(result, {
-            aroundNext: (callback) => runWithAutoInstrumentationSuppressed(callback),
-            onChunk: (chunk) => handleMultiAgentStreamEvent(
-              state,
-              chunk,
-              this.activeChildParents
-            ),
-            onComplete: () => {
-              finalizeMultiAgentStream(state, this.activeChildParents);
-              states.delete(event);
-            },
-            onError: (error) => {
-              finalizeMultiAgentStream(state, this.activeChildParents, error);
-              states.delete(event);
-            }
-          });
-          return;
-        }
-        finalizeMultiAgentStream(
-          state,
-          this.activeChildParents,
-          void 0,
-          result
+  interceptMultiAgentStream(channel, operation) {
+    this.unsubscribers.push(
+      channel.intercept(
+        (target, thisArg, args, additional) => instrumentStrandsStreamInvocation({
+          finalize: (state, error, output) => finalizeMultiAgentStream(
+            state,
+            this.activeChildParents,
+            error,
+            output
+          ),
+          handleChunk: (state, chunk) => handleMultiAgentStreamEvent(state, chunk, this.activeChildParents),
+          invoke: () => Reflect.apply(target, thisArg, args),
+          name: "Strands multi-agent",
+          start: () => startMultiAgentStream(
+            args[0],
+            extractOrchestrator(additional.orchestrator, thisArg),
+            operation,
+            this.activeChildParents
+          )
+        })
+      )
+    );
+  }
+};
+function instrumentStrandsStreamInvocation(options) {
+  let state;
+  try {
+    state = options.start();
+  } catch (error) {
+    debugLogger.error(`Error starting ${options.name} instrumentation:`, error);
+  }
+  let result;
+  try {
+    result = runWithAutoInstrumentationSuppressed(options.invoke);
+  } catch (error) {
+    if (state) {
+      try {
+        options.finalize(state, error);
+      } catch (instrumentationError) {
+        debugLogger.error(
+          `Error handling ${options.name} instrumentation failure:`,
+          instrumentationError
         );
-        states.delete(event);
-      },
-      error: (event) => {
-        const state = states.get(event);
-        if (!state || !event.error) {
-          return;
-        }
-        finalizeMultiAgentStream(state, this.activeChildParents, event.error);
-        states.delete(event);
       }
-    };
-    tracingChannel.subscribe(handlers);
-    this.unsubscribers.push(() => {
-      unbindAutoInstrumentationSuppression?.();
-      tracingChannel.unsubscribe(handlers);
-    });
+    }
+    throw error;
   }
-};
-function startAgentStream(event, activeChildParents) {
-  const agent = extractAgent(event);
+  if (state) {
+    try {
+      if (isAsyncIterable(result)) {
+        patchStreamIfNeeded(result, {
+          aroundNext: (callback) => runWithAutoInstrumentationSuppressed(callback),
+          onChunk: (chunk) => options.handleChunk(state, chunk),
+          onComplete: () => options.finalize(state),
+          onError: (error) => options.finalize(state, error)
+        });
+      } else {
+        options.finalize(state, void 0, result);
+      }
+    } catch (error) {
+      debugLogger.error(
+        `Error finalizing ${options.name} instrumentation:`,
+        error
+      );
+    }
+  }
+  return result;
+}
+function startAgentStream(input, agent, activeChildParents) {
   const model = agent?.model;
   const metadata = {
     ...extractAgentMetadata2(agent),
     ...extractModelMetadata3(model),
     "strands.operation": "Agent.stream",
-    provider: extractProvider(model),
-    ...event.moduleVersion ? { "strands_agent_sdk.version": event.moduleVersion } : {}
+    provider: extractProvider(model)
   };
   const parentSpan = agent ? getOnlyChildParent(activeChildParents, agent) : void 0;
   const attachmentCache = createStrandsAttachmentCache();
-  const input = processStrandsInputAttachments(
-    event.arguments[0],
-    attachmentCache
-  );
+  const processedInput = processStrandsInputAttachments(input, attachmentCache);
   const span = parentSpan ? withCurrent(
     parentSpan,
     () => startSpan(
       withSpanInstrumentationName(
         {
           event: {
-            input,
+            input: processedInput,
             metadata
           },
           name: formatAgentSpanName(agent),
@@ -32627,7 +33247,7 @@ function startAgentStream(event, activeChildParents) {
     withSpanInstrumentationName(
       {
         event: {
-          input,
+          input: processedInput,
           metadata
         },
         name: formatAgentSpanName(agent),
@@ -32645,23 +33265,21 @@ function startAgentStream(event, activeChildParents) {
     startTime: getCurrentUnixTimestamp()
   };
 }
-function startMultiAgentStream(event, operation, activeChildParents) {
-  const orchestrator = extractOrchestrator(event);
+function startMultiAgentStream(input, orchestrator, operation, activeChildParents) {
   const metadata = {
     "strands.operation": operation,
     provider: "strands",
-    ...orchestrator?.id ? { "strands.orchestrator.id": orchestrator.id } : {},
-    ...event.moduleVersion ? { "strands_agent_sdk.version": event.moduleVersion } : {}
+    ...orchestrator?.id ? { "strands.orchestrator.id": orchestrator.id } : {}
   };
   const parentSpan = orchestrator ? getOnlyChildParent(activeChildParents, orchestrator) : void 0;
-  const input = processStrandsInputAttachments(event.arguments[0]);
+  const processedInput = processStrandsInputAttachments(input);
   const span = parentSpan ? withCurrent(
     parentSpan,
     () => startSpan(
       withSpanInstrumentationName(
         {
           event: {
-            input,
+            input: processedInput,
             metadata
           },
           name: operation === "Graph.stream" ? "Strands Graph" : "Strands Swarm",
@@ -32674,7 +33292,7 @@ function startMultiAgentStream(event, operation, activeChildParents) {
     withSpanInstrumentationName(
       {
         event: {
-          input,
+          input: processedInput,
           metadata
         },
         name: operation === "Graph.stream" ? "Strands Graph" : "Strands Swarm",
@@ -33045,12 +33663,12 @@ function finalizeMultiAgentStream(state, activeChildParents, error, output) {
   });
   state.span.end();
 }
-function extractAgent(event) {
-  const candidate = event.agent ?? event.self;
+function extractAgent(agent, self) {
+  const candidate = agent ?? self;
   return isObject(candidate) && typeof candidate.stream === "function" ? candidate : void 0;
 }
-function extractOrchestrator(event) {
-  const candidate = event.orchestrator ?? event.self;
+function extractOrchestrator(orchestrator, self) {
+  const candidate = orchestrator ?? self;
   return isObject(candidate) && typeof candidate.stream === "function" ? candidate : void 0;
 }
 function extractAgentMetadata2(agent) {
@@ -33437,6 +34055,296 @@ function logInstrumentationError5(context, error) {
   debugLogger.debug(`${context}:`, error);
 }
 
+// src/instrumentation/plugins/voyageai-plugin.ts
+var RERANK_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
+  "model",
+  "returnDocuments",
+  "topK",
+  "truncation"
+]);
+var VoyageAIPlugin = class extends BasePlugin {
+  onEnable() {
+    this.unsubscribers.push(
+      interceptVoyageAICall(
+        voyageAIChannels.embed,
+        "voyageai.embed",
+        extractTextEmbeddingInput,
+        summarizeEmbeddingOutput,
+        extractEmbeddingUsageMetrics
+      ),
+      interceptVoyageAICall(
+        voyageAIChannels.multimodalEmbed,
+        "voyageai.multimodalEmbed",
+        extractMultimodalEmbeddingInput,
+        summarizeEmbeddingOutput,
+        extractEmbeddingUsageMetrics
+      ),
+      interceptVoyageAICall(
+        voyageAIChannels.rerank,
+        "voyageai.rerank",
+        extractRerankInput,
+        summarizeRerankOutput
+      ),
+      interceptVoyageAICall(
+        voyageAIChannels.contextualizedEmbed,
+        "voyageai.contextualizedEmbed",
+        extractContextualizedEmbeddingInput,
+        summarizeContextualizedEmbeddingOutput,
+        extractEmbeddingUsageMetrics
+      )
+    );
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+};
+function interceptVoyageAICall(channel, name, extractInput2, extractOutput2, extractMetrics2 = extractUsageMetrics3) {
+  return channel.intercept((target, thisArg, args) => {
+    const invokeTarget = () => Reflect.apply(target, thisArg, args);
+    if (isAutoInstrumentationSuppressed()) {
+      return invokeTarget();
+    }
+    let span;
+    try {
+      const { input, metadata } = extractInput2(args);
+      span = startSpan(
+        withSpanInstrumentationName(
+          {
+            event: { input, metadata },
+            name,
+            spanAttributes: { type: "llm" /* LLM */ }
+          },
+          INSTRUMENTATION_NAMES.VOYAGEAI
+        )
+      );
+    } catch (error) {
+      debugLogger.error(`Error starting span for ${name}:`, error);
+      return invokeTarget();
+    }
+    let result;
+    try {
+      result = withCurrent(
+        span,
+        () => runWithAutoInstrumentationSuppressed(invokeTarget)
+      );
+    } catch (error) {
+      finishVoyageAISpan(span, name, () => span.log({ error }));
+      throw error;
+    }
+    void Promise.resolve(result).then(
+      (value) => finishVoyageAISpan(span, name, () => {
+        const metadata = extractResponseMetadata3(value);
+        span.log({
+          output: extractOutput2(value),
+          ...metadata ? { metadata } : {},
+          metrics: extractMetrics2(value)
+        });
+      }),
+      (error) => finishVoyageAISpan(span, name, () => span.log({ error }))
+    );
+    return result;
+  });
+}
+function finishVoyageAISpan(span, name, log2) {
+  try {
+    log2();
+  } catch (error) {
+    debugLogger.error(`Error logging span for ${name}:`, error);
+  }
+  try {
+    span.end();
+  } catch (error) {
+    debugLogger.error(`Error ending span for ${name}:`, error);
+  }
+}
+function getRequestArg2(args) {
+  if (Array.isArray(args)) {
+    return isObject(args[0]) ? args[0] : void 0;
+  }
+  if (!isObject(args)) {
+    return void 0;
+  }
+  const firstArg = Reflect.get(args, "0");
+  return isObject(firstArg) ? firstArg : void 0;
+}
+function pickMetadata(request, allowlist) {
+  const metadata = {};
+  if (request) {
+    for (const key of allowlist) {
+      if (!Object.hasOwn(request, key)) {
+        continue;
+      }
+      const value = request[key];
+      if (value !== void 0) {
+        metadata[key] = value;
+      }
+    }
+  }
+  return {
+    ...metadata,
+    provider: "voyage"
+  };
+}
+function buildEmbeddingInput(inputs, request) {
+  const outputDimensions = request?.outputDimension;
+  return {
+    inputs,
+    ...typeof outputDimensions === "number" && Number.isFinite(outputDimensions) ? { output_dimensions: outputDimensions } : {}
+  };
+}
+function embeddingMetadata(request) {
+  return {
+    ...typeof request?.model === "string" ? { model: request.model } : {},
+    provider: "voyage"
+  };
+}
+function extractTextEmbeddingInput(args) {
+  const request = getRequestArg2(args);
+  const rawInput = request?.input;
+  const values = Array.isArray(rawInput) ? rawInput : [rawInput];
+  return {
+    input: buildEmbeddingInput(
+      values.flatMap(
+        (value) => typeof value === "string" ? [{ content: value }] : []
+      ),
+      request
+    ),
+    metadata: embeddingMetadata(request)
+  };
+}
+function extractContextualizedEmbeddingInput(args) {
+  const request = getRequestArg2(args);
+  const rawInputs = request?.inputs;
+  const values = Array.isArray(rawInputs) ? rawInputs.flatMap((value) => Array.isArray(value) ? value : [value]) : [];
+  return {
+    input: buildEmbeddingInput(
+      values.flatMap(
+        (value) => typeof value === "string" ? [{ content: value }] : []
+      ),
+      request
+    ),
+    metadata: embeddingMetadata(request)
+  };
+}
+function extractMultimodalEmbeddingInput(args) {
+  const request = getRequestArg2(args);
+  const rawInputs = request?.inputs;
+  const inputs = Array.isArray(rawInputs) ? rawInputs.map((rawInput) => {
+    const rawContent = isObject(rawInput) ? rawInput.content : void 0;
+    return {
+      content: Array.isArray(rawContent) ? rawContent.flatMap(normalizeMultimodalContentPart) : []
+    };
+  }) : [];
+  const input = buildEmbeddingInput(inputs, request);
+  const processedInput = processInputAttachments(input);
+  return {
+    input: hasInlineEmbeddingMedia(processedInput) ? input : processedInput,
+    metadata: embeddingMetadata(request)
+  };
+}
+function normalizeMultimodalContentPart(part) {
+  if (!isObject(part) || typeof part.type !== "string") {
+    return [];
+  }
+  if (part.type === "text") {
+    return typeof part.text === "string" ? [{ type: "text", text: part.text }] : [];
+  }
+  const camelCaseField = {
+    image_base64: "imageBase64",
+    image_url: "imageUrl",
+    video_base64: "videoBase64",
+    video_url: "videoUrl"
+  };
+  const field = camelCaseField[part.type];
+  if (!field) {
+    return [];
+  }
+  const data = typeof part[field] === "string" ? part[field] : part[part.type];
+  if (typeof data !== "string") {
+    return [];
+  }
+  return part.type.startsWith("image_") ? [{ type: "image_url", image_url: { url: data } }] : [{ type: "file", file: { file_data: data } }];
+}
+function hasInlineEmbeddingMedia(input) {
+  return input.inputs.some(
+    ({ content }) => Array.isArray(content) ? content.some((part) => {
+      const value = part.type === "image_url" ? part.image_url.url : part.type === "file" ? part.file.file_data : void 0;
+      return typeof value === "string" && value.startsWith("data:");
+    }) : false
+  );
+}
+function extractRerankInput(args) {
+  const request = getRequestArg2(args);
+  const documents = request?.documents;
+  return {
+    input: {
+      documents,
+      query: request?.query
+    },
+    metadata: {
+      ...pickMetadata(request, RERANK_METADATA_ALLOWLIST),
+      ...Array.isArray(documents) ? { document_count: documents.length } : {}
+    }
+  };
+}
+function extractResponseMetadata3(result) {
+  if (!isObject(result)) {
+    return void 0;
+  }
+  const rawResponse = isObject(result.rawResponse) ? result.rawResponse : void 0;
+  const model = typeof result.model === "string" ? result.model : typeof rawResponse?.model === "string" ? rawResponse.model : void 0;
+  return model ? { model } : void 0;
+}
+function summarizeEmbeddingOutput(result) {
+  return {
+    count: isObject(result) && Array.isArray(result.data) ? result.data.length : 0
+  };
+}
+function summarizeRerankOutput(result) {
+  if (!isObject(result) || !Array.isArray(result.data)) {
+    return void 0;
+  }
+  return result.data.slice(0, 100).map((item) => ({
+    index: isObject(item) ? item.index : void 0,
+    relevance_score: isObject(item) ? (typeof item.relevanceScore === "number" ? item.relevanceScore : item.relevance_score) ?? null : null
+  }));
+}
+function summarizeContextualizedEmbeddingOutput(result) {
+  if (!isObject(result)) {
+    return { count: 0 };
+  }
+  if (Array.isArray(result.results)) {
+    return {
+      count: result.results.reduce(
+        (count, item) => count + (isObject(item) && Array.isArray(item.embeddings) ? item.embeddings.length : 0),
+        0
+      )
+    };
+  }
+  if (!Array.isArray(result.data)) {
+    return { count: 0 };
+  }
+  return {
+    count: result.data.reduce(
+      (count, item) => count + (isObject(item) && Array.isArray(item.data) ? item.data.length : 0),
+      0
+    )
+  };
+}
+function extractEmbeddingUsageMetrics(result) {
+  const metrics = extractUsageMetrics3(result);
+  return typeof metrics.tokens === "number" ? { prompt_tokens: metrics.tokens, tokens: metrics.tokens } : {};
+}
+function extractUsageMetrics3(result) {
+  if (!isObject(result)) {
+    return {};
+  }
+  const rawResponse = isObject(result.rawResponse) ? result.rawResponse : void 0;
+  const usage = isObject(result.usage) ? result.usage : isObject(rawResponse?.usage) ? rawResponse.usage : void 0;
+  const tokens = typeof result.totalTokens === "number" ? result.totalTokens : usage?.totalTokens ?? usage?.total_tokens;
+  return typeof tokens === "number" && Number.isFinite(tokens) && tokens >= 0 ? { tokens } : {};
+}
+
 // src/instrumentation/plugins/cloudflare-ai-chat-instrumentation.ts
 var wrappedTurnRunners = /* @__PURE__ */ new WeakSet();
 var wrappedResponseHooks = /* @__PURE__ */ new WeakSet();
@@ -33987,6 +34895,7 @@ var BraintrustPlugin = class extends BasePlugin {
   langSmithPlugin = null;
   piCodingAgentPlugin = null;
   strandsAgentSDKPlugin = null;
+  voyageAIPlugin = null;
   cloudflareAIChatPlugin = null;
   cloudflareAgentsPlugin = null;
   constructor(config = {}) {
@@ -34061,6 +34970,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.coherePlugin = new CoherePlugin();
       this.coherePlugin.enable();
     }
+    if (integrations.voyageai !== false) {
+      this.voyageAIPlugin = new VoyageAIPlugin();
+      this.voyageAIPlugin.enable();
+    }
     if (integrations.groq !== false) {
       this.groqPlugin = new GroqPlugin();
       this.groqPlugin.enable();
@@ -34177,6 +35090,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.coherePlugin.disable();
       this.coherePlugin = null;
     }
+    if (this.voyageAIPlugin) {
+      this.voyageAIPlugin.disable();
+      this.voyageAIPlugin = null;
+    }
     if (this.groqPlugin) {
       this.groqPlugin.disable();
       this.groqPlugin = null;
@@ -34543,7 +35460,7 @@ function wrapMastraAgent(agent, _options) {
 
 // src/node/config.ts
 import { AsyncLocalStorage } from "node:async_hooks";
-import * as path from "node:path";
+import * as path2 from "node:path";
 import * as fs from "node:fs/promises";
 import * as os from "node:os";
 import * as fsSync from "node:fs";
@@ -34552,34 +35469,93 @@ import { promisify } from "node:util";
 import * as zlib from "node:zlib";
 import * as dotenv from "dotenv";
 
-// src/gitutil.ts
-import { simpleGit } from "simple-git";
-var COMMON_BASE_BRANCHES = ["main", "master", "develop"];
-async function currentRepo() {
-  try {
-    const git = simpleGit();
-    if (await git.checkIsRepo()) {
-      return git;
-    } else {
-      return null;
+// src/git-command.ts
+import { execFile } from "node:child_process";
+import { constants } from "node:fs";
+import { access } from "node:fs/promises";
+import * as path from "node:path";
+var GIT_EXECUTABLE_NAMES = process.platform === "win32" ? ["git.exe", "git"] : ["git"];
+var GIT_MAX_BUFFER_BYTES = 10 * 1024 * 1024;
+var gitExecutablePromise;
+function executableSearchPath() {
+  return Object.entries(process.env).find(
+    ([name]) => name.toUpperCase() === "PATH"
+  )?.[1];
+}
+async function findGitExecutable(searchPath = executableSearchPath()) {
+  if (!searchPath) {
+    return void 0;
+  }
+  for (const rawSearchDir of searchPath.split(path.delimiter)) {
+    const searchDir = rawSearchDir.trim().replace(/^"(.*)"$/, "$1");
+    if (!path.isAbsolute(searchDir)) {
+      continue;
+    }
+    for (const executableName of GIT_EXECUTABLE_NAMES) {
+      const candidate = path.join(searchDir, executableName);
+      try {
+        await access(
+          candidate,
+          process.platform === "win32" ? constants.F_OK : constants.X_OK
+        );
+        return candidate;
+      } catch {
+      }
     }
-  } catch {
-    return null;
   }
+  return void 0;
+}
+async function resolveGitExecutable() {
+  gitExecutablePromise ??= findGitExecutable();
+  return await gitExecutablePromise;
 }
+async function runGitCommand(args, options = {}) {
+  const executable = await resolveGitExecutable();
+  if (!executable) {
+    throw new Error("Could not find a git executable on PATH");
+  }
+  return await new Promise((resolve, reject) => {
+    execFile(
+      executable,
+      args,
+      {
+        cwd: options.cwd,
+        encoding: "utf8",
+        maxBuffer: GIT_MAX_BUFFER_BYTES
+      },
+      (error, stdout) => {
+        if (error) {
+          reject(error);
+        } else {
+          resolve(stdout);
+        }
+      }
+    );
+  });
+}
+
+// src/gitutil.ts
+var COMMON_BASE_BRANCHES = ["main", "master", "develop"];
 var _baseBranch = null;
 async function getBaseBranch(remote = void 0) {
   if (_baseBranch === null) {
-    const git = await currentRepo();
-    if (git === null) {
+    const repoPath = await currentRepoPath();
+    if (!repoPath) {
       throw new Error("Not in a git repo");
     }
-    const remoteName = remote ?? (await git.getRemotes())[0]?.name;
+    const runGit = async (args) => await runGitCommand(args, { cwd: repoPath });
+    const remoteName = remote ?? (await runGit(["remote"])).trim().split(/\r?\n/)[0];
     if (!remoteName) {
       throw new Error("No remote found");
     }
     let branch = null;
-    const repoBranches = new Set((await git.branchLocal()).all);
+    const repoBranches = new Set(
+      (await runGit([
+        "for-each-ref",
+        "--format=%(refname:short)",
+        "refs/heads/"
+      ])).trim().split(/\r?\n/)
+    );
     const matchingBaseBranches = COMMON_BASE_BRANCHES.filter(
       (b) => repoBranches.has(b)
     );
@@ -34587,7 +35563,7 @@ async function getBaseBranch(remote = void 0) {
       branch = matchingBaseBranches[0];
     } else {
       try {
-        const remoteInfo = await git.remote(["show", remoteName]);
+        const remoteInfo = await runGit(["remote", "show", remoteName]);
         if (!remoteInfo) {
           throw new Error(`Could not find remote ${remoteName}`);
         }
@@ -34605,27 +35581,28 @@ async function getBaseBranch(remote = void 0) {
   return _baseBranch;
 }
 async function getBaseBranchAncestor(remote = void 0) {
-  const git = await currentRepo();
-  if (git === null) {
+  const repoPath = await currentRepoPath();
+  if (!repoPath) {
     throw new Error("Not in a git repo");
   }
   const { remote: remoteName, branch: baseBranch } = await getBaseBranch(remote);
-  const isDirty = (await git.diffSummary()).files.length > 0;
+  const isDirty = (await runGitCommand(["diff", "--name-only"], {
+    cwd: repoPath
+  })).trim().length > 0;
   const head = isDirty ? "HEAD" : "HEAD^";
   try {
-    const ancestor = await git.raw([
-      "merge-base",
-      head,
-      `${remoteName}/${baseBranch}`
-    ]);
+    const ancestor = await runGitCommand(
+      ["merge-base", head, `${remoteName}/${baseBranch}`],
+      { cwd: repoPath }
+    );
     return ancestor.trim();
   } catch {
     return void 0;
   }
 }
 async function getPastNAncestors(n = 1e3, remote = void 0) {
-  const git = await currentRepo();
-  if (git === null) {
+  const repoPath = await currentRepoPath();
+  if (!repoPath) {
     return [];
   }
   let ancestor = void 0;
@@ -34640,8 +35617,10 @@ async function getPastNAncestors(n = 1e3, remote = void 0) {
   if (!ancestor) {
     return [];
   }
-  const commits = await git.log({ from: ancestor, to: "HEAD", maxCount: n });
-  return commits.all.slice(0, n).map((c) => c.hash);
+  const commits = (await runGitCommand(["rev-list", `--max-count=${n}`, `${ancestor}..HEAD`], {
+    cwd: repoPath
+  })).trim();
+  return commits ? commits.split(/\r?\n/).slice(0, n) : [];
 }
 async function attempt(fn) {
   try {
@@ -34672,9 +35651,14 @@ async function getRepoInfo(settings) {
   });
   return sanitized;
 }
+async function currentRepoPath() {
+  return await attempt(
+    async () => (await runGitCommand(["rev-parse", "--show-toplevel"])).trim()
+  );
+}
 async function repoInfo() {
-  const git = await currentRepo();
-  if (git === null) {
+  const repoPath = await currentRepoPath();
+  if (!repoPath) {
     return void 0;
   }
   let commit = void 0;
@@ -34685,29 +35669,32 @@ async function repoInfo() {
   let tag = void 0;
   let branch = void 0;
   let git_diff = void 0;
-  const dirty = (await git.diffSummary()).files.length > 0;
-  commit = await attempt(async () => await git.revparse(["HEAD"]));
+  const runGit = async (args) => await runGitCommand(args, { cwd: repoPath });
+  const dirty = (await runGit(["diff", "--name-only"])).trim().length > 0;
+  commit = await attempt(
+    async () => (await runGit(["rev-parse", "HEAD"])).trim()
+  );
   commit_message = await attempt(
-    async () => (await git.raw(["log", "-1", "--pretty=%B"])).trim()
+    async () => (await runGit(["log", "-1", "--pretty=%B"])).trim()
   );
   commit_time = await attempt(
-    async () => (await git.raw(["log", "-1", "--pretty=%cI"])).trim()
+    async () => (await runGit(["log", "-1", "--pretty=%cI"])).trim()
   );
   author_name = await attempt(
-    async () => (await git.raw(["log", "-1", "--pretty=%aN"])).trim()
+    async () => (await runGit(["log", "-1", "--pretty=%aN"])).trim()
   );
   author_email = await attempt(
-    async () => (await git.raw(["log", "-1", "--pretty=%aE"])).trim()
+    async () => (await runGit(["log", "-1", "--pretty=%aE"])).trim()
   );
   tag = await attempt(
-    async () => (await git.raw(["describe", "--tags", "--exact-match", "--always"])).trim()
+    async () => (await runGit(["describe", "--tags", "--exact-match", "--always"])).trim()
   );
   branch = await attempt(
-    async () => (await git.raw(["rev-parse", "--abbrev-ref", "HEAD"])).trim()
+    async () => (await runGit(["rev-parse", "--abbrev-ref", "HEAD"])).trim()
   );
   if (dirty) {
     git_diff = await attempt(
-      async () => truncateToByteLimit(await git.raw(["--no-ext-diff", "diff", "HEAD"]))
+      async () => truncateToByteLimit(await runGit(["diff", "--no-ext-diff", "HEAD"]))
     );
   }
   return {
@@ -34788,9 +35775,9 @@ function configureNode() {
       return value;
     }
     const envPaths = [];
-    for (let dir = process.cwd(), depth = 0; depth <= BRAINTRUST_ENV_SEARCH_PARENT_LIMIT; dir = path.dirname(dir), depth++) {
-      envPaths.push(path.join(dir, ".env.braintrust"));
-      if (path.dirname(dir) === dir) {
+    for (let dir = process.cwd(), depth = 0; depth <= BRAINTRUST_ENV_SEARCH_PARENT_LIMIT; dir = path2.dirname(dir), depth++) {
+      envPaths.push(path2.join(dir, ".env.braintrust"));
+      if (path2.dirname(dir) === dir) {
         break;
       }
     }
@@ -34832,10 +35819,10 @@ function configureNode() {
   isomorph_default.processOn = (event, handler) => {
     process.on(event, handler);
   };
-  isomorph_default.basename = path.basename;
+  isomorph_default.basename = path2.basename;
   isomorph_default.writeln = (text) => process.stdout.write(text + "\n");
-  isomorph_default.pathJoin = path.join;
-  isomorph_default.pathDirname = path.dirname;
+  isomorph_default.pathJoin = path2.join;
+  isomorph_default.pathDirname = path2.dirname;
   isomorph_default.mkdir = fs.mkdir;
   isomorph_default.writeFile = fs.writeFile;
   isomorph_default.readFile = fs.readFile;
@@ -34870,8 +35857,8 @@ function configureNode() {
   registry.enable();
 }
 function getNearestBraintrustEnvValue(name) {
-  for (let dir = process.cwd(), depth = 0; depth <= BRAINTRUST_ENV_SEARCH_PARENT_LIMIT; dir = path.dirname(dir), depth++) {
-    const envPath = path.join(dir, ".env.braintrust");
+  for (let dir = process.cwd(), depth = 0; depth <= BRAINTRUST_ENV_SEARCH_PARENT_LIMIT; dir = path2.dirname(dir), depth++) {
+    const envPath = path2.join(dir, ".env.braintrust");
     try {
       const parsed = dotenv.parse(fsSync.readFileSync(envPath, "utf8"));
       const value = parsed[name];
@@ -34881,7 +35868,7 @@ function getNearestBraintrustEnvValue(name) {
         return void 0;
       }
     }
-    if (path.dirname(dir) === dir) {
+    if (path2.dirname(dir) === dir) {
       break;
     }
   }
@@ -34930,7 +35917,6 @@ async function summarizeAndFlush(experiment, options) {
 }
 
 export {
-  debugLogger,
   IDGenerator,
   UUIDGenerator,
   OTELIDGenerator,
@@ -34939,9 +35925,10 @@ export {
   TRACESTATE_HEADER,
   BAGGAGE_HEADER,
   BRAINTRUST_PARENT_KEY,
+  uint8ArrayToBase64,
+  base64ToUint8Array,
   spanObjectTypeV3ToTypedString,
   isObject,
-  mergeDicts,
   slugify,
   SpanComponentsV4,
   makeScorerPropagatedEvent,
@@ -34992,6 +35979,7 @@ export {
   ReadonlyAttachment,
   JSONAttachment,
   updateSpan,
+  _internalResumeSpan,
   spanComponentsToObjectId,
   ERR_PERMALINK,
   permalink,
@@ -35053,6 +36041,7 @@ export {
   X_CACHED_HEADER,
   parseCachedHeader,
   finalizeAnthropicTokens,
+  toNumericMetrics,
   extractAnthropicCacheTokens,
   collectAnthropicSession,
   zodToJsonSchema,