braintrust 3.9.0 → 3.10.0

package/dist/cli.js

@@ -1232,7 +1232,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "3.9.0",
+      version: "3.10.0",
       description: "SDK for integrating Braintrust",
       repository: {
         type: "git",
@@ -1311,13 +1311,13 @@ var require_package = __commonJS({
           types: "./dist/auto-instrumentations/bundler/vite.d.ts",
           import: "./dist/auto-instrumentations/bundler/vite.mjs",
           module: "./dist/auto-instrumentations/bundler/vite.mjs",
-          require: "./dist/auto-instrumentations/bundler/vite.js"
+          require: "./dist/auto-instrumentations/bundler/vite.cjs"
         },
         "./webpack": {
           types: "./dist/auto-instrumentations/bundler/webpack.d.ts",
           import: "./dist/auto-instrumentations/bundler/webpack.mjs",
           module: "./dist/auto-instrumentations/bundler/webpack.mjs",
-          require: "./dist/auto-instrumentations/bundler/webpack.js"
+          require: "./dist/auto-instrumentations/bundler/webpack.cjs"
         },
         "./webpack-loader": {
           types: "./dist/auto-instrumentations/bundler/webpack-loader.d.ts",
@@ -1327,13 +1327,13 @@ var require_package = __commonJS({
           types: "./dist/auto-instrumentations/bundler/esbuild.d.ts",
           import: "./dist/auto-instrumentations/bundler/esbuild.mjs",
           module: "./dist/auto-instrumentations/bundler/esbuild.mjs",
-          require: "./dist/auto-instrumentations/bundler/esbuild.js"
+          require: "./dist/auto-instrumentations/bundler/esbuild.cjs"
         },
         "./rollup": {
           types: "./dist/auto-instrumentations/bundler/rollup.d.ts",
           import: "./dist/auto-instrumentations/bundler/rollup.mjs",
           module: "./dist/auto-instrumentations/bundler/rollup.mjs",
-          require: "./dist/auto-instrumentations/bundler/rollup.js"
+          require: "./dist/auto-instrumentations/bundler/rollup.cjs"
         }
       },
       files: [
@@ -1345,7 +1345,7 @@ var require_package = __commonJS({
         build: 'cross-env NODE_OPTIONS="--max-old-space-size=8192" tsup',
         "check:typings": "tsc --noEmit",
         watch: "tsup --watch",
-        clean: "rm -r dist/* && rm -r dev/dist/*",
+        clean: "rm -r dist/* && rm -r dev/dist/* && rm -r util/dist/*",
         docs: "typedoc --options typedoc.json src/node/index.ts",
         test: 'vitest run --exclude "src/wrappers/**/*.test.ts" --exclude "src/otel/**/*.test.ts" --exclude "smoke/**/*.test.ts" --exclude "src/zod/**/*.test.ts" --exclude "tests/api-compatibility/**"',
         "test:core": "pnpm prune && pnpm test",
@@ -3319,6 +3319,7 @@ var TopicMapData = import_v36.z.object({
   report_key: import_v36.z.string().optional(),
   topic_names: import_v36.z.record(import_v36.z.string()).optional(),
   generation_settings: TopicMapGenerationSettings.optional(),
+  disable_reconciliation: import_v36.z.boolean().optional(),
   distance_threshold: import_v36.z.number().optional()
 });
 var BatchedFacetData = import_v36.z.object({
@@ -3649,28 +3650,6 @@ var EnvVar = import_v36.z.object({
   secret_type: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
   secret_category: import_v36.z.enum(["env_var", "ai_provider", "sandbox_provider"]).optional().default("env_var")
 });
-var EvalStatusPageTheme = import_v36.z.enum(["light", "dark"]);
-var EvalStatusPageConfig = import_v36.z.object({
-  score_columns: import_v36.z.union([import_v36.z.array(import_v36.z.string()), import_v36.z.null()]),
-  metric_columns: import_v36.z.union([import_v36.z.array(import_v36.z.string()), import_v36.z.null()]),
-  grouping_field: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]),
-  filter: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]),
-  sort_by: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]),
-  sort_order: import_v36.z.union([import_v36.z.enum(["asc", "desc"]), import_v36.z.null()]),
-  api_key: import_v36.z.union([import_v36.z.string(), import_v36.z.null()])
-}).partial();
-var EvalStatusPage = import_v36.z.object({
-  id: import_v36.z.string().uuid(),
-  project_id: import_v36.z.string().uuid(),
-  user_id: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
-  created: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
-  deleted_at: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
-  name: import_v36.z.string(),
-  description: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
-  logo_url: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
-  theme: EvalStatusPageTheme,
-  config: EvalStatusPageConfig
-});
 var RepoInfo = import_v36.z.union([
   import_v36.z.object({
     commit: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]),
@@ -3697,6 +3676,15 @@ var Experiment = import_v36.z.object({
   deleted_at: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
   dataset_id: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
   dataset_version: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
+  internal_metadata: import_v36.z.union([
+    import_v36.z.object({
+      dataset_filter: import_v36.z.union([
+        import_v36.z.object({}).partial().passthrough(),
+        import_v36.z.null()
+      ])
+    }).partial().passthrough(),
+    import_v36.z.null()
+  ]).optional(),
   parameters_id: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
   parameters_version: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]).optional(),
   public: import_v36.z.boolean(),
@@ -4433,14 +4421,21 @@ var ProjectAutomation = import_v36.z.object({
         import_v36.z.object({ type: import_v36.z.literal("log_spans") }),
         import_v36.z.object({ type: import_v36.z.literal("btql_query"), btql_query: import_v36.z.string() })
       ]),
+      scope: import_v36.z.union([SpanScope, TraceScope, GroupScope, import_v36.z.null()]).optional(),
       export_path: import_v36.z.string(),
       format: import_v36.z.enum(["jsonl", "parquet"]),
       interval_seconds: import_v36.z.number().gte(1).lte(2592e3),
-      credentials: import_v36.z.object({
-        type: import_v36.z.literal("aws_iam"),
-        role_arn: import_v36.z.string(),
-        external_id: import_v36.z.string()
-      }),
+      credentials: import_v36.z.union([
+        import_v36.z.object({
+          type: import_v36.z.literal("aws_iam"),
+          role_arn: import_v36.z.string(),
+          external_id: import_v36.z.string()
+        }),
+        import_v36.z.object({
+          type: import_v36.z.literal("gcp_service_account"),
+          service_account_email: import_v36.z.string()
+        })
+      ]),
       batch_size: import_v36.z.union([import_v36.z.number(), import_v36.z.null()]).optional()
     }),
     import_v36.z.object({
@@ -4820,7 +4815,7 @@ var ViewOptions = import_v36.z.union([
       import_v36.z.object({ from: import_v36.z.string(), to: import_v36.z.string() }),
       import_v36.z.null()
     ]),
-    queryShape: import_v36.z.union([import_v36.z.enum(["traces", "spans"]), import_v36.z.null()]),
+    queryShape: import_v36.z.union([import_v36.z.enum(["traces", "spans", "topics"]), import_v36.z.null()]),
     cluster: import_v36.z.union([import_v36.z.string(), import_v36.z.null()]),
     freezeColumns: import_v36.z.union([import_v36.z.boolean(), import_v36.z.null()])
   }).partial(),
@@ -5834,6 +5829,19 @@ var RESET_CONTEXT_MANAGER_STATE = Symbol.for(
   "braintrust.resetContextManagerState"
 );
 var DEFAULT_MAX_REQUEST_SIZE = 6 * 1024 * 1024;
+var datasetSnapshotRegisterResponseSchema = import_v38.z.object({
+  dataset_snapshot: DatasetSnapshot,
+  found_existing: import_v38.z.boolean().optional()
+});
+var datasetRestorePreviewResultSchema = import_v38.z.object({
+  rows_to_restore: import_v38.z.number(),
+  rows_to_delete: import_v38.z.number()
+});
+var datasetRestoreResultSchema = import_v38.z.object({
+  xact_id: import_v38.z.string().nullable(),
+  rows_restored: import_v38.z.number(),
+  rows_deleted: import_v38.z.number()
+});
 var parametersRowSchema = import_v38.z.object({
   id: import_v38.z.string().uuid(),
   _xact_id: import_v38.z.string(),
@@ -7843,6 +7851,33 @@ Error: ${errorText}`;
     this.queue.enforceQueueSizeLimit(enforce);
   }
 };
+function isDatasetSnapshotNameLookup(lookup) {
+  return "snapshotName" in lookup;
+}
+function assertDatasetSnapshotLookup(lookup) {
+  const hasSnapshotName = lookup.snapshotName !== void 0;
+  const hasXactId = lookup.xactId !== void 0;
+  if (hasSnapshotName === hasXactId) {
+    throw new Error("Exactly one of snapshotName or xactId must be provided");
+  }
+}
+function getExperimentDatasetFilter({
+  dataset,
+  _internal_btql
+}) {
+  if (_internal_btql !== void 0) {
+    return _internal_btql;
+  }
+  if (!(dataset instanceof Dataset2)) {
+    return void 0;
+  }
+  const datasetFilter = Reflect.get(dataset, "_internal_btql");
+  return isObject(datasetFilter) ? datasetFilter : void 0;
+}
+function getInternalBtqlLimit(internalBtql) {
+  const limit = internalBtql?.["limit"];
+  return typeof limit === "number" ? limit : void 0;
+}
 function init(projectOrOptions, optionalOptions) {
   const options = (() => {
     if (typeof projectOrOptions === "string") {
@@ -7861,6 +7896,7 @@ function init(projectOrOptions, optionalOptions) {
     experiment,
     description,
     dataset,
+    _internal_btql,
     parameters,
     baseExperiment,
     isPublic,
@@ -7969,16 +8005,24 @@ function init(projectOrOptions, optionalOptions) {
         args["ancestor_commits"] = await isomorph_default.getPastNAncestors();
       }
       if (dataset !== void 0) {
-        if ("id" in dataset && typeof dataset.id === "string" && !("__braintrust_dataset_marker" in dataset)) {
-          args["dataset_id"] = dataset.id;
-          if ("version" in dataset && dataset.version !== void 0) {
-            args["dataset_version"] = dataset.version;
-          }
-        } else {
-          args["dataset_id"] = await dataset.id;
-          args["dataset_version"] = await dataset.version();
+        const datasetSelection = await serializeDatasetForExperiment({
+          dataset,
+          state
+        });
+        args["dataset_id"] = datasetSelection.datasetId;
+        if (datasetSelection.datasetVersion !== void 0) {
+          args["dataset_version"] = datasetSelection.datasetVersion;
         }
       }
+      const datasetFilter = getExperimentDatasetFilter({
+        dataset,
+        _internal_btql
+      });
+      if (datasetFilter !== void 0) {
+        args["internal_metadata"] = {
+          dataset_filter: datasetFilter
+        };
+      }
       if (parameters !== void 0) {
         if (RemoteEvalParameters.isParameters(parameters)) {
           args["parameters_id"] = parameters.id;
@@ -8032,13 +8076,160 @@ function init(projectOrOptions, optionalOptions) {
   const ret = new Experiment2(
     state,
     lazyMetadata,
-    dataset !== void 0 && "version" in dataset ? dataset : void 0
+    dataset !== void 0 ? dataset : void 0
   );
   if (options.setCurrent ?? true) {
     state.currentExperiment = ret;
   }
   return ret;
 }
+async function getDatasetSnapshots(params) {
+  const { state, datasetId } = params;
+  return DatasetSnapshot.array().parse(
+    await state.appConn().post_json("api/dataset_snapshot/get", {
+      dataset_id: datasetId,
+      ..."snapshotName" in params ? { name: params.snapshotName } : {},
+      ..."xactId" in params ? { xact_id: params.xactId } : {}
+    })
+  );
+}
+async function getDatasetSnapshot(params) {
+  assertDatasetSnapshotLookup(params);
+  const snapshots = await getDatasetSnapshots(params);
+  if (snapshots.length > 1) {
+    throw new Error(
+      isDatasetSnapshotNameLookup(params) ? `Expected a unique dataset snapshot named "${params.snapshotName}" for ${params.datasetId}` : `Expected a unique dataset snapshot for xact_id "${params.xactId}" in ${params.datasetId}`
+    );
+  }
+  return snapshots[0];
+}
+function normalizeDatasetSelection({
+  version: version2,
+  environment,
+  snapshotName
+}) {
+  if (version2 !== void 0) {
+    return { version: version2 };
+  }
+  if (snapshotName !== void 0) {
+    return { snapshotName };
+  }
+  if (environment !== void 0) {
+    return { environment };
+  }
+  return {};
+}
+async function resolveDatasetSnapshotName({
+  state,
+  datasetId,
+  snapshotName
+}) {
+  const match = await getDatasetSnapshot({
+    state,
+    datasetId,
+    snapshotName
+  });
+  if (match === void 0) {
+    throw new Error(
+      `Dataset snapshot "${snapshotName}" not found for ${datasetId}`
+    );
+  }
+  return match.xact_id;
+}
+async function resolveDatasetSnapshotNameForMetadata({
+  state,
+  lazyMetadata,
+  snapshotName
+}) {
+  const metadata = await lazyMetadata.get();
+  return await resolveDatasetSnapshotName({
+    state,
+    datasetId: metadata.dataset.id,
+    snapshotName
+  });
+}
+async function resolveDatasetEnvironment({
+  state,
+  datasetId,
+  environment
+}) {
+  const environmentObjectPath = `environment-object/dataset/${datasetId}/${encodeURIComponent(environment)}`;
+  const response = state.orgName == null ? await state.apiConn().get_json(environmentObjectPath) : await state.apiConn().get_json(environmentObjectPath, {
+    org_name: state.orgName
+  });
+  return import_v38.z.object({ object_version: import_v38.z.string() }).parse(response).object_version;
+}
+async function resolveDatasetEnvironmentForMetadata({
+  state,
+  lazyMetadata,
+  environment
+}) {
+  const metadata = await lazyMetadata.get();
+  return await resolveDatasetEnvironment({
+    state,
+    datasetId: metadata.dataset.id,
+    environment
+  });
+}
+async function serializeDatasetForExperiment({
+  dataset,
+  state
+}) {
+  if (!Dataset2.isDataset(dataset)) {
+    const selection2 = normalizeDatasetSelection(dataset);
+    if (selection2.version !== void 0) {
+      return {
+        datasetId: dataset.id,
+        datasetVersion: selection2.version
+      };
+    }
+    if (selection2.snapshotName !== void 0) {
+      return {
+        datasetId: dataset.id,
+        datasetVersion: await resolveDatasetSnapshotName({
+          state,
+          datasetId: dataset.id,
+          snapshotName: selection2.snapshotName
+        })
+      };
+    }
+    if (selection2.environment !== void 0) {
+      return {
+        datasetId: dataset.id,
+        datasetVersion: await resolveDatasetEnvironment({
+          state,
+          datasetId: dataset.id,
+          environment: selection2.environment
+        })
+      };
+    }
+    return {
+      datasetId: dataset.id
+    };
+  }
+  const evalData = await dataset.toEvalData();
+  const selection = normalizeDatasetSelection({
+    version: evalData.dataset_version,
+    environment: evalData.dataset_environment,
+    snapshotName: evalData.dataset_snapshot_name
+  });
+  if (selection.version !== void 0) {
+    return {
+      datasetId: evalData.dataset_id,
+      datasetVersion: selection.version
+    };
+  }
+  const datasetVersion = await dataset.version();
+  if (datasetVersion !== void 0) {
+    return {
+      datasetId: evalData.dataset_id,
+      datasetVersion
+    };
+  }
+  return {
+    datasetId: evalData.dataset_id
+  };
+}
 function initDataset(projectOrOptions, optionalOptions) {
   const options = (() => {
     if (typeof projectOrOptions === "string") {
@@ -8057,6 +8248,8 @@ function initDataset(projectOrOptions, optionalOptions) {
     dataset,
     description,
     version: version2,
+    snapshotName,
+    environment,
     appUrl,
     apiKey,
     orgName,
@@ -8068,6 +8261,14 @@ function initDataset(projectOrOptions, optionalOptions) {
     state: stateArg,
     _internal_btql
   } = options;
+  const selection = normalizeDatasetSelection({
+    version: version2,
+    environment,
+    snapshotName
+  });
+  const normalizedVersion = selection.version;
+  const normalizedEnvironment = selection.environment;
+  const normalizedSnapshotName = selection.snapshotName;
   const state = stateArg ?? _globalState;
   const lazyMetadata = new LazyValue(
     async () => {
@@ -8101,13 +8302,38 @@ function initDataset(projectOrOptions, optionalOptions) {
       };
     }
   );
-  return new Dataset2(
+  const resolvedVersion = normalizedVersion !== void 0 ? normalizedVersion : normalizedSnapshotName !== void 0 ? new LazyValue(async () => {
+    return await resolveDatasetSnapshotNameForMetadata({
+      state,
+      lazyMetadata,
+      snapshotName: normalizedSnapshotName
+    });
+  }) : normalizedEnvironment !== void 0 ? new LazyValue(async () => {
+    return await resolveDatasetEnvironmentForMetadata({
+      state,
+      lazyMetadata,
+      environment: normalizedEnvironment
+    });
+  }) : void 0;
+  const datasetObject = new Dataset2(
     stateArg ?? _globalState,
     lazyMetadata,
-    version2,
+    typeof resolvedVersion === "string" ? resolvedVersion : void 0,
     legacy,
-    _internal_btql
+    _internal_btql,
+    resolvedVersion instanceof LazyValue || normalizedEnvironment !== void 0 || normalizedSnapshotName !== void 0 ? {
+      ...resolvedVersion instanceof LazyValue ? {
+        lazyPinnedVersion: resolvedVersion
+      } : {},
+      ...normalizedEnvironment !== void 0 ? {
+        pinnedEnvironment: normalizedEnvironment
+      } : {},
+      ...normalizedSnapshotName !== void 0 ? {
+        pinnedSnapshotName: normalizedSnapshotName
+      } : {}
+    } : void 0
   );
+  return datasetObject;
 }
 async function computeLoggerMetadata(state, {
   project_name,
@@ -8599,11 +8825,20 @@ var ObjectFetcher = class {
   async getState() {
     throw new Error("ObjectFetcher subclasses must have a 'getState' method");
   }
+  getPinnedVersion() {
+    return this.pinnedVersion;
+  }
+  setPinnedVersion(pinnedVersion) {
+    this.pinnedVersion = pinnedVersion;
+  }
+  getInternalBtql() {
+    return this._internal_btql;
+  }
   async *fetchRecordsFromApi(batchSize) {
     const state = await this.getState();
     const objectId = await this.id;
     const batchLimit = batchSize ?? DEFAULT_FETCH_BATCH_SIZE;
-    const internalLimit = this._internal_btql?.limit;
+    const internalLimit = getInternalBtqlLimit(this._internal_btql);
     const limit = batchSize !== void 0 ? batchSize : internalLimit ?? batchLimit;
     const internalBtqlWithoutReservedQueryKeys = Object.fromEntries(
       Object.entries(this._internal_btql ?? {}).filter(
@@ -9417,7 +9652,7 @@ function splitLoggingData({
   return [serializableInternalData, lazyInternalData];
 }
 var Dataset2 = class extends ObjectFetcher {
-  constructor(state, lazyMetadata, pinnedVersion, legacy, _internal_btql) {
+  constructor(state, lazyMetadata, pinnedVersion, legacy, _internal_btql, pinState) {
     const isLegacyDataset = legacy ?? DEFAULT_IS_LEGACY_DATASET;
     if (isLegacyDataset) {
       debugLogger.forState(state).warn(
@@ -9438,10 +9673,16 @@ var Dataset2 = class extends ObjectFetcher {
     );
     this.state = state;
     this.lazyMetadata = lazyMetadata;
+    this.lazyPinnedVersion = pinState?.lazyPinnedVersion;
+    this.pinnedEnvironment = pinState?.pinnedEnvironment;
+    this.pinnedSnapshotName = pinState?.pinnedSnapshotName;
   }
   lazyMetadata;
   __braintrust_dataset_marker = true;
   newRecords = 0;
+  lazyPinnedVersion;
+  pinnedEnvironment;
+  pinnedSnapshotName;
   get id() {
     return (async () => {
       return (await this.lazyMetadata.get()).dataset.id;
@@ -9460,10 +9701,40 @@ var Dataset2 = class extends ObjectFetcher {
   get loggingState() {
     return this.state;
   }
+  async toEvalData() {
+    await this.getState();
+    const metadata = await this.lazyMetadata.get();
+    const pinnedVersion = this.getPinnedVersion();
+    const internalBtql = this.getInternalBtql();
+    return {
+      dataset_id: metadata.dataset.id,
+      ...this.pinnedEnvironment !== void 0 ? {
+        dataset_environment: this.pinnedEnvironment
+      } : {},
+      ...this.pinnedEnvironment === void 0 && this.pinnedSnapshotName !== void 0 ? {
+        dataset_snapshot_name: this.pinnedSnapshotName
+      } : {},
+      ...this.pinnedEnvironment === void 0 && this.pinnedSnapshotName === void 0 && pinnedVersion !== void 0 ? {
+        dataset_version: pinnedVersion
+      } : {},
+      ...internalBtql !== void 0 ? { _internal_btql: internalBtql } : {}
+    };
+  }
   async getState() {
     await this.lazyMetadata.get();
+    if (this.lazyPinnedVersion !== void 0 && this.getPinnedVersion() === void 0) {
+      this.setPinnedVersion(await this.lazyPinnedVersion.get());
+    }
     return this.state;
   }
+  async version(options) {
+    const pinnedVersion = this.getPinnedVersion();
+    if (pinnedVersion !== void 0) {
+      return pinnedVersion;
+    }
+    await this.getState();
+    return await super.version(options);
+  }
   validateEvent({
     metadata,
     expected,
@@ -9599,6 +9870,88 @@ var Dataset2 = class extends ObjectFetcher {
     this.state.bgLogger().log([args]);
     return id;
   }
+  async createSnapshot({
+    name,
+    description,
+    update
+  }) {
+    await this.flush();
+    const state = await this.getState();
+    const datasetId = await this.id;
+    const currentVersion = await this.version();
+    if (currentVersion === void 0) {
+      throw new Error("Cannot create snapshot: dataset has no version");
+    }
+    const response = await state.appConn().post_json("api/dataset_snapshot/register", {
+      dataset_id: datasetId,
+      dataset_snapshot_name: name,
+      description,
+      xact_id: currentVersion,
+      update
+    });
+    return datasetSnapshotRegisterResponseSchema.parse(response).dataset_snapshot;
+  }
+  async listSnapshots() {
+    const state = await this.getState();
+    return await getDatasetSnapshots({
+      state,
+      datasetId: await this.id
+    });
+  }
+  async getSnapshot(lookup) {
+    const state = await this.getState();
+    const datasetId = await this.id;
+    return await getDatasetSnapshot({
+      state,
+      datasetId,
+      ...lookup
+    });
+  }
+  async updateSnapshot(snapshotId, {
+    name,
+    description
+  }) {
+    const state = await this.getState();
+    return DatasetSnapshot.parse(
+      await state.appConn().post_json("api/dataset_snapshot/patch_id", {
+        id: snapshotId,
+        name,
+        description
+      })
+    );
+  }
+  async deleteSnapshot(snapshotId) {
+    const state = await this.getState();
+    return DatasetSnapshot.parse(
+      await state.appConn().post_json("api/dataset_snapshot/delete_id", {
+        id: snapshotId
+      })
+    );
+  }
+  async restorePreview({
+    version: version2
+  }) {
+    await this.flush();
+    const state = await this.getState();
+    const datasetId = await this.id;
+    return datasetRestorePreviewResultSchema.parse(
+      await state.apiConn().post_json(`v1/dataset/${datasetId}/restore/preview`, {
+        version: version2
+      })
+    );
+  }
+  async restore({
+    version: version2
+  }) {
+    await this.flush();
+    const state = await this.getState();
+    const datasetId = await this.id;
+    return datasetRestoreResultSchema.parse(
+      await state.apiConn().post_json(`v1/dataset/${datasetId}/restore`, {
+        version: version2
+      })
+    );
+  }
   /**
    * Summarize the dataset, including high level metrics about its size and other metadata.
    * @param summarizeData Whether to summarize the data. If false, only the metadata will be returned.
@@ -17782,6 +18135,9 @@ function llmParentKey(parentToolUseId) {
 function isSubAgentDelegationToolName(toolName) {
   return toolName === "Agent" || toolName === "Task";
 }
+function shouldParentToolAsTaskSibling(toolName) {
+  return toolName === "Agent" || toolName === "Task" || toolName === "Bash";
+}
 function filterSerializableOptions(options) {
   const allowedKeys = [
     "model",
@@ -17881,6 +18237,11 @@ function resolveTaskToolUseId(taskIdToToolUseId, message) {
   }
   return void 0;
 }
+function seedTaskToolUseIdMapping(taskIdToToolUseId, message) {
+  if (typeof message.task_id === "string" && typeof message.tool_use_id === "string") {
+    taskIdToToolUseId.set(message.task_id, message.tool_use_id);
+  }
+}
 function extractUsageFromMessage(message) {
   const metrics = {};
   let usage;
@@ -18076,7 +18437,10 @@ function createToolTracingHooks(resolveParentSpan, activeToolSpans, mcpServers,
         }
       },
       name: parsed.displayName,
-      parent: await resolveParentSpan(toolUseID),
+      parent: await resolveParentSpan(toolUseID, {
+        agentId: input.agent_id,
+        preferTaskSiblingParent: shouldParentToolAsTaskSibling(parsed.toolName)
+      }),
       spanAttributes: { type: "tool" /* TOOL */ }
     });
     activeToolSpans.set(toolUseID, toolSpan);
@@ -18366,6 +18730,7 @@ async function finalizeCurrentMessageGroup(state) {
   state.currentMessages.length = 0;
 }
 function maybeTrackToolUseContext(state, message) {
+  seedTaskToolUseIdMapping(state.taskIdToToolUseId, message);
   if (message.type !== "assistant" || !Array.isArray(message.message?.content)) {
     return;
   }
@@ -18420,6 +18785,34 @@ async function ensureSubAgentSpan(subAgentDetailsByToolUseId, rootSpan, activeTo
   subAgentSpans.set(parentToolUseId, subAgentSpan);
   return subAgentSpan;
 }
+async function ensureActiveLlmSpanForParentToolUse(rootSpan, activeLlmSpansByParentToolUse, subAgentDetailsByToolUseId, activeToolSpans, subAgentSpans, parentToolUseId, startTime) {
+  const parentKey = llmParentKey(parentToolUseId);
+  const existingLlmSpan = activeLlmSpansByParentToolUse.get(parentKey);
+  if (existingLlmSpan) {
+    return existingLlmSpan;
+  }
+  let llmParentSpan = await rootSpan.export();
+  if (parentToolUseId) {
+    const subAgentSpan = await ensureSubAgentSpan(
+      subAgentDetailsByToolUseId,
+      rootSpan,
+      activeToolSpans,
+      subAgentSpans,
+      parentToolUseId
+    );
+    llmParentSpan = await subAgentSpan.export();
+  }
+  const llmSpan = startSpan({
+    name: "anthropic.messages.create",
+    parent: llmParentSpan,
+    spanAttributes: {
+      type: "llm" /* LLM */
+    },
+    startTime
+  });
+  activeLlmSpansByParentToolUse.set(parentKey, llmSpan);
+  return llmSpan;
+}
 async function maybeHandleTaskLifecycleMessage(state, message) {
   if (message.type !== "system") {
     return false;
@@ -18529,29 +18922,15 @@ async function handleStreamMessage(state, message) {
   }
   if (message.type === "assistant" && message.message?.usage) {
     const parentToolUseId = message.parent_tool_use_id ?? null;
-    const parentKey = llmParentKey(parentToolUseId);
-    if (!state.activeLlmSpansByParentToolUse.has(parentKey)) {
-      let llmParentSpan = await state.span.export();
-      if (parentToolUseId) {
-        const subAgentSpan = await ensureSubAgentSpan(
-          state.subAgentDetailsByToolUseId,
-          state.span,
-          state.activeToolSpans,
-          state.subAgentSpans,
-          parentToolUseId
-        );
-        llmParentSpan = await subAgentSpan.export();
-      }
-      const llmSpan = startSpan({
-        name: "anthropic.messages.create",
-        parent: llmParentSpan,
-        spanAttributes: {
-          type: "llm" /* LLM */
-        },
-        startTime: state.currentMessageStartTime
-      });
-      state.activeLlmSpansByParentToolUse.set(parentKey, llmSpan);
-    }
+    await ensureActiveLlmSpanForParentToolUse(
+      state.span,
+      state.activeLlmSpansByParentToolUse,
+      state.subAgentDetailsByToolUseId,
+      state.activeToolSpans,
+      state.subAgentSpans,
+      parentToolUseId,
+      state.currentMessageStartTime
+    );
     state.currentMessages.push(message);
   }
   if (message.type !== "result" || !message.usage) {
@@ -18698,10 +19077,35 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
         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) => {
-          const parentToolUseId = toolUseToParent.get(toolUseID) ?? null;
+        const resolveToolUseParentSpan = async (toolUseID, context2) => {
+          const trackedParentToolUseId = toolUseToParent.get(toolUseID);
+          const parentToolUseId = trackedParentToolUseId ?? (context2?.agentId ? taskIdToToolUseId.get(context2.agentId) ?? null : null);
           const parentKey = llmParentKey(parentToolUseId);
           const activeLlmSpan = activeLlmSpansByParentToolUse.get(parentKey);
+          if (context2?.preferTaskSiblingParent) {
+            if (!activeLlmSpan) {
+              await ensureActiveLlmSpanForParentToolUse(
+                span,
+                activeLlmSpansByParentToolUse,
+                subAgentDetailsByToolUseId,
+                activeToolSpans,
+                subAgentSpans,
+                parentToolUseId,
+                getCurrentUnixTimestamp()
+              );
+            }
+            if (parentToolUseId) {
+              const subAgentSpan = await ensureSubAgentSpan(
+                subAgentDetailsByToolUseId,
+                span,
+                activeToolSpans,
+                subAgentSpans,
+                parentToolUseId
+              );
+              return subAgentSpan.export();
+            }
+            return span.export();
+          }
           if (activeLlmSpan) {
             return activeLlmSpan.export();
           }
@@ -18829,66 +19233,944 @@ var ClaudeAgentSDKPlugin = class extends BasePlugin {
   }
 };
 
-// src/instrumentation/plugins/google-genai-channels.ts
-var googleGenAIChannels = defineChannels("@google/genai", {
-  generateContent: channel({
-    channelName: "models.generateContent",
+// src/instrumentation/plugins/cursor-sdk-channels.ts
+var cursorSDKChannels = defineChannels("@cursor/sdk", {
+  create: channel({
+    channelName: "Agent.create",
     kind: "async"
   }),
-  generateContentStream: channel({
-    channelName: "models.generateContentStream",
+  resume: channel({
+    channelName: "Agent.resume",
     kind: "async"
   }),
-  embedContent: channel({
-    channelName: "models.embedContent",
+  prompt: channel({
+    channelName: "Agent.prompt",
+    kind: "async"
+  }),
+  send: channel({
+    channelName: "agent.send",
     kind: "async"
   })
 });
 
-// src/instrumentation/plugins/google-genai-plugin.ts
-var GOOGLE_GENAI_INTERNAL_CONTEXT = {
-  caller_filename: "<node-internal>",
-  caller_functionname: "<node-internal>",
-  caller_lineno: 0
-};
-function createWrapperParityEvent(args) {
-  return {
-    context: GOOGLE_GENAI_INTERNAL_CONTEXT,
-    input: args.input,
-    metadata: args.metadata
-  };
-}
-var GoogleGenAIPlugin = class extends BasePlugin {
+// src/instrumentation/plugins/cursor-sdk-plugin.ts
+var PATCHED_AGENT = Symbol.for("braintrust.cursor-sdk.auto-patched-agent");
+var PATCHED_RUN = Symbol.for("braintrust.cursor-sdk.patched-run");
+var WRAPPED_AGENT = Symbol.for("braintrust.cursor-sdk.wrapped-agent");
+var CursorSDKPlugin = class extends BasePlugin {
+  promptDepth = 0;
   onEnable() {
-    this.subscribeToGoogleGenAIChannels();
+    this.subscribeToAgentFactories();
+    this.subscribeToPrompt();
+    this.subscribeToSend();
   }
   onDisable() {
-    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+    for (const unsubscribe of this.unsubscribers) {
+      unsubscribe();
+    }
+    this.unsubscribers = [];
+    this.promptDepth = 0;
   }
-  subscribeToGoogleGenAIChannels() {
-    this.subscribeToGenerateContentChannel();
-    this.subscribeToGenerateContentStreamChannel();
-    this.subscribeToEmbedContentChannel();
+  subscribeToAgentFactories() {
+    this.subscribeToAgentFactory(cursorSDKChannels.create);
+    this.subscribeToAgentFactory(cursorSDKChannels.resume);
   }
-  subscribeToGenerateContentChannel() {
-    const tracingChannel2 = googleGenAIChannels.generateContent.tracingChannel();
+  subscribeToAgentFactory(channel2) {
+    const tracingChannel2 = channel2.tracingChannel();
+    const handlers = {
+      asyncEnd: (event) => {
+        patchCursorAgentInPlace(event.result);
+      },
+      error: () => {
+      }
+    };
+    tracingChannel2.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      tracingChannel2.unsubscribe(handlers);
+    });
+  }
+  subscribeToPrompt() {
+    const channel2 = cursorSDKChannels.prompt.tracingChannel();
     const states = /* @__PURE__ */ new WeakMap();
-    const unbindCurrentSpanStore = bindCurrentSpanStoreToStart2(
-      tracingChannel2,
-      states,
-      (event) => {
-        const params = event.arguments[0];
-        const input = serializeGenerateContentInput(params);
-        const metadata = extractGenerateContentMetadata(params);
+    const handlers = {
+      start: (event) => {
+        this.promptDepth += 1;
+        const message = event.arguments[0];
+        const options = event.arguments[1];
+        const metadata = {
+          ...extractAgentOptionsMetadata(options),
+          "cursor_sdk.operation": "Agent.prompt",
+          provider: "cursor",
+          ...event.moduleVersion ? { "cursor_sdk.version": event.moduleVersion } : {}
+        };
         const span = startSpan({
-          name: "generate_content",
-          spanAttributes: {
-            type: "llm" /* LLM */
-          },
-          event: createWrapperParityEvent({ input, metadata })
+          name: "Cursor Agent",
+          spanAttributes: { type: "task" /* TASK */ }
         });
-        return {
-          span,
+        const startTime = getCurrentUnixTimestamp();
+        safeLog(span, {
+          input: sanitizeUserMessage(message),
+          metadata
+        });
+        states.set(event, { metadata, span, startTime });
+      },
+      asyncEnd: (event) => {
+        this.promptDepth = Math.max(0, this.promptDepth - 1);
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        try {
+          safeLog(state.span, {
+            metadata: {
+              ...state.metadata,
+              ...extractRunResultMetadata(event.result)
+            },
+            metrics: buildDurationMetrics(state.startTime),
+            output: event.result?.result ?? event.result
+          });
+        } finally {
+          state.span.end();
+          states.delete(event);
+        }
+      },
+      error: (event) => {
+        this.promptDepth = Math.max(0, this.promptDepth - 1);
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        safeLog(state.span, { error: event.error.message });
+        state.span.end();
+        states.delete(event);
+      }
+    };
+    channel2.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      channel2.unsubscribe(handlers);
+    });
+  }
+  subscribeToSend() {
+    const channel2 = cursorSDKChannels.send.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const handlers = {
+      start: (event) => {
+        if (this.promptDepth > 0) {
+          return;
+        }
+        const message = event.arguments[0];
+        const sendOptions = event.arguments[1];
+        const agent = event.agent;
+        const metadata = {
+          ...extractSendMetadata(sendOptions),
+          ...agent ? extractAgentMetadata(agent) : {},
+          "cursor_sdk.operation": "agent.send",
+          provider: "cursor",
+          ...event.moduleVersion ? { "cursor_sdk.version": event.moduleVersion } : {}
+        };
+        const span = startSpan({
+          name: "Cursor Agent",
+          spanAttributes: { type: "task" /* TASK */ }
+        });
+        const startTime = getCurrentUnixTimestamp();
+        safeLog(span, {
+          input: sanitizeUserMessage(message),
+          metadata
+        });
+        const state = {
+          activeToolSpans: /* @__PURE__ */ new Map(),
+          agent,
+          conversationText: [],
+          deltaText: [],
+          finalized: false,
+          input: message,
+          metadata,
+          metrics: {},
+          span,
+          startTime,
+          streamMessages: [],
+          streamText: [],
+          stepText: [],
+          taskText: []
+        };
+        if (hasCursorCallbacks(sendOptions)) {
+          event.arguments[1] = wrapSendOptionsCallbacks(sendOptions, state);
+        }
+        states.set(event, state);
+      },
+      asyncEnd: (event) => {
+        const state = states.get(event);
+        if (!state) {
+          return;
+        }
+        if (!event.result) {
+          return;
+        }
+        state.run = event.result;
+        state.metadata = {
+          ...state.metadata,
+          ...extractRunMetadata(event.result)
+        };
+        patchCursorRun(event.result, state);
+      },
+      error: (event) => {
+        const state = states.get(event);
+        if (!state || !event.error) {
+          return;
+        }
+        safeLog(state.span, { error: event.error.message });
+        endOpenToolSpans(state, event.error.message);
+        state.span.end();
+        state.finalized = true;
+        states.delete(event);
+      }
+    };
+    channel2.subscribe(handlers);
+    this.unsubscribers.push(() => {
+      channel2.unsubscribe(handlers);
+    });
+  }
+};
+function patchCursorAgentInPlace(agent) {
+  if (!agent || typeof agent !== "object") {
+    return;
+  }
+  const agentRecord = agent;
+  if (agentRecord[PATCHED_AGENT] || agentRecord[WRAPPED_AGENT] || typeof agentRecord.send !== "function") {
+    return;
+  }
+  const originalSend = agentRecord.send.bind(agentRecord);
+  try {
+    Object.defineProperty(agentRecord, PATCHED_AGENT, {
+      configurable: false,
+      enumerable: false,
+      value: true
+    });
+    Object.defineProperty(agentRecord, "send", {
+      configurable: true,
+      value(message, options) {
+        const args = [message, options];
+        return cursorSDKChannels.send.tracePromise(
+          () => originalSend(...args),
+          {
+            agent: agentRecord,
+            arguments: args,
+            operation: "send"
+          }
+        );
+      },
+      writable: true
+    });
+  } catch {
+  }
+}
+function wrapSendOptionsCallbacks(options, state) {
+  const originalOnDelta = options.onDelta;
+  const originalOnStep = options.onStep;
+  return {
+    ...options,
+    async onDelta(args) {
+      try {
+        await handleInteractionUpdate(state, args.update);
+      } catch (error2) {
+        logInstrumentationError("Cursor SDK onDelta", error2);
+      }
+      if (originalOnDelta) {
+        return originalOnDelta(args);
+      }
+    },
+    async onStep(args) {
+      try {
+        handleStepUpdate(state, args.step);
+      } catch (error2) {
+        logInstrumentationError("Cursor SDK onStep", error2);
+      }
+      if (originalOnStep) {
+        return originalOnStep(args);
+      }
+    }
+  };
+}
+function hasCursorCallbacks(options) {
+  return !!options && (typeof options.onDelta === "function" || typeof options.onStep === "function");
+}
+function patchCursorRun(run2, state) {
+  if (!run2 || typeof run2 !== "object") {
+    return;
+  }
+  const runRecord = run2;
+  if (runRecord[PATCHED_RUN]) {
+    return;
+  }
+  try {
+    Object.defineProperty(runRecord, PATCHED_RUN, {
+      configurable: false,
+      enumerable: false,
+      value: true
+    });
+    if (typeof runRecord.stream === "function") {
+      const originalStream = runRecord.stream.bind(runRecord);
+      Object.defineProperty(runRecord, "stream", {
+        configurable: true,
+        value() {
+          const stream = originalStream();
+          return patchCursorStream(stream, state);
+        },
+        writable: true
+      });
+    }
+    if (typeof runRecord.wait === "function") {
+      const originalWait = runRecord.wait.bind(runRecord);
+      Object.defineProperty(runRecord, "wait", {
+        configurable: true,
+        async value() {
+          try {
+            const result = await originalWait();
+            state.lastResult = result;
+            await finalizeCursorRun(state, { result });
+            return result;
+          } catch (error2) {
+            await finalizeCursorRun(state, { error: error2 });
+            throw error2;
+          }
+        },
+        writable: true
+      });
+    }
+    if (typeof runRecord.conversation === "function") {
+      const originalConversation = runRecord.conversation.bind(runRecord);
+      Object.defineProperty(runRecord, "conversation", {
+        configurable: true,
+        async value() {
+          try {
+            const conversation = await originalConversation();
+            await handleConversation(state, conversation);
+            await finalizeCursorRun(state);
+            return conversation;
+          } catch (error2) {
+            await finalizeCursorRun(state, { error: error2 });
+            throw error2;
+          }
+        },
+        writable: true
+      });
+    }
+  } catch {
+    void finalizeCursorRun(state, { output: run2 });
+  }
+}
+async function* patchCursorStream(stream, state) {
+  try {
+    for await (const message of stream) {
+      try {
+        await handleStreamMessage2(state, message);
+      } catch (error2) {
+        logInstrumentationError("Cursor SDK stream", error2);
+      }
+      yield message;
+    }
+    await finalizeCursorRun(state);
+  } catch (error2) {
+    await finalizeCursorRun(state, { error: error2 });
+    throw error2;
+  }
+}
+async function handleInteractionUpdate(state, update) {
+  switch (update.type) {
+    case "text-delta":
+      if (typeof update.text === "string") {
+        state.deltaText.push(update.text);
+      }
+      return;
+    case "token-delta":
+      if (typeof update.tokens === "number") {
+        state.metrics["cursor_sdk.delta_tokens"] = (state.metrics["cursor_sdk.delta_tokens"] ?? 0) + update.tokens;
+      }
+      return;
+    case "tool-call-started":
+    case "partial-tool-call":
+    case "tool-call-completed":
+      await handleToolUpdate(
+        state,
+        update
+      );
+      return;
+    case "turn-ended":
+      addUsageMetrics(
+        state.metrics,
+        update.usage
+      );
+      return;
+    case "summary":
+      if (typeof update.summary === "string") {
+        state.taskText.push(update.summary);
+      }
+      return;
+    case "step-completed":
+      if (typeof update.stepDurationMs === "number") {
+        state.metrics["cursor_sdk.step_duration_ms"] = (state.metrics["cursor_sdk.step_duration_ms"] ?? 0) + update.stepDurationMs;
+      }
+      state.metrics["cursor_sdk.steps"] = (state.metrics["cursor_sdk.steps"] ?? 0) + 1;
+      return;
+    default:
+      return;
+  }
+}
+async function handleToolUpdate(state, update) {
+  const callId = update.callId;
+  if (!callId) {
+    return;
+  }
+  const toolCall = update.toolCall;
+  const name = extractToolName(toolCall);
+  const args = extractToolArgs(toolCall);
+  const result = extractToolResult(toolCall);
+  if (update.type === "tool-call-started" || update.type === "partial-tool-call") {
+    if (!state.activeToolSpans.has(callId)) {
+      state.activeToolSpans.set(
+        callId,
+        await startToolSpan(state, {
+          args,
+          callId,
+          name,
+          status: "running",
+          toolCall
+        })
+      );
+    }
+    return;
+  }
+  const toolState = state.activeToolSpans.get(callId) ?? await startToolSpan(state, {
+    args,
+    callId,
+    name,
+    status: "completed",
+    toolCall
+  });
+  finishToolSpan(toolState, {
+    error: toolCall?.status === "error" ? stringifyUnknown(result) : void 0,
+    metadata: {
+      "cursor_sdk.tool.status": toolCall?.status ?? "completed"
+    },
+    output: result
+  });
+  state.activeToolSpans.delete(callId);
+}
+async function handleStreamMessage2(state, message) {
+  state.streamMessages.push(message);
+  if (message.type === "system") {
+    const systemMessage = message;
+    state.metadata = {
+      ...state.metadata,
+      ...extractModelMetadata(systemMessage.model),
+      ...systemMessage.agent_id ? { "cursor_sdk.agent_id": systemMessage.agent_id } : {},
+      ...systemMessage.run_id ? { "cursor_sdk.run_id": systemMessage.run_id } : {},
+      ...systemMessage.tools ? { "cursor_sdk.tools": systemMessage.tools } : {}
+    };
+    return;
+  }
+  if (message.type === "assistant") {
+    const assistantMessage = message;
+    for (const block of assistantMessage.message?.content ?? []) {
+      if (block?.type === "text" && typeof block.text === "string") {
+        state.streamText.push(block.text);
+      } else if (block?.type === "tool_use" && block.id) {
+        state.activeToolSpans.set(
+          block.id,
+          await startToolSpan(state, {
+            args: block.input,
+            callId: block.id,
+            name: block.name,
+            status: "running"
+          })
+        );
+      }
+    }
+    return;
+  }
+  if (message.type === "tool_call") {
+    await handleToolMessage(
+      state,
+      message
+    );
+    return;
+  }
+  if (message.type === "task" && typeof message.text === "string") {
+    state.taskText.push(message.text);
+    return;
+  }
+  if (message.type === "status" && message.status) {
+    state.metadata["cursor_sdk.status"] = message.status;
+  }
+}
+async function handleToolMessage(state, message) {
+  const callId = message.call_id;
+  if (!callId) {
+    return;
+  }
+  if (message.status === "running") {
+    if (!state.activeToolSpans.has(callId)) {
+      state.activeToolSpans.set(
+        callId,
+        await startToolSpan(state, {
+          args: message.args,
+          callId,
+          name: message.name,
+          status: message.status,
+          truncated: message.truncated
+        })
+      );
+    }
+    return;
+  }
+  const toolState = state.activeToolSpans.get(callId) ?? await startToolSpan(state, {
+    args: message.args,
+    callId,
+    name: message.name,
+    status: message.status,
+    truncated: message.truncated
+  });
+  finishToolSpan(toolState, {
+    error: message.status === "error" ? stringifyUnknown(message.result) : void 0,
+    metadata: {
+      "cursor_sdk.tool.status": message.status
+    },
+    output: message.result
+  });
+  state.activeToolSpans.delete(callId);
+}
+async function handleConversation(state, turns) {
+  state.conversationOutput = turns;
+  for (const turn of turns) {
+    if (turn.type === "agentConversationTurn") {
+      for (const step of turn.turn?.steps ?? []) {
+        await handleConversationStep(state, step);
+      }
+    } else if (turn.type === "shellConversationTurn") {
+      const command = turn.turn?.shellCommand?.command;
+      if (command) {
+        const callId = `shell:${state.activeToolSpans.size}:${command}`;
+        const toolState = await startToolSpan(state, {
+          args: turn.turn?.shellCommand,
+          callId,
+          name: "shell",
+          status: "completed"
+        });
+        finishToolSpan(toolState, {
+          metadata: { "cursor_sdk.tool.status": "completed" },
+          output: turn.turn?.shellOutput
+        });
+      }
+    }
+  }
+}
+async function handleConversationStep(state, step) {
+  if (step.type === "assistantMessage" && typeof step.message?.text === "string") {
+    state.conversationText.push(step.message.text);
+    return;
+  }
+  if (step.type !== "toolCall") {
+    return;
+  }
+  const toolCall = step.message;
+  const callId = typeof toolCall?.callId === "string" ? toolCall.callId : `conversation-tool:${state.activeToolSpans.size}`;
+  const toolState = await startToolSpan(state, {
+    args: extractToolArgs(toolCall),
+    callId,
+    name: extractToolName(toolCall),
+    status: toolCall?.status,
+    toolCall
+  });
+  finishToolSpan(toolState, {
+    error: toolCall?.status === "error" ? stringifyUnknown(toolCall.result) : void 0,
+    metadata: {
+      "cursor_sdk.tool.status": toolCall?.status ?? "completed"
+    },
+    output: extractToolResult(toolCall)
+  });
+}
+function handleStepUpdate(state, step) {
+  state.metrics["cursor_sdk.steps"] = (state.metrics["cursor_sdk.steps"] ?? 0) + 1;
+  if (step.type) {
+    const stepTypes = state.metadata["cursor_sdk.step_types"];
+    if (Array.isArray(stepTypes)) {
+      if (!stepTypes.includes(step.type)) {
+        stepTypes.push(step.type);
+      }
+    } else {
+      state.metadata["cursor_sdk.step_types"] = [step.type];
+    }
+  }
+  if (step.type === "assistantMessage" && typeof step.message?.text === "string") {
+    state.stepText.push(step.message.text);
+  }
+}
+async function startToolSpan(state, args) {
+  const name = args.name || "unknown";
+  const metadata = {
+    "cursor_sdk.tool.status": args.status,
+    "gen_ai.tool.call.id": args.callId,
+    "gen_ai.tool.name": name
+  };
+  if (args.truncated?.args !== void 0) {
+    metadata["cursor_sdk.tool.args_truncated"] = args.truncated.args;
+  }
+  if (args.truncated?.result !== void 0) {
+    metadata["cursor_sdk.tool.result_truncated"] = args.truncated.result;
+  }
+  const span = startSpan({
+    event: {
+      input: args.args,
+      metadata
+    },
+    name: `tool: ${name}`,
+    parent: await state.span.export(),
+    spanAttributes: { type: "tool" /* TOOL */ }
+  });
+  let subAgentSpan;
+  if (isSubAgentToolName(name)) {
+    subAgentSpan = startSpan({
+      event: {
+        input: args.args,
+        metadata: {
+          "cursor_sdk.subagent.tool_call_id": args.callId,
+          "gen_ai.tool.name": name
+        }
+      },
+      name: formatSubAgentSpanName2(args.toolCall, args.args),
+      parent: await span.export(),
+      spanAttributes: { type: "task" /* TASK */ }
+    });
+  }
+  return { span, subAgentSpan };
+}
+function finishToolSpan(toolState, result) {
+  try {
+    if (result.error) {
+      safeLog(toolState.span, {
+        error: result.error,
+        metadata: result.metadata,
+        output: result.output
+      });
+      if (toolState.subAgentSpan) {
+        safeLog(toolState.subAgentSpan, {
+          error: result.error,
+          metadata: result.metadata,
+          output: result.output
+        });
+      }
+    } else {
+      safeLog(toolState.span, {
+        metadata: result.metadata,
+        output: result.output
+      });
+      if (toolState.subAgentSpan) {
+        safeLog(toolState.subAgentSpan, {
+          metadata: result.metadata,
+          output: result.output
+        });
+      }
+    }
+  } finally {
+    toolState.subAgentSpan?.end();
+    toolState.span.end();
+  }
+}
+async function finalizeCursorRun(state, params = {}) {
+  if (state.finalized) {
+    return;
+  }
+  state.finalized = true;
+  const error2 = params.error;
+  const result = params.result ?? state.lastResult;
+  const output = params.output ?? result?.result ?? state.run?.result ?? (state.conversationText.length > 0 ? state.conversationText.join("\n") : void 0) ?? state.conversationOutput ?? (state.streamText.length > 0 ? state.streamText.join("") : void 0) ?? (state.deltaText.length > 0 ? state.deltaText.join("") : void 0) ?? (state.stepText.length > 0 ? state.stepText.join("\n") : void 0) ?? (state.taskText.length > 0 ? state.taskText.join("\n") : void 0);
+  try {
+    const metadata = {
+      ...state.metadata,
+      ...state.run ? extractRunMetadata(state.run) : {},
+      ...result ? extractRunResultMetadata(result) : {}
+    };
+    if (error2) {
+      safeLog(state.span, {
+        error: error2 instanceof Error ? error2.message : String(error2),
+        metadata,
+        metrics: {
+          ...cleanMetrics(state.metrics),
+          ...buildDurationMetrics(state.startTime)
+        },
+        output
+      });
+    } else {
+      safeLog(state.span, {
+        metadata,
+        metrics: {
+          ...cleanMetrics(state.metrics),
+          ...buildDurationMetrics(state.startTime)
+        },
+        output
+      });
+    }
+  } finally {
+    endOpenToolSpans(state);
+    state.span.end();
+  }
+}
+function endOpenToolSpans(state, error2) {
+  for (const [, toolState] of state.activeToolSpans) {
+    finishToolSpan(toolState, { error: error2 });
+  }
+  state.activeToolSpans.clear();
+}
+function sanitizeUserMessage(message) {
+  if (typeof message === "string" || message === void 0) {
+    return message;
+  }
+  return {
+    ...message,
+    images: message.images?.map((image) => {
+      const imageRecord = image;
+      return {
+        ...typeof imageRecord.url === "string" ? { url: imageRecord.url } : {},
+        ...typeof imageRecord.mimeType === "string" ? { mimeType: imageRecord.mimeType } : {},
+        ...image.dimension ? { dimension: image.dimension } : {},
+        hasData: typeof imageRecord.data === "string"
+      };
+    })
+  };
+}
+function extractAgentOptionsMetadata(options) {
+  if (!options) {
+    return {};
+  }
+  return {
+    ...extractModelMetadata(options.model),
+    ...options.agentId ? { "cursor_sdk.agent_id": options.agentId } : {},
+    ...options.name ? { "cursor_sdk.agent_name": options.name } : {},
+    ...options.local ? {
+      "cursor_sdk.runtime": "local",
+      "cursor_sdk.local.cwd": Array.isArray(options.local.cwd) ? options.local.cwd.join(",") : options.local.cwd
+    } : {},
+    ...options.cloud ? {
+      "cursor_sdk.runtime": "cloud",
+      "cursor_sdk.cloud.auto_create_pr": options.cloud.autoCreatePR,
+      "cursor_sdk.cloud.env_type": options.cloud.env?.type,
+      "cursor_sdk.cloud.env_name": options.cloud.env?.name,
+      "cursor_sdk.cloud.repos": options.cloud.repos?.map((repo) => repo.url).filter((url) => typeof url === "string")
+    } : {}
+  };
+}
+function extractSendMetadata(options) {
+  if (!options) {
+    return {};
+  }
+  return {
+    ...extractModelMetadata(options.model),
+    ...options.local?.force !== void 0 ? { "cursor_sdk.local.force": options.local.force } : {}
+  };
+}
+function extractAgentMetadata(agent) {
+  return {
+    ...agent.agentId ? { "cursor_sdk.agent_id": agent.agentId } : {},
+    ...extractModelMetadata(agent.model)
+  };
+}
+function extractRunMetadata(run2) {
+  if (!run2) {
+    return {};
+  }
+  return {
+    ...run2.id ? { "cursor_sdk.run_id": run2.id } : {},
+    ...run2.agentId ? { "cursor_sdk.agent_id": run2.agentId } : {},
+    ...run2.status ? { "cursor_sdk.status": run2.status } : {},
+    ...run2.durationMs !== void 0 ? { "cursor_sdk.duration_ms": run2.durationMs } : {},
+    ...extractModelMetadata(run2.model),
+    ...extractGitMetadata(run2.git)
+  };
+}
+function extractRunResultMetadata(result) {
+  if (!result) {
+    return {};
+  }
+  return {
+    ...result.id ? { "cursor_sdk.run_id": result.id } : {},
+    ...result.status ? { "cursor_sdk.status": result.status } : {},
+    ...result.durationMs !== void 0 ? { "cursor_sdk.duration_ms": result.durationMs } : {},
+    ...extractModelMetadata(result.model),
+    ...extractGitMetadata(result.git)
+  };
+}
+function extractGitMetadata(git) {
+  const branches = git?.branches;
+  if (!branches || branches.length === 0) {
+    return {};
+  }
+  return {
+    "cursor_sdk.git.branches": branches.map((branch) => ({
+      branch: branch.branch,
+      prUrl: branch.prUrl,
+      repoUrl: branch.repoUrl
+    }))
+  };
+}
+function extractModelMetadata(model) {
+  if (!model?.id) {
+    return {};
+  }
+  return {
+    model: model.id,
+    "cursor_sdk.model": model.id,
+    ...model.params ? { "cursor_sdk.model.params": model.params } : {}
+  };
+}
+function addUsageMetrics(metrics, usage) {
+  if (!usage) {
+    return;
+  }
+  if (usage.inputTokens !== void 0) {
+    metrics.prompt_tokens = (metrics.prompt_tokens ?? 0) + usage.inputTokens;
+  }
+  if (usage.outputTokens !== void 0) {
+    metrics.completion_tokens = (metrics.completion_tokens ?? 0) + usage.outputTokens;
+  }
+  if (usage.cacheReadTokens !== void 0) {
+    metrics.prompt_cached_tokens = (metrics.prompt_cached_tokens ?? 0) + usage.cacheReadTokens;
+  }
+  if (usage.cacheWriteTokens !== void 0) {
+    metrics.prompt_cache_creation_tokens = (metrics.prompt_cache_creation_tokens ?? 0) + usage.cacheWriteTokens;
+  }
+  metrics.tokens = (metrics.prompt_tokens ?? 0) + (metrics.completion_tokens ?? 0) + (metrics.prompt_cached_tokens ?? 0) + (metrics.prompt_cache_creation_tokens ?? 0);
+}
+function buildDurationMetrics(startTime) {
+  const end = getCurrentUnixTimestamp();
+  return {
+    duration: end - startTime,
+    end,
+    start: startTime
+  };
+}
+function extractToolName(toolCall) {
+  if (!toolCall) {
+    return "unknown";
+  }
+  if (typeof toolCall.name === "string") {
+    return toolCall.name;
+  }
+  if (typeof toolCall.type === "string") {
+    return toolCall.type;
+  }
+  return "unknown";
+}
+function extractToolArgs(toolCall) {
+  return toolCall && "args" in toolCall ? toolCall.args : void 0;
+}
+function extractToolResult(toolCall) {
+  return toolCall && "result" in toolCall ? toolCall.result : void 0;
+}
+function isSubAgentToolName(name) {
+  return name === "Agent" || name === "Task" || name === "task";
+}
+function formatSubAgentSpanName2(toolCall, args) {
+  const details = toolCall ?? args;
+  const description = getString(details, "description") ?? getString(details, "subagent_type") ?? getString(details, "type") ?? getString(details, "name");
+  return description ? `Agent: ${description}` : "Agent: sub-agent";
+}
+function getString(obj, key) {
+  const value = obj?.[key];
+  return typeof value === "string" ? value : void 0;
+}
+function stringifyUnknown(value) {
+  if (value instanceof Error) {
+    return value.message;
+  }
+  if (typeof value === "string") {
+    return value;
+  }
+  try {
+    return JSON.stringify(value);
+  } catch {
+    return String(value);
+  }
+}
+function safeLog(span, event) {
+  try {
+    span.log(event);
+  } catch (error2) {
+    logInstrumentationError("Cursor SDK span log", error2);
+  }
+}
+function logInstrumentationError(context2, error2) {
+  debugLogger.error(`Error processing ${context2}:`, error2);
+}
+function cleanMetrics(metrics) {
+  const cleaned = {};
+  for (const [key, value] of Object.entries(metrics)) {
+    if (value !== void 0 && Number.isFinite(value)) {
+      cleaned[key] = value;
+    }
+  }
+  return cleaned;
+}
+
+// src/instrumentation/plugins/google-genai-channels.ts
+var googleGenAIChannels = defineChannels("@google/genai", {
+  generateContent: channel({
+    channelName: "models.generateContent",
+    kind: "async"
+  }),
+  generateContentStream: channel({
+    channelName: "models.generateContentStream",
+    kind: "async"
+  }),
+  embedContent: channel({
+    channelName: "models.embedContent",
+    kind: "async"
+  })
+});
+
+// src/instrumentation/plugins/google-genai-plugin.ts
+var GOOGLE_GENAI_INTERNAL_CONTEXT = {
+  caller_filename: "<node-internal>",
+  caller_functionname: "<node-internal>",
+  caller_lineno: 0
+};
+function createWrapperParityEvent(args) {
+  return {
+    context: GOOGLE_GENAI_INTERNAL_CONTEXT,
+    input: args.input,
+    metadata: args.metadata
+  };
+}
+var GoogleGenAIPlugin = class extends BasePlugin {
+  onEnable() {
+    this.subscribeToGoogleGenAIChannels();
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+  subscribeToGoogleGenAIChannels() {
+    this.subscribeToGenerateContentChannel();
+    this.subscribeToGenerateContentStreamChannel();
+    this.subscribeToEmbedContentChannel();
+  }
+  subscribeToGenerateContentChannel() {
+    const tracingChannel2 = googleGenAIChannels.generateContent.tracingChannel();
+    const states = /* @__PURE__ */ new WeakMap();
+    const unbindCurrentSpanStore = bindCurrentSpanStoreToStart2(
+      tracingChannel2,
+      states,
+      (event) => {
+        const params = event.arguments[0];
+        const input = serializeGenerateContentInput(params);
+        const metadata = extractGenerateContentMetadata(params);
+        const span = startSpan({
+          name: "generate_content",
+          spanAttributes: {
+            type: "llm" /* LLM */
+          },
+          event: createWrapperParityEvent({ input, metadata })
+        });
+        return {
+          span,
           startTime: getCurrentUnixTimestamp()
         };
       }
@@ -18921,7 +20203,7 @@ var GoogleGenAIPlugin = class extends BasePlugin {
           const responseMetadata = extractResponseMetadata(event.result);
           spanState.span.log({
             ...responseMetadata ? { metadata: responseMetadata } : {},
-            metrics: cleanMetrics(
+            metrics: cleanMetrics2(
               extractGenerateContentMetrics(
                 event.result,
                 spanState.startTime
@@ -19022,7 +20304,7 @@ var GoogleGenAIPlugin = class extends BasePlugin {
           const output = summarizeEmbedContentOutput(event.result);
           spanState.span.log({
             ...output ? { output } : {},
-            metrics: cleanMetrics(
+            metrics: cleanMetrics2(
               extractEmbedContentMetrics(event.result, spanState.startTime)
             )
           });
@@ -19119,7 +20401,7 @@ function patchGoogleGenAIStreamingResult(args) {
       );
       span.log({
         ...responseMetadata ? { metadata: responseMetadata } : {},
-        metrics: cleanMetrics(metricsWithoutEnd),
+        metrics: cleanMetrics2(metricsWithoutEnd),
         output: options.result.aggregated
       });
       span.end(typeof end === "number" ? { endTime: end } : void 0);
@@ -19544,7 +20826,7 @@ function aggregateGenerateContentChunks(chunks, startTime, firstTokenTime) {
   }
   return { aggregated, metrics };
 }
-function cleanMetrics(metrics) {
+function cleanMetrics2(metrics) {
   const cleaned = {};
   for (const [key, value] of Object.entries(metrics)) {
     if (value !== null && value !== void 0) {
@@ -19628,6 +20910,8 @@ var REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
   "stop",
   "stream",
   "temperature",
+  "tool_choice",
+  "tools",
   "top_p"
 ]);
 var RESPONSE_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
@@ -19890,7 +21174,10 @@ function aggregateChatCompletionChunks2(chunks) {
   for (const chunk of chunks) {
     for (const choice of chunk.choices ?? []) {
       const index = typeof choice.index === "number" ? choice.index : 0;
-      const existing = aggregatedChoices.get(index) ?? { content: "" };
+      const existing = aggregatedChoices.get(index) ?? {
+        content: "",
+        toolCallsByIndex: /* @__PURE__ */ new Map()
+      };
       const delta = isObject(choice.delta) ? choice.delta : void 0;
       const message = isObject(choice.message) ? choice.message : void 0;
       if (typeof delta?.content === "string") {
@@ -19906,6 +21193,10 @@ function aggregateChatCompletionChunks2(chunks) {
       if (choice.finish_reason !== void 0) {
         existing.finish_reason = choice.finish_reason;
       }
+      const toolCallDeltas = getChatToolCallDeltas(delta) ?? getChatToolCallDeltas(message);
+      if (toolCallDeltas) {
+        mergeChatToolCallDeltas(existing.toolCallsByIndex, toolCallDeltas);
+      }
       aggregatedChoices.set(index, existing);
     }
   }
@@ -19914,12 +21205,72 @@ function aggregateChatCompletionChunks2(chunks) {
       index,
       message: {
         content: choice.content,
-        role: choice.role ?? "assistant"
+        role: choice.role ?? "assistant",
+        ...choice.toolCallsByIndex.size > 0 ? {
+          tool_calls: [...choice.toolCallsByIndex.entries()].sort(([leftIndex], [rightIndex]) => leftIndex - rightIndex).map(([, toolCall]) => toolCall)
+        } : {}
       },
       ...choice.finish_reason !== void 0 ? { finish_reason: choice.finish_reason } : {}
     }))
   };
 }
+function getChatToolCallDeltas(value) {
+  if (!Array.isArray(value?.tool_calls)) {
+    return void 0;
+  }
+  const toolCalls = value.tool_calls.filter((toolCall) => isObject(toolCall));
+  return toolCalls.length > 0 ? toolCalls : void 0;
+}
+function mergeChatToolCallDeltas(toolCallsByIndex, toolCallDeltas) {
+  for (const toolDelta of toolCallDeltas) {
+    const toolIndex = typeof toolDelta.index === "number" && toolDelta.index >= 0 ? toolDelta.index : 0;
+    const existing = toolCallsByIndex.get(toolIndex);
+    if (!existing) {
+      toolCallsByIndex.set(toolIndex, createChatToolCall(toolDelta));
+      continue;
+    }
+    mergeChatToolCall(existing, toolDelta);
+  }
+}
+function createChatToolCall(toolDelta) {
+  const toolFunction = isObject(toolDelta.function) ? toolDelta.function : {};
+  const toolCallFunction = {
+    arguments: typeof toolFunction.arguments === "string" ? toolFunction.arguments : ""
+  };
+  if (typeof toolFunction.name === "string") {
+    toolCallFunction.name = toolFunction.name;
+  }
+  const toolCall = {
+    function: toolCallFunction
+  };
+  if (typeof toolDelta.id === "string") {
+    toolCall.id = toolDelta.id;
+  }
+  if (typeof toolDelta.type === "string") {
+    toolCall.type = toolDelta.type;
+  }
+  return toolCall;
+}
+function mergeChatToolCall(existing, toolDelta) {
+  const currentFunction = isObject(existing.function) ? existing.function : {};
+  const deltaFunction = isObject(toolDelta.function) ? toolDelta.function : {};
+  const currentArguments = typeof currentFunction.arguments === "string" ? currentFunction.arguments : "";
+  const deltaArguments = typeof deltaFunction.arguments === "string" ? deltaFunction.arguments : "";
+  if (typeof toolDelta.id === "string" && typeof existing.id !== "string") {
+    existing.id = toolDelta.id;
+  }
+  if (typeof toolDelta.type === "string" && typeof existing.type !== "string") {
+    existing.type = toolDelta.type;
+  }
+  const nextFunction = {
+    ...currentFunction,
+    arguments: `${currentArguments}${deltaArguments}`
+  };
+  if (typeof deltaFunction.name === "string" && typeof currentFunction.name !== "string") {
+    nextFunction.name = deltaFunction.name;
+  }
+  existing.function = nextFunction;
+}
 function aggregateTextGenerationStreamChunks(chunks) {
   if (chunks.length === 0) {
     return void 0;
@@ -21964,6 +23315,8 @@ var MISTRAL_REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
   "presence_penalty",
   "randomSeed",
   "random_seed",
+  "reasoningEffort",
+  "reasoning_effort",
   "responseFormat",
   "response_format",
   "safePrompt",
@@ -22104,6 +23457,98 @@ function extractDeltaText(content) {
   }).filter((part) => part.length > 0);
   return textParts.length > 0 ? textParts.join("") : void 0;
 }
+function normalizeMistralTextContentPart(part) {
+  if (!isObject(part) || part.type !== "text" || typeof part.text !== "string") {
+    return void 0;
+  }
+  return {
+    type: "text",
+    text: part.text
+  };
+}
+function normalizeMistralThinkingContentPart(part) {
+  if (!isObject(part) || part.type !== "thinking") {
+    return void 0;
+  }
+  const thinking = Array.isArray(part.thinking) ? part.thinking.map((thinkingPart) => normalizeMistralTextContentPart(thinkingPart)).filter(
+    (thinkingPart) => thinkingPart !== void 0 && typeof thinkingPart.text === "string"
+  ) : [];
+  return {
+    type: "thinking",
+    thinking
+  };
+}
+function normalizeMistralContentParts(content) {
+  if (!Array.isArray(content)) {
+    return [];
+  }
+  return content.map((part) => {
+    return normalizeMistralTextContentPart(part) || normalizeMistralThinkingContentPart(part);
+  }).filter((part) => part !== void 0);
+}
+function mergeMistralTextSegments(left, right) {
+  const merged = left.map((part) => ({ ...part }));
+  for (const part of right) {
+    const lastPart = merged[merged.length - 1];
+    if (lastPart && lastPart.type === "text" && typeof lastPart.text === "string" && typeof part.text === "string") {
+      lastPart.text += part.text;
+      continue;
+    }
+    merged.push({ ...part });
+  }
+  return merged;
+}
+function mergeMistralContentParts(left, right) {
+  const merged = [...(left || []).map((part) => structuredClone(part))];
+  for (const part of right) {
+    const lastPart = merged[merged.length - 1];
+    if (part.type === "text" && lastPart?.type === "text" && typeof lastPart.text === "string" && typeof part.text === "string") {
+      lastPart.text += part.text;
+      continue;
+    }
+    if (part.type === "thinking" && lastPart?.type === "thinking" && Array.isArray(lastPart.thinking) && Array.isArray(part.thinking)) {
+      lastPart.thinking = mergeMistralTextSegments(
+        lastPart.thinking,
+        part.thinking
+      );
+      continue;
+    }
+    merged.push(structuredClone(part));
+  }
+  return merged;
+}
+function appendMistralContent(accumulator, content) {
+  if (typeof content === "string") {
+    if (accumulator.contentParts) {
+      accumulator.contentParts = mergeMistralContentParts(
+        accumulator.contentParts,
+        [{ type: "text", text: content }]
+      );
+      return;
+    }
+    accumulator.content = `${accumulator.content || ""}${content}`;
+    return;
+  }
+  const normalizedContentParts = normalizeMistralContentParts(content);
+  if (normalizedContentParts.length === 0) {
+    return;
+  }
+  const hasStructuredContent = normalizedContentParts.some(
+    (part) => part.type !== "text"
+  );
+  if (!accumulator.contentParts && !hasStructuredContent) {
+    const text = extractDeltaText(content);
+    if (text) {
+      accumulator.content = `${accumulator.content || ""}${text}`;
+    }
+    return;
+  }
+  accumulator.contentParts = mergeMistralContentParts(
+    accumulator.contentParts || (accumulator.content ? [{ type: "text", text: accumulator.content }] : []),
+    normalizedContentParts
+  );
+  delete accumulator.content;
+}
 function getDeltaToolCalls(delta) {
   const toolCalls = Array.isArray(delta.toolCalls) && delta.toolCalls || Array.isArray(delta.tool_calls) && delta.tool_calls || [];
   return toolCalls.filter((toolCall) => isObject(toolCall));
@@ -22272,10 +23717,7 @@ function aggregateMistralStreamChunks(chunks) {
         if (!accumulator.role && typeof delta.role === "string") {
           accumulator.role = delta.role;
         }
-        const deltaText = extractDeltaText(delta.content);
-        if (deltaText) {
-          accumulator.content = `${accumulator.content || ""}${deltaText}`;
-        }
+        appendMistralContent(accumulator, delta.content);
         accumulator.toolCalls = mergeToolCallDeltas(
           accumulator.toolCalls,
           getDeltaToolCalls(delta)
@@ -22293,7 +23735,7 @@ function aggregateMistralStreamChunks(chunks) {
     index: choice.index,
     message: {
       ...choice.role ? { role: choice.role } : {},
-      content: choice.content ?? null,
+      content: choice.contentParts ?? choice.content ?? null,
       ...choice.toolCalls ? { toolCalls: choice.toolCalls } : {}
     },
     ...choice.finishReason !== void 0 ? { finishReason: choice.finishReason } : {}
@@ -22532,7 +23974,7 @@ var GoogleADKPlugin = class extends BasePlugin {
       start: (event) => {
         const req = event.arguments[0] ?? {};
         const tool = event.self;
-        const toolName = extractToolName(req, tool);
+        const toolName = extractToolName2(req, tool);
         const parentSpan = findToolParentSpan(
           req,
           this.activeAgentSpans,
@@ -22571,7 +24013,7 @@ var GoogleADKPlugin = class extends BasePlugin {
           metrics.duration = end - state.startTime;
           state.span.log({
             output: event.result,
-            metrics: cleanMetrics2(metrics)
+            metrics: cleanMetrics3(metrics)
           });
         } finally {
           state.span.end();
@@ -22790,7 +24232,7 @@ function extractToolCallId(req) {
   const toolContext = req.toolContext;
   return toolContext?.functionCallId;
 }
-function extractToolName(req, tool) {
+function extractToolName2(req, tool) {
   if (typeof tool?.name === "string" && tool.name.length > 0) {
     return tool.name;
   }
@@ -22838,7 +24280,7 @@ function finalizeRunnerSpan(state, activeRunnerSpans) {
     }
     state.span.log({
       output: lastEvent ? extractEventOutput(lastEvent) : void 0,
-      metrics: cleanMetrics2(metrics)
+      metrics: cleanMetrics3(metrics)
     });
   } finally {
     cleanupActiveRunnerSpan(state, activeRunnerSpans);
@@ -22855,7 +24297,7 @@ function finalizeAgentSpan(state, activeAgentSpans) {
     metrics.duration = end - state.startTime;
     state.span.log({
       output: lastEvent ? extractEventOutput(lastEvent) : void 0,
-      metrics: cleanMetrics2(metrics)
+      metrics: cleanMetrics3(metrics)
     });
   } finally {
     cleanupActiveAgentSpan(state, activeAgentSpans);
@@ -22941,7 +24383,7 @@ function populateUsageMetrics2(metrics, usage) {
     metrics.completion_reasoning_tokens = usage.thoughtsTokenCount;
   }
 }
-function cleanMetrics2(metrics) {
+function cleanMetrics3(metrics) {
   const cleaned = {};
   for (const [key, value] of Object.entries(metrics)) {
     if (value !== null && value !== void 0) {
@@ -23069,6 +24511,11 @@ var CHAT_REQUEST_METADATA_ALLOWLIST = /* @__PURE__ */ new Set([
   "strictTools",
   "strict_tools",
   "temperature",
+  "thinking",
+  "thinkingTokenBudget",
+  "thinkingType",
+  "thinking_token_budget",
+  "thinking_type",
   "toolChoice",
   "tool_choice"
 ]);
@@ -23289,6 +24736,11 @@ function mergeUsageMetrics(metrics, usage) {
       "tokens",
       tokenContainer.totalTokens ?? tokenContainer.total_tokens
     );
+    setMetricIfNumber(
+      metrics,
+      "reasoning_tokens",
+      tokenContainer.reasoningTokens ?? tokenContainer.reasoning_tokens ?? tokenContainer.thinkingTokens ?? tokenContainer.thinking_tokens
+    );
   }
   const billedUnits = (isObject(source.billedUnits) ? source.billedUnits : void 0) || (isObject(source.billed_units) ? source.billed_units : void 0);
   if (billedUnits) {
@@ -23369,13 +24821,87 @@ function extractV8DeltaText(chunk) {
   }
   return void 0;
 }
+function getV8ContentIndex(chunk) {
+  return typeof chunk.index === "number" ? chunk.index : 0;
+}
+function toContentBlockType(value) {
+  return value === "text" || value === "thinking" ? value : void 0;
+}
+function getOrCreateContentBlock(contentBlocksByIndex, contentBlockOrder, index) {
+  if (!contentBlockOrder.includes(index)) {
+    contentBlockOrder.push(index);
+  }
+  if (!(index in contentBlocksByIndex)) {
+    contentBlocksByIndex[index] = {
+      text: "",
+      thinking: ""
+    };
+  }
+  return contentBlocksByIndex[index];
+}
+function appendV8ContentBlock(contentBlocksByIndex, contentBlockOrder, index, content) {
+  if (typeof content === "string") {
+    const block2 = getOrCreateContentBlock(
+      contentBlocksByIndex,
+      contentBlockOrder,
+      index
+    );
+    block2.type ??= "text";
+    block2.text += content;
+    return;
+  }
+  if (!isObject(content)) {
+    return;
+  }
+  const block = getOrCreateContentBlock(
+    contentBlocksByIndex,
+    contentBlockOrder,
+    index
+  );
+  const contentType = toContentBlockType(content.type);
+  if (contentType) {
+    block.type = contentType;
+  }
+  if (typeof content.text === "string") {
+    block.type ??= "text";
+    block.text += content.text;
+  }
+  if (typeof content.thinking === "string") {
+    block.type ??= "thinking";
+    block.thinking += content.thinking;
+  }
+}
+function serializeAggregatedContentBlocks(contentBlocksByIndex, contentBlockOrder) {
+  return contentBlockOrder.sort((left, right) => left - right).flatMap((index) => {
+    const block = contentBlocksByIndex[index];
+    if (!block) {
+      return [];
+    }
+    if (block.type === "thinking" && block.thinking.length > 0) {
+      return [{ type: "thinking", thinking: block.thinking }];
+    }
+    if (block.text.length > 0) {
+      return [{ type: "text", text: block.text }];
+    }
+    if (block.thinking.length > 0) {
+      return [{ type: "thinking", thinking: block.thinking }];
+    }
+    return [];
+  });
+}
+function hasThinkingContent(contentBlocks) {
+  return contentBlocks.some((block) => block.type === "thinking");
+}
 function aggregateCohereChatStreamChunks(chunks) {
   const textDeltas = [];
+  const contentBlocksByIndex = {};
+  const contentBlockOrder = [];
   const toolCallsByIndex = {};
   const toolCallOrder = [];
   let terminalResponse;
   let role;
   let finishReason;
+  let toolPlan = "";
   let metadata = {};
   let metrics = {};
   for (const chunk of chunks) {
@@ -23433,12 +24959,36 @@ function aggregateCohereChatStreamChunks(chunks) {
       continue;
     }
     if (eventType === "content-delta") {
+      appendV8ContentBlock(
+        contentBlocksByIndex,
+        contentBlockOrder,
+        getV8ContentIndex(chunk),
+        isObject(chunk.delta) && isObject(chunk.delta.message) ? chunk.delta.message.content : void 0
+      );
       const text = extractV8DeltaText(chunk);
       if (text) {
         textDeltas.push(text);
       }
       continue;
     }
+    if (eventType === "content-start") {
+      appendV8ContentBlock(
+        contentBlocksByIndex,
+        contentBlockOrder,
+        getV8ContentIndex(chunk),
+        isObject(chunk.delta) && isObject(chunk.delta.message) ? chunk.delta.message.content : void 0
+      );
+      continue;
+    }
+    if (eventType === "tool-plan-delta") {
+      if (isObject(chunk.delta) && isObject(chunk.delta.message)) {
+        const deltaToolPlan = typeof chunk.delta.message.toolPlan === "string" ? chunk.delta.message.toolPlan : typeof chunk.delta.message.tool_plan === "string" ? chunk.delta.message.tool_plan : void 0;
+        if (deltaToolPlan) {
+          toolPlan += deltaToolPlan;
+        }
+      }
+      continue;
+    }
     if (eventType === "tool-call-start") {
       const toolCalls = isObject(chunk.delta) && isObject(chunk.delta.message) ? toToolCallArray(
         Array.isArray(chunk.delta.message.toolCalls) ? chunk.delta.message.toolCalls : chunk.delta.message.toolCalls ?? chunk.delta.message.tool_calls
@@ -23492,13 +25042,26 @@ function aggregateCohereChatStreamChunks(chunks) {
     }
   }
   const mergedToolCalls = toolCallOrder.sort((left, right) => left - right).map((index) => toolCallsByIndex[index]).filter((toolCall) => isObject(toolCall));
+  const aggregatedContentBlocks = serializeAggregatedContentBlocks(
+    contentBlocksByIndex,
+    contentBlockOrder
+  );
   let output = extractCohereChatOutput(terminalResponse);
   if (output === void 0) {
     const mergedText = textDeltas.join("");
-    if (mergedToolCalls.length > 0 || role || mergedText.length > 0) {
+    const shouldUseStructuredContent = hasThinkingContent(aggregatedContentBlocks) || toolPlan.length > 0;
+    if (shouldUseStructuredContent) {
+      output = {
+        ...role ? { role } : {},
+        ...aggregatedContentBlocks.length > 0 ? { content: aggregatedContentBlocks } : {},
+        ...toolPlan.length > 0 ? { toolPlan } : {},
+        ...mergedToolCalls.length > 0 ? { toolCalls: mergedToolCalls } : {}
+      };
+    } else if (mergedToolCalls.length > 0 || role || mergedText.length > 0 || aggregatedContentBlocks.length > 0) {
+      const textContent = mergedText.length > 0 ? mergedText : aggregatedContentBlocks[0]?.type === "text" ? aggregatedContentBlocks[0].text : void 0;
       output = {
         ...role ? { role } : {},
-        ...mergedText.length > 0 ? { content: mergedText } : {},
+        ...textContent ? { content: textContent } : {},
         ...mergedToolCalls.length > 0 ? { toolCalls: mergedToolCalls } : {}
       };
     }
@@ -23516,6 +25079,98 @@ function aggregateCohereChatStreamChunks(chunks) {
   };
 }
 
+// src/instrumentation/plugins/groq-channels.ts
+var groqChannels = defineChannels("groq-sdk", {
+  chatCompletionsCreate: channel({
+    channelName: "chat.completions.create",
+    kind: "async"
+  }),
+  embeddingsCreate: channel(
+    {
+      channelName: "embeddings.create",
+      kind: "async"
+    }
+  )
+});
+
+// src/instrumentation/plugins/groq-plugin.ts
+var GroqPlugin = class extends BasePlugin {
+  onEnable() {
+    this.unsubscribers.push(
+      traceStreamingChannel(groqChannels.chatCompletionsCreate, {
+        name: "groq.chat.completions.create",
+        type: "llm" /* LLM */,
+        extractInput: ([params]) => {
+          const { messages, ...metadata } = params;
+          return {
+            input: processInputAttachments(messages),
+            metadata: { ...metadata, provider: "groq" }
+          };
+        },
+        extractOutput: (result) => result?.choices,
+        extractMetrics: (result, startTime) => {
+          const metrics = parseGroqMetrics(result);
+          if (startTime) {
+            metrics.time_to_first_token = getCurrentUnixTimestamp() - startTime;
+          }
+          return metrics;
+        },
+        aggregateChunks: aggregateGroqChatCompletionChunks
+      })
+    );
+    this.unsubscribers.push(
+      traceAsyncChannel(groqChannels.embeddingsCreate, {
+        name: "groq.embeddings.create",
+        type: "llm" /* LLM */,
+        extractInput: ([params]) => {
+          const { input, ...metadata } = params;
+          return {
+            input,
+            metadata: { ...metadata, provider: "groq" }
+          };
+        },
+        extractOutput: (result) => {
+          const embedding = result?.data?.[0]?.embedding;
+          return Array.isArray(embedding) ? { embedding_length: embedding.length } : void 0;
+        },
+        extractMetrics: (result) => parseGroqMetrics(result)
+      })
+    );
+  }
+  onDisable() {
+    this.unsubscribers = unsubscribeAll(this.unsubscribers);
+  }
+};
+function parseGroqMetrics(result) {
+  const metrics = parseMetricsFromUsage(result?.usage);
+  const xGroq = result?.x_groq;
+  if (!xGroq || typeof xGroq !== "object") {
+    return metrics;
+  }
+  const extraUsage = "usage" in xGroq ? xGroq.usage : void 0;
+  if (!extraUsage || typeof extraUsage !== "object") {
+    return metrics;
+  }
+  const dramCachedTokens = extraUsage["dram_cached_tokens"];
+  const sramCachedTokens = extraUsage["sram_cached_tokens"];
+  return {
+    ...metrics,
+    ...typeof dramCachedTokens === "number" ? { dram_cached_tokens: dramCachedTokens } : {},
+    ...typeof sramCachedTokens === "number" ? { sram_cached_tokens: sramCachedTokens } : {}
+  };
+}
+function aggregateGroqChatCompletionChunks(chunks, streamResult, endEvent) {
+  const aggregated = aggregateChatCompletionChunks(
+    chunks,
+    streamResult,
+    endEvent
+  );
+  return {
+    metrics: aggregated.metrics,
+    output: aggregated.output
+  };
+}
+
 // src/instrumentation/braintrust-plugin.ts
 var BraintrustPlugin = class extends BasePlugin {
   config;
@@ -23523,6 +25178,7 @@ var BraintrustPlugin = class extends BasePlugin {
   anthropicPlugin = null;
   aiSDKPlugin = null;
   claudeAgentSDKPlugin = null;
+  cursorSDKPlugin = null;
   googleGenAIPlugin = null;
   huggingFacePlugin = null;
   openRouterPlugin = null;
@@ -23530,6 +25186,7 @@ var BraintrustPlugin = class extends BasePlugin {
   mistralPlugin = null;
   googleADKPlugin = null;
   coherePlugin = null;
+  groqPlugin = null;
   constructor(config3 = {}) {
     super();
     this.config = config3;
@@ -23552,6 +25209,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.claudeAgentSDKPlugin = new ClaudeAgentSDKPlugin();
       this.claudeAgentSDKPlugin.enable();
     }
+    if (integrations.cursorSDK !== false && integrations.cursor !== false) {
+      this.cursorSDKPlugin = new CursorSDKPlugin();
+      this.cursorSDKPlugin.enable();
+    }
     if (integrations.googleGenAI !== false && integrations.google !== false) {
       this.googleGenAIPlugin = new GoogleGenAIPlugin();
       this.googleGenAIPlugin.enable();
@@ -23580,6 +25241,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.coherePlugin = new CoherePlugin();
       this.coherePlugin.enable();
     }
+    if (integrations.groq !== false) {
+      this.groqPlugin = new GroqPlugin();
+      this.groqPlugin.enable();
+    }
   }
   onDisable() {
     if (this.openaiPlugin) {
@@ -23598,6 +25263,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.claudeAgentSDKPlugin.disable();
       this.claudeAgentSDKPlugin = null;
     }
+    if (this.cursorSDKPlugin) {
+      this.cursorSDKPlugin.disable();
+      this.cursorSDKPlugin = null;
+    }
     if (this.googleGenAIPlugin) {
       this.googleGenAIPlugin.disable();
       this.googleGenAIPlugin = null;
@@ -23626,6 +25295,10 @@ var BraintrustPlugin = class extends BasePlugin {
       this.coherePlugin.disable();
       this.coherePlugin = null;
     }
+    if (this.groqPlugin) {
+      this.groqPlugin.disable();
+      this.groqPlugin = null;
+    }
   }
 };
 
@@ -23699,6 +25372,8 @@ var PluginRegistry = class {
       google: true,
       huggingface: true,
       claudeAgentSDK: true,
+      cursor: true,
+      cursorSDK: true,
       openrouter: true,
       openrouterAgent: true,
       mistral: true,
@@ -23715,7 +25390,11 @@ var PluginRegistry = class {
     if (disabledList) {
       const disabled = disabledList.split(",").map((s) => s.trim().toLowerCase()).filter((s) => s.length > 0);
       for (const sdk of disabled) {
-        integrations[sdk] = false;
+        if (sdk === "cursor-sdk") {
+          integrations.cursorSDK = false;
+        } else {
+          integrations[sdk] = false;
+        }
       }
     }
     return { integrations };
@@ -25751,6 +27430,8 @@ async function getDataset(state, data) {
       state,
       project: data.project_name,
       dataset: data.dataset_name,
+      version: data.dataset_version ?? void 0,
+      environment: data.dataset_environment ?? void 0,
       _internal_btql: data._internal_btql ?? void 0
     });
   } else if ("dataset_id" in data) {
@@ -25762,6 +27443,8 @@ async function getDataset(state, data) {
       state,
       projectId: datasetInfo.projectId,
       dataset: datasetInfo.dataset,
+      version: data.dataset_version ?? void 0,
+      environment: data.dataset_environment ?? void 0,
       _internal_btql: data._internal_btql ?? void 0
     });
   } else {
@@ -26303,7 +27986,7 @@ async function collectFiles(inputPath, mode) {
       const prefix = mode === "eval" ? ".eval" : "";
       console.warn(
         warning2(
-          `Reading ${inputPath} because it was specified directly. Rename it to end in ${prefix}.ts or .${prefix}.js to include it automatically when you specify a directory.`
+          `Reading ${inputPath} because it was specified directly. Rename it to end in ${prefix}.ts or ${prefix}.js to include it automatically when you specify a directory.`
         )
       );
     }