braintrust 3.28.0 → 3.30.0

package/dist/workerd.mjs

@@ -659,8 +659,6 @@ function newGlobalTracingChannel(nameOrChannels) {
 var DefaultAsyncLocalStorage = class {
   constructor() {
   }
-  enterWith(_) {
-  }
   run(_, callback) {
     return callback();
   }
@@ -2477,7 +2475,9 @@ var AclObjectType = z6.union([
     "project_log",
     "org_project",
     "org_audit_logs",
-    "project_group"
+    "project_group",
+    "ai_secret",
+    "org_ai_secret"
   ]),
   z6.null()
 ]);
@@ -3310,7 +3310,7 @@ var PromptParserNullish = z6.union([
   }),
   z6.null()
 ]);
-var PreprocessorSavedFunctionId = z6.union([
+var PreprocessorId = z6.union([
   z6.object({
     type: z6.literal("function"),
     id: z6.string(),
@@ -3321,6 +3321,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([
@@ -3328,7 +3329,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"]),
@@ -3530,7 +3531,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"]),
@@ -4474,6 +4475,7 @@ var View = z6.object({
   ]),
   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(),
@@ -5177,44 +5179,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;
@@ -5228,58 +5258,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
+      });
     }
   }
 };
@@ -5583,6 +5646,7 @@ var INSTRUMENTATION_NAMES = {
   CLOUDFLARE_THINK: "cloudflare-think",
   COHERE: "cohere",
   CURSOR_SDK: "cursor-sdk",
+  DEEPSEEK_HARNESS: "deepseek-harness",
   EVE: "eve",
   FLUE: "flue",
   GENKIT: "genkit",
@@ -5608,7 +5672,7 @@ var INSTRUMENTATION_NAMES = {
 var INTERNAL_SPAN_INSTRUMENTATION_NAME = /* @__PURE__ */ Symbol.for(
   "braintrust.spanInstrumentationName"
 );
-var SDK_VERSION = true ? "3.28.0" : "0.0.0";
+var SDK_VERSION = true ? "3.30.0" : "0.0.0";
 function withSpanInstrumentationName(args, instrumentationName) {
   return {
     ...args,
@@ -5727,6 +5791,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()
@@ -5940,12 +6014,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;
@@ -5982,12 +6097,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
@@ -5996,43 +6113,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();
@@ -6057,6 +6137,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;
@@ -6105,6 +6280,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
     );
@@ -6344,36 +6521,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;
   }
@@ -6413,9 +6639,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,
@@ -6423,8 +6649,14 @@ 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(path, params, config, retries = 0) {
     const { headers, ...rest } = config || {};
@@ -6434,8 +6666,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, path), {
+        let response;
+        try {
+          response = await this_fetch(_urljoin(this_base_url, path), {
             method: "POST",
             headers: {
               Accept: "application/json",
@@ -6446,8 +6679,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);
@@ -6472,7 +6711,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(
@@ -6495,7 +6734,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")]() {
@@ -8543,6 +8782,7 @@ function initDataset(projectOrOptions, optionalOptions) {
   const {
     project,
     dataset,
+    datasetId,
     description,
     version,
     snapshotName,
@@ -8558,6 +8798,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,
@@ -8578,6 +8823,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,
@@ -8731,6 +9005,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,
@@ -8755,43 +9047,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,
@@ -8807,6 +9103,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.`);
@@ -8830,9 +9127,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
       );
@@ -8864,46 +9161,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,
@@ -8919,6 +9220,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.`);
@@ -8942,9 +9244,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
       );
@@ -8985,6 +9287,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",
@@ -9016,16 +9364,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) {
@@ -9528,11 +9878,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;
@@ -9547,12 +9897,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;
@@ -9569,6 +9919,15 @@ function _internalStartSpanWithInitialMerge(args) {
     [INITIAL_SPAN_WRITE_AS_MERGE]: true
   }).span;
 }
+function _internalStartSpanWithInitialMergeAndParentSpanIds(args) {
+  return startSpanAndIsLogger(
+    {
+      ...args,
+      [INITIAL_SPAN_WRITE_AS_MERGE]: true
+    },
+    { useParentSpanIdsForObjectParent: true }
+  ).span;
+}
 function _internalStartSpanWithContext(args, context) {
   return startSpanAndIsLogger({
     ...args,
@@ -9582,7 +9941,7 @@ async function flush(options) {
 function setFetch(fetch2) {
   _internalGetGlobalState().setFetch(fetch2);
 }
-function startSpanAndIsLogger(args) {
+function startSpanAndIsLogger(args, internalOptions) {
   const state = args?.state ?? _globalState;
   const { parentObject, propagatedState } = getSpanParentObjectAndPropagatedState({
     asyncFlush: args?.asyncFlush,
@@ -9596,7 +9955,7 @@ function startSpanAndIsLogger(args) {
     ) ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
-    } : void 0;
+    } : internalOptions?.useParentSpanIdsForObjectParent ? args?.parentSpanIds : void 0;
     const { parent: _ignoredParent, ...spanArgs } = args ?? {};
     const span = new SpanImpl({
       state,
@@ -9678,27 +10037,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();
@@ -11604,6 +11964,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(
       {
@@ -11644,6 +12012,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;
@@ -12464,56 +12836,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
@@ -13103,6 +13433,131 @@ function unsubscribeAll(unsubscribers) {
   return [];
 }
 
+// src/instrumentation/core/channel-definitions.ts
+function channel(spec) {
+  return spec;
+}
+function defineChannels(pkg, channels, options) {
+  const { instrumentationName } = options;
+  return Object.fromEntries(
+    Object.entries(channels).map(([key, spec]) => {
+      const fullChannelName = `orchestrion:${pkg}:${spec.channelName}`;
+      if (spec.kind === "async") {
+        const asyncSpec = spec;
+        const tracingChannel2 = () => isomorph_default.newTracingChannel(
+          fullChannelName
+        );
+        const intercept2 = (interceptor) => {
+          const hook = tracingChannel2();
+          return typeof hook.intercept === "function" ? hook.intercept(interceptor) : () => {
+          };
+        };
+        return [
+          key,
+          {
+            ...asyncSpec,
+            instrumentationName,
+            intercept: intercept2,
+            invoke: (target, thisArg, args, additional) => {
+              const hook = tracingChannel2();
+              return typeof hook.invoke === "function" ? hook.invoke(target, thisArg, args, additional) : Reflect.apply(target, thisArg, args);
+            },
+            tracingChannel: tracingChannel2,
+            tracePromise: (fn, context) => tracingChannel2().tracePromise(
+              fn,
+              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+              context
+            )
+          }
+        ];
+      }
+      const syncSpec = spec;
+      const tracingChannel = () => isomorph_default.newTracingChannel(
+        fullChannelName
+      );
+      const intercept = (interceptor) => {
+        const hook = tracingChannel();
+        return typeof hook.intercept === "function" ? hook.intercept(interceptor) : () => {
+        };
+      };
+      return [
+        key,
+        {
+          ...syncSpec,
+          instrumentationName,
+          intercept,
+          invoke: (target, thisArg, args, additional) => {
+            const hook = tracingChannel();
+            return typeof hook.invoke === "function" ? hook.invoke(target, thisArg, args, additional) : Reflect.apply(target, thisArg, args);
+          },
+          tracingChannel,
+          traceSync: (fn, context) => tracingChannel().traceSync(
+            fn,
+            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+            context
+          )
+        }
+      ];
+    })
+  );
+}
+
+// src/instrumentation/plugins/openai-channels.ts
+var openAIChannels = defineChannels(
+  "openai",
+  {
+    filesCreateTraced: channel({
+      channelName: "files.create-traced",
+      kind: "async"
+    }),
+    batchesRetrieveTraced: channel({
+      channelName: "batches.retrieve-traced",
+      kind: "async"
+    }),
+    batchesCompleteTrace: channel({
+      channelName: "batches.complete-trace",
+      kind: "async"
+    }),
+    chatCompletionsCreate: channel({
+      channelName: "chat.completions.create",
+      kind: "async"
+    }),
+    embeddingsCreate: channel({
+      channelName: "embeddings.create",
+      kind: "async"
+    }),
+    betaChatCompletionsParse: channel({
+      channelName: "beta.chat.completions.parse",
+      kind: "async"
+    }),
+    betaChatCompletionsStream: channel({
+      channelName: "beta.chat.completions.stream",
+      kind: "sync-stream"
+    }),
+    moderationsCreate: channel({
+      channelName: "moderations.create",
+      kind: "async"
+    }),
+    responsesCreate: channel({
+      channelName: "responses.create",
+      kind: "async"
+    }),
+    responsesStream: channel({
+      channelName: "responses.stream",
+      kind: "sync-stream"
+    }),
+    responsesParse: channel({
+      channelName: "responses.parse",
+      kind: "async"
+    }),
+    responsesCompact: channel({
+      channelName: "responses.compact",
+      kind: "async"
+    })
+  },
+  { instrumentationName: INSTRUMENTATION_NAMES.OPENAI }
+);
+
 // src/wrappers/attachment-utils.ts
 function getExtensionFromMediaType(mediaType) {
   const extensionMap = {
@@ -13278,118 +13733,91 @@ function processInputAttachments(input) {
   return processNode(input);
 }
 
-// src/instrumentation/core/channel-definitions.ts
-function channel(spec) {
-  return spec;
-}
-function defineChannels(pkg, channels, options) {
-  const { instrumentationName } = options;
-  return Object.fromEntries(
-    Object.entries(channels).map(([key, spec]) => {
-      const fullChannelName = `orchestrion:${pkg}:${spec.channelName}`;
-      if (spec.kind === "async") {
-        const asyncSpec = spec;
-        const tracingChannel2 = () => isomorph_default.newTracingChannel(
-          fullChannelName
-        );
-        const intercept2 = (interceptor) => {
-          const hook = tracingChannel2();
-          return typeof hook.intercept === "function" ? hook.intercept(interceptor) : () => {
-          };
-        };
-        return [
-          key,
-          {
-            ...asyncSpec,
-            instrumentationName,
-            intercept: intercept2,
-            invoke: (target, thisArg, args, additional) => {
-              const hook = tracingChannel2();
-              return typeof hook.invoke === "function" ? hook.invoke(target, thisArg, args, additional) : Reflect.apply(target, thisArg, args);
-            },
-            tracingChannel: tracingChannel2,
-            tracePromise: (fn, context) => tracingChannel2().tracePromise(
-              fn,
-              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-              context
-            )
-          }
-        ];
+// src/instrumentation/plugins/openai-span-data.ts
+var OPENAI_METADATA_KEYS = [
+  "model",
+  "temperature",
+  "top_p",
+  "max_tokens",
+  "frequency_penalty",
+  "presence_penalty",
+  "stop",
+  "response_format",
+  "tools",
+  "tool_choice",
+  "parallel_tool_calls",
+  "max_tool_calls"
+];
+function batchMetadata(params) {
+  const metadata = { provider: "openai" };
+  for (const key of OPENAI_METADATA_KEYS) {
+    try {
+      const value = Reflect.get(params, key);
+      if (value !== void 0) {
+        metadata[key] = value;
       }
-      const syncSpec = spec;
-      const tracingChannel = () => isomorph_default.newTracingChannel(
-        fullChannelName
-      );
-      const intercept = (interceptor) => {
-        const hook = tracingChannel();
-        return typeof hook.intercept === "function" ? hook.intercept(interceptor) : () => {
-        };
-      };
-      return [
-        key,
-        {
-          ...syncSpec,
-          instrumentationName,
-          intercept,
-          invoke: (target, thisArg, args, additional) => {
-            const hook = tracingChannel();
-            return typeof hook.invoke === "function" ? hook.invoke(target, thisArg, args, additional) : Reflect.apply(target, thisArg, args);
-          },
-          tracingChannel,
-          traceSync: (fn, context) => tracingChannel().traceSync(
-            fn,
-            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-            context
-          )
-        }
-      ];
-    })
-  );
+    } catch {
+    }
+  }
+  return metadata;
+}
+function extractOpenAIBatchInput(endpoint, params) {
+  const input = endpoint === "/v1/chat/completions" ? params.messages : params.input;
+  return {
+    input: processInputAttachments(input),
+    metadata: batchMetadata(params)
+  };
+}
+function extractOpenAIChatInput(params) {
+  const { messages, ...metadata } = params;
+  return {
+    input: processInputAttachments(messages),
+    metadata: { ...metadata, provider: "openai" }
+  };
+}
+function extractOpenAIResponsesInput(params) {
+  const { input, ...metadata } = params;
+  return {
+    input: processInputAttachments(input),
+    metadata: { ...metadata, provider: "openai" }
+  };
+}
+function extractOpenAIResponsesMetadata(result) {
+  if (!result) {
+    return void 0;
+  }
+  const { output: _output, usage: _usage, ...metadata } = result;
+  return Object.keys(metadata).length > 0 ? metadata : void 0;
+}
+function processImagesInOutput(output) {
+  if (Array.isArray(output)) {
+    return output.map(processImagesInOutput);
+  }
+  if (isObject(output) && output.type === "image_generation_call" && typeof output.result === "string" && output.result) {
+    const fileExtension = output.output_format || "png";
+    const contentType = `image/${fileExtension}`;
+    const baseFilename = typeof output.revised_prompt === "string" ? output.revised_prompt.slice(0, 50).replace(/[^a-zA-Z0-9]/g, "_") : "generated_image";
+    let binaryString;
+    try {
+      binaryString = atob(output.result);
+    } catch {
+      return output;
+    }
+    const bytes = new Uint8Array(binaryString.length);
+    for (let i = 0; i < binaryString.length; i++) {
+      bytes[i] = binaryString.charCodeAt(i);
+    }
+    return {
+      ...output,
+      result: new Attachment({
+        data: new Blob([bytes], { type: contentType }),
+        filename: `${baseFilename}.${fileExtension}`,
+        contentType
+      })
+    };
+  }
+  return output;
 }
-
-// src/instrumentation/plugins/openai-channels.ts
-var openAIChannels = defineChannels(
-  "openai",
-  {
-    chatCompletionsCreate: channel({
-      channelName: "chat.completions.create",
-      kind: "async"
-    }),
-    embeddingsCreate: channel({
-      channelName: "embeddings.create",
-      kind: "async"
-    }),
-    betaChatCompletionsParse: channel({
-      channelName: "beta.chat.completions.parse",
-      kind: "async"
-    }),
-    betaChatCompletionsStream: channel({
-      channelName: "beta.chat.completions.stream",
-      kind: "sync-stream"
-    }),
-    moderationsCreate: channel({
-      channelName: "moderations.create",
-      kind: "async"
-    }),
-    responsesCreate: channel({
-      channelName: "responses.create",
-      kind: "async"
-    }),
-    responsesStream: channel({
-      channelName: "responses.stream",
-      kind: "sync-stream"
-    }),
-    responsesParse: channel({
-      channelName: "responses.parse",
-      kind: "async"
-    }),
-    responsesCompact: channel({
-      channelName: "responses.compact",
-      kind: "async"
-    })
-  },
-  { instrumentationName: INSTRUMENTATION_NAMES.OPENAI }
-);
 
 // src/openai-utils.ts
 var BRAINTRUST_CACHED_STREAM_METRIC = "__braintrust_cached_metric";
@@ -13446,23 +13874,573 @@ function getCachedMetricFromHeaders(headers) {
   return parseCachedHeader(headers.get(LEGACY_CACHED_HEADER));
 }
 
+// src/instrumentation/plugins/openai-batch-instrumentation.ts
+var SUPPORTED_ENDPOINTS = /* @__PURE__ */ new Set(["/v1/chat/completions", "/v1/responses"]);
+var TERMINAL_STATUSES = /* @__PURE__ */ new Set([
+  "completed",
+  "failed",
+  "expired",
+  "cancelled"
+]);
+var pendingBatchTraces = /* @__PURE__ */ new Map();
+function read(value, key) {
+  if (!isObject(value)) {
+    return void 0;
+  }
+  try {
+    return Reflect.get(value, key);
+  } catch {
+    return void 0;
+  }
+}
+function isBatchRecordIterable(value) {
+  if (value === null || typeof value !== "object" && typeof value !== "function") {
+    return false;
+  }
+  return typeof read(value, Symbol.iterator) === "function" || typeof read(value, Symbol.asyncIterator) === "function";
+}
+function validCustomId(value) {
+  return typeof value === "string" && value.length > 0 && value.length <= 64;
+}
+function logBatchInstrumentationError(context, error) {
+  debugLogger.debug(`OpenAI Batch instrumentation ${context}:`, error);
+}
+async function exportParent(parent) {
+  if ("toStr" in parent && typeof parent.toStr === "function") {
+    return parent.toStr();
+  }
+  if ("export" in parent && typeof parent.export === "function") {
+    return await parent.export();
+  }
+  return void 0;
+}
+async function deterministicDigest(namespace, ...parts) {
+  const encoded = new TextEncoder().encode(
+    [namespace, ...parts].map((part) => `${part.length}:${part}`).join("\0")
+  );
+  return new Uint8Array(
+    await globalThis.crypto.subtle.digest("SHA-256", encoded)
+  );
+}
+function digestHex(bytes, length) {
+  return Array.from(bytes.slice(0, length)).map((byte) => byte.toString(16).padStart(2, "0")).join("");
+}
+function digestUuid(bytes) {
+  const hex = digestHex(bytes, 16);
+  return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
+}
+async function batchSpanIds(inputFileId) {
+  const [row, span, root] = await Promise.all([
+    deterministicDigest("openai:batch:row", inputFileId),
+    deterministicDigest("openai:batch:span", inputFileId),
+    deterministicDigest("openai:batch:root", inputFileId)
+  ]);
+  return {
+    rowId: digestUuid(row),
+    spanId: digestHex(span, 8),
+    rootSpanId: digestHex(root, 16)
+  };
+}
+async function childSpanIds(inputFileId, customId) {
+  const [row, span] = await Promise.all([
+    deterministicDigest("openai:batch:child:row", inputFileId, customId),
+    deterministicDigest("openai:batch:child:span", inputFileId, customId)
+  ]);
+  return { rowId: digestUuid(row), spanId: digestHex(span, 8) };
+}
+async function startBatchSpan(context) {
+  const ids = await batchSpanIds(context.inputFileId);
+  const parent = SpanComponentsV4.fromStr(context.parent);
+  const hasParentSpan = Boolean(
+    parent.data.row_id && parent.data.span_id && parent.data.root_span_id
+  );
+  return withCurrent(
+    NOOP_SPAN,
+    () => _internalStartSpanWithInitialMergeAndParentSpanIds(
+      withSpanInstrumentationName(
+        {
+          name: "openai.batch",
+          type: "task" /* TASK */,
+          parent: context.parent,
+          ...!hasParentSpan ? {
+            parentSpanIds: {
+              parentSpanIds: [],
+              rootSpanId: ids.rootSpanId
+            }
+          } : {},
+          spanId: ids.spanId,
+          startTime: context.taskStartTime,
+          event: {
+            id: ids.rowId,
+            metadata: {
+              endpoint: context.endpoint,
+              input_file_id: context.inputFileId,
+              provider: "openai"
+            }
+          }
+        },
+        INSTRUMENTATION_NAMES.OPENAI
+      )
+    )
+  );
+}
+async function startBatchChild(context, taskParent, input) {
+  const ids = await childSpanIds(context.inputFileId, input.customId);
+  return withCurrent(
+    NOOP_SPAN,
+    () => _internalStartSpanWithInitialMerge(
+      withSpanInstrumentationName(
+        {
+          name: context.endpoint === "/v1/chat/completions" ? "Chat Completion" : "openai.responses.create",
+          type: "llm" /* LLM */,
+          parent: taskParent,
+          spanId: ids.spanId,
+          startTime: context.childStartTime,
+          event: {
+            id: ids.rowId,
+            ...input.spanData?.input !== void 0 ? { input: input.spanData.input } : {},
+            metadata: {
+              ...input.spanData?.metadata,
+              custom_id: input.customId,
+              provider: "openai"
+            }
+          }
+        },
+        INSTRUMENTATION_NAMES.OPENAI
+      )
+    )
+  );
+}
+async function* jsonlRecords(file, onIssue = () => {
+}) {
+  const resolvedFile = await file;
+  if (typeof resolvedFile === "string") {
+    for (const line of resolvedFile.split("\n")) {
+      if (!line.trim()) {
+        continue;
+      }
+      try {
+        yield JSON.parse(line.replace(/\r$/, ""));
+      } catch (error) {
+        logBatchInstrumentationError("skipped malformed JSONL", error);
+        onIssue(new Error("OpenAI Batch file contains malformed JSONL"));
+      }
+    }
+    return;
+  }
+  const body = read(resolvedFile, "body");
+  const getReader = read(body, "getReader");
+  if (isObject(body) && typeof getReader === "function") {
+    let reader;
+    try {
+      reader = Reflect.apply(getReader, body, []);
+      const decoder2 = new TextDecoder();
+      let pending = "";
+      while (true) {
+        const readChunk = read(reader, "read");
+        if (typeof readChunk !== "function") {
+          throw new Error("Response body stream has no read method");
+        }
+        const chunk = await Reflect.apply(readChunk, reader, []);
+        if (!isObject(chunk)) {
+          throw new Error("Response body stream returned an invalid chunk");
+        }
+        if (chunk.done === true) {
+          pending += decoder2.decode();
+          break;
+        }
+        if (!(chunk.value instanceof Uint8Array)) {
+          throw new Error("Response body stream returned a non-byte chunk");
+        }
+        pending += decoder2.decode(chunk.value, { stream: true });
+        let newline = pending.indexOf("\n");
+        while (newline !== -1) {
+          const line = pending.slice(0, newline).replace(/\r$/, "");
+          pending = pending.slice(newline + 1);
+          if (line.trim()) {
+            try {
+              yield JSON.parse(line);
+            } catch (error) {
+              logBatchInstrumentationError("skipped malformed JSONL", error);
+              onIssue(new Error("OpenAI Batch file contains malformed JSONL"));
+            }
+          }
+          newline = pending.indexOf("\n");
+        }
+      }
+      if (pending.trim()) {
+        try {
+          yield JSON.parse(pending.replace(/\r$/, ""));
+        } catch (error) {
+          logBatchInstrumentationError("skipped malformed JSONL", error);
+          onIssue(new Error("OpenAI Batch file contains malformed JSONL"));
+        }
+      }
+    } catch (error) {
+      logBatchInstrumentationError("could not read JSONL response body", error);
+      onIssue(new Error("OpenAI Batch response body could not be read"));
+    } finally {
+      const releaseLock = read(reader, "releaseLock");
+      if (typeof releaseLock === "function") {
+        try {
+          Reflect.apply(releaseLock, reader, []);
+        } catch (error) {
+          logBatchInstrumentationError(
+            "could not release stream reader",
+            error
+          );
+        }
+      }
+    }
+    return;
+  }
+  if (isBatchRecordIterable(resolvedFile)) {
+    for await (const record of resolvedFile) {
+      yield record;
+    }
+    return;
+  }
+  logBatchInstrumentationError("skipped invalid JSONL source", resolvedFile);
+  onIssue(new Error("OpenAI Batch file source is invalid"));
+}
+async function readBatchInputs(file) {
+  const inputs = /* @__PURE__ */ new Map();
+  const issues = [];
+  let endpoint;
+  try {
+    for await (const value of jsonlRecords(
+      file,
+      (issue) => issues.push(issue)
+    )) {
+      const customId = read(value, "custom_id");
+      const url = read(value, "url");
+      const body = read(value, "body");
+      if (!validCustomId(customId) || inputs.has(customId) || read(value, "method") !== "POST" || typeof url !== "string" || !SUPPORTED_ENDPOINTS.has(url) || endpoint !== void 0 && endpoint !== url || !isObject(body)) {
+        issues.push(new Error("OpenAI Batch input contains an invalid record"));
+        continue;
+      }
+      endpoint = url;
+      let spanData;
+      try {
+        spanData = extractOpenAIBatchInput(url, body);
+      } catch (error) {
+        logBatchInstrumentationError("could not extract batch input", error);
+      }
+      inputs.set(customId, { customId, spanData });
+    }
+  } catch (error) {
+    logBatchInstrumentationError("could not process input file", error);
+    issues.push(new Error("OpenAI Batch input file could not be processed"));
+  }
+  if (!endpoint || inputs.size === 0) {
+    issues.push(new Error("OpenAI Batch input contains no supported records"));
+  }
+  return { endpoint, inputs, issues };
+}
+async function writePendingSpans(trace, endTime) {
+  const task = await startBatchSpan(trace.context);
+  const taskParent = await task.export();
+  for (const input of trace.inputs.values()) {
+    try {
+      const child = await startBatchChild(trace.context, taskParent, input);
+      if (endTime !== void 0) {
+        child.end({ endTime });
+      }
+    } catch (error) {
+      logBatchInstrumentationError("could not write batch request span", error);
+    }
+  }
+  if (endTime !== void 0) {
+    if (trace.status && trace.status !== "completed") {
+      task.log({ error: new Error(`OpenAI Batch ${trace.status}`) });
+    }
+    task.end({ endTime });
+  }
+}
+var interceptOpenAIFilesCreateTraced = async (target, thisArg, args) => {
+  const startTime = getCurrentUnixTimestamp();
+  const inputPromise = readBatchInputs(args[0].inputFileContent);
+  const parentPromise = exportParent(args[0].parent);
+  const file = await Reflect.apply(target, thisArg, args);
+  try {
+    const inputFileId = read(file, "id");
+    const [inputData, exportedParent2] = await Promise.all([
+      inputPromise,
+      parentPromise
+    ]);
+    if (typeof inputFileId !== "string" || !exportedParent2 || !inputData.endpoint || inputData.issues.length > 0) {
+      if (inputData.issues[0]) {
+        logBatchInstrumentationError(
+          "skipped invalid input file",
+          inputData.issues[0]
+        );
+      }
+      return file;
+    }
+    const trace = {
+      context: {
+        inputFileId,
+        endpoint: inputData.endpoint,
+        parent: exportedParent2,
+        taskStartTime: startTime,
+        childStartTime: startTime
+      },
+      inputs: inputData.inputs
+    };
+    pendingBatchTraces.set(inputFileId, trace);
+    await writePendingSpans(trace);
+  } catch (error) {
+    logBatchInstrumentationError("could not start batch spans", error);
+  }
+  return file;
+};
+function validTimestamp(value) {
+  return typeof value === "number" && Number.isFinite(value) && value >= 0;
+}
+function terminalEndTime(batch, startTime) {
+  let timestamp;
+  switch (batch.status) {
+    case "completed":
+      timestamp = batch.completed_at;
+      break;
+    case "failed":
+      timestamp = batch.failed_at;
+      break;
+    case "expired":
+      timestamp = batch.expired_at;
+      break;
+    default:
+      timestamp = batch.cancelled_at;
+  }
+  return validTimestamp(timestamp) ? Math.max(timestamp, startTime) : Math.max(getCurrentUnixTimestamp(), startTime);
+}
+async function updateBatchTimestamps(batch) {
+  const trace = pendingBatchTraces.get(batch.input_file_id);
+  if (!trace || trace.context.endpoint !== batch.endpoint) {
+    return;
+  }
+  if (validTimestamp(batch.created_at)) {
+    trace.context.taskStartTime = batch.created_at;
+  }
+  trace.context.childStartTime = validTimestamp(batch.in_progress_at) ? Math.max(batch.in_progress_at, trace.context.taskStartTime) : trace.context.taskStartTime;
+  trace.status = batch.status;
+  if (TERMINAL_STATUSES.has(batch.status)) {
+    trace.endTime = terminalEndTime(batch, trace.context.childStartTime);
+  }
+  await writePendingSpans(trace, trace.endTime);
+}
+var interceptOpenAIBatchesRetrieveTraced = (target, thisArg, args) => Promise.resolve(Reflect.apply(target, thisArg, args)).then(async (batch) => {
+  try {
+    await updateBatchTimestamps(batch);
+  } catch (error) {
+    logBatchInstrumentationError("could not update batch timestamps", error);
+  }
+  return batch;
+});
+function errorFromResult(result) {
+  const error = read(result.value, "error");
+  if (isObject(error)) {
+    const message = read(error, "message");
+    return new Error(
+      typeof message === "string" ? message : "OpenAI Batch request failed"
+    );
+  }
+  const response = read(result.value, "response");
+  const statusCode = read(response, "status_code");
+  if (result.source === "error" || typeof statusCode === "number" && (statusCode < 200 || statusCode >= 300)) {
+    const message = read(read(read(response, "body"), "error"), "message");
+    return new Error(
+      typeof message === "string" ? message : "OpenAI Batch request failed"
+    );
+  }
+  return void 0;
+}
+async function completeBatchResult(context, taskParent, endTime, input, result) {
+  const child = await startBatchChild(context, taskParent, input);
+  try {
+    const resultError = errorFromResult(result);
+    const responseBody = read(read(result.value, "response"), "body");
+    if (resultError) {
+      child.log({ error: resultError });
+    } else if (isObject(responseBody)) {
+      const model = read(responseBody, "model");
+      child.log({
+        output: context.endpoint === "/v1/chat/completions" ? read(responseBody, "choices") : processImagesInOutput(read(responseBody, "output")),
+        ...typeof model === "string" ? { metadata: { model } } : {},
+        metrics: parseMetricsFromUsage(read(responseBody, "usage"))
+      });
+    } else {
+      child.log({ error: new Error("OpenAI Batch response body is missing") });
+    }
+  } catch (error) {
+    child.log({ error });
+  } finally {
+    child.end({ endTime });
+  }
+}
+async function completeResultFile({
+  context,
+  endTime,
+  file,
+  inputs,
+  issues,
+  seen,
+  source,
+  taskParent
+}) {
+  if (file === void 0) {
+    return;
+  }
+  for await (const value of jsonlRecords(file, (issue) => issues.push(issue))) {
+    const customId = read(value, "custom_id");
+    if (!validCustomId(customId) || !isObject(value)) {
+      issues.push(
+        new Error("OpenAI Batch result is missing a valid custom_id")
+      );
+      continue;
+    }
+    if (seen.has(customId)) {
+      issues.push(new Error("OpenAI Batch result contains a duplicate"));
+      continue;
+    }
+    const input = inputs.get(customId);
+    if (!input) {
+      issues.push(
+        new Error("OpenAI Batch result does not match an input record")
+      );
+      continue;
+    }
+    seen.add(customId);
+    await completeBatchResult(context, taskParent, endTime, input, {
+      value,
+      source
+    });
+  }
+}
+function sameInputs(first, second) {
+  return first.size === second.size && [...first.keys()].every((customId) => second.has(customId));
+}
+async function contextForCompletion(inputFileId, inputData) {
+  if (!inputData.endpoint || inputData.issues.length > 0) {
+    return void 0;
+  }
+  const existing = pendingBatchTraces.get(inputFileId);
+  if (existing) {
+    if (existing.context.endpoint !== inputData.endpoint || !sameInputs(existing.inputs, inputData.inputs)) {
+      return void 0;
+    }
+    return { ...existing, inputs: inputData.inputs };
+  }
+  const parent = await exportParent(getSpanParentObject());
+  if (!parent) {
+    return void 0;
+  }
+  const startTime = getCurrentUnixTimestamp();
+  return {
+    context: {
+      inputFileId,
+      endpoint: inputData.endpoint,
+      parent,
+      taskStartTime: startTime,
+      childStartTime: startTime
+    },
+    inputs: inputData.inputs
+  };
+}
+async function completeBatch(args) {
+  const inputData = await readBatchInputs(args.inputFileContent);
+  const trace = await contextForCompletion(args.inputFileId, inputData);
+  if (!trace) {
+    logBatchInstrumentationError(
+      "left batch spans pending",
+      inputData.issues[0] ?? new Error("OpenAI Batch input does not match")
+    );
+    return;
+  }
+  const endTime = trace.endTime ?? getCurrentUnixTimestamp();
+  const task = await startBatchSpan(trace.context);
+  const taskParent = await task.export();
+  const issues = [];
+  const seen = /* @__PURE__ */ new Set();
+  await Promise.all([
+    completeResultFile({
+      context: trace.context,
+      endTime,
+      file: args.outputFileContent,
+      inputs: trace.inputs,
+      issues,
+      seen,
+      source: "output",
+      taskParent
+    }),
+    completeResultFile({
+      context: trace.context,
+      endTime,
+      file: args.errorFileContent,
+      inputs: trace.inputs,
+      issues,
+      seen,
+      source: "error",
+      taskParent
+    })
+  ]);
+  if (issues.length > 0) {
+    logBatchInstrumentationError("left batch spans pending", issues[0]);
+    return;
+  }
+  const missing = [...trace.inputs.values()].filter(
+    ({ customId }) => !seen.has(customId)
+  );
+  if (!trace.status || trace.status === "completed") {
+    if (missing.length > 0) {
+      logBatchInstrumentationError(
+        "left batch spans pending",
+        new Error("OpenAI Batch result files are incomplete")
+      );
+      return;
+    }
+  } else {
+    for (const input of missing) {
+      const child = await startBatchChild(trace.context, taskParent, input);
+      child.log({ error: new Error(`OpenAI Batch ${trace.status}`) });
+      child.end({ endTime });
+    }
+    task.log({ error: new Error(`OpenAI Batch ${trace.status}`) });
+  }
+  task.end({ endTime });
+  pendingBatchTraces.delete(args.inputFileId);
+}
+var interceptOpenAIBatchTraceComplete = (target, thisArg, args) => Promise.resolve(Reflect.apply(target, thisArg, args)).then(async (result) => {
+  try {
+    await completeBatch(args[0]);
+  } catch (error) {
+    logBatchInstrumentationError("could not complete batch", error);
+  }
+  return result;
+});
+
 // src/instrumentation/plugins/openai-plugin.ts
 var OpenAIPlugin = class extends BasePlugin {
   constructor() {
     super();
   }
   onEnable() {
+    this.unsubscribers.push(
+      openAIChannels.filesCreateTraced.intercept(
+        interceptOpenAIFilesCreateTraced
+      ),
+      openAIChannels.batchesRetrieveTraced.intercept(
+        interceptOpenAIBatchesRetrieveTraced
+      ),
+      openAIChannels.batchesCompleteTrace.intercept(
+        interceptOpenAIBatchTraceComplete
+      )
+    );
     this.unsubscribers.push(
       traceStreamingChannel(openAIChannels.chatCompletionsCreate, {
         name: "Chat Completion",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { messages, ...metadata } = params;
-          return {
-            input: processInputAttachments(messages),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIChatInput(params),
         extractOutput: (result) => {
           return result?.choices;
         },
@@ -13508,13 +14486,7 @@ var OpenAIPlugin = class extends BasePlugin {
       traceStreamingChannel(openAIChannels.betaChatCompletionsParse, {
         name: "Chat Completion",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { messages, ...metadata } = params;
-          return {
-            input: processInputAttachments(messages),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIChatInput(params),
         extractOutput: (result) => {
           return result?.choices;
         },
@@ -13536,13 +14508,7 @@ var OpenAIPlugin = class extends BasePlugin {
       traceSyncStreamChannel(openAIChannels.betaChatCompletionsStream, {
         name: "Chat Completion",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { messages, ...metadata } = params;
-          return {
-            input: processInputAttachments(messages),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        }
+        extractInput: ([params]) => extractOpenAIChatInput(params)
       })
     );
     this.unsubscribers.push(
@@ -13572,23 +14538,11 @@ var OpenAIPlugin = class extends BasePlugin {
       traceStreamingChannel(openAIChannels.responsesCreate, {
         name: "openai.responses.create",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { input, ...metadata } = params;
-          return {
-            input: processInputAttachments(input),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIResponsesInput(params),
         extractOutput: (result) => {
           return processImagesInOutput(result?.output);
         },
-        extractMetadata: (result) => {
-          if (!result) {
-            return void 0;
-          }
-          const { output: _output, usage: _usage, ...metadata } = result;
-          return Object.keys(metadata).length > 0 ? metadata : void 0;
-        },
+        extractMetadata: (result) => extractOpenAIResponsesMetadata(result),
         extractMetrics: (result, startTime, endEvent) => {
           const metrics = withCachedMetric(
             parseMetricsFromUsage(result?.usage),
@@ -13607,13 +14561,7 @@ var OpenAIPlugin = class extends BasePlugin {
       traceSyncStreamChannel(openAIChannels.responsesStream, {
         name: "openai.responses.create",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { input, ...metadata } = params;
-          return {
-            input: processInputAttachments(input),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIResponsesInput(params),
         extractFromEvent: (event) => {
           if (event.type !== "response.completed" || !event.response) {
             return {};
@@ -13636,23 +14584,11 @@ var OpenAIPlugin = class extends BasePlugin {
       traceStreamingChannel(openAIChannels.responsesParse, {
         name: "openai.responses.parse",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { input, ...metadata } = params;
-          return {
-            input: processInputAttachments(input),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIResponsesInput(params),
         extractOutput: (result) => {
           return processImagesInOutput(result?.output);
         },
-        extractMetadata: (result) => {
-          if (!result) {
-            return void 0;
-          }
-          const { output: _output, usage: _usage, ...metadata } = result;
-          return Object.keys(metadata).length > 0 ? metadata : void 0;
-        },
+        extractMetadata: (result) => extractOpenAIResponsesMetadata(result),
         extractMetrics: (result, startTime, endEvent) => {
           const metrics = withCachedMetric(
             parseMetricsFromUsage(result?.usage),
@@ -13671,23 +14607,11 @@ var OpenAIPlugin = class extends BasePlugin {
       traceAsyncChannel(openAIChannels.responsesCompact, {
         name: "openai.responses.compact",
         type: "llm" /* LLM */,
-        extractInput: ([params]) => {
-          const { input, ...metadata } = params;
-          return {
-            input: processInputAttachments(input),
-            metadata: { ...metadata, provider: "openai" }
-          };
-        },
+        extractInput: ([params]) => extractOpenAIResponsesInput(params),
         extractOutput: (result) => {
           return processImagesInOutput(result?.output);
         },
-        extractMetadata: (result) => {
-          if (!result) {
-            return void 0;
-          }
-          const { output: _output, usage: _usage, ...metadata } = result;
-          return Object.keys(metadata).length > 0 ? metadata : void 0;
-        },
+        extractMetadata: (result) => extractOpenAIResponsesMetadata(result),
         extractMetrics: (result, startTime, endEvent) => {
           const metrics = withCachedMetric(
             parseMetricsFromUsage(result?.usage),
@@ -13743,35 +14667,6 @@ function withCachedMetric(metrics, result, endEvent) {
     cached
   };
 }
-function processImagesInOutput(output) {
-  if (Array.isArray(output)) {
-    return output.map(processImagesInOutput);
-  }
-  if (isObject(output)) {
-    if (output.type === "image_generation_call" && output.result && typeof output.result === "string") {
-      const fileExtension = output.output_format || "png";
-      const contentType = `image/${fileExtension}`;
-      const baseFilename = output.revised_prompt && typeof output.revised_prompt === "string" ? output.revised_prompt.slice(0, 50).replace(/[^a-zA-Z0-9]/g, "_") : "generated_image";
-      const filename = `${baseFilename}.${fileExtension}`;
-      const binaryString = atob(output.result);
-      const bytes = new Uint8Array(binaryString.length);
-      for (let i = 0; i < binaryString.length; i++) {
-        bytes[i] = binaryString.charCodeAt(i);
-      }
-      const blob = new Blob([bytes], { type: contentType });
-      const attachment = new Attachment({
-        data: blob,
-        filename,
-        contentType
-      });
-      return {
-        ...output,
-        result: attachment
-      };
-    }
-  }
-  return output;
-}
 function mergeLogprobTokens(existing, incoming) {
   if (incoming === void 0) {
     return existing;
@@ -13802,13 +14697,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 toolCalls2 = 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: toolCalls2.length > 0 ? toolCalls2 : 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) {
@@ -13817,62 +14732,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) {
@@ -15208,6 +16136,12 @@ function parseMetricsFromUsage2(usage) {
       }
     }
   }
+  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") {
@@ -16164,7 +17098,6 @@ function endHarnessTurn(parent) {
 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();
@@ -16207,9 +17140,6 @@ function braintrustAISDKTelemetry() {
       return;
     }
     operations.delete(operationKey);
-    if (workflowOperationKeyStore.getStore() === operationKey) {
-      workflowOperationKeyStore.enterWith(void 0);
-    }
     const keys = operationKeysByCallId.get(state.callId);
     if (!keys) {
       return;
@@ -16257,14 +17187,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);
@@ -16278,17 +17201,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) {
@@ -16298,8 +17217,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;
@@ -16502,9 +17421,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;
@@ -16976,6 +17892,10 @@ var aiSDKChannels = defineChannels(
       channelName: "generateText",
       kind: "async"
     }),
+    generateImage: channel({
+      channelName: "generateImage",
+      kind: "async"
+    }),
     streamText: channel({
       channelName: "streamText",
       kind: "async"
@@ -17163,7 +18083,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(
@@ -17191,6 +18111,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",
@@ -17680,26 +18612,29 @@ function subscribeToHarnessContinuation(continuationChannel, defaultDenyOutputPa
     channel2.unsubscribe(handlers);
   };
 }
-function subscribeToAISDKV7TelemetryDispatcher() {
-  const channel2 = 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;
     }
-  };
-  channel2.subscribe(handlers);
-  return () => {
-    channel2.unsubscribe(handlers);
-  };
+  );
 }
 function patchAISDKV7TelemetryDispatcher(dispatcher, telemetry, telemetryOptions) {
   if (!isObject(dispatcher)) {
@@ -18024,16 +18959,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
         });
@@ -18087,6 +19016,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 {
@@ -18225,6 +19173,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 {
@@ -19600,6 +20554,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;
@@ -20100,118 +21105,22 @@ var claudeAgentSDKChannels = defineChannels(
 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
-  );
+var localToolContextStore = isomorph_default.newAsyncLocalStorage();
+var localToolParentResolversByToolUseId = /* @__PURE__ */ new Map();
+function runWithClaudeLocalToolContext(callback, resolver) {
+  return localToolContextStore.run(resolver, callback);
 }
-function ensureClaudeLocalToolContext() {
-  const existing = localToolContextStore.getStore();
-  if (existing) {
-    return existing;
-  }
-  const created = {};
-  localToolContextStore.enterWith(created);
-  return created;
+function registerClaudeLocalToolParentResolver(toolUseId, resolver) {
+  localToolParentResolversByToolUseId.set(toolUseId, resolver);
 }
-function setClaudeLocalToolParentResolver(resolver) {
-  fallbackLocalToolParentResolver = resolver;
-  const context = ensureClaudeLocalToolContext();
-  if (!context) {
-    return;
+function getClaudeLocalToolParentResolver(toolUseId) {
+  const currentResolver = localToolContextStore.getStore();
+  if (!toolUseId) {
+    return currentResolver;
   }
-  context.resolveLocalToolParent = resolver;
-}
-function getClaudeLocalToolParentResolver() {
-  return localToolContextStore.getStore()?.resolveLocalToolParent ?? fallbackLocalToolParentResolver;
-}
-function isAsyncIterable3(value) {
-  return value !== null && typeof value === "object" && Symbol.asyncIterator in value && typeof value[Symbol.asyncIterator] === "function";
-}
-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;
-  }
-  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
@@ -20237,7 +21146,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;
@@ -20764,6 +21673,7 @@ function createToolTracingHooks(resolveParentSpan, taskIdToToolUseId, toolUseToP
       }
     }
     if (skipLocalToolHooks && (isLocalToolUse(input.tool_name, mcpServers) || localToolHookNames.has(input.tool_name))) {
+      registerClaudeLocalToolParentResolver(toolUseID, resolveParentSpan);
       return {};
     }
     const parsed = parseToolName(input.tool_name);
@@ -21474,7 +22384,7 @@ async function finalizeQuerySpan(state) {
 }
 var ClaudeAgentSDKPlugin = class extends BasePlugin {
   onEnable() {
-    this.subscribeToQuery();
+    this.interceptQuery();
   }
   onDisable() {
     for (const unsubscribe of this.unsubscribers) {
@@ -21482,211 +22392,218 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
     }
     this.unsubscribers = [];
   }
-  subscribeToQuery() {
-    const channel2 = 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,
-          taskIdToToolUseId,
-          toolUseToParent,
-          activeToolSpans,
-          localToolHookNames,
-          skipLocalToolHooks,
-          subAgentDetailsByToolUseId,
-          subAgentSpans,
-          endedSubAgentSpans
-        );
-        params.options = optionsWithHooks;
-        event.arguments[0] = params;
-        spans.set(event, {
-          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(),
-          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);
       }
     };
-    channel2.subscribe(handlers);
-    this.unsubscribers.push(() => {
-      channel2.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;
+        }
+      })
+    );
   }
 };
 
@@ -25800,7 +26717,7 @@ function patchOpenRouterCallModelResult(args) {
         span,
         () => originalMethod.apply(resultLike, args2)
       );
-      if (!isAsyncIterable4(stream)) {
+      if (!isAsyncIterable3(stream)) {
         return stream;
       }
       return wrapAsyncIterableWithSpan({
@@ -25995,7 +26912,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) {
@@ -26873,7 +27790,7 @@ function patchOpenRouterCallModelResult2(args) {
         span,
         () => originalMethod.apply(resultLike, args2)
       );
-      if (!isAsyncIterable5(stream)) {
+      if (!isAsyncIterable4(stream)) {
         return stream;
       }
       return wrapAsyncIterableWithSpan2({
@@ -27068,7 +27985,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) {
@@ -33060,12 +33977,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 || {};
@@ -33652,6 +34571,9 @@ var piCodingAgentChannels = defineChannels(
 // 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();
@@ -33732,6 +34654,7 @@ function startPiPromptRun(event, onFinalize) {
     return void 0;
   }
   installPiAgentInstrumentation(agent);
+  wrapPiToolExecutors(agent.state?.tools);
   const metadata = {
     ...extractSessionMetadata(session),
     ...extractPromptOptionsMetadata(event.arguments[1]),
@@ -33776,7 +34699,7 @@ function extractSession(event) {
   return isObject(candidate) && typeof candidate.prompt === "function" ? candidate : void 0;
 }
 function isPiAgent(value) {
-  return isObject(value) && typeof value.streamFn === "function" && typeof value.subscribe === "function";
+  return isObject(value) && (typeof value.streamFunction === "function" || typeof value.streamFn === "function") && typeof value.subscribe === "function";
 }
 function promptContextStore() {
   piPromptContextStore ??= isomorph_default.newAsyncLocalStorage();
@@ -33786,17 +34709,21 @@ function currentPiPromptState() {
   return promptContextStore().getStore();
 }
 function installPiAgentInstrumentation(agent) {
+  const property = typeof agent.streamFunction === "function" ? "streamFunction" : "streamFn";
+  const streamFunction = agent[property];
   const existing = piAgentPatchStates.get(agent);
-  if (!existing || agent.streamFn !== existing.wrappedStreamFn) {
+  if (streamFunction && (!existing || existing.property !== property || streamFunction !== existing.wrappedStreamFunction)) {
     const patchState = {
-      originalStreamFn: agent.streamFn,
-      wrappedStreamFn: agent.streamFn
+      originalStreamFunction: streamFunction,
+      property,
+      wrappedStreamFunction: streamFunction
     };
-    patchState.wrappedStreamFn = makeInstrumentedStreamFn(
+    patchState.wrappedStreamFunction = makeInstrumentedStreamFunction(
       agent,
-      patchState.originalStreamFn
+      patchState.originalStreamFunction,
+      property
     );
-    agent.streamFn = patchState.wrappedStreamFn;
+    agent[property] = patchState.wrappedStreamFunction;
     piAgentPatchStates.set(agent, patchState);
   }
   if (piAgentEventSubscriptions.has(agent)) {
@@ -33821,14 +34748,21 @@ function installPiAgentInstrumentation(agent) {
     logInstrumentationError4("Pi Coding Agent event subscription", error);
   }
 }
-function makeInstrumentedStreamFn(agent, originalStreamFn) {
-  return async function instrumentedPiStreamFn(model, context, options) {
-    const invokeOriginal = () => Reflect.apply(originalStreamFn, this, [model, context, options]);
+function makeInstrumentedStreamFunction(agent, originalStreamFunction, property) {
+  return async function instrumentedPiStreamFunction(model, context, options) {
+    const invokeOriginal = () => Reflect.apply(originalStreamFunction, this, [model, context, options]);
     const state = currentPiPromptState();
     if (!state || state.agent !== agent || state.finalized) {
       return invokeOriginal();
     }
-    const llmState = await startPiLlmSpan(state, model, context, options);
+    wrapPiToolExecutors(context.tools);
+    const llmState = await startPiLlmSpan(
+      state,
+      model,
+      context,
+      property,
+      options
+    );
     try {
       const stream = await runWithAutoInstrumentationSuppressed(invokeOriginal);
       return patchAssistantMessageStream(stream, state, llmState);
@@ -33838,12 +34772,39 @@ function makeInstrumentedStreamFn(agent, originalStreamFn) {
     }
   };
 }
-async function startPiLlmSpan(state, model, context, options) {
+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, property, options) {
   const metadata = {
     ...extractModelMetadata2(model),
     ...extractStreamOptionsMetadata(options),
     ...extractToolMetadata(context.tools),
-    "pi_coding_agent.operation": "agent.streamFn"
+    "pi_coding_agent.operation": `agent.${property}`
   };
   const span = startSpan(
     withSpanInstrumentationName(
@@ -34009,35 +34970,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);
@@ -34058,11 +35010,7 @@ function finishPiToolSpan(state, event) {
       output: event.result
     });
   } finally {
-    try {
-      toolState.span.end();
-    } finally {
-      toolState.restoreAutoInstrumentation?.();
-    }
+    toolState.span.end();
   }
 }
 function finishPiPromptRun(state, error) {
@@ -34112,10 +35060,10 @@ function finishPiLlmSpan(promptState, llmState, message, error) {
     ...cleanMetrics5(llmState.metrics),
     ...buildDurationMetrics3(llmState.startTime)
   };
-  const usageMetrics = extractUsageMetrics2(message?.usage);
-  if (Object.keys(usageMetrics).length > 0) {
+  const usageMetrics2 = extractUsageMetrics2(message?.usage);
+  if (Object.keys(usageMetrics2).length > 0) {
     promptState.collectedLlmUsageMetrics = true;
-    addMetrics(promptState.metrics, usageMetrics);
+    addMetrics(promptState.metrics, usageMetrics2);
   }
   try {
     safeLog4(llmState.span, {
@@ -34133,14 +35081,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();
 }
@@ -34459,12 +35403,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"
     );
@@ -34475,122 +35419,92 @@ var StrandsAgentSDKPlugin = class extends BasePlugin {
     }
     this.unsubscribers = [];
   }
-  subscribeToAgentStream() {
-    const channel2 = strandsAgentSDKChannels.agentStream.tracingChannel();
-    const states = /* @__PURE__ */ new WeakMap();
-    const unbindAutoInstrumentationSuppression = bindAutoInstrumentationSuppressionToStart(channel2);
-    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);
-      }
-    };
-    channel2.subscribe(handlers);
-    this.unsubscribers.push(() => {
-      unbindAutoInstrumentationSuppression?.();
-      channel2.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(channel2, operation) {
-    const tracingChannel = channel2.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(channel2, operation) {
+    this.unsubscribers.push(
+      channel2.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),
@@ -34600,17 +35514,14 @@ function startAgentStream(event, activeChildParents) {
   };
   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),
@@ -34623,7 +35534,7 @@ function startAgentStream(event, activeChildParents) {
     withSpanInstrumentationName(
       {
         event: {
-          input,
+          input: processedInput,
           metadata
         },
         name: formatAgentSpanName(agent),
@@ -34641,22 +35552,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 } : {}
   };
   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",
@@ -34669,7 +35579,7 @@ function startMultiAgentStream(event, operation, activeChildParents) {
     withSpanInstrumentationName(
       {
         event: {
-          input,
+          input: processedInput,
           metadata
         },
         name: operation === "Graph.stream" ? "Strands Graph" : "Strands Swarm",
@@ -35040,12 +35950,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) {
@@ -36910,6 +37820,7 @@ __export(exports_exports, {
   braintrustStreamChunkSchema: () => braintrustStreamChunkSchema,
   buildLocalSummary: () => buildLocalSummary,
   collectAnthropicSession: () => collectAnthropicSession,
+  completeOpenAIBatchTrace: () => completeOpenAIBatchTrace,
   configureInstrumentation: () => configureInstrumentation,
   constructLogs3OverflowRequest: () => constructLogs3OverflowRequest,
   createFinalValuePassThroughStream: () => createFinalValuePassThroughStream,
@@ -36948,6 +37859,8 @@ __export(exports_exports, {
   loginToState: () => loginToState,
   logs3OverflowUploadSchema: () => logs3OverflowUploadSchema,
   newId: () => newId,
+  openaiBatchesRetrieveTraced: () => openaiBatchesRetrieveTraced,
+  openaiFilesCreateTraced: () => openaiFilesCreateTraced,
   parseCachedHeader: () => parseCachedHeader,
   parseTemplateFormat: () => parseTemplateFormat,
   permalink: () => permalink,
@@ -37112,6 +38025,260 @@ async function registerSandbox(options) {
   };
 }
 
+// src/wrappers/openai-promise-utils.ts
+function splitSpanInfo(allParams) {
+  const { span_info, ...params } = allParams;
+  return {
+    params,
+    span_info
+  };
+}
+function createChannelContext(_channel, params, span_info) {
+  return {
+    arguments: (
+      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+      [params]
+    ),
+    span_info
+  };
+}
+async function tracePromiseWithResponse(channel2, traceContext, apiPromise) {
+  let enhancedResponse;
+  const tracePromise = (
+    // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+    channel2.tracePromise
+  );
+  const data = await tracePromise(async () => {
+    enhancedResponse = await apiPromise.withResponse();
+    traceContext.response = enhancedResponse.response;
+    return enhancedResponse.data;
+  }, traceContext);
+  if (!enhancedResponse) {
+    throw new Error("Expected withResponse() to provide response");
+  }
+  return {
+    data,
+    response: enhancedResponse.response,
+    request_id: enhancedResponse.request_id
+  };
+}
+async function tracePromiseAsResponse(channel2, traceContext, apiPromise) {
+  const tracePromise = (
+    // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+    channel2.tracePromise
+  );
+  let response;
+  await tracePromise(async () => {
+    response = await apiPromise.asResponse();
+    traceContext.response = response;
+    return void 0;
+  }, traceContext);
+  if (!response) {
+    throw new Error("Expected asResponse() to provide response");
+  }
+  return response;
+}
+function createLazyAPIPromise(ensureExecuted, ensureResponse, getAPIPromise) {
+  let firstConsumption;
+  let enhancedResponsePromise;
+  let dataPromise;
+  let responsePromise;
+  const withResponse = () => {
+    firstConsumption ??= "data";
+    enhancedResponsePromise ??= firstConsumption === "data" ? ensureExecuted() : getAPIPromise().withResponse();
+    return enhancedResponsePromise;
+  };
+  const asResponse = () => {
+    firstConsumption ??= "response";
+    responsePromise ??= firstConsumption === "response" ? ensureResponse() : getAPIPromise().asResponse();
+    return responsePromise;
+  };
+  return new Proxy({}, {
+    get(target, prop, receiver) {
+      if (prop === "withResponse") {
+        return withResponse;
+      }
+      if (prop === "asResponse") {
+        return asResponse;
+      }
+      if (prop === "then" || prop === "catch" || prop === "finally" || prop in Promise.prototype) {
+        dataPromise ??= withResponse().then((result) => result.data);
+        const value = Reflect.get(dataPromise, prop, receiver);
+        return typeof value === "function" ? value.bind(dataPromise) : value;
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+}
+
+// src/openai-batch.ts
+function read2(value, key) {
+  if (value === null || typeof value !== "object" && typeof value !== "function") {
+    return void 0;
+  }
+  try {
+    return Reflect.get(value, key);
+  } catch {
+    return void 0;
+  }
+}
+function isAsyncIterable5(value) {
+  return typeof read2(value, Symbol.asyncIterator) === "function";
+}
+function teeAsyncIterable(source) {
+  const iterator = source[Symbol.asyncIterator]();
+  const stream = new ReadableStream({
+    async pull(controller) {
+      try {
+        const result = await iterator.next();
+        if (result.done) {
+          controller.close();
+          return;
+        }
+        if (!(result.value instanceof Uint8Array)) {
+          throw new TypeError(
+            "OpenAI upload streams must yield Uint8Array chunks"
+          );
+        }
+        controller.enqueue(result.value);
+      } catch (error) {
+        controller.error(error);
+      }
+    },
+    async cancel(reason) {
+      await iterator.return?.(reason);
+    }
+  });
+  const [uploadStream, trace] = stream.tee();
+  const upload = {
+    async *[Symbol.asyncIterator]() {
+      const reader = uploadStream.getReader();
+      try {
+        while (true) {
+          const result = await reader.read();
+          if (result.done) {
+            return;
+          }
+          yield result.value;
+        }
+      } finally {
+        reader.releaseLock();
+      }
+    }
+  };
+  const path = read2(source, "path");
+  if (typeof path === "string") {
+    Object.defineProperty(upload, "path", { value: path });
+  }
+  return { upload, trace: new Response(trace) };
+}
+function teeUpload(upload) {
+  if (upload instanceof Blob) {
+    return { upload, trace: new Response(upload) };
+  }
+  if (upload instanceof Response) {
+    return { upload, trace: upload.clone() };
+  }
+  if (isAsyncIterable5(upload)) {
+    return teeAsyncIterable(upload);
+  }
+  return void 0;
+}
+function openaiFilesCreateTraced(files) {
+  return (params, options) => {
+    const parent = getSpanParentObject();
+    const purpose = read2(params, "purpose");
+    const file = read2(params, "file");
+    if (purpose !== "batch") {
+      return files.create(params, options);
+    }
+    let branches;
+    try {
+      branches = teeUpload(file);
+    } catch {
+      return files.create(params, options);
+    }
+    if (!branches) {
+      return files.create(params, options);
+    }
+    const tracedParams = {
+      ...Object(params),
+      file: branches.upload
+    };
+    const apiPromise = files.create(tracedParams, options);
+    let enhancedResponse;
+    const tracedResponse = openAIChannels.filesCreateTraced.invoke(
+      async () => {
+        enhancedResponse = await apiPromise.withResponse();
+        return enhancedResponse.data;
+      },
+      files,
+      [{ params, inputFileContent: branches.trace, parent }],
+      {}
+    ).then((data) => {
+      if (!enhancedResponse) {
+        throw new Error(
+          "Expected OpenAI withResponse() to provide a response"
+        );
+      }
+      return { ...enhancedResponse, data };
+    });
+    return createLazyAPIPromise(
+      () => tracedResponse,
+      () => tracedResponse.then(({ response }) => response),
+      () => apiPromise
+    );
+  };
+}
+function openaiBatchesRetrieveTraced(batches) {
+  return (batchId, options) => {
+    const apiPromise = batches.retrieve(batchId, options);
+    let enhancedResponse;
+    const tracedResponse = openAIChannels.batchesRetrieveTraced.invoke(
+      async () => {
+        enhancedResponse = await apiPromise.withResponse();
+        return enhancedResponse.data;
+      },
+      batches,
+      [{ batchId }],
+      {}
+    ).then((data) => {
+      if (!enhancedResponse) {
+        throw new Error(
+          "Expected OpenAI withResponse() to provide a response"
+        );
+      }
+      return { ...enhancedResponse, data };
+    });
+    return createLazyAPIPromise(
+      () => tracedResponse,
+      () => tracedResponse.then(({ response }) => response),
+      () => apiPromise
+    );
+  };
+}
+async function completeOpenAIBatchTrace(args) {
+  const inputFileContent = Promise.resolve(args.inputFileContent);
+  const outputFileContent = args.outputFileContent === void 0 ? void 0 : Promise.resolve(args.outputFileContent);
+  const errorFileContent = args.errorFileContent === void 0 ? void 0 : Promise.resolve(args.errorFileContent);
+  void inputFileContent.catch(() => void 0);
+  void outputFileContent?.catch(() => void 0);
+  void errorFileContent?.catch(() => void 0);
+  await openAIChannels.batchesCompleteTrace.invoke(
+    async () => void 0,
+    void 0,
+    [
+      {
+        ...args,
+        inputFileContent,
+        outputFileContent,
+        errorFileContent
+      }
+    ],
+    {}
+  );
+}
+
 // src/functions/invoke.ts
 async function invoke(args) {
   const {
@@ -37214,58 +38381,6 @@ function initFunction({
   return f;
 }
 
-// src/wrappers/openai-promise-utils.ts
-function splitSpanInfo(allParams) {
-  const { span_info, ...params } = allParams;
-  return {
-    params,
-    span_info
-  };
-}
-function createChannelContext(_channel, params, span_info) {
-  return {
-    arguments: (
-      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-      [params]
-    ),
-    span_info
-  };
-}
-async function tracePromiseWithResponse(channel2, traceContext, apiPromise) {
-  let enhancedResponse;
-  const tracePromise = (
-    // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
-    channel2.tracePromise
-  );
-  const data = await tracePromise(async () => {
-    enhancedResponse = await apiPromise.withResponse();
-    traceContext.response = enhancedResponse.response;
-    return enhancedResponse.data;
-  }, traceContext);
-  if (!enhancedResponse) {
-    throw new Error("Expected withResponse() to provide response");
-  }
-  return { data, response: enhancedResponse.response };
-}
-function createLazyAPIPromise(ensureExecuted) {
-  let dataPromise = null;
-  return new Proxy({}, {
-    get(target, prop, receiver) {
-      if (prop === "withResponse") {
-        return () => ensureExecuted();
-      }
-      if (prop === "then" || prop === "catch" || prop === "finally" || prop in Promise.prototype) {
-        if (!dataPromise) {
-          dataPromise = ensureExecuted().then((result) => result.data);
-        }
-        const value = Reflect.get(dataPromise, prop, receiver);
-        return typeof value === "function" ? value.bind(dataPromise) : value;
-      }
-      return Reflect.get(target, prop, receiver);
-    }
-  });
-}
-
 // src/wrappers/oai_responses.ts
 function responsesProxy(openai) {
   if (!openai.responses) {
@@ -37302,17 +38417,33 @@ function wrapResponsesAsync(target, channel2) {
   return (allParams, options) => {
     const { span_info, params } = splitSpanInfo(allParams);
     let executionPromise = null;
+    let apiPromise = null;
+    const getAPIPromise = () => {
+      apiPromise ??= target(params, options);
+      return apiPromise;
+    };
     const ensureExecuted = () => {
       if (!executionPromise) {
         executionPromise = (async () => {
           const traceContext = createChannelContext(channel2, params, span_info);
-          const apiPromise = target(params, options);
-          return tracePromiseWithResponse(channel2, traceContext, apiPromise);
+          return tracePromiseWithResponse(
+            channel2,
+            traceContext,
+            getAPIPromise()
+          );
         })();
       }
       return executionPromise;
     };
-    return createLazyAPIPromise(ensureExecuted);
+    return createLazyAPIPromise(
+      ensureExecuted,
+      () => tracePromiseAsResponse(
+        channel2,
+        createChannelContext(channel2, params, span_info),
+        getAPIPromise()
+      ),
+      getAPIPromise
+    );
   };
 }
 function wrapResponsesSyncStream(target, channel2) {
@@ -37458,6 +38589,11 @@ function wrapChatCompletion(completion) {
       allParams
     );
     let executionPromise = null;
+    let apiPromise = null;
+    const getAPIPromise = () => {
+      apiPromise ??= completion(params, options);
+      return apiPromise;
+    };
     const ensureExecuted = () => {
       if (!executionPromise) {
         executionPromise = (async () => {
@@ -37467,32 +38603,44 @@ function wrapChatCompletion(completion) {
             span_info
           );
           if (params.stream) {
-            const completionPromise = completion(
-              params,
-              options
+            const completionPromise = (
+              // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+              getAPIPromise()
             );
-            const { data: data2, response: response2 } = await tracePromiseWithResponse(
+            const { data: data2, response: response2, request_id: request_id2 } = await tracePromiseWithResponse(
               openAIChannels.chatCompletionsCreate,
               traceContext,
               completionPromise
             );
-            return { data: data2, response: response2 };
+            return { data: data2, response: response2, request_id: request_id2 };
           }
-          const completionResponse = completion(
-            params,
-            options
+          const completionResponse = (
+            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+            getAPIPromise()
           );
-          const { data, response } = await tracePromiseWithResponse(
+          const { data, response, request_id } = await tracePromiseWithResponse(
             openAIChannels.chatCompletionsCreate,
             traceContext,
             completionResponse
           );
-          return { data, response };
+          return { data, response, request_id };
         })();
       }
       return executionPromise;
     };
-    return createLazyAPIPromise(ensureExecuted);
+    return createLazyAPIPromise(
+      ensureExecuted,
+      () => tracePromiseAsResponse(
+        openAIChannels.chatCompletionsCreate,
+        createChannelContext(
+          openAIChannels.chatCompletionsCreate,
+          params,
+          span_info
+        ),
+        getAPIPromise()
+      ),
+      getAPIPromise
+    );
   };
 }
 function createEndpointProxy(target, wrapperFn) {
@@ -37510,6 +38658,11 @@ function wrapApiCreateWithChannel(create, channel2) {
   return (allParams, options) => {
     const { span_info, params } = splitSpanInfo(allParams);
     let executionPromise = null;
+    let apiPromise = null;
+    const getAPIPromise = () => {
+      apiPromise ??= create(params, options);
+      return apiPromise;
+    };
     const ensureExecuted = () => {
       if (!executionPromise) {
         executionPromise = (async () => {
@@ -37517,13 +38670,21 @@ function wrapApiCreateWithChannel(create, channel2) {
           return tracePromiseWithResponse(
             channel2,
             traceContext,
-            create(params, options)
+            getAPIPromise()
           );
         })();
       }
       return executionPromise;
     };
-    return createLazyAPIPromise(ensureExecuted);
+    return createLazyAPIPromise(
+      ensureExecuted,
+      () => tracePromiseAsResponse(
+        channel2,
+        createChannelContext(channel2, params, span_info),
+        getAPIPromise()
+      ),
+      getAPIPromise
+    );
   };
 }
 var wrapEmbeddings = (create) => wrapApiCreateWithChannel(create, openAIChannels.embeddingsCreate);
@@ -37559,6 +38720,13 @@ function wrapAISDK(aiSDK, options = {}) {
       switch (prop) {
         case "generateText":
           return wrapGenerateText(typedAISDK.generateText, options, typedAISDK);
+        case "generateImage":
+        case "experimental_generateImage":
+          return typeof original === "function" ? wrapGenerateImage(
+            original,
+            options,
+            typedAISDK
+          ) : original;
         case "streamText":
           return wrapStreamText(typedAISDK.streamText, options, typedAISDK);
         case "generateObject":
@@ -37795,6 +38963,32 @@ var wrapGenerateObject = (generateObject, options = {}, aiSDK) => {
     options
   );
 };
+var wrapGenerateImage = (generateImage, options = {}, aiSDK) => {
+  return makeGenerateImageWrapper(generateImage, { aiSDK }, options);
+};
+var makeGenerateImageWrapper = (generateImage, contextOptions = {}, options = {}) => {
+  const wrapper = async function(allParams) {
+    const { span_info, ...params } = allParams;
+    const tracedParams = { ...params };
+    return aiSDKChannels.generateImage.tracePromise(
+      () => generateImage(tracedParams),
+      createAISDKChannelContext(tracedParams, {
+        aiSDK: contextOptions.aiSDK,
+        denyOutputPaths: options.denyOutputPaths,
+        self: contextOptions.self,
+        span_info: mergeSpanInfo(span_info, {
+          name: "generateImage",
+          spanType: contextOptions.spanType
+        })
+      })
+    );
+  };
+  Object.defineProperty(wrapper, "name", {
+    value: "generateImage",
+    writable: false
+  });
+  return wrapper;
+};
 var makeEmbedWrapper = (channel2, name, embed, contextOptions = {}, options = {}) => {
   const wrapper = async function(allParams) {
     const { span_info, ...params } = allParams;
@@ -38545,7 +39739,7 @@ function braintrustEveHook(options) {
     }
   };
 }
-function braintrustEveInstrumentation(options) {
+function createLegacyEveInstrumentation(options) {
   const state = options.defineState(EVE_TRACE_STATE_KEY, emptyEveTraceState);
   return {
     events: {
@@ -39933,8 +41127,10 @@ function capturedModelInput(modelInput) {
   const value = [];
   if (typeof instructions === "string") {
     value.push({ content: instructions, role: "system" });
-  } else if (instructions) {
+  } else if (Array.isArray(instructions)) {
     value.push(...instructions.map(capturedEveModelMessage));
+  } else if (instructions) {
+    value.push(capturedEveModelMessage(instructions));
   }
   value.push(...messages.map(capturedEveModelMessage));
   try {
@@ -40178,6 +41374,472 @@ async function deterministicEveId(...parts) {
   return `${hex.slice(0, 8)}-${hex.slice(8, 12)}-${hex.slice(12, 16)}-${hex.slice(16, 20)}-${hex.slice(20)}`;
 }
 
+// src/instrumentation/plugins/eve-provider.ts
+var MAX_EVE_PROVIDER_CACHE_ENTRIES = 1e4;
+function createEveInstrumentationProvider(options = {}) {
+  const bridge = new EveProviderBridge(options.metadata);
+  return {
+    // Eve versions before tracePolicy use capture to decide whether provider
+    // events include content. Newer versions prefer tracePolicy when both are
+    // present, so keep the deprecated field for backwards compatibility.
+    capture: "content",
+    tracePolicy: () => ({
+      emit: true,
+      recordInputs: true,
+      recordOutputs: true
+    }),
+    events: {
+      "action.completed": (event, context) => bridge.handleActionTerminal(event, context),
+      "action.failed": (event, context) => bridge.handleActionTerminal(event, context),
+      "action.started": (event, context) => bridge.handleActionStarted(event, context),
+      "model.call.completed": (event, context) => bridge.handleModelTerminal(event, context),
+      "model.call.failed": (event, context) => bridge.handleModelTerminal(event, context),
+      "model.call.started": (event, context) => bridge.handleModelStarted(event, context),
+      "step.attempt.completed": (event) => bridge.handleStepAttemptTerminal(event),
+      "step.attempt.failed": (event) => bridge.handleStepAttemptTerminal(event),
+      "turn.cancelled": (event, context) => bridge.handleTurnTerminal(event, context),
+      "turn.completed": (event, context) => bridge.handleTurnTerminal(event, context),
+      "turn.failed": (event, context) => bridge.handleTurnTerminal(event, context),
+      "turn.started": (event, context) => bridge.handleTurnStarted(event, context)
+    },
+    flush,
+    setup: options.setup
+  };
+}
+var EveProviderBridge = class {
+  constructor(metadata) {
+    this.metadata = metadata;
+  }
+  metadata;
+  activeActions = /* @__PURE__ */ new Map();
+  activeModels = /* @__PURE__ */ new Map();
+  modelsByAttempt = /* @__PURE__ */ new Map();
+  settledOperations = new LRUCache({
+    max: MAX_EVE_PROVIDER_CACHE_ENTRIES
+  });
+  turns = new LRUCache({
+    max: MAX_EVE_PROVIDER_CACHE_ENTRIES
+  });
+  async handleTurnStarted(event, context) {
+    await this.contain("turn start", async () => {
+      if (this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set({ skip: true });
+        return;
+      }
+      const { rowId, spanId } = await generateEveIds(
+        "turn",
+        event.idempotencyKey
+      );
+      const key = turnKey3(event.sessionId, event.turnId);
+      const stored = storedSpan(context);
+      if (stored && "exported" in stored && stored.rootSpanId) {
+        this.turns.set(key, { rootSpanId: stored.rootSpanId, spanId });
+        return;
+      }
+      let rootSpanId = spanId;
+      let parentSpanId;
+      if (event.parentLineage) {
+        const parentTurnKey = turnKey3(
+          event.parentLineage.sessionId,
+          event.parentLineage.turnId
+        );
+        const parentTurn = this.turns.get(parentTurnKey);
+        rootSpanId = parentTurn?.rootSpanId ?? // Eve propagates the root session through every nested subagent. Use
+        // it instead of the immediate parent session when no ancestor ran
+        // in-process.
+        (await generateEveIds(
+          "turn",
+          turnIdempotencyKey(
+            event.rootSessionId,
+            event.parentLineage.turnId
+          )
+        )).spanId;
+        parentSpanId = (await generateEveIds(
+          "subagent",
+          actionIdempotencyKey(
+            event.parentLineage.sessionId,
+            event.parentLineage.turnId,
+            event.parentLineage.callId
+          )
+        )).spanId;
+      }
+      const metadata = this.spanMetadata(event.sessionId);
+      const span = await this.startSpan(
+        context,
+        {
+          event: { id: rowId, metadata },
+          name: "eve.turn",
+          parentSpanIds: parentSpanId ? { rootSpanId, spanId: parentSpanId } : { parentSpanIds: [], rootSpanId },
+          spanAttributes: { type: "task" /* TASK */ },
+          spanId
+        },
+        rootSpanId
+      );
+      span?.log({ metadata });
+      this.turns.set(key, {
+        rootSpanId,
+        spanId
+      });
+    });
+  }
+  async handleTurnTerminal(event, context) {
+    await this.contain("turn terminal", async () => {
+      if (this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set(void 0);
+        return;
+      }
+      const error = event.type === "turn.failed" ? event.error : void 0;
+      this.drainActionsForTurn(event.sessionId, event.turnId, error);
+      const stored = storedSpan(context);
+      if (stored && "exported" in stored) {
+        updateSpan({
+          exported: stored.exported,
+          ...error !== void 0 ? { error } : {},
+          metrics: { end: Date.now() / 1e3 }
+        });
+      }
+      this.settledOperations.set(event.idempotencyKey, true);
+      context.state.set(void 0);
+      this.turns.delete(turnKey3(event.sessionId, event.turnId));
+    });
+  }
+  async handleModelStarted(event, context) {
+    await this.contain("model start", async () => {
+      if (this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set({ skip: true });
+        return;
+      }
+      const parent = await this.parentForScope(event.scope);
+      const { rowId, spanId } = await generateEveIds(
+        "step",
+        event.idempotencyKey
+      );
+      const input = event.input ? capturedModelInput(event.input) : void 0;
+      const metadata = {
+        ...this.spanMetadata(event.scope.sessionId),
+        model: event.model.modelId,
+        provider: event.model.provider
+      };
+      const span = await this.startSpan(context, {
+        event: {
+          id: rowId,
+          ...input !== void 0 ? { input } : {},
+          metadata
+        },
+        name: "eve.step",
+        parentSpanIds: parent,
+        spanAttributes: { type: "llm" /* LLM */ },
+        spanId
+      });
+      if (!span) return;
+      span.log({ ...input !== void 0 ? { input } : {}, metadata });
+      this.activeModels.set(event.idempotencyKey, {
+        span,
+        turnKey: turnKey3(event.scope.sessionId, event.scope.turnId)
+      });
+      const keys = this.modelsByAttempt.get(event.scope.attemptId) ?? /* @__PURE__ */ new Set();
+      keys.add(event.idempotencyKey);
+      this.modelsByAttempt.set(event.scope.attemptId, keys);
+    });
+  }
+  async handleModelTerminal(event, context) {
+    await this.contain("model terminal", async () => {
+      if (this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set(void 0);
+        return;
+      }
+      const active = this.activeModels.get(event.idempotencyKey);
+      if (active) {
+        const span = active.span;
+        if (event.type === "model.call.failed") {
+          if (event.error !== void 0) span.log({ error: event.error });
+        } else {
+          span.log({
+            metrics: usageMetrics(event.usage),
+            output: modelOutput(event)
+          });
+        }
+        span.end();
+      } else {
+        const stored = storedSpan(context);
+        if (stored && "exported" in stored) {
+          updateSpan({
+            exported: stored.exported,
+            ...event.type === "model.call.failed" ? event.error !== void 0 ? { error: event.error } : {} : { output: modelOutput(event) },
+            metrics: {
+              ...event.type === "model.call.completed" ? usageMetrics(event.usage) : {},
+              end: Date.now() / 1e3
+            }
+          });
+        }
+      }
+      this.settledOperations.set(event.idempotencyKey, true);
+      this.forgetModel(event.scope.attemptId, event.idempotencyKey);
+      context.state.set(void 0);
+    });
+  }
+  async handleActionStarted(event, context) {
+    await this.contain("action start", async () => {
+      if (event.kind === "load-skill" || this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set({ skip: true });
+        return;
+      }
+      const parent = await this.parentForScope(event.scope);
+      const { rowId, spanId } = await generateEveIds(
+        event.kind === "subagent-call" ? "subagent" : "tool",
+        event.idempotencyKey
+      );
+      const metadata = {
+        ...this.spanMetadata(event.scope.sessionId),
+        "eve.action_kind": event.kind
+      };
+      const span = await this.startSpan(context, {
+        event: {
+          id: rowId,
+          ...event.input !== void 0 ? { input: event.input } : {},
+          metadata
+        },
+        name: event.name,
+        parentSpanIds: parent,
+        spanAttributes: { type: "tool" /* TOOL */ },
+        spanId
+      });
+      if (!span) return;
+      span.log({
+        ...event.input !== void 0 ? { input: event.input } : {},
+        metadata
+      });
+      this.activeActions.set(event.idempotencyKey, {
+        span,
+        turnKey: turnKey3(event.scope.sessionId, event.scope.turnId)
+      });
+    });
+  }
+  async handleActionTerminal(event, context) {
+    await this.contain("action terminal", async () => {
+      if (this.settledOperations.has(event.idempotencyKey)) {
+        context.state.set(void 0);
+        return;
+      }
+      const stored = storedSpan(context);
+      if (stored?.skip) {
+        context.state.set(void 0);
+        return;
+      }
+      const active = this.activeActions.get(event.idempotencyKey);
+      const end = finiteTimestamp(event.acceptedAtMs) ?? Date.now();
+      if (active) {
+        const span = active.span;
+        if (event.type === "action.failed") {
+          span.log({
+            error: event.error ?? new Error(event.errorCode ?? `Eve action ${event.outcome}`)
+          });
+        } else if (event.output.type === "error") {
+          span.log({
+            error: event.output.error ?? new Error("Eve action returned an error")
+          });
+        } else {
+          span.log({ output: event.output.output });
+        }
+        span.end({ endTime: end / 1e3 });
+      } else {
+        const stored2 = storedSpan(context);
+        if (stored2 && "exported" in stored2) {
+          updateSpan({
+            exported: stored2.exported,
+            ...event.type === "action.failed" ? {
+              error: event.error ?? new Error(event.errorCode ?? `Eve action ${event.outcome}`)
+            } : event.output.type === "error" ? {
+              error: event.output.error ?? new Error("Eve action returned an error")
+            } : { output: event.output.output },
+            metrics: { end: end / 1e3 }
+          });
+        }
+      }
+      this.settledOperations.set(event.idempotencyKey, true);
+      this.activeActions.delete(event.idempotencyKey);
+      context.state.set(void 0);
+    });
+  }
+  handleStepAttemptTerminal(event) {
+    void this.contain("step attempt terminal", () => {
+      const keys = this.modelsByAttempt.get(event.scope.attemptId);
+      if (!keys) return;
+      for (const key of keys) {
+        const active = this.activeModels.get(key);
+        if (!active) continue;
+        if (event.type === "step.attempt.failed" && event.error !== void 0) {
+          active.span.log({ error: event.error });
+        }
+        active.span.end();
+        this.activeModels.delete(key);
+      }
+      this.modelsByAttempt.delete(event.scope.attemptId);
+    });
+  }
+  async parentForScope(scope) {
+    const key = turnKey3(scope.sessionId, scope.turnId);
+    const known = this.turns.get(key);
+    if (known) {
+      return { rootSpanId: known.rootSpanId, spanId: known.spanId };
+    }
+    const [{ spanId }, { spanId: rootSpanId }] = await Promise.all([
+      generateEveIds("turn", turnIdempotencyKey(scope.sessionId, scope.turnId)),
+      generateEveIds(
+        "turn",
+        turnIdempotencyKey(
+          scope.rootSessionId ?? scope.sessionId,
+          scope.turnId
+        )
+      )
+    ]);
+    return { rootSpanId, spanId };
+  }
+  async startSpan(context, args, rootSpanId) {
+    const span = withCurrent(
+      NOOP_SPAN,
+      () => _internalStartSpanWithInitialMerge(
+        withSpanInstrumentationName(args ?? {}, INSTRUMENTATION_NAMES.EVE)
+      )
+    );
+    try {
+      context.state.set({
+        exported: await span.export(),
+        ...rootSpanId ? { rootSpanId } : {}
+      });
+    } catch (error) {
+      debugLogger.warn("Error exporting Eve provider span:", error);
+    }
+    return span;
+  }
+  drainActionsForTurn(sessionId, turnId, error) {
+    const key = turnKey3(sessionId, turnId);
+    for (const [idempotencyKey, active] of this.activeActions) {
+      if (active.turnKey !== key) continue;
+      if (error !== void 0) active.span.log({ error });
+      active.span.end();
+      this.activeActions.delete(idempotencyKey);
+    }
+  }
+  forgetModel(attemptId, idempotencyKey) {
+    this.activeModels.delete(idempotencyKey);
+    const keys = this.modelsByAttempt.get(attemptId);
+    keys?.delete(idempotencyKey);
+    if (keys?.size === 0) this.modelsByAttempt.delete(attemptId);
+  }
+  spanMetadata(sessionId) {
+    return {
+      ...this.metadata ?? {},
+      "eve.session_id": sessionId
+    };
+  }
+  async contain(operation, fn) {
+    try {
+      await fn();
+    } catch (error) {
+      debugLogger.warn(`Error in Eve provider ${operation}:`, error);
+    }
+  }
+};
+function storedSpan(context) {
+  const value = context.state.get();
+  if (!isObject(value)) return void 0;
+  if (value["skip"] === true) return { skip: true };
+  return typeof value["exported"] === "string" ? {
+    exported: value["exported"],
+    ...typeof value["rootSpanId"] === "string" && value["rootSpanId"].length > 0 ? { rootSpanId: value["rootSpanId"] } : {}
+  } : void 0;
+}
+function modelOutput(event) {
+  const content = event.content ?? [];
+  let text = "";
+  const reasoning = [];
+  const toolCalls2 = [];
+  for (const part of content) {
+    if (part.type === "text") {
+      text += part.text;
+    } else if (part.type === "reasoning" && part.text.trim().length > 0) {
+      reasoning.push({ content: part.text });
+    } else if (part.type === "tool-call") {
+      toolCalls2.push({
+        function: {
+          arguments: safeJsonStringify(part.input),
+          name: part.toolName
+        },
+        id: part.callId,
+        type: "function"
+      });
+    }
+  }
+  return [
+    {
+      finish_reason: normalizeFinishReason3(event.finishReason),
+      index: 0,
+      message: {
+        content: text || null,
+        ...reasoning.length > 0 ? { reasoning } : {},
+        role: "assistant",
+        ...toolCalls2.length > 0 ? { tool_calls: toolCalls2 } : {}
+      }
+    }
+  ];
+}
+function usageMetrics(usage) {
+  const promptTokens = nonNegativeNumber(usage.inputTokens);
+  const completionTokens = nonNegativeNumber(usage.outputTokens);
+  const cachedTokens = nonNegativeNumber(
+    usage.inputTokenDetails?.cacheReadTokens
+  );
+  const cacheCreationTokens = nonNegativeNumber(
+    usage.inputTokenDetails?.cacheWriteTokens
+  );
+  return {
+    ...promptTokens !== void 0 ? { prompt_tokens: promptTokens } : {},
+    ...completionTokens !== void 0 ? { completion_tokens: completionTokens } : {},
+    ...promptTokens !== void 0 && completionTokens !== void 0 ? { tokens: promptTokens + completionTokens } : {},
+    ...cachedTokens !== void 0 ? { prompt_cached_tokens: cachedTokens } : {},
+    ...cacheCreationTokens !== void 0 ? { prompt_cache_creation_tokens: cacheCreationTokens } : {}
+  };
+}
+function nonNegativeNumber(value) {
+  return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : void 0;
+}
+function finiteTimestamp(value) {
+  return typeof value === "number" && Number.isFinite(value) ? value : void 0;
+}
+function normalizeFinishReason3(value) {
+  if (value === "content-filter") return "content_filter";
+  if (value === "tool-calls") return "tool_calls";
+  return value;
+}
+function safeJsonStringify(value) {
+  try {
+    return JSON.stringify(value) ?? "null";
+  } catch {
+    return "null";
+  }
+}
+function turnKey3(sessionId, turnId) {
+  return `${sessionId}:${turnId}`;
+}
+function turnIdempotencyKey(sessionId, turnId) {
+  return `turn:${sessionId}:${turnId}`;
+}
+function actionIdempotencyKey(sessionId, turnId, callId) {
+  return `action:${sessionId}:${turnId}:${callId}`;
+}
+
+// src/instrumentation/plugins/eve-instrumentation.ts
+var EVE_INSTRUMENTATION_PROVIDER = /* @__PURE__ */ Symbol.for("eve.instrumentation.provider");
+function braintrustEveInstrumentation(options) {
+  const definition = "defineState" in options ? createLegacyEveInstrumentation(options) : createEveInstrumentationProvider(options);
+  const declaration = {
+    ...definition,
+    [EVE_INSTRUMENTATION_PROVIDER]: true
+  };
+  return declaration;
+}
+
 // src/typed-instrumentation-helpers.ts
 var TypedApplyProxy = Proxy;
 
@@ -40579,11 +42241,11 @@ function wrapClaudeAgentQuery(queryFn, defaultThis) {
         }
       };
       const invocationTarget = thisArg === proxy || thisArg === void 0 ? defaultThis ?? thisArg : thisArg;
-      return claudeAgentSDKChannels.query.traceSync(
-        () => Reflect.apply(target, invocationTarget, [wrappedParams]),
-        // The channel carries no extra context fields, but the generated
-        // StartOf<> type for Record<string, never> is overly strict here.
-        { arguments: [wrappedParams] }
+      return claudeAgentSDKChannels.query.invoke(
+        target,
+        invocationTarget,
+        [wrappedParams],
+        {}
       );
     }
   });
@@ -41198,13 +42860,11 @@ function wrapAgentInstance(agent) {
       if (prop === "stream" && typeof value === "function") {
         return function(args, options) {
           const callArgs = [args, options];
-          return strandsAgentSDKChannels.agentStream.traceSync(
-            () => Reflect.apply(value, target, callArgs),
-            {
-              agent: proxy,
-              arguments: callArgs,
-              self: proxy
-            }
+          return strandsAgentSDKChannels.agentStream.invoke(
+            value,
+            target,
+            callArgs,
+            { agent: proxy }
           );
         };
       }
@@ -41235,12 +42895,12 @@ function wrapMultiAgentInstance(orchestrator, kind) {
         return function(input, options) {
           const callArgs = [input, options];
           const channel2 = kind === "graph" ? strandsAgentSDKChannels.graphStream : strandsAgentSDKChannels.swarmStream;
-          return channel2.traceSync(
-            () => Reflect.apply(value, target, callArgs),
+          return channel2.invoke(
+            value,
+            target,
+            callArgs,
             {
-              arguments: callArgs,
-              orchestrator: proxy,
-              self: proxy
+              orchestrator: proxy
             }
           );
         };
@@ -43192,9 +44852,6 @@ var VitestContextManager = class {
   getCurrentContext() {
     return this.contextStorage.getStore();
   }
-  setContext(context) {
-    this.contextStorage.enterWith(context);
-  }
   runInContext(context, callback) {
     return this.contextStorage.run(context, callback);
   }
@@ -43580,8 +45237,7 @@ function wrapDescribe(originalDescribe, config, afterAll) {
         if (config.onProgress) {
           config.onProgress({ type: "suite_start", suiteName });
         }
-        contextManager.setContext(lazyContext);
-        factory();
+        contextManager.runInContext(lazyContext, factory);
         if (afterAll) {
           afterAll(async () => {
             await flushExperimentWithSync(context, config);
@@ -48770,6 +50426,7 @@ export {
   braintrustStreamChunkSchema,
   buildLocalSummary,
   collectAnthropicSession,
+  completeOpenAIBatchTrace,
   configureInstrumentation,
   constructLogs3OverflowRequest,
   createFinalValuePassThroughStream,
@@ -48809,6 +50466,8 @@ export {
   loginToState,
   logs3OverflowUploadSchema,
   newId,
+  openaiBatchesRetrieveTraced,
+  openaiFilesCreateTraced,
   parseCachedHeader,
   parseTemplateFormat,
   permalink,