braintrust 0.0.134 → 0.0.136

package/dist/browser.js

@@ -5461,8 +5461,8 @@ function tryMakeUuid(s) {
 var ENCODING_VERSION_NUMBER = 1;
 var INVALID_ENCODING_ERRMSG = "SpanComponents string is not properly encoded. This may be due to a version mismatch between the SDK library used to export the span and the library used to decode it. Please make sure you are using the same SDK version across the board";
 var SpanObjectTypeV1 = /* @__PURE__ */ ((SpanObjectTypeV12) => {
-  SpanObjectTypeV12[SpanObjectTypeV12["EXPERIMENT"] = 0] = "EXPERIMENT";
-  SpanObjectTypeV12[SpanObjectTypeV12["PROJECT_LOGS"] = 1] = "PROJECT_LOGS";
+  SpanObjectTypeV12[SpanObjectTypeV12["EXPERIMENT"] = 1] = "EXPERIMENT";
+  SpanObjectTypeV12[SpanObjectTypeV12["PROJECT_LOGS"] = 2] = "PROJECT_LOGS";
   return SpanObjectTypeV12;
 })(SpanObjectTypeV1 || {});
 var SpanObjectTypeV1EnumSchema = z.nativeEnum(SpanObjectTypeV1);
@@ -5562,9 +5562,9 @@ var SpanComponentsV1 = class _SpanComponentsV1 {
   }
   objectIdFields() {
     switch (this.objectType) {
-      case 0:
-        return { experiment_id: this.objectId };
       case 1:
+        return { experiment_id: this.objectId };
+      case 2:
         return { project_id: this.objectId, log_id: "g" };
       default:
         throw new Error("Impossible");
@@ -5594,8 +5594,8 @@ var ENCODING_VERSION_NUMBER2 = 2;
 var INVALID_ENCODING_ERRMSG2 = `SpanComponents string is not properly encoded. This library only supports encoding versions up to ${ENCODING_VERSION_NUMBER2}. Please make sure the SDK library used to decode the SpanComponents is at least as new as any library used to encode it.`;
 var INTEGER_ENCODING_NUM_BYTES = 4;
 var SpanObjectTypeV2 = /* @__PURE__ */ ((SpanObjectTypeV22) => {
-  SpanObjectTypeV22[SpanObjectTypeV22["EXPERIMENT"] = 0] = "EXPERIMENT";
-  SpanObjectTypeV22[SpanObjectTypeV22["PROJECT_LOGS"] = 1] = "PROJECT_LOGS";
+  SpanObjectTypeV22[SpanObjectTypeV22["EXPERIMENT"] = 1] = "EXPERIMENT";
+  SpanObjectTypeV22[SpanObjectTypeV22["PROJECT_LOGS"] = 2] = "PROJECT_LOGS";
   return SpanObjectTypeV22;
 })(SpanObjectTypeV2 || {});
 var SpanObjectTypeV2EnumSchema = z.nativeEnum(SpanObjectTypeV2);
@@ -5766,9 +5766,9 @@ var SpanComponentsV2 = class _SpanComponentsV2 {
       );
     }
     switch (this.objectType) {
-      case 0:
-        return { experiment_id: this.objectId };
       case 1:
+        return { experiment_id: this.objectId };
+      case 2:
         return { project_id: this.objectId, log_id: "g" };
       default:
         throw new Error("Impossible");
@@ -5841,13 +5841,15 @@ var objectTypes = z.enum([
   "acl",
   "user",
   "project_score",
-  "project_tag"
+  "project_tag",
+  "function"
 ]);
 var objectTypesWithEvent = z.enum([
   "project",
   "experiment",
   "dataset",
   "prompt",
+  "function",
   "prompt_session"
 ]);
 function getEventObjectType(objectType2) {
@@ -6147,17 +6149,20 @@ var datasetSchema = z.strictObject({
   deleted_at: datasetBaseSchema.shape.deleted_at,
   user_id: datasetBaseSchema.shape.user_id
 }).openapi("Dataset");
+var validRuntimesEnum = z.enum(["node"]);
+var runtimeContextSchema = z.strictObject({
+  runtime: validRuntimesEnum,
+  version: z.string()
+});
 var promptBaseSchema = generateBaseTableSchema("prompt");
-var promptSchema = z.strictObject({
+var promptSchemaObject = z.strictObject({
   id: promptBaseSchema.shape.id,
   // This has to be copy/pasted because zod blows up when there are circular dependencies
   _xact_id: z.string().describe(
     `The transaction id of an event is unique to the network operation that processed the event insertion. Transaction ids are monotonically increasing over time and can be used to retrieve a versioned snapshot of the prompt (see the \`version\` parameter)`
   ),
   project_id: promptBaseSchema.shape.project_id,
-  log_id: z.literal("p").describe(
-    "A literal 'p' which identifies the object as a project prompt"
-  ),
+  log_id: z.literal("p").describe("A literal 'p' which identifies the object as a project prompt"),
   org_id: organizationSchema.shape.id,
   name: promptBaseSchema.shape.name,
   slug: z.string().describe("Unique identifier for the prompt"),
@@ -6166,7 +6171,46 @@ var promptSchema = z.strictObject({
   prompt_data: promptDataSchema.nullish().describe("The prompt, model, and its parameters"),
   tags: z.array(z.string()).nullish().describe("A list of tags for the prompt"),
   metadata: promptBaseSchema.shape.metadata
-}).openapi("Prompt");
+});
+var promptSchema = promptSchemaObject.openapi("Prompt");
+var codeBundleSchema = z.strictObject({
+  runtime_context: z.strictObject({
+    runtime: validRuntimesEnum,
+    version: z.string()
+  }),
+  // This should be a union, once we support code living in different places
+  // Other options should be:
+  //  - a "handler" function that has some signature [does AWS lambda assume it's always called "handler"?]
+  location: z.strictObject({
+    type: z.literal("experiment"),
+    eval_name: z.string(),
+    position: z.union([
+      z.literal("task"),
+      z.strictObject({ score: z.number() })
+    ])
+  }),
+  bundle_id: z.string()
+});
+var functionDataSchema = z.union([
+  z.strictObject({
+    type: z.literal("prompt")
+    // For backwards compatibility reasons, this is hoisted out and stored
+    // in the outer object
+  }),
+  z.strictObject({
+    type: z.literal("code"),
+    data: codeBundleSchema
+  }),
+  z.strictObject({
+    type: z.literal("global"),
+    name: z.string()
+  })
+]);
+var functionSchema2 = promptSchemaObject.merge(
+  z.strictObject({
+    function_data: functionDataSchema
+  })
+).openapi("Function");
 var repoInfoSchema = z.strictObject({
   commit: z.string().nullish().describe("SHA of most recent commit"),
   branch: z.string().nullish().describe("Name of the branch the most recent commit belongs to"),
@@ -6248,8 +6292,6 @@ var aclObjectTypeEnum = z.enum([
   "dataset",
   "prompt",
   "prompt_session",
-  "project_score",
-  "project_tag",
   "group",
   "role",
   "org_member",
@@ -6456,12 +6498,27 @@ var createPromptSchema = promptSchema.omit({
   created: true,
   metadata: true
 }).openapi("CreatePrompt");
+var createFunctionSchema = functionSchema2.omit({
+  id: true,
+  _xact_id: true,
+  org_id: true,
+  log_id: true,
+  created: true,
+  metadata: true
+}).openapi("CreateFunction");
 var patchPromptSchema = z.strictObject({
   name: promptSchema.shape.name.nullish(),
   description: promptSchema.shape.description.nullish(),
   prompt_data: promptSchema.shape.prompt_data.nullish(),
   tags: promptSchema.shape.tags.nullish()
 }).openapi("PatchPrompt");
+var patchFunctionSchema = z.strictObject({
+  name: functionSchema2.shape.name.nullish(),
+  description: functionSchema2.shape.description.nullish(),
+  prompt_data: functionSchema2.shape.prompt_data.nullish(),
+  function_data: functionSchema2.shape.function_data.nullish(),
+  tags: functionSchema2.shape.tags.nullish()
+}).openapi("PatchFunction");
 var createRoleBaseSchema = generateBaseTableOpSchema("role");
 var createRoleSchema = z.strictObject({
   name: roleSchema.shape.name,
@@ -6793,6 +6850,7 @@ var promptSessionEventSchema = z.strictObject({
   completion: customTypes.any.describe("Data about the completion"),
   tags: promptSessionEventBaseSchema.shape.tags
 }).openapi("PromptSessionEvent");
+var functionEventBaseSchema = generateBaseEventOpSchema("function");
 var projectLogsEventBaseSchema = generateBaseEventOpSchema("project");
 var projectLogsEventSchema = z.strictObject({
   id: projectLogsEventBaseSchema.shape.id,
@@ -6975,6 +7033,17 @@ var feedbackPromptRequestSchema = makeFeedbackRequestSchema(
   "prompt",
   feedbackPromptItemSchema
 );
+var feedbackFunctionRequestBaseSchema = generateBaseEventFeedbackSchema("function");
+var feedbackFunctionItemSchema = z.strictObject({
+  id: feedbackFunctionRequestBaseSchema.shape.id,
+  comment: feedbackFunctionRequestBaseSchema.shape.comment,
+  metadata: feedbackFunctionRequestBaseSchema.shape.metadata,
+  source: feedbackFunctionRequestBaseSchema.shape.source
+}).openapi("FeedbackFunctionItem");
+var feedbackFunctionRequestSchema = makeFeedbackRequestSchema(
+  "function",
+  feedbackFunctionItemSchema
+);
 var feedbackPromptSessionRequestBaseSchema = generateBaseEventFeedbackSchema("prompt_session");
 var feedbackPromptSessionItemSchema = z.strictObject({
   id: feedbackPromptSessionRequestBaseSchema.shape.id,
@@ -7025,6 +7094,14 @@ var eventObjectSchemas = {
     feedbackItem: feedbackPromptItemSchema,
     feedbackRequest: feedbackPromptRequestSchema
   },
+  function: {
+    event: functionSchema2,
+    fetchResponse: void 0,
+    insertEvent: void 0,
+    insertRequest: void 0,
+    feedbackItem: feedbackFunctionItemSchema,
+    feedbackRequest: feedbackFunctionRequestSchema
+  },
   prompt_session: {
     event: promptSessionEventSchema,
     fetchResponse: void 0,
@@ -7646,9 +7723,25 @@ var NoopSpan = class {
   close(args) {
     return this.end(args);
   }
+  setAttributes(_args) {
+  }
 };
 var NOOP_SPAN = new NoopSpan();
 var BraintrustState = class {
+  constructor(loginParams) {
+    this.loginParams = loginParams;
+    this.id = (/* @__PURE__ */ new Date()).toLocaleString();
+    this.currentExperiment = void 0;
+    this.currentLogger = void 0;
+    this.currentSpan = isomorph_default.newAsyncLocalStorage();
+    const defaultGetLogConn = async () => {
+      await this.login({});
+      return this.logConn();
+    };
+    this._bgLogger = new BackgroundLogger(new LazyValue(defaultGetLogConn));
+    this.resetLoginInfo();
+    globalThis.__inherited_braintrust_state = this;
+  }
   id;
   currentExperiment;
   // Note: the value of IsAsyncFlush doesn't really matter here, since we
@@ -7659,7 +7752,7 @@ var BraintrustState = class {
   // This is preferable to replacing the whole logger, which would create the
   // possibility of multiple loggers floating around, which may not log in a
   // deterministic order.
-  _globalBgLogger;
+  _bgLogger;
   appUrl = null;
   appPublicUrl = null;
   loginToken = null;
@@ -7670,23 +7763,9 @@ var BraintrustState = class {
   gitMetadataSettings;
   _apiConn = null;
   _logConn = null;
-  constructor() {
-    this.id = v4_default();
-    this.currentExperiment = void 0;
-    this.currentLogger = void 0;
-    this.currentSpan = isomorph_default.newAsyncLocalStorage();
-    const defaultGetLogConn = async () => {
-      await login();
-      return this.logConn();
-    };
-    this._globalBgLogger = new BackgroundLogger(
-      new LazyValue(defaultGetLogConn)
-    );
-    this.resetLoginInfo();
-    globalThis.__inherited_braintrust_state = this;
-  }
   resetLoginInfo() {
     this.appUrl = null;
+    this.appPublicUrl = null;
     this.loginToken = null;
     this.orgId = null;
     this.orgName = null;
@@ -7696,6 +7775,28 @@ var BraintrustState = class {
     this._apiConn = null;
     this._logConn = null;
   }
+  copyLoginInfo(other) {
+    this.appUrl = other.appUrl;
+    this.appPublicUrl = other.appPublicUrl;
+    this.loginToken = other.loginToken;
+    this.orgId = other.orgId;
+    this.orgName = other.orgName;
+    this.logUrl = other.logUrl;
+    this.loggedIn = other.loggedIn;
+    this.gitMetadataSettings = other.gitMetadataSettings;
+    this._apiConn = other._apiConn;
+    this._logConn = other._logConn;
+  }
+  async login(loginParams) {
+    if (this.logUrl && !loginParams.forceLogin) {
+      return;
+    }
+    const newState = await loginToState({
+      ...this.loginParams,
+      ...loginParams
+    });
+    this.copyLoginInfo(newState);
+  }
   apiConn() {
     if (!this._apiConn) {
       if (!this.appUrl) {
@@ -7714,22 +7815,24 @@ var BraintrustState = class {
     }
     return this._logConn;
   }
-  globalBgLogger() {
-    return this._globalBgLogger;
+  bgLogger() {
+    return this._bgLogger;
   }
   // Should only be called by the login function.
   loginReplaceLogConn(logConn) {
-    this._globalBgLogger.internalReplaceLogConn(logConn);
+    this._bgLogger.internalReplaceLogConn(logConn);
   }
 };
-var _state;
+var _globalState;
 function _internalSetInitialState() {
-  if (_state) {
+  if (_globalState) {
     throw new Error("Cannot set initial state more than once");
   }
-  _state = globalThis.__inherited_braintrust_state || new BraintrustState();
+  _globalState = globalThis.__inherited_braintrust_state || new BraintrustState({
+    /*empty login options*/
+  });
 }
-var _internalGetGlobalState = () => _state;
+var _internalGetGlobalState = () => _globalState;
 var FailedHTTPResponse = class extends Error {
   status;
   text;
@@ -7850,7 +7953,7 @@ var HTTPConnection = class _HTTPConnection {
     return await resp.json();
   }
 };
-function logFeedbackImpl(parentObjectType, parentObjectId, {
+function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,
   expected,
   scores,
@@ -7893,7 +7996,7 @@ function logFeedbackImpl(parentObjectType, parentObjectId, {
         [IS_MERGE_FIELD]: true
       };
     });
-    _state.globalBgLogger().log([record]);
+    state.bgLogger().log([record]);
   }
   if (!isEmpty(comment)) {
     const record = new LazyValue(async () => {
@@ -7913,10 +8016,10 @@ function logFeedbackImpl(parentObjectType, parentObjectId, {
         [AUDIT_METADATA_FIELD]: metadata
       };
     });
-    _state.globalBgLogger().log([record]);
+    state.bgLogger().log([record]);
   }
 }
-function spanComponentsToObjectIdLambda(components) {
+function spanComponentsToObjectIdLambda(state, components) {
   if (components.objectId) {
     const ret = components.objectId;
     return async () => ret;
@@ -7932,8 +8035,9 @@ function spanComponentsToObjectIdLambda(components) {
         "Impossible: computeObjectMetadataArgs not supported for experiments"
       );
     case SpanObjectTypeV2.PROJECT_LOGS:
-      const args = components.computeObjectMetadataArgs;
-      return async () => (await computeLoggerMetadata(args)).project.id;
+      return async () => (await computeLoggerMetadata(state, {
+        ...components.computeObjectMetadataArgs
+      })).project.id;
     default:
       const x = components.objectType;
       throw new Error(`Unknown object type: ${x}`);
@@ -7952,7 +8056,10 @@ function startSpanParentArgs(args) {
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.objectType}`
       );
     }
-    const parentComponentsObjectIdLambda = spanComponentsToObjectIdLambda(parentComponents);
+    const parentComponentsObjectIdLambda = spanComponentsToObjectIdLambda(
+      args.state,
+      parentComponents
+    );
     const computeParentObjectId = async () => {
       const parentComponentsObjectId = await parentComponentsObjectIdLambda();
       if (await args.parentObjectId.get() !== parentComponentsObjectId) {
@@ -7981,6 +8088,7 @@ function startSpanParentArgs(args) {
   };
 }
 var Logger = class {
+  state;
   lazyMetadata;
   _asyncFlush;
   computeMetadataArgs;
@@ -7989,13 +8097,14 @@ var Logger = class {
   calledStartSpan;
   // For type identification.
   kind = "logger";
-  constructor(lazyMetadata, logOptions = {}) {
+  constructor(state, lazyMetadata, logOptions = {}) {
     this.lazyMetadata = lazyMetadata;
     this._asyncFlush = logOptions.asyncFlush;
     this.computeMetadataArgs = logOptions.computeMetadataArgs;
     this.lastStartTime = getCurrentUnixTimestamp();
     this.lazyId = new LazyValue(async () => await this.id);
     this.calledStartSpan = false;
+    this.state = state;
   }
   get org_id() {
     return (async () => {
@@ -8087,7 +8196,9 @@ var Logger = class {
   }
   startSpanImpl(args) {
     return new SpanImpl({
+      state: this.state,
       ...startSpanParentArgs({
+        state: this.state,
         parent: args?.parent,
         parentObjectType: this.parentObjectType(),
         parentObjectId: this.lazyId,
@@ -8110,7 +8221,7 @@ var Logger = class {
    * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
    */
   logFeedback(event) {
-    logFeedbackImpl(this.parentObjectType(), this.lazyId, event);
+    logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
   /**
    * Return a serialized representation of the logger that can be used to start subspans in other places. See `Span.start_span` for more details.
@@ -8133,7 +8244,7 @@ var Logger = class {
    * Flush any pending logs to the server.
    */
   async flush() {
-    return await _state.globalBgLogger().flush();
+    return await this.state.bgLogger().flush();
   }
   get asyncFlush() {
     return this._asyncFlush;
@@ -8498,29 +8609,27 @@ function init(projectOrOptions, optionalOptions) {
     gitMetadataSettings,
     projectId,
     baseExperimentId,
-    repoInfo
+    repoInfo,
+    state: stateArg
   } = options;
   if (open && update) {
     throw new Error("Cannot open and update an experiment at the same time");
   }
+  const state = stateArg ?? _globalState;
   if (open) {
     if (isEmpty(experiment)) {
       throw new Error(`Cannot open an experiment without specifying its name`);
     }
     const lazyMetadata2 = new LazyValue(
       async () => {
-        await login({
-          orgName,
-          apiKey,
-          appUrl
-        });
+        await state.login({ apiKey, appUrl, orgName });
         const args = {
           project_name: project,
           project_id: projectId,
-          org_name: _state.orgName,
+          org_name: state.orgName,
           experiment_name: experiment
         };
-        const response = await _state.apiConn().post_json("api/experiment/get", args);
+        const response = await state.apiConn().post_json("api/experiment/get", args);
         if (response.length === 0) {
           throw new Error(
             `Experiment ${experiment} not found in project ${projectId ?? project}.`
@@ -8542,20 +8651,17 @@ function init(projectOrOptions, optionalOptions) {
       }
     );
     return new ReadonlyExperiment(
+      stateArg ?? _globalState,
       lazyMetadata2
     );
   }
   const lazyMetadata = new LazyValue(
     async () => {
-      await login({
-        orgName,
-        apiKey,
-        appUrl
-      });
+      await state.login({ apiKey, appUrl, orgName });
       const args = {
         project_name: project,
         project_id: projectId,
-        org_id: _state.orgId,
+        org_id: state.orgId,
         update
       };
       if (experiment) {
@@ -8569,7 +8675,7 @@ function init(projectOrOptions, optionalOptions) {
           return repoInfo;
         }
         let mergedGitMetadataSettings = {
-          ..._state.gitMetadataSettings || {
+          ...state.gitMetadataSettings || {
             collect: "all"
           }
         };
@@ -8604,7 +8710,7 @@ function init(projectOrOptions, optionalOptions) {
       let response = null;
       while (true) {
         try {
-          response = await _state.apiConn().post_json("api/experiment/register", args);
+          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")) {
@@ -8631,9 +8737,9 @@ function init(projectOrOptions, optionalOptions) {
       };
     }
   );
-  const ret = new Experiment(lazyMetadata, dataset);
+  const ret = new Experiment(state, lazyMetadata, dataset);
   if (options.setCurrent ?? true) {
-    _state.currentExperiment = ret;
+    state.currentExperiment = ret;
   }
   return ret;
 }
@@ -8688,23 +8794,25 @@ function initDataset(projectOrOptions, optionalOptions) {
     apiKey,
     orgName,
     projectId,
-    useOutput: legacy
+    useOutput: legacy,
+    state: stateArg
   } = options;
+  const state = stateArg ?? _globalState;
   const lazyMetadata = new LazyValue(
     async () => {
-      await login({
+      await state.login({
         orgName,
         apiKey,
         appUrl
       });
       const args = {
-        org_id: _state.orgId,
+        org_id: state.orgId,
         project_name: project,
         project_id: projectId,
         dataset_name: dataset,
         description
       };
-      const response = await _state.apiConn().post_json("api/dataset/register", args);
+      const response = await state.apiConn().post_json("api/dataset/register", args);
       return {
         project: {
           id: response.project.id,
@@ -8719,7 +8827,7 @@ function initDataset(projectOrOptions, optionalOptions) {
       };
     }
   );
-  return new Dataset(lazyMetadata, version, legacy);
+  return new Dataset(stateArg ?? _globalState, lazyMetadata, version, legacy);
 }
 function withDataset(project, callback, options = {}) {
   console.warn(
@@ -8728,14 +8836,13 @@ function withDataset(project, callback, options = {}) {
   const dataset = initDataset(project, options);
   return callback(dataset);
 }
-async function computeLoggerMetadata({
+async function computeLoggerMetadata(state, {
   project_name,
   project_id
 }) {
-  await login();
-  const org_id = _state.orgId;
-  if (project_id === void 0) {
-    const response = await _state.apiConn().post_json("api/project/register", {
+  const org_id = state.orgId;
+  if (isEmpty(project_id)) {
+    const response = await state.apiConn().post_json("api/project/register", {
       project_name: project_name || GLOBAL_PROJECT,
       org_id
     });
@@ -8747,8 +8854,8 @@ async function computeLoggerMetadata({
         fullInfo: response.project
       }
     };
-  } else if (project_name === void 0) {
-    const response = await _state.apiConn().get_json("api/project", {
+  } else if (isEmpty(project_name)) {
+    const response = await state.apiConn().get_json("api/project", {
       id: project_id
     });
     return {
@@ -8774,29 +8881,31 @@ function initLogger(options = {}) {
     appUrl,
     apiKey,
     orgName,
-    forceLogin
+    forceLogin,
+    state: stateArg
   } = options || {};
   const computeMetadataArgs = {
     project_name: projectName,
     project_id: projectId
   };
+  const state = stateArg ?? _globalState;
   const lazyMetadata = new LazyValue(
     async () => {
-      await login({
+      await state.login({
         orgName,
         apiKey,
         appUrl,
         forceLogin
       });
-      return computeLoggerMetadata(computeMetadataArgs);
+      return computeLoggerMetadata(state, computeMetadataArgs);
     }
   );
-  const ret = new Logger(lazyMetadata, {
+  const ret = new Logger(state, lazyMetadata, {
     asyncFlush,
     computeMetadataArgs
   });
   if (options.setCurrent ?? true) {
-    _state.currentLogger = ret;
+    state.currentLogger = ret;
   }
   return ret;
 }
@@ -8809,7 +8918,8 @@ async function loadPrompt({
   noTrace = false,
   appUrl,
   apiKey,
-  orgName
+  orgName,
+  state: stateArg
 }) {
   if (isEmpty(projectName) && isEmpty(projectId)) {
     throw new Error("Must specify either projectName or projectId");
@@ -8817,7 +8927,8 @@ async function loadPrompt({
   if (isEmpty(slug)) {
     throw new Error("Must specify slug");
   }
-  await login({
+  const state = stateArg ?? _globalState;
+  state.login({
     orgName,
     apiKey,
     appUrl
@@ -8828,7 +8939,7 @@ async function loadPrompt({
     slug,
     version
   };
-  const response = await _state.logConn().get_json("v1/prompt", args);
+  const response = await state.logConn().get_json("v1/prompt", args);
   if (!("objects" in response) || response.objects.length === 0) {
     throw new Error(
       `Prompt ${slug} not found in ${[projectName ?? projectId]}`
@@ -8843,7 +8954,7 @@ async function loadPrompt({
 }
 async function login(options = {}) {
   let { forceLogin = false } = options || {};
-  if (_state.loggedIn && !forceLogin) {
+  if (_globalState.loggedIn && !forceLogin) {
     let checkUpdatedParam2 = function(varname, arg, orig) {
       if (!isEmpty(arg) && !isEmpty(orig) && arg !== orig) {
         throw new Error(
@@ -8852,28 +8963,32 @@ async function login(options = {}) {
       }
     };
     var checkUpdatedParam = checkUpdatedParam2;
-    checkUpdatedParam2("appUrl", options.appUrl, _state.appUrl);
+    checkUpdatedParam2("appUrl", options.appUrl, _globalState.appUrl);
     checkUpdatedParam2(
       "apiKey",
       options.apiKey ? HTTPConnection.sanitize_token(options.apiKey) : void 0,
-      _state.loginToken
+      _globalState.loginToken
     );
-    checkUpdatedParam2("orgName", options.orgName, _state.orgName);
-    return;
+    checkUpdatedParam2("orgName", options.orgName, _globalState.orgName);
+    return _globalState;
   }
+  await _globalState.login(options);
+}
+async function loginToState(options = {}) {
   const {
     appUrl = isomorph_default.getEnv("BRAINTRUST_APP_URL") || "https://www.braintrust.dev",
     apiKey = isomorph_default.getEnv("BRAINTRUST_API_KEY"),
     orgName = isomorph_default.getEnv("BRAINTRUST_ORG_NAME")
   } = options || {};
   const appPublicUrl = isomorph_default.getEnv("BRAINTRUST_APP_PUBLIC_URL") || appUrl;
-  _state.resetLoginInfo();
-  _state.appUrl = appUrl;
-  _state.appPublicUrl = appPublicUrl;
+  const state = new BraintrustState(options);
+  state.resetLoginInfo();
+  state.appUrl = appUrl;
+  state.appPublicUrl = appPublicUrl;
   let conn = null;
   if (apiKey !== void 0) {
     const resp = await checkResponse(
-      await fetch(_urljoin(_state.appUrl, `/api/apikey/login`), {
+      await fetch(_urljoin(state.appUrl, `/api/apikey/login`), {
         method: "POST",
         headers: {
           "Content-Type": "application/json"
@@ -8884,8 +8999,8 @@ async function login(options = {}) {
       })
     );
     const info = await resp.json();
-    _check_org_info(info.org_info, orgName);
-    conn = _state.logConn();
+    _check_org_info(state, info.org_info, orgName);
+    conn = state.logConn();
     conn.set_token(apiKey);
   } else {
     throw new Error(
@@ -8896,10 +9011,11 @@ async function login(options = {}) {
     throw new Error("Conn should be set at this point (a bug)");
   }
   conn.make_long_lived();
-  _state.apiConn().set_token(apiKey);
-  _state.loginToken = conn.token;
-  _state.loggedIn = true;
-  _state.loginReplaceLogConn(conn);
+  state.apiConn().set_token(apiKey);
+  state.loginToken = conn.token;
+  state.loggedIn = true;
+  state.loginReplaceLogConn(conn);
+  return state;
 }
 function log(event) {
   console.warn(
@@ -8921,17 +9037,21 @@ async function summarize(options = {}) {
   }
   return await e.summarize(options);
 }
-function currentExperiment() {
-  return _state.currentExperiment;
+function currentExperiment(options) {
+  const state = options?.state ?? _globalState;
+  return state.currentExperiment;
 }
 function currentLogger(options) {
-  return castLogger(_state.currentLogger, options?.asyncFlush);
+  const state = options?.state ?? _globalState;
+  return castLogger(state.currentLogger, options?.asyncFlush);
 }
-function currentSpan() {
-  return _state.currentSpan.getStore() ?? NOOP_SPAN;
+function currentSpan(options) {
+  const state = options?.state ?? _globalState;
+  return state.currentSpan.getStore() ?? NOOP_SPAN;
 }
 function getSpanParentObject(options) {
-  const parentSpan = currentSpan();
+  const state = options?.state ?? _globalState;
+  const parentSpan = currentSpan({ state });
   if (!Object.is(parentSpan, NOOP_SPAN)) {
     return parentSpan;
   }
@@ -8972,10 +9092,12 @@ function traced(callback, args) {
 function startSpan(args) {
   return startSpanAndIsLogger(args).span;
 }
-async function flush() {
-  return await _state.globalBgLogger().flush();
+async function flush(options) {
+  const state = options?.state ?? _globalState;
+  return await state.bgLogger().flush();
 }
 function startSpanAndIsLogger(args) {
+  const state = args?.state ?? _globalState;
   if (args?.parent) {
     const components = SpanComponentsV2.fromStr(args?.parent);
     const parentSpanIds = components.rowIds ? {
@@ -8983,9 +9105,12 @@ function startSpanAndIsLogger(args) {
       rootSpanId: components.rowIds.rootSpanId
     } : void 0;
     const span = new SpanImpl({
+      state,
       ...args,
       parentObjectType: components.objectType,
-      parentObjectId: new LazyValue(spanComponentsToObjectIdLambda(components)),
+      parentObjectId: new LazyValue(
+        spanComponentsToObjectIdLambda(state, components)
+      ),
       parentComputeObjectMetadataArgs: components.computeObjectMetadataArgs,
       parentSpanIds
     });
@@ -9001,23 +9126,23 @@ function startSpanAndIsLogger(args) {
     return { span, isLogger: parentObject.kind === "logger" };
   }
 }
-function withCurrent(span, callback) {
-  return _state.currentSpan.run(span, () => callback(span));
+function withCurrent(span, callback, state = _globalState) {
+  return state.currentSpan.run(span, () => callback(span));
 }
-function _check_org_info(org_info, org_name) {
+function _check_org_info(state, org_info, org_name) {
   if (org_info.length === 0) {
     throw new Error("This user is not part of any organizations.");
   }
   for (const org of org_info) {
     if (org_name === void 0 || org.name === org_name) {
-      _state.orgId = org.id;
-      _state.orgName = org.name;
-      _state.logUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
-      _state.gitMetadataSettings = org.git_metadata || void 0;
+      state.orgId = org.id;
+      state.orgName = org.name;
+      state.logUrl = isomorph_default.getEnv("BRAINTRUST_API_URL") ?? org.api_url;
+      state.gitMetadataSettings = org.git_metadata || void 0;
       break;
     }
   }
-  if (_state.orgId === void 0) {
+  if (state.orgId === void 0) {
     throw new Error(
       `Organization ${org_name} not found. Must be one of ${org_info.map((x) => x.name).join(", ")}`
     );
@@ -9176,15 +9301,17 @@ var Experiment = class extends ObjectFetcher {
   lastStartTime;
   lazyId;
   calledStartSpan;
+  state;
   // For type identification.
   kind = "experiment";
-  constructor(lazyMetadata, dataset) {
+  constructor(state, lazyMetadata, dataset) {
     super("experiment", void 0);
     this.lazyMetadata = lazyMetadata;
     this.dataset = dataset;
     this.lastStartTime = getCurrentUnixTimestamp();
     this.lazyId = new LazyValue(async () => await this.id);
     this.calledStartSpan = false;
+    this.state = state;
   }
   get id() {
     return (async () => {
@@ -9206,7 +9333,7 @@ var Experiment = class extends ObjectFetcher {
   }
   async getState() {
     await this.lazyMetadata.get();
-    return _state;
+    return this.state;
   }
   /**
    * Log a single event to the experiment. The event will be batched and uploaded behind the scenes.
@@ -9268,7 +9395,9 @@ var Experiment = class extends ObjectFetcher {
   }
   startSpanImpl(args) {
     return new SpanImpl({
+      state: this.state,
       ...startSpanParentArgs({
+        state: this.state,
         parent: args?.parent,
         parentObjectType: this.parentObjectType(),
         parentObjectId: this.lazyId,
@@ -9362,7 +9491,7 @@ var Experiment = class extends ObjectFetcher {
    * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
    */
   logFeedback(event) {
-    logFeedbackImpl(this.parentObjectType(), this.lazyId, event);
+    logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
   /**
    * Return a serialized representation of the experiment that can be used to start subspans in other places. See `Span.start_span` for more details.
@@ -9377,7 +9506,7 @@ var Experiment = class extends ObjectFetcher {
    * Flush any pending rows to the server.
    */
   async flush() {
-    return await _state.globalBgLogger().flush();
+    return await this.state.bgLogger().flush();
   }
   /**
    * This function is deprecated. You can simply remove it from your code.
@@ -9390,8 +9519,9 @@ var Experiment = class extends ObjectFetcher {
   }
 };
 var ReadonlyExperiment = class extends ObjectFetcher {
-  constructor(lazyMetadata) {
+  constructor(state, lazyMetadata) {
     super("experiment", void 0);
+    this.state = state;
     this.lazyMetadata = lazyMetadata;
   }
   get id() {
@@ -9406,7 +9536,7 @@ var ReadonlyExperiment = class extends ObjectFetcher {
   }
   async getState() {
     await this.lazyMetadata.get();
-    return _state;
+    return this.state;
   }
   async *asDataset() {
     const records = this.fetch();
@@ -9432,10 +9562,13 @@ var ReadonlyExperiment = class extends ObjectFetcher {
   }
 };
 var executionCounter = 0;
+function newId() {
+  return v4_default();
+}
 var SpanImpl = class _SpanImpl {
+  state;
   // `internalData` contains fields that are not part of the "user-sanitized"
   // set of fields which we want to log in just one of the span rows.
-  internalData;
   isMerge;
   loggedEndTime;
   // For internal use only.
@@ -9448,6 +9581,7 @@ var SpanImpl = class _SpanImpl {
   spanParents;
   kind = "span";
   constructor(args) {
+    this.state = args.state;
     const spanAttributes = args.spanAttributes ?? {};
     const event = args.event ?? {};
     const type = args.type ?? (args.parentSpanIds ? void 0 : args.defaultRootType);
@@ -9470,7 +9604,7 @@ var SpanImpl = class _SpanImpl {
       }
       return "subspan";
     })();
-    this.internalData = {
+    const internalData = {
       metrics: {
         start: args.startTime ?? getCurrentUnixTimestamp()
       },
@@ -9494,17 +9628,25 @@ var SpanImpl = class _SpanImpl {
     }
     this.isMerge = false;
     const { id: _id, ...eventRest } = event;
-    this.log(eventRest);
+    this.logInternal({ event: eventRest, internalData });
     this.isMerge = true;
   }
   get id() {
     return this._id;
   }
+  setAttributes(args) {
+    this.logInternal({ internalData: { span_attributes: args } });
+  }
   log(event) {
-    const sanitized = validateAndSanitizeExperimentLogPartialArgs(event);
-    let sanitizedAndInternalData = { ...this.internalData };
+    this.logInternal({ event });
+  }
+  logInternal({
+    event,
+    internalData
+  }) {
+    const sanitized = validateAndSanitizeExperimentLogPartialArgs(event ?? {});
+    let sanitizedAndInternalData = { ...internalData };
     mergeDicts(sanitizedAndInternalData, sanitized);
-    this.internalData = {};
     let partialRecord = {
       id: this.id,
       span_id: this.spanId,
@@ -9528,10 +9670,10 @@ var SpanImpl = class _SpanImpl {
         objectId: await this.parentObjectId.get()
       }).objectIdFields()
     });
-    _state.globalBgLogger().log([new LazyValue(computeRecord)]);
+    this.state.bgLogger().log([new LazyValue(computeRecord)]);
   }
   logFeedback(event) {
-    logFeedbackImpl(this.parentObjectType, this.parentObjectId, {
+    logFeedbackImpl(this.state, this.parentObjectType, this.parentObjectId, {
       ...event,
       id: this.id
     });
@@ -9553,8 +9695,10 @@ var SpanImpl = class _SpanImpl {
   startSpan(args) {
     const parentSpanIds = args?.parent ? void 0 : { spanId: this.spanId, rootSpanId: this.rootSpanId };
     return new _SpanImpl({
+      state: this.state,
       ...args,
       ...startSpanParentArgs({
+        state: this.state,
         parent: args?.parent,
         parentObjectType: this.parentObjectType,
         parentObjectId: this.parentObjectId,
@@ -9565,13 +9709,14 @@ var SpanImpl = class _SpanImpl {
   }
   end(args) {
     let endTime;
+    let internalData = {};
     if (!this.loggedEndTime) {
       endTime = args?.endTime ?? getCurrentUnixTimestamp();
-      this.internalData = { metrics: { end: endTime } };
+      internalData = { metrics: { end: endTime } };
     } else {
       endTime = this.loggedEndTime;
     }
-    this.log({});
+    this.logInternal({ internalData });
     return endTime;
   }
   async export() {
@@ -9594,15 +9739,14 @@ var SpanImpl = class _SpanImpl {
     }).toStr();
   }
   async flush() {
-    return await _state.globalBgLogger().flush();
+    return await this.state.bgLogger().flush();
   }
   close(args) {
     return this.end(args);
   }
 };
 var Dataset = class extends ObjectFetcher {
-  lazyMetadata;
-  constructor(lazyMetadata, pinnedVersion, legacy) {
+  constructor(state, lazyMetadata, pinnedVersion, legacy) {
     const isLegacyDataset = legacy ?? DEFAULT_IS_LEGACY_DATASET;
     if (isLegacyDataset) {
       console.warn(
@@ -9614,8 +9758,10 @@ var Dataset = class extends ObjectFetcher {
       pinnedVersion,
       (r) => ensureDatasetRecord(r, isLegacyDataset)
     );
+    this.state = state;
     this.lazyMetadata = lazyMetadata;
   }
+  lazyMetadata;
   get id() {
     return (async () => {
       return (await this.lazyMetadata.get()).dataset.id;
@@ -9633,7 +9779,7 @@ var Dataset = class extends ObjectFetcher {
   }
   async getState() {
     await this.lazyMetadata.get();
-    return _state;
+    return this.state;
   }
   /**
    * Insert a single record to the dataset. The record will be batched and uploaded behind the scenes. If you pass in an `id`,
@@ -9684,7 +9830,7 @@ var Dataset = class extends ObjectFetcher {
       created: (/* @__PURE__ */ new Date()).toISOString(),
       metadata
     }));
-    _state.globalBgLogger().log([args]);
+    this.state.bgLogger().log([args]);
     return rowId;
   }
   delete(id) {
@@ -9694,7 +9840,7 @@ var Dataset = class extends ObjectFetcher {
       created: (/* @__PURE__ */ new Date()).toISOString(),
       _object_delete: true
     }));
-    _state.globalBgLogger().log([args]);
+    this.state.bgLogger().log([args]);
     return id;
   }
   /**
@@ -9735,7 +9881,7 @@ var Dataset = class extends ObjectFetcher {
    * Flush any pending rows to the server.
    */
   async flush() {
-    return await _state.globalBgLogger().flush();
+    return await this.state.bgLogger().flush();
   }
   /**
    * This function is deprecated. You can simply remove it from your code.
@@ -9820,13 +9966,18 @@ var Prompt = class {
     if (!prompt) {
       throw new Error("Empty prompt");
     }
+    const dictArgParsed = z.record(z.unknown()).safeParse(buildArgs);
+    const variables = {
+      input: buildArgs,
+      ...dictArgParsed.success ? dictArgParsed.data : {}
+    };
     if (flavor === "chat") {
       if (prompt.type !== "chat") {
         throw new Error(
           "Prompt is a completion prompt. Use buildCompletion() instead"
         );
       }
-      const render3 = (template) => mustache_default.render(template, buildArgs, void 0, {
+      const render3 = (template) => mustache_default.render(template, variables, void 0, {
         escape: (v) => typeof v === "string" ? v : JSON.stringify(v)
       });
       const messages = (prompt.messages || []).map((m) => ({
@@ -9841,7 +9992,7 @@ var Prompt = class {
         messages,
         ...prompt.tools ? {
           tools: toolsSchema.parse(
-            JSON.parse(mustache_default.render(prompt.tools, buildArgs))
+            JSON.parse(mustache_default.render(prompt.tools, variables))
           )
         } : void 0
       };
@@ -9852,7 +10003,7 @@ var Prompt = class {
       return {
         ...params,
         ...spanInfo,
-        prompt: mustache_default.render(prompt.content, buildArgs)
+        prompt: mustache_default.render(prompt.content, variables)
       };
     } else {
       throw new Error("never!");
@@ -10002,20 +10153,29 @@ function wrapBetaChatCompletion(completion) {
     return ret;
   };
 }
-var X_CACHED_HEADER = "x-cached";
+var LEGACY_CACHED_HEADER = "x-cached";
+var X_CACHED_HEADER = "x-bt-cached";
 function parseCachedHeader(value) {
-  return isEmpty(value) ? void 0 : value.toLowerCase() === "true" ? 1 : 0;
+  return isEmpty(value) ? void 0 : ["true", "hit"].includes(value.toLowerCase()) ? 1 : 0;
 }
 function logHeaders(response, span) {
   const cachedHeader = response.headers.get(X_CACHED_HEADER);
   if (isEmpty(cachedHeader)) {
-    return;
-  }
-  span.log({
-    metrics: {
-      cached: parseCachedHeader(cachedHeader)
+    const legacyCacheHeader = response.headers.get(LEGACY_CACHED_HEADER);
+    if (!isEmpty(legacyCacheHeader)) {
+      span.log({
+        metrics: {
+          cached: parseCachedHeader(legacyCacheHeader)
+        }
+      });
     }
-  });
+  } else {
+    span.log({
+      metrics: {
+        cached: parseCachedHeader(cachedHeader)
+      }
+    });
+  }
 }
 function wrapChatCompletion(completion) {
   return async (allParams, options) => {
@@ -10135,7 +10295,16 @@ function postprocessStreamingResults(allResults) {
   let content = void 0;
   let tool_calls = void 0;
   let finish_reason = void 0;
+  let metrics = {};
   for (const result of allResults) {
+    if (result.usage) {
+      metrics = {
+        ...metrics,
+        tokens: result.usage.total_tokens,
+        prompt_tokens: result.usage.prompt_tokens,
+        completion_tokens: result.usage.completion_tokens
+      };
+    }
     const delta = result.choices?.[0]?.delta;
     if (!delta) {
       continue;
@@ -10163,18 +10332,21 @@ function postprocessStreamingResults(allResults) {
       }
     }
   }
-  return [
-    {
-      index: 0,
-      message: {
-        role,
-        content,
-        tool_calls
-      },
-      logprobs: null,
-      finish_reason
-    }
-  ];
+  return {
+    metrics,
+    output: [
+      {
+        index: 0,
+        message: {
+          role,
+          content,
+          tool_calls
+        },
+        logprobs: null,
+        finish_reason
+      }
+    ]
+  };
 }
 var WrapperStream = class {
   span;
@@ -10203,7 +10375,7 @@ var WrapperStream = class {
         yield item;
       }
       this.span.log({
-        output: postprocessStreamingResults(allResults)
+        ...postprocessStreamingResults(allResults)
       });
     } finally {
       this.span.end();
@@ -10214,8 +10386,10 @@ var WrapperStream = class {
 // src/browser.ts
 configureBrowser();
 export {
+  BraintrustState,
   Dataset,
   Experiment,
+  LEGACY_CACHED_HEADER,
   Logger,
   NOOP_SPAN,
   NoopSpan,
@@ -10237,6 +10411,8 @@ export {
   loadPrompt,
   log,
   login,
+  loginToState,
+  newId,
   parseCachedHeader,
   startSpan,
   summarize,