braintrust 0.0.93 → 0.0.94

package/dist/cli.js

@@ -9065,7 +9065,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "0.0.93",
+      version: "0.0.94",
       description: "SDK for integrating Braintrust",
       main: "./dist/index.js",
       browser: {
@@ -10644,6 +10644,21 @@ function getCurrentUnixTimestamp() {
 function isEmpty(a) {
   return a === void 0 || a === null;
 }
+var LazyValue = class {
+  constructor(callable) {
+    this.value = {
+      hasComputed: false
+    };
+    this.callable = callable;
+  }
+  async get() {
+    if (this.value.hasComputed) {
+      return this.value.val;
+    }
+    this.value = { hasComputed: true, val: await this.callable() };
+    return this.value.val;
+  }
+};
 
 // src/logger.ts
 var NoopSpan = class {
@@ -10858,25 +10873,25 @@ function logFeedbackImpl(bgLogger, parentIds, {
   updateEvent = Object.fromEntries(
     Object.entries(updateEvent).filter(([_, v]) => !isEmpty(v))
   );
-  const trueParentIds = (async () => {
-    const { kind, ...ids } = await parentIds;
+  const trueParentIds = new LazyValue(async () => {
+    const { kind, ...ids } = await parentIds.get();
     return ids;
-  })();
+  });
   if (Object.keys(updateEvent).length > 0) {
-    const record = (async () => {
+    const record = new LazyValue(async () => {
       return {
         id,
         ...updateEvent,
-        ...await trueParentIds,
+        ...await trueParentIds.get(),
         [AUDIT_SOURCE_FIELD]: source,
         [AUDIT_METADATA_FIELD]: metadata,
         [IS_MERGE_FIELD]: true
       };
-    })();
+    });
     bgLogger.log([record]);
   }
   if (!isEmpty(comment)) {
-    const record = (async () => {
+    const record = new LazyValue(async () => {
       return {
         id: v4_default(),
         created: (/* @__PURE__ */ new Date()).toISOString(),
@@ -10888,11 +10903,11 @@ function logFeedbackImpl(bgLogger, parentIds, {
         comment: {
           text: comment
         },
-        ...await trueParentIds,
+        ...await trueParentIds.get(),
         [AUDIT_SOURCE_FIELD]: source,
         [AUDIT_METADATA_FIELD]: metadata
       };
-    })();
+    });
     bgLogger.log([record]);
   }
 }
@@ -10924,9 +10939,20 @@ var BackgroundLogger = class {
   }
   async flush_once(batchSize = DefaultBatchSize) {
     this.active_flush_resolved = false;
-    const itemPromises = this.items;
+    const itemLazyValues = this.items;
     this.items = [];
-    const allItems = mergeRowBatch(await Promise.all(itemPromises)).reverse();
+    const allItems = await (async () => {
+      try {
+        const itemPromises = itemLazyValues.map((x) => x.get());
+        return mergeRowBatch(await Promise.all(itemPromises)).reverse();
+      } catch (e) {
+        console.warn(
+          "Encountered error when constructing records to flush:\n",
+          e
+        );
+        return [];
+      }
+    })();
     let postPromises = [];
     while (true) {
       const items = [];
@@ -10951,9 +10977,7 @@ var BackgroundLogger = class {
           for (let i = 0; i < NumRetries; i++) {
             const startTime = now();
             try {
-              return (await (await this.logConn).post_json("logs", itemsS)).map(
-                (res) => res.id
-              );
+              return (await (await this.logConn.get()).post_json("logs", itemsS)).map((res) => res.id);
             } catch (e) {
               const retryingText = i + 1 === NumRetries ? "" : " Retrying";
               const errMsg = (() => {
@@ -11006,82 +11030,86 @@ function init(project, options = {}) {
     metadata,
     gitMetadataSettings
   } = options || {};
-  const lazyMetadata = (async () => {
-    await login({
-      orgName,
-      apiKey,
-      appUrl
-    });
-    const args = {
-      project_name: project,
-      org_id: _state.orgId
-    };
-    if (experiment) {
-      args["experiment_name"] = experiment;
-    }
-    if (description) {
-      args["description"] = description;
-    }
-    if (update) {
-      args["update"] = update;
-    }
-    let mergedGitMetadataSettings = {
-      ..._state.gitMetadataSettings || {
-        collect: "all"
+  const lazyMetadata = new LazyValue(
+    async () => {
+      await login({
+        orgName,
+        apiKey,
+        appUrl
+      });
+      const args = {
+        project_name: project,
+        org_id: _state.orgId
+      };
+      if (experiment) {
+        args["experiment_name"] = experiment;
       }
-    };
-    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;
-    } else {
-      args["ancestor_commits"] = await isomorph_default.getPastNAncestors();
-    }
-    if (dataset !== void 0) {
-      args["dataset_id"] = dataset.id;
-      args["dataset_version"] = await dataset.version();
-    }
-    if (isPublic !== void 0) {
-      args["public"] = isPublic;
-    }
-    if (metadata) {
-      args["metadata"] = metadata;
-    }
-    let response = null;
-    while (true) {
-      try {
-        response = await _state.apiConn().post_json("api/experiment/register", args);
-        break;
-      } catch (e) {
-        if (args["base_experiment"] && `${"data" in e && e.data}`.includes("base experiment")) {
-          console.warn(`Base experiment ${args["base_experiment"]} not found.`);
-          delete args["base_experiment"];
-        } else {
-          throw e;
+      if (description) {
+        args["description"] = description;
+      }
+      if (update) {
+        args["update"] = update;
+      }
+      let mergedGitMetadataSettings = {
+        ..._state.gitMetadataSettings || {
+          collect: "all"
         }
+      };
+      if (gitMetadataSettings) {
+        mergedGitMetadataSettings = mergeGitMetadataSettings(
+          mergedGitMetadataSettings,
+          gitMetadataSettings
+        );
       }
-    }
-    return {
-      project: {
-        id: response.project.id,
-        name: response.project.name,
-        fullInfo: response.project
-      },
-      experiment: {
-        id: response.experiment.id,
-        name: response.experiment.name,
-        fullInfo: response.experiment
+      const repoStatus2 = await isomorph_default.getRepoStatus(gitMetadataSettings);
+      if (repoStatus2) {
+        args["repo_info"] = repoStatus2;
       }
-    };
-  })();
+      if (baseExperiment) {
+        args["base_experiment"] = baseExperiment;
+      } else {
+        args["ancestor_commits"] = await isomorph_default.getPastNAncestors();
+      }
+      if (dataset !== void 0) {
+        args["dataset_id"] = dataset.id;
+        args["dataset_version"] = await dataset.version();
+      }
+      if (isPublic !== void 0) {
+        args["public"] = isPublic;
+      }
+      if (metadata) {
+        args["metadata"] = metadata;
+      }
+      let response = null;
+      while (true) {
+        try {
+          response = await _state.apiConn().post_json("api/experiment/register", args);
+          break;
+        } catch (e) {
+          if (args["base_experiment"] && `${"data" in e && e.data}`.includes("base experiment")) {
+            console.warn(
+              `Base experiment ${args["base_experiment"]} not found.`
+            );
+            delete args["base_experiment"];
+          } else {
+            throw e;
+          }
+        }
+      }
+      return {
+        project: {
+          id: response.project.id,
+          name: response.project.name,
+          fullInfo: response.project
+        },
+        experiment: {
+          id: response.experiment.id,
+          name: response.experiment.name,
+          fullInfo: response.experiment
+        }
+      };
+    }
+  );
   const ret = new Experiment(lazyMetadata, dataset);
   if (options.setCurrent ?? true) {
     _state.currentExperiment = ret;
@@ -11232,27 +11260,29 @@ var Experiment = class {
     this.kind = "experiment";
     this.lazyMetadata = lazyMetadata;
     this.dataset = dataset;
-    const logConn = this.getState().then((state) => state.logConn());
+    const logConn = new LazyValue(
+      () => this.getState().then((state) => state.logConn())
+    );
     this.bgLogger = new BackgroundLogger(logConn);
     this.lastStartTime = getCurrentUnixTimestamp();
   }
   get id() {
     return (async () => {
-      return (await this.lazyMetadata).experiment.id;
+      return (await this.lazyMetadata.get()).experiment.id;
     })();
   }
   get name() {
     return (async () => {
-      return (await this.lazyMetadata).experiment.name;
+      return (await this.lazyMetadata.get()).experiment.name;
     })();
   }
   get project() {
     return (async () => {
-      return (await this.lazyMetadata).project;
+      return (await this.lazyMetadata.get()).project;
     })();
   }
   async getState() {
-    await this.lazyMetadata;
+    await this.lazyMetadata.get();
     return _state;
   }
   /**
@@ -11312,7 +11342,7 @@ var Experiment = class {
   startSpan(args) {
     const { name, ...argsRest } = args ?? {};
     return new SpanImpl({
-      parentIds: this.lazyParentIds(),
+      parentIds: new LazyValue(() => this.lazyParentIds()),
       bgLogger: this.bgLogger,
       name: name ?? "root",
       ...argsRest
@@ -11386,7 +11416,11 @@ var Experiment = class {
    * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
    */
   logFeedback(event) {
-    logFeedbackImpl(this.bgLogger, this.lazyParentIds(), event);
+    logFeedbackImpl(
+      this.bgLogger,
+      new LazyValue(() => this.lazyParentIds()),
+      event
+    );
   }
   /**
    * Flush any pending rows to the server.
@@ -11467,18 +11501,18 @@ var SpanImpl = class _SpanImpl {
     if (sanitizedAndInternalData.metrics?.end) {
       this.loggedEndTime = sanitizedAndInternalData.metrics?.end;
     }
-    const parentIds = (async () => {
-      const { kind, ...ids } = await this.parentIds;
+    const parentIds = new LazyValue(async () => {
+      const { kind, ...ids } = await this.parentIds.get();
       return ids;
-    })();
-    const record = (async () => {
+    });
+    const record = new LazyValue(async () => {
       return {
         ...sanitizedAndInternalData,
         ...this.rowIds,
-        ...await parentIds,
+        ...await parentIds.get(),
         [IS_MERGE_FIELD]: this.isMerge
       };
-    })();
+    });
     this.bgLogger.log([record]);
   }
   logFeedback(event) {
@@ -11532,26 +11566,28 @@ var Dataset = class {
     this._fetchedData = void 0;
     this.lazyMetadata = lazyMetadata;
     this.pinnedVersion = pinnedVersion;
-    const logConn = this.getState().then((state) => state.logConn());
+    const logConn = new LazyValue(
+      () => this.getState().then((state) => state.logConn())
+    );
     this.bgLogger = new BackgroundLogger(logConn);
   }
   get id() {
     return (async () => {
-      return (await this.lazyMetadata).dataset.id;
+      return (await this.lazyMetadata.get()).dataset.id;
     })();
   }
   get name() {
     return (async () => {
-      return (await this.lazyMetadata).dataset.name;
+      return (await this.lazyMetadata.get()).dataset.name;
     })();
   }
   get project() {
     return (async () => {
-      return (await this.lazyMetadata).project;
+      return (await this.lazyMetadata.get()).project;
     })();
   }
   async getState() {
-    await this.lazyMetadata;
+    await this.lazyMetadata.get();
     return _state;
   }
   /**
@@ -11582,7 +11618,7 @@ var Dataset = class {
       }
     }
     const rowId = id || v4_default();
-    const args = (async () => ({
+    const args = new LazyValue(async () => ({
       id: rowId,
       inputs: input,
       output,
@@ -11590,18 +11626,18 @@ var Dataset = class {
       dataset_id: await this.id,
       created: (/* @__PURE__ */ new Date()).toISOString(),
       metadata
-    }))();
+    }));
     this.bgLogger.log([args]);
     return rowId;
   }
   delete(id) {
-    const args = (async () => ({
+    const args = new LazyValue(async () => ({
       id,
       project_id: (await this.project).id,
       dataset_id: await this.id,
       created: (/* @__PURE__ */ new Date()).toISOString(),
       _object_delete: true
-    }))();
+    }));
     this.bgLogger.log([args]);
     return id;
   }
@@ -16054,7 +16090,9 @@ 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 {