braintrust 0.0.91 → 0.0.93

package/dist/browser.js

@@ -12,7 +12,7 @@ var DefaultAsyncLocalStorage = class {
   }
 };
 var iso = {
-  getRepoStatus: async () => void 0,
+  getRepoStatus: async (_settings) => void 0,
   getPastNAncestors: async () => [],
   getEnv: (_name) => void 0,
   getCallerLocation: () => void 0,
@@ -140,6 +140,24 @@ var SpanTypeAttribute = /* @__PURE__ */ ((SpanTypeAttribute2) => {
   SpanTypeAttribute2["TOOL"] = "tool";
   return SpanTypeAttribute2;
 })(SpanTypeAttribute || {});
+function mergeGitMetadataSettings(s1, s2) {
+  var _a;
+  if (s1.collect === "all") {
+    return s2;
+  } else if (s2.collect === "all") {
+    return s1;
+  } else if (s1.collect === "none") {
+    return s1;
+  } else if (s2.collect === "none") {
+    return s2;
+  }
+  const fields = ((_a = s1.fields) != null ? _a : []).filter((f) => {
+    var _a2;
+    return ((_a2 = s2.fields) != null ? _a2 : []).includes(f);
+  });
+  const collect = fields.length > 0 ? "some" : "none";
+  return { collect, fields };
+}
 
 // src/util.ts
 var GLOBAL_PROJECT = "Global";
@@ -194,7 +212,7 @@ var NoopSpan = class {
 var NOOP_SPAN = new NoopSpan();
 var BraintrustState = class {
   constructor() {
-    this.apiUrl = null;
+    this.appUrl = null;
     this.loginToken = null;
     this.orgId = null;
     this.orgName = null;
@@ -210,21 +228,22 @@ var BraintrustState = class {
     globalThis.__inherited_braintrust_state = this;
   }
   resetLoginInfo() {
-    this.apiUrl = null;
+    this.appUrl = null;
     this.loginToken = null;
     this.orgId = null;
     this.orgName = null;
     this.logUrl = null;
     this.loggedIn = false;
+    this.gitMetadataSettings = void 0;
     this._apiConn = null;
     this._logConn = null;
   }
   apiConn() {
     if (!this._apiConn) {
-      if (!this.apiUrl) {
-        throw new Error("Must initialize apiUrl before requesting apiConn");
+      if (!this.appUrl) {
+        throw new Error("Must initialize appUrl before requesting apiConn");
       }
-      this._apiConn = new HTTPConnection(this.apiUrl);
+      this._apiConn = new HTTPConnection(this.appUrl);
     }
     return this._apiConn;
   }
@@ -444,12 +463,12 @@ var Logger = class {
    * Log a single event. The event will be batched and uploaded behind the scenes if `logOptions.asyncFlush` is true.
    *
    * @param event The event to log.
-   * @param event.input: The arguments that uniquely define a user input (an arbitrary, JSON serializable object).
-   * @param event.output: The output of your application, including post-processing (an arbitrary, JSON serializable object), that allows you to determine whether the result is correct or not. For example, in an app that generates SQL queries, the `output` should be the _result_ of the SQL query generated by the model, not the query itself, because there may be multiple valid queries that answer a single question.
-   * @param event.expected: The ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
-   * @param event.scores: A dictionary of numeric values (between 0 and 1) to log. The scores should give you a variety of signals that help you determine how accurate the outputs are compared to what you expect and diagnose failures. For example, a summarization app might have one score that tells you how accurate the summary is, and another that measures the word similarity between the generated and grouth truth summary. The word similarity score could help you determine whether the summarization was covering similar concepts or not. You can use these scores to help you sort, filter, and compare logs.
+   * @param event.input: (Optional) the arguments that uniquely define a user input (an arbitrary, JSON serializable object).
+   * @param event.output: (Optional) the output of your application, including post-processing (an arbitrary, JSON serializable object), that allows you to determine whether the result is correct or not. For example, in an app that generates SQL queries, the `output` should be the _result_ of the SQL query generated by the model, not the query itself, because there may be multiple valid queries that answer a single question.
+   * @param event.expected: (Optional) the ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
+   * @param event.scores: (Optional) a dictionary of numeric values (between 0 and 1) to log. The scores should give you a variety of signals that help you determine how accurate the outputs are compared to what you expect and diagnose failures. For example, a summarization app might have one score that tells you how accurate the summary is, and another that measures the word similarity between the generated and grouth truth summary. The word similarity score could help you determine whether the summarization was covering similar concepts or not. You can use these scores to help you sort, filter, and compare logs.
    * @param event.metadata: (Optional) a dictionary with additional data about the test example, model outputs, or just about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any JSON-serializable type, but its keys must be strings.
-   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end", "caller_functionname", "caller_filename", "caller_lineno".
+   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end".
    * @param event.id: (Optional) a unique identifier for the event. If you don't provide one, BrainTrust will generate one for you.
    * :returns: The `id` of the logged event.
    */
@@ -503,7 +522,9 @@ var Logger = class {
     };
   }
   /**
-   * Lower-level alternative to `traced`, which does not automatically end the span or mark it as current.
+   * Lower-level alternative to `traced`. This allows you to start a span yourself, and can be useful in situations
+   * where you cannot use callbacks. However, spans started with `startSpan` will not be marked as the "current span",
+   * so `currentSpan()` and `traced()` will be no-ops. If you want to mark a span as current, use `traced` instead.
    *
    * See `traced` for full details.
    */
@@ -654,16 +675,17 @@ function init(project, options = {}) {
     baseExperiment,
     isPublic,
     update,
-    apiUrl,
+    appUrl,
     apiKey,
     orgName,
-    metadata
+    metadata,
+    gitMetadataSettings
   } = options || {};
   const lazyMetadata = (async () => {
     await login({
       orgName,
       apiKey,
-      apiUrl
+      appUrl
     });
     const args = {
       project_name: project,
@@ -678,7 +700,18 @@ function init(project, options = {}) {
     if (update) {
       args["update"] = update;
     }
-    const repoStatus = await isomorph_default.getRepoStatus();
+    let mergedGitMetadataSettings = {
+      ..._state.gitMetadataSettings || {
+        collect: "all"
+      }
+    };
+    if (gitMetadataSettings) {
+      mergedGitMetadataSettings = mergeGitMetadataSettings(
+        mergedGitMetadataSettings,
+        gitMetadataSettings
+      );
+    }
+    const repoStatus = await isomorph_default.getRepoStatus(gitMetadataSettings);
     if (repoStatus) {
       args["repo_info"] = repoStatus;
     }
@@ -745,12 +778,12 @@ function withLogger(callback, options = {}) {
   return callback(logger);
 }
 function initDataset(project, options = {}) {
-  const { dataset, description, version, apiUrl, apiKey, orgName } = options || {};
+  const { dataset, description, version, appUrl, apiKey, orgName } = options || {};
   const lazyMetadata = (async () => {
     await login({
       orgName,
       apiKey,
-      apiUrl
+      appUrl
     });
     const args = {
       org_id: _state.orgId,
@@ -786,7 +819,7 @@ function initLogger(options = {}) {
     projectName,
     projectId,
     asyncFlush,
-    apiUrl,
+    appUrl,
     apiKey,
     orgName,
     forceLogin
@@ -795,7 +828,7 @@ function initLogger(options = {}) {
     await login({
       orgName,
       apiKey,
-      apiUrl,
+      appUrl,
       forceLogin
     });
     const org_id = _state.orgId;
@@ -841,7 +874,7 @@ function initLogger(options = {}) {
 }
 async function login(options = {}) {
   const {
-    apiUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") || "https://www.braintrustdata.com",
+    appUrl = isomorph_default.getEnv("BRAINTRUST_APP_URL") || "https://www.braintrustdata.com",
     apiKey = isomorph_default.getEnv("BRAINTRUST_API_KEY"),
     orgName = isomorph_default.getEnv("BRAINTRUST_ORG_NAME")
   } = options || {};
@@ -850,11 +883,11 @@ async function login(options = {}) {
     return;
   }
   _state.resetLoginInfo();
-  _state.apiUrl = apiUrl;
+  _state.appUrl = appUrl;
   let conn = null;
   if (apiKey !== void 0) {
     const resp = await checkResponse(
-      await fetch(_urljoin(_state.apiUrl, `/api/apikey/login`), {
+      await fetch(_urljoin(_state.appUrl, `/api/apikey/login`), {
         method: "POST",
         headers: {
           "Content-Type": "application/json"
@@ -971,7 +1004,8 @@ function _check_org_info(org_info, org_name) {
     if (org_name === void 0 || org.name === org_name) {
       _state.orgId = org.id;
       _state.orgName = org.name;
-      _state.logUrl = isomorph_default.getEnv("BRAINTRUST_LOG_URL") ?? org.api_url;
+      _state.logUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
+      _state.gitMetadataSettings = org.git_metadata || void 0;
       break;
     }
   }
@@ -1040,6 +1074,9 @@ function validateAndSanitizeExperimentLogFullArgs(event, hasDataset) {
       "Exactly one of input or inputs (deprecated) must be specified. Prefer input."
     );
   }
+  if (!event.output) {
+    throw new Error("output must be specified");
+  }
   if (!event.scores) {
     throw new Error("scores must be specified");
   }
@@ -1087,10 +1124,10 @@ var Experiment = class {
    * @param event The event to log.
    * @param event.input: The arguments that uniquely define a test case (an arbitrary, JSON serializable object). Later on, Braintrust will use the `input` to know whether two test cases are the same between experiments, so they should not contain experiment-specific state. A simple rule of thumb is that if you run the same experiment twice, the `input` should be identical.
    * @param event.output: The output of your application, including post-processing (an arbitrary, JSON serializable object), that allows you to determine whether the result is correct or not. For example, in an app that generates SQL queries, the `output` should be the _result_ of the SQL query generated by the model, not the query itself, because there may be multiple valid queries that answer a single question.
-   * @param event.expected: The ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate your experiments while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
+   * @param event.expected: (Optional) The ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate your experiments while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
    * @param event.scores: A dictionary of numeric values (between 0 and 1) to log. The scores should give you a variety of signals that help you determine how accurate the outputs are compared to what you expect and diagnose failures. For example, a summarization app might have one score that tells you how accurate the summary is, and another that measures the word similarity between the generated and grouth truth summary. The word similarity score could help you determine whether the summarization was covering similar concepts or not. You can use these scores to help you sort, filter, and compare experiments.
    * @param event.metadata: (Optional) a dictionary with additional data about the test example, model outputs, or just about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any JSON-serializable type, but its keys must be strings.
-   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end", "caller_functionname", "caller_filename", "caller_lineno".
+   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end".
    * @param event.id: (Optional) a unique identifier for the event. If you don't provide one, BrainTrust will generate one for you.
    * @param event.dataset_record_id: (Optional) the id of the dataset record that this event is associated with. This field is required if and only if the experiment is associated with a dataset.
    * @param event.inputs: (Deprecated) the same as `input` (will be removed in a future version).
@@ -1129,7 +1166,9 @@ var Experiment = class {
     };
   }
   /**
-   * Lower-level alternative to `traced`, which does not automatically end the span or mark it as current.
+   * Lower-level alternative to `traced`. This allows you to start a span yourself, and can be useful in situations
+   * where you cannot use callbacks. However, spans started with `startSpan` will not be marked as the "current span",
+   * so `currentSpan()` and `traced()` will be no-ops. If you want to mark a span as current, use `traced` instead.
    *
    * See `traced` for full details.
    */
@@ -1154,7 +1193,7 @@ var Experiment = class {
     let { summarizeScores = true, comparisonExperimentId = void 0 } = options || {};
     await this.bgLogger.flush();
     const state = await this.getState();
-    const projectUrl = `${state.apiUrl}/app/${encodeURIComponent(
+    const projectUrl = `${state.appUrl}/app/${encodeURIComponent(
       state.orgName
     )}/p/${encodeURIComponent((await this.project).name)}`;
     const experimentUrl = `${projectUrl}/${encodeURIComponent(
@@ -1250,9 +1289,9 @@ var SpanImpl = class _SpanImpl {
     })();
     this.internalData = {
       metrics: {
-        start: args.startTime ?? getCurrentUnixTimestamp(),
-        ...callerLocation
+        start: args.startTime ?? getCurrentUnixTimestamp()
       },
+      context: { ...callerLocation },
       span_attributes: { ...args.spanAttributes, name },
       created: (/* @__PURE__ */ new Date()).toISOString()
     };
@@ -1439,7 +1478,7 @@ var Dataset = class {
     let { summarizeData = true } = options || {};
     await this.bgLogger.flush();
     const state = await this.getState();
-    const projectUrl = `${state.apiUrl}/app/${encodeURIComponent(
+    const projectUrl = `${state.appUrl}/app/${encodeURIComponent(
       state.orgName
     )}/p/${encodeURIComponent((await this.project).name)}`;
     const datasetUrl = `${projectUrl}/d/${encodeURIComponent(await this.name)}`;

package/dist/cli.js

@@ -9065,7 +9065,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "0.0.91",
+      version: "0.0.93",
       description: "SDK for integrating Braintrust",
       main: "./dist/index.js",
       browser: {
@@ -9108,7 +9108,7 @@ var require_package = __commonJS({
         typescript: "^5.3.3"
       },
       dependencies: {
-        "@braintrust/core": "^0.0.12",
+        "@braintrust/core": "^0.0.13",
         argparse: "^2.0.1",
         chalk: "^4.1.2",
         "cli-progress": "^3.12.0",
@@ -10578,6 +10578,24 @@ var SpanTypeAttribute = /* @__PURE__ */ ((SpanTypeAttribute2) => {
   SpanTypeAttribute2["TOOL"] = "tool";
   return SpanTypeAttribute2;
 })(SpanTypeAttribute || {});
+function mergeGitMetadataSettings(s1, s2) {
+  var _a2;
+  if (s1.collect === "all") {
+    return s2;
+  } else if (s2.collect === "all") {
+    return s1;
+  } else if (s1.collect === "none") {
+    return s1;
+  } else if (s2.collect === "none") {
+    return s2;
+  }
+  const fields = ((_a2 = s1.fields) != null ? _a2 : []).filter((f) => {
+    var _a22;
+    return ((_a22 = s2.fields) != null ? _a22 : []).includes(f);
+  });
+  const collect = fields.length > 0 ? "some" : "none";
+  return { collect, fields };
+}
 
 // src/isomorph.ts
 var DefaultAsyncLocalStorage = class {
@@ -10593,7 +10611,7 @@ var DefaultAsyncLocalStorage = class {
   }
 };
 var iso = {
-  getRepoStatus: async () => void 0,
+  getRepoStatus: async (_settings) => void 0,
   getPastNAncestors: async () => [],
   getEnv: (_name) => void 0,
   getCallerLocation: () => void 0,
@@ -10655,7 +10673,7 @@ var NoopSpan = class {
 var NOOP_SPAN = new NoopSpan();
 var BraintrustState = class {
   constructor() {
-    this.apiUrl = null;
+    this.appUrl = null;
     this.loginToken = null;
     this.orgId = null;
     this.orgName = null;
@@ -10671,21 +10689,22 @@ var BraintrustState = class {
     globalThis.__inherited_braintrust_state = this;
   }
   resetLoginInfo() {
-    this.apiUrl = null;
+    this.appUrl = null;
     this.loginToken = null;
     this.orgId = null;
     this.orgName = null;
     this.logUrl = null;
     this.loggedIn = false;
+    this.gitMetadataSettings = void 0;
     this._apiConn = null;
     this._logConn = null;
   }
   apiConn() {
     if (!this._apiConn) {
-      if (!this.apiUrl) {
-        throw new Error("Must initialize apiUrl before requesting apiConn");
+      if (!this.appUrl) {
+        throw new Error("Must initialize appUrl before requesting apiConn");
       }
-      this._apiConn = new HTTPConnection(this.apiUrl);
+      this._apiConn = new HTTPConnection(this.appUrl);
     }
     return this._apiConn;
   }
@@ -10981,16 +11000,17 @@ function init(project, options = {}) {
     baseExperiment,
     isPublic,
     update,
-    apiUrl,
+    appUrl,
     apiKey,
     orgName,
-    metadata
+    metadata,
+    gitMetadataSettings
   } = options || {};
   const lazyMetadata = (async () => {
     await login({
       orgName,
       apiKey,
-      apiUrl
+      appUrl
     });
     const args = {
       project_name: project,
@@ -11005,9 +11025,20 @@ function init(project, options = {}) {
     if (update) {
       args["update"] = update;
     }
-    const repoStatus = await isomorph_default.getRepoStatus();
-    if (repoStatus) {
-      args["repo_info"] = repoStatus;
+    let mergedGitMetadataSettings = {
+      ..._state.gitMetadataSettings || {
+        collect: "all"
+      }
+    };
+    if (gitMetadataSettings) {
+      mergedGitMetadataSettings = mergeGitMetadataSettings(
+        mergedGitMetadataSettings,
+        gitMetadataSettings
+      );
+    }
+    const repoStatus2 = await isomorph_default.getRepoStatus(gitMetadataSettings);
+    if (repoStatus2) {
+      args["repo_info"] = repoStatus2;
     }
     if (baseExperiment) {
       args["base_experiment"] = baseExperiment;
@@ -11059,7 +11090,7 @@ function init(project, options = {}) {
 }
 async function login(options = {}) {
   const {
-    apiUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") || "https://www.braintrustdata.com",
+    appUrl = isomorph_default.getEnv("BRAINTRUST_APP_URL") || "https://www.braintrustdata.com",
     apiKey = isomorph_default.getEnv("BRAINTRUST_API_KEY"),
     orgName = isomorph_default.getEnv("BRAINTRUST_ORG_NAME")
   } = options || {};
@@ -11068,11 +11099,11 @@ async function login(options = {}) {
     return;
   }
   _state.resetLoginInfo();
-  _state.apiUrl = apiUrl;
+  _state.appUrl = appUrl;
   let conn = null;
   if (apiKey !== void 0) {
     const resp = await checkResponse(
-      await fetch(_urljoin(_state.apiUrl, `/api/apikey/login`), {
+      await fetch(_urljoin(_state.appUrl, `/api/apikey/login`), {
         method: "POST",
         headers: {
           "Content-Type": "application/json"
@@ -11110,7 +11141,8 @@ function _check_org_info(org_info, org_name) {
     if (org_name === void 0 || org.name === org_name) {
       _state.orgId = org.id;
       _state.orgName = org.name;
-      _state.logUrl = isomorph_default.getEnv("BRAINTRUST_LOG_URL") ?? org.api_url;
+      _state.logUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
+      _state.gitMetadataSettings = org.git_metadata || void 0;
       break;
     }
   }
@@ -11179,6 +11211,9 @@ function validateAndSanitizeExperimentLogFullArgs(event, hasDataset) {
       "Exactly one of input or inputs (deprecated) must be specified. Prefer input."
     );
   }
+  if (!event.output) {
+    throw new Error("output must be specified");
+  }
   if (!event.scores) {
     throw new Error("scores must be specified");
   }
@@ -11226,10 +11261,10 @@ var Experiment = class {
    * @param event The event to log.
    * @param event.input: The arguments that uniquely define a test case (an arbitrary, JSON serializable object). Later on, Braintrust will use the `input` to know whether two test cases are the same between experiments, so they should not contain experiment-specific state. A simple rule of thumb is that if you run the same experiment twice, the `input` should be identical.
    * @param event.output: The output of your application, including post-processing (an arbitrary, JSON serializable object), that allows you to determine whether the result is correct or not. For example, in an app that generates SQL queries, the `output` should be the _result_ of the SQL query generated by the model, not the query itself, because there may be multiple valid queries that answer a single question.
-   * @param event.expected: The ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate your experiments while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
+   * @param event.expected: (Optional) The ground truth value (an arbitrary, JSON serializable object) that you'd compare to `output` to determine if your `output` value is correct or not. Braintrust currently does not compare `output` to `expected` for you, since there are so many different ways to do that correctly. Instead, these values are just used to help you navigate your experiments while digging into analyses. However, we may later use these values to re-score outputs or fine-tune your models.
    * @param event.scores: A dictionary of numeric values (between 0 and 1) to log. The scores should give you a variety of signals that help you determine how accurate the outputs are compared to what you expect and diagnose failures. For example, a summarization app might have one score that tells you how accurate the summary is, and another that measures the word similarity between the generated and grouth truth summary. The word similarity score could help you determine whether the summarization was covering similar concepts or not. You can use these scores to help you sort, filter, and compare experiments.
    * @param event.metadata: (Optional) a dictionary with additional data about the test example, model outputs, or just about anything else that's relevant, that you can use to help find and analyze examples later. For example, you could log the `prompt`, example's `id`, or anything else that would be useful to slice/dice later. The values in `metadata` can be any JSON-serializable type, but its keys must be strings.
-   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end", "caller_functionname", "caller_filename", "caller_lineno".
+   * @param event.metrics: (Optional) a dictionary of metrics to log. The following keys are populated automatically: "start", "end".
    * @param event.id: (Optional) a unique identifier for the event. If you don't provide one, BrainTrust will generate one for you.
    * @param event.dataset_record_id: (Optional) the id of the dataset record that this event is associated with. This field is required if and only if the experiment is associated with a dataset.
    * @param event.inputs: (Deprecated) the same as `input` (will be removed in a future version).
@@ -11268,7 +11303,9 @@ var Experiment = class {
     };
   }
   /**
-   * Lower-level alternative to `traced`, which does not automatically end the span or mark it as current.
+   * Lower-level alternative to `traced`. This allows you to start a span yourself, and can be useful in situations
+   * where you cannot use callbacks. However, spans started with `startSpan` will not be marked as the "current span",
+   * so `currentSpan()` and `traced()` will be no-ops. If you want to mark a span as current, use `traced` instead.
    *
    * See `traced` for full details.
    */
@@ -11293,7 +11330,7 @@ var Experiment = class {
     let { summarizeScores = true, comparisonExperimentId = void 0 } = options || {};
     await this.bgLogger.flush();
     const state = await this.getState();
-    const projectUrl = `${state.apiUrl}/app/${encodeURIComponent(
+    const projectUrl = `${state.appUrl}/app/${encodeURIComponent(
       state.orgName
     )}/p/${encodeURIComponent((await this.project).name)}`;
     const experimentUrl = `${projectUrl}/${encodeURIComponent(
@@ -11389,9 +11426,9 @@ var SpanImpl = class _SpanImpl {
     })();
     this.internalData = {
       metrics: {
-        start: args.startTime ?? getCurrentUnixTimestamp(),
-        ...callerLocation
+        start: args.startTime ?? getCurrentUnixTimestamp()
       },
+      context: { ...callerLocation },
       span_attributes: { ...args.spanAttributes, name },
       created: (/* @__PURE__ */ new Date()).toISOString()
     };
@@ -11578,7 +11615,7 @@ var Dataset = class {
     let { summarizeData = true } = options || {};
     await this.bgLogger.flush();
     const state = await this.getState();
-    const projectUrl = `${state.apiUrl}/app/${encodeURIComponent(
+    const projectUrl = `${state.appUrl}/app/${encodeURIComponent(
       state.orgName
     )}/p/${encodeURIComponent((await this.project).name)}`;
     const datasetUrl = `${projectUrl}/d/${encodeURIComponent(await this.name)}`;
@@ -16017,9 +16054,7 @@ async function getBaseBranchAncestor(remote = void 0) {
   if (git === null) {
     throw new Error("Not in a git repo");
   }
-  const { remote: remoteName, branch: baseBranch } = await getBaseBranch(
-    remote
-  );
+  const { remote: remoteName, branch: baseBranch } = await getBaseBranch(remote);
   const isDirty = (await git.diffSummary()).files.length > 0;
   const head = isDirty ? "HEAD" : "HEAD^";
   try {
@@ -16068,7 +16103,21 @@ function truncateToByteLimit(s, byteLimit = 65536) {
   const truncated = encoded.subarray(0, byteLimit);
   return new TextDecoder().decode(truncated);
 }
-async function getRepoStatus() {
+async function getRepoStatus(settings) {
+  if (settings && settings.collect === "none") {
+    return void 0;
+  }
+  const repo = await repoStatus();
+  if (!repo || !settings || settings.collect === "all") {
+    return repo;
+  }
+  let sanitized = {};
+  settings.fields?.forEach((field) => {
+    sanitized = { ...sanitized, [field]: repo[field] };
+  });
+  return sanitized;
+}
+async function repoStatus() {
   const git = await currentRepo();
   if (git === null) {
     return void 0;
@@ -16511,7 +16560,7 @@ async function run(args) {
     verbose: args.verbose,
     apiKey: args.api_key,
     orgName: args.org_name,
-    apiUrl: args.api_url,
+    appUrl: args.app_url,
     noSendLogs: !!args.no_send_logs,
     terminateOnFailure: !!args.terminate_on_failure,
     watch: !!args.watch,
@@ -16525,7 +16574,7 @@ async function run(args) {
       await login({
         apiKey: args.api_key,
         orgName: args.org_name,
-        apiUrl: args.api_url
+        appUrl: args.app_url
       });
     }
     if (args.watch) {
@@ -16563,8 +16612,8 @@ async function main() {
   parser_run.add_argument("--org-name", {
     help: "The name of a specific organization to connect to. This is useful if you belong to multiple."
   });
-  parser_run.add_argument("--api-url", {
-    help: "Specify a custom braintrust api url. Defaults to https://www.braintrustdata.com. This is only necessary if you are using an experimental version of Braintrust"
+  parser_run.add_argument("--app-url", {
+    help: "Specify a custom braintrust app url. Defaults to https://www.braintrustdata.com. This is only necessary if you are using an experimental version of Braintrust"
   });
   parser_run.add_argument("--watch", {
     action: "store_true",

package/dist/gitutil.d.ts

@@ -1,27 +1,7 @@
+import { GitMetadataSettings, RepoStatus } from "@braintrust/core";
 /**
  * Information about the current HEAD of the repo.
  */
-export interface RepoStatus {
-    commit?: string;
-    branch?: string;
-    tag?: string;
-    dirty: boolean;
-    author_name?: string;
-    author_email?: string;
-    commit_message?: string;
-    commit_time?: string;
-    git_diff?: string;
-}
 export declare function currentRepo(): Promise<import("simple-git").SimpleGit | null>;
 export declare function getPastNAncestors(n?: number, remote?: string | undefined): Promise<string[]>;
-export declare function getRepoStatus(): Promise<{
-    commit: string | undefined;
-    branch: string | undefined;
-    tag: string | undefined;
-    dirty: boolean;
-    author_name: string | undefined;
-    author_email: string | undefined;
-    commit_message: string | undefined;
-    commit_time: string | undefined;
-    git_diff: string | undefined;
-} | undefined>;
+export declare function getRepoStatus(settings?: GitMetadataSettings): Promise<RepoStatus | undefined>;