braintrust 1.0.0 → 1.0.2

package/dist/index.js

@@ -6476,7 +6476,7 @@ function validateAndSanitizeExperimentLogFullArgs(event, hasDataset) {
   }
   return event;
 }
-var INTERNAL_BTQL_LIMIT = 1e3;
+var DEFAULT_FETCH_BATCH_SIZE = 1e3;
 var MAX_BTQL_ITERATIONS = 1e4;
 var ObjectFetcher = (_class9 = class {
   constructor(objectType, pinnedVersion, mutateRecord, _internal_btql) {;_class9.prototype.__init40.call(this);
@@ -6492,9 +6492,10 @@ var ObjectFetcher = (_class9 = class {
   async getState() {
     throw new Error("ObjectFetcher subclasses must have a 'getState' method");
   }
-  async *fetchRecordsFromApi() {
+  async *fetchRecordsFromApi(batchSize) {
     const state = await this.getState();
     const objectId = await this.id;
+    const limit = _nullishCoalesce(batchSize, () => ( DEFAULT_FETCH_BATCH_SIZE));
     let cursor = void 0;
     let iterations = 0;
     while (true) {
@@ -6522,7 +6523,7 @@ var ObjectFetcher = (_class9 = class {
               ]
             },
             cursor,
-            limit: INTERNAL_BTQL_LIMIT
+            limit
           },
           use_columnstore: false,
           brainstore_realtime: true,
@@ -6548,24 +6549,31 @@ var ObjectFetcher = (_class9 = class {
       }
     }
   }
-  async *fetch() {
+  /**
+   * Fetch all records from the object.
+   *
+   * @param options Optional parameters for fetching.
+   * @param options.batchSize The number of records to fetch per request. Defaults to 1000.
+   * @returns An async generator of records.
+   */
+  async *fetch(options) {
     if (this._fetchedData !== void 0) {
       for (const record of this._fetchedData) {
         yield record;
       }
       return;
     }
-    for await (const record of this.fetchRecordsFromApi()) {
+    for await (const record of this.fetchRecordsFromApi(_optionalChain([options, 'optionalAccess', _64 => _64.batchSize]))) {
       yield record;
     }
   }
   [Symbol.asyncIterator]() {
     return this.fetch();
   }
-  async fetchedData() {
+  async fetchedData(options) {
     if (this._fetchedData === void 0) {
       const data = [];
-      for await (const record of this.fetchRecordsFromApi()) {
+      for await (const record of this.fetchRecordsFromApi(_optionalChain([options, 'optionalAccess', _65 => _65.batchSize]))) {
         data.push(record);
       }
       this._fetchedData = data;
@@ -6575,12 +6583,12 @@ var ObjectFetcher = (_class9 = class {
   clearCache() {
     this._fetchedData = void 0;
   }
-  async version() {
+  async version(options) {
     if (this.pinnedVersion !== void 0) {
       return this.pinnedVersion;
     } else {
       let maxVersion = void 0;
-      for await (const record of this.fetch()) {
+      for await (const record of this.fetch(options)) {
         const xactId = String(_nullishCoalesce(record[TRANSACTION_ID_FIELD], () => ( "0")));
         if (maxVersion === void 0 || xactId > maxVersion) {
           maxVersion = xactId;
@@ -6660,7 +6668,7 @@ var Experiment2 = (_class10 = class extends ObjectFetcher {
    * @returns The `id` of the logged event.
    */
   log(event, options) {
-    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _64 => _64.allowConcurrentWithSpans])) {
+    if (this.calledStartSpan && !_optionalChain([options, 'optionalAccess', _66 => _66.allowConcurrentWithSpans])) {
       throw new Error(
         "Cannot run toplevel `log` method while using spans. To log to the span, call `experiment.traced` and then log with `span.log`"
       );
@@ -6713,12 +6721,12 @@ var Experiment2 = (_class10 = class extends ObjectFetcher {
       state: this.state,
       ...startSpanParentArgs({
         state: this.state,
-        parent: _optionalChain([args, 'optionalAccess', _65 => _65.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _67 => _67.parent]),
         parentObjectType: this.parentObjectType(),
         parentObjectId: this.lazyId,
         parentComputeObjectMetadataArgs: void 0,
         parentSpanIds: void 0,
-        propagatedEvent: _optionalChain([args, 'optionalAccess', _66 => _66.propagatedEvent])
+        propagatedEvent: _optionalChain([args, 'optionalAccess', _68 => _68.propagatedEvent])
       }),
       defaultRootType: "eval" /* EVAL */
     });
@@ -6888,8 +6896,8 @@ var ReadonlyExperiment = class extends ObjectFetcher {
     await this.lazyMetadata.get();
     return this.state;
   }
-  async *asDataset() {
-    const records = this.fetch();
+  async *asDataset(options) {
+    const records = this.fetch(options);
     for await (const record of records) {
       if (record.root_span_id !== record.span_id) {
         continue;
@@ -7056,10 +7064,10 @@ var SpanImpl = (_class11 = class _SpanImpl {
       ...serializableInternalData,
       [IS_MERGE_FIELD]: this.isMerge
     });
-    if (_optionalChain([partialRecord, 'access', _67 => _67.metrics, 'optionalAccess', _68 => _68.end])) {
-      this.loggedEndTime = _optionalChain([partialRecord, 'access', _69 => _69.metrics, 'optionalAccess', _70 => _70.end]);
+    if (_optionalChain([partialRecord, 'access', _69 => _69.metrics, 'optionalAccess', _70 => _70.end])) {
+      this.loggedEndTime = _optionalChain([partialRecord, 'access', _71 => _71.metrics, 'optionalAccess', _72 => _72.end]);
     }
-    if ((_nullishCoalesce(partialRecord.tags, () => ( []))).length > 0 && _optionalChain([this, 'access', _71 => _71._spanParents, 'optionalAccess', _72 => _72.length])) {
+    if ((_nullishCoalesce(partialRecord.tags, () => ( []))).length > 0 && _optionalChain([this, 'access', _73 => _73._spanParents, 'optionalAccess', _74 => _74.length])) {
       throw new Error("Tags can only be logged to the root span");
     }
     const computeRecord = async () => ({
@@ -7104,18 +7112,18 @@ var SpanImpl = (_class11 = class _SpanImpl {
     );
   }
   startSpan(args) {
-    const parentSpanIds = _optionalChain([args, 'optionalAccess', _73 => _73.parent]) ? void 0 : { spanId: this._spanId, rootSpanId: this._rootSpanId };
+    const parentSpanIds = _optionalChain([args, 'optionalAccess', _75 => _75.parent]) ? void 0 : { spanId: this._spanId, rootSpanId: this._rootSpanId };
     return new _SpanImpl({
       state: this._state,
       ...args,
       ...startSpanParentArgs({
         state: this._state,
-        parent: _optionalChain([args, 'optionalAccess', _74 => _74.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _76 => _76.parent]),
         parentObjectType: this.parentObjectType,
         parentObjectId: this.parentObjectId,
         parentComputeObjectMetadataArgs: this.parentComputeObjectMetadataArgs,
         parentSpanIds,
-        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _75 => _75.propagatedEvent]), () => ( this.propagatedEvent))
+        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _77 => _77.propagatedEvent]), () => ( this.propagatedEvent))
       })
     });
   }
@@ -7129,12 +7137,12 @@ var SpanImpl = (_class11 = class _SpanImpl {
       ...args,
       ...startSpanParentArgs({
         state: this._state,
-        parent: _optionalChain([args, 'optionalAccess', _76 => _76.parent]),
+        parent: _optionalChain([args, 'optionalAccess', _78 => _78.parent]),
         parentObjectType: this.parentObjectType,
         parentObjectId: this.parentObjectId,
         parentComputeObjectMetadataArgs: this.parentComputeObjectMetadataArgs,
         parentSpanIds,
-        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _77 => _77.propagatedEvent]), () => ( this.propagatedEvent))
+        propagatedEvent: _nullishCoalesce(_optionalChain([args, 'optionalAccess', _79 => _79.propagatedEvent]), () => ( this.propagatedEvent))
       }),
       spanId
     });
@@ -7143,7 +7151,7 @@ var SpanImpl = (_class11 = class _SpanImpl {
     let endTime;
     let internalData = {};
     if (!this.loggedEndTime) {
-      endTime = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _78 => _78.endTime]), () => ( getCurrentUnixTimestamp()));
+      endTime = _nullishCoalesce(_optionalChain([args, 'optionalAccess', _80 => _80.endTime]), () => ( getCurrentUnixTimestamp()));
       internalData = { metrics: { end: endTime } };
     } else {
       endTime = this.loggedEndTime;
@@ -7186,8 +7194,8 @@ var SpanImpl = (_class11 = class _SpanImpl {
     const args = this.parentComputeObjectMetadataArgs;
     switch (this.parentObjectType) {
       case 2 /* PROJECT_LOGS */: {
-        const projectID = _optionalChain([args, 'optionalAccess', _79 => _79.project_id]) || this.parentObjectId.getSync().value;
-        const projectName = _optionalChain([args, 'optionalAccess', _80 => _80.project_name]);
+        const projectID = _optionalChain([args, 'optionalAccess', _81 => _81.project_id]) || this.parentObjectId.getSync().value;
+        const projectName = _optionalChain([args, 'optionalAccess', _82 => _82.project_name]);
         if (projectID) {
           return `${baseUrl}/object?object_type=project_logs&object_id=${projectID}&id=${this._id}`;
         } else if (projectName) {
@@ -7197,7 +7205,7 @@ var SpanImpl = (_class11 = class _SpanImpl {
         }
       }
       case 1 /* EXPERIMENT */: {
-        const expID = _optionalChain([args, 'optionalAccess', _81 => _81.experiment_id]) || _optionalChain([this, 'access', _82 => _82.parentObjectId, 'optionalAccess', _83 => _83.getSync, 'call', _84 => _84(), 'optionalAccess', _85 => _85.value]);
+        const expID = _optionalChain([args, 'optionalAccess', _83 => _83.experiment_id]) || _optionalChain([this, 'access', _84 => _84.parentObjectId, 'optionalAccess', _85 => _85.getSync, 'call', _86 => _86(), 'optionalAccess', _87 => _87.value]);
         if (!expID) {
           return getErrPermlink("provide-experiment-id");
         } else {
@@ -7653,13 +7661,13 @@ var Prompt2 = (_class13 = class _Prompt {
     return "slug" in this.metadata ? this.metadata.slug : this.metadata.id;
   }
   get prompt() {
-    return _optionalChain([this, 'access', _86 => _86.getParsedPromptData, 'call', _87 => _87(), 'optionalAccess', _88 => _88.prompt]);
+    return _optionalChain([this, 'access', _88 => _88.getParsedPromptData, 'call', _89 => _89(), 'optionalAccess', _90 => _90.prompt]);
   }
   get version() {
     return this.metadata[TRANSACTION_ID_FIELD];
   }
   get options() {
-    return _optionalChain([this, 'access', _89 => _89.getParsedPromptData, 'call', _90 => _90(), 'optionalAccess', _91 => _91.options]) || {};
+    return _optionalChain([this, 'access', _91 => _91.getParsedPromptData, 'call', _92 => _92(), 'optionalAccess', _93 => _93.options]) || {};
   }
   get promptData() {
     return this.getParsedPromptData();
@@ -7810,7 +7818,7 @@ var Prompt2 = (_class13 = class _Prompt {
       return {
         type: "chat",
         messages,
-        ..._optionalChain([prompt, 'access', _92 => _92.tools, 'optionalAccess', _93 => _93.trim, 'call', _94 => _94()]) ? {
+        ..._optionalChain([prompt, 'access', _94 => _94.tools, 'optionalAccess', _95 => _95.trim, 'call', _96 => _96()]) ? {
           tools: render(prompt.tools)
         } : void 0
       };
@@ -7920,9 +7928,9 @@ async function getPromptVersions(projectId, promptId) {
     );
   }
   const result = await response.json();
-  return _optionalChain([result, 'access', _95 => _95.data, 'optionalAccess', _96 => _96.filter, 'call', _97 => _97(
-    (entry) => ["upsert", "merge"].includes(_optionalChain([entry, 'access', _98 => _98.audit_data, 'optionalAccess', _99 => _99.action]))
-  ), 'access', _100 => _100.map, 'call', _101 => _101((entry) => prettifyXact(entry._xact_id))]) || [];
+  return _optionalChain([result, 'access', _97 => _97.data, 'optionalAccess', _98 => _98.filter, 'call', _99 => _99(
+    (entry) => ["upsert", "merge"].includes(_optionalChain([entry, 'access', _100 => _100.audit_data, 'optionalAccess', _101 => _101.action]))
+  ), 'access', _102 => _102.map, 'call', _103 => _103((entry) => prettifyXact(entry._xact_id))]) || [];
 }
 function resetIdGenStateForTests() {
   const state = _internalGetGlobalState();
@@ -7985,6 +7993,7 @@ __export(exports_node_exports, {
   CodeFunction: () => CodeFunction,
   CodePrompt: () => CodePrompt,
   ContextManager: () => ContextManager,
+  DEFAULT_FETCH_BATCH_SIZE: () => DEFAULT_FETCH_BATCH_SIZE,
   Dataset: () => Dataset2,
   ERR_PERMALINK: () => ERR_PERMALINK,
   Eval: () => Eval,
@@ -7993,7 +8002,6 @@ __export(exports_node_exports, {
   ExternalAttachment: () => ExternalAttachment,
   FailedHTTPResponse: () => FailedHTTPResponse,
   IDGenerator: () => IDGenerator,
-  INTERNAL_BTQL_LIMIT: () => INTERNAL_BTQL_LIMIT,
   JSONAttachment: () => JSONAttachment,
   LEGACY_CACHED_HEADER: () => LEGACY_CACHED_HEADER,
   LazyValue: () => LazyValue,
@@ -8219,7 +8227,7 @@ function parseSpanFromResponseCreateParams(params) {
 }
 function parseEventFromResponseCreateResult(result) {
   const data = {};
-  if (_optionalChain([result, 'optionalAccess', _102 => _102.output]) !== void 0) {
+  if (_optionalChain([result, 'optionalAccess', _104 => _104.output]) !== void 0) {
     data.output = processImagesInOutput(result.output);
   }
   if (result) {
@@ -8228,7 +8236,7 @@ function parseEventFromResponseCreateResult(result) {
       data.metadata = metadata;
     }
   }
-  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _103 => _103.usage]));
+  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _105 => _105.usage]));
   return data;
 }
 function processImagesInOutput(output) {
@@ -8282,7 +8290,7 @@ function parseSpanFromResponseParseParams(params) {
 }
 function parseEventFromResponseParseResult(result) {
   const data = {};
-  if (_optionalChain([result, 'optionalAccess', _104 => _104.output]) !== void 0) {
+  if (_optionalChain([result, 'optionalAccess', _106 => _106.output]) !== void 0) {
     data.output = processImagesInOutput(result.output);
   }
   if (result) {
@@ -8291,7 +8299,7 @@ function parseEventFromResponseParseResult(result) {
       data.metadata = metadata;
     }
   }
-  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _105 => _105.usage]));
+  data.metrics = parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _107 => _107.usage]));
   return data;
 }
 function traceResponseCreateStream(stream, timedSpan) {
@@ -8308,7 +8316,7 @@ function traceResponseCreateStream(stream, timedSpan) {
       return result;
     }
     const item = result.value;
-    if (!item || !_optionalChain([item, 'optionalAccess', _106 => _106.type]) || !_optionalChain([item, 'optionalAccess', _107 => _107.response])) {
+    if (!item || !_optionalChain([item, 'optionalAccess', _108 => _108.type]) || !_optionalChain([item, 'optionalAccess', _109 => _109.response])) {
       return result;
     }
     const event = parseLogFromItem(item);
@@ -8319,14 +8327,14 @@ function traceResponseCreateStream(stream, timedSpan) {
   };
 }
 function parseLogFromItem(item) {
-  if (!item || !_optionalChain([item, 'optionalAccess', _108 => _108.type]) || !_optionalChain([item, 'optionalAccess', _109 => _109.response])) {
+  if (!item || !_optionalChain([item, 'optionalAccess', _110 => _110.type]) || !_optionalChain([item, 'optionalAccess', _111 => _111.response])) {
     return {};
   }
   const response = item.response;
   switch (item.type) {
     case "response.completed":
       const data = {};
-      if (_optionalChain([response, 'optionalAccess', _110 => _110.output]) !== void 0) {
+      if (_optionalChain([response, 'optionalAccess', _112 => _112.output]) !== void 0) {
         data.output = processImagesInOutput(response.output);
       }
       if (response) {
@@ -8335,7 +8343,7 @@ function parseLogFromItem(item) {
           data.metadata = metadata;
         }
       }
-      data.metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _111 => _111.usage]));
+      data.metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _113 => _113.usage]));
       return data;
     default:
       return {};
@@ -8464,6 +8472,12 @@ function apiPromiseProxy(apiPromise, span, onThen, traceStreamFunc) {
           );
         };
       }
+      if (name === "withResponse") {
+        return async () => {
+          const { data, response } = await target.withResponse();
+          return { data: onThen(data), response };
+        };
+      }
       return Reflect.get(target, name, receiver);
     }
   });
@@ -8509,6 +8523,8 @@ function wrapOpenAIv4(openai) {
         return wrapChatCompletion(baseVal.bind(target));
       } else if (name === "parse") {
         return wrapBetaChatCompletionParse(baseVal.bind(target));
+      } else if (name === "stream") {
+        return wrapBetaChatCompletionStream(baseVal.bind(target));
       }
       return baseVal;
     }
@@ -8524,8 +8540,8 @@ function wrapOpenAIv4(openai) {
   const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
   const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy2;
-  if (_optionalChain([openai, 'access', _112 => _112.beta, 'optionalAccess', _113 => _113.chat, 'optionalAccess', _114 => _114.completions, 'optionalAccess', _115 => _115.stream])) {
-    const betaChatCompletionProxy = new Proxy(_optionalChain([openai, 'optionalAccess', _116 => _116.beta, 'optionalAccess', _117 => _117.chat, 'access', _118 => _118.completions]), {
+  if (_optionalChain([openai, 'access', _114 => _114.beta, 'optionalAccess', _115 => _115.chat, 'optionalAccess', _116 => _116.completions, 'optionalAccess', _117 => _117.stream])) {
+    const betaChatCompletionProxy = new Proxy(_optionalChain([openai, 'optionalAccess', _118 => _118.beta, 'optionalAccess', _119 => _119.chat, 'access', _120 => _120.completions]), {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -8573,7 +8589,7 @@ function wrapOpenAIv4(openai) {
   });
 }
 function logCompletionResponse(startTime, response, span) {
-  const metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _119 => _119.usage]));
+  const metrics = parseMetricsFromUsage(_optionalChain([response, 'optionalAccess', _121 => _121.usage]));
   metrics.time_to_first_token = getCurrentUnixTimestamp() - startTime;
   span.log({
     output: response.choices,
@@ -8803,7 +8819,7 @@ function parseChatCompletionParams(params) {
 function processEmbeddingResponse(result, span) {
   span.log({
     output: { embedding_length: result.data[0].embedding.length },
-    metrics: parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _120 => _120.usage]))
+    metrics: parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _122 => _122.usage]))
   });
 }
 function processModerationResponse(result, span) {
@@ -8833,10 +8849,10 @@ function postprocessStreamingResults(allResults) {
     if (result.usage) {
       metrics = {
         ...metrics,
-        ...parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _121 => _121.usage]))
+        ...parseMetricsFromUsage(_optionalChain([result, 'optionalAccess', _123 => _123.usage]))
       };
     }
-    const delta = _optionalChain([result, 'access', _122 => _122.choices, 'optionalAccess', _123 => _123[0], 'optionalAccess', _124 => _124.delta]);
+    const delta = _optionalChain([result, 'access', _124 => _124.choices, 'optionalAccess', _125 => _125[0], 'optionalAccess', _126 => _126.delta]);
     if (!delta) {
       continue;
     }
@@ -9068,24 +9084,67 @@ function wrapAISDK(aiSDK, options = {}) {
       const original = Reflect.get(target, prop, receiver);
       switch (prop) {
         case "generateText":
-          return wrapGenerateText(original, options);
+          return wrapGenerateText(original, options, aiSDK);
         case "streamText":
-          return wrapStreamText(original, options);
+          return wrapStreamText(original, options, aiSDK);
         case "generateObject":
-          return wrapGenerateObject(original, options);
+          return wrapGenerateObject(original, options, aiSDK);
         case "streamObject":
-          return wrapStreamObject(original, options);
+          return wrapStreamObject(original, options, aiSDK);
+        case "Agent":
+        case "Experimental_Agent":
+        case "ToolLoopAgent":
+          return original ? wrapAgentClass(original, options) : original;
       }
       return original;
     }
   });
 }
-var wrapGenerateText = (generateText, options = {}) => {
-  return async function wrappedGenerateText(params) {
+var wrapAgentClass = (AgentClass, options = {}) => {
+  return new Proxy(AgentClass, {
+    construct(target, args) {
+      const instance = new target(...args);
+      return new Proxy(instance, {
+        get(instanceTarget, prop, instanceReceiver) {
+          const original = Reflect.get(instanceTarget, prop, instanceReceiver);
+          if (prop === "generate") {
+            return wrapAgentGenerate(original, instanceTarget, options);
+          }
+          if (prop === "stream") {
+            return wrapAgentStream(original, instanceTarget, options);
+          }
+          return original;
+        }
+      });
+    }
+  });
+};
+var wrapAgentGenerate = (generate, instance, options = {}) => {
+  return async (params) => makeGenerateTextWrapper(
+    "Agent.generate",
+    options,
+    generate.bind(instance)
+    // as of v5 this is just streamText under the hood
+    // Follows what the AI SDK does under the hood when calling generateText
+  )({ ...instance.settings, ...params });
+};
+var wrapAgentStream = (stream, instance, options = {}) => {
+  return (params) => makeStreamTextWrapper(
+    "Agent.stream",
+    options,
+    stream.bind(instance),
+    // as of v5 this is just streamText under the hood
+    void 0
+    // aiSDK not needed since model is already on instance
+  )({ ...instance.settings, ...params });
+};
+var makeGenerateTextWrapper = (name, options, generateText, aiSDK) => {
+  const wrapper = async function(params) {
     return traced(
       async (span) => {
         const result = await generateText({
           ...params,
+          model: wrapModel(params.model, aiSDK),
           tools: wrapTools(params.tools)
         });
         span.log({
@@ -9095,7 +9154,7 @@ var wrapGenerateText = (generateText, options = {}) => {
         return result;
       },
       {
-        name: "generateText",
+        name,
         spanAttributes: {
           type: "llm" /* LLM */
         },
@@ -9112,17 +9171,176 @@ var wrapGenerateText = (generateText, options = {}) => {
       }
     );
   };
+  Object.defineProperty(wrapper, "name", { value: name, writable: false });
+  return wrapper;
 };
-var wrapGenerateObject = (generateObject, options = {}) => {
+var resolveModel = (model, ai) => {
+  if (typeof model !== "string") {
+    return model;
+  }
+  const provider = _nullishCoalesce(_nullishCoalesce(globalThis.AI_SDK_DEFAULT_PROVIDER, () => ( _optionalChain([ai, 'optionalAccess', _127 => _127.gateway]))), () => ( null));
+  if (provider && typeof provider.languageModel === "function") {
+    return provider.languageModel(model);
+  }
+  return model;
+};
+var wrapModel = (model, ai) => {
+  const resolvedModel = resolveModel(model, ai);
+  if (!resolvedModel || typeof resolvedModel !== "object" || typeof resolvedModel.doGenerate !== "function") {
+    return model;
+  }
+  if (resolvedModel._braintrustWrapped) {
+    return resolvedModel;
+  }
+  const originalDoGenerate = resolvedModel.doGenerate.bind(resolvedModel);
+  const originalDoStream = _optionalChain([resolvedModel, 'access', _128 => _128.doStream, 'optionalAccess', _129 => _129.bind, 'call', _130 => _130(resolvedModel)]);
+  const wrappedDoGenerate = async (options) => {
+    return traced(
+      async (span) => {
+        const result = await originalDoGenerate(options);
+        span.log({
+          output: await processOutput(result),
+          metrics: extractTokenMetrics(result)
+        });
+        return result;
+      },
+      {
+        name: "doGenerate",
+        spanAttributes: {
+          type: "llm" /* LLM */
+        },
+        event: {
+          input: processInputAttachments2(options),
+          metadata: {
+            model: resolvedModel.modelId,
+            provider: resolvedModel.provider,
+            braintrust: {
+              integration_name: "ai-sdk",
+              sdk_language: "typescript"
+            }
+          }
+        }
+      }
+    );
+  };
+  const wrappedDoStream = async (options) => {
+    const span = startSpan({
+      name: "doStream",
+      spanAttributes: {
+        type: "llm" /* LLM */
+      },
+      event: {
+        input: processInputAttachments2(options),
+        metadata: {
+          model: resolvedModel.modelId,
+          provider: resolvedModel.provider,
+          braintrust: {
+            integration_name: "ai-sdk",
+            sdk_language: "typescript"
+          }
+        }
+      }
+    });
+    const result = await originalDoStream(options);
+    const output = {};
+    let text = "";
+    let reasoning = "";
+    const toolCalls = [];
+    let object = void 0;
+    const extractTextDelta = (chunk) => {
+      if (typeof chunk.textDelta === "string") return chunk.textDelta;
+      if (typeof chunk.delta === "string") return chunk.delta;
+      if (typeof chunk.text === "string") return chunk.text;
+      if (typeof chunk.content === "string") return chunk.content;
+      return "";
+    };
+    const transformStream = new TransformStream({
+      async transform(chunk, controller) {
+        switch (chunk.type) {
+          case "text-delta":
+            text += extractTextDelta(chunk);
+            break;
+          case "reasoning-delta":
+            if (chunk.delta) {
+              reasoning += chunk.delta;
+            } else if (chunk.text) {
+              reasoning += chunk.text;
+            }
+            break;
+          case "tool-call":
+            toolCalls.push(chunk);
+            break;
+          case "object":
+            object = chunk.object;
+            break;
+          case "raw":
+            if (chunk.rawValue) {
+              const rawVal = chunk.rawValue;
+              if (_optionalChain([rawVal, 'access', _131 => _131.delta, 'optionalAccess', _132 => _132.content])) {
+                text += rawVal.delta.content;
+              } else if (_optionalChain([rawVal, 'access', _133 => _133.choices, 'optionalAccess', _134 => _134[0], 'optionalAccess', _135 => _135.delta, 'optionalAccess', _136 => _136.content])) {
+                text += rawVal.choices[0].delta.content;
+              } else if (typeof rawVal.text === "string") {
+                text += rawVal.text;
+              } else if (typeof rawVal.content === "string") {
+                text += rawVal.content;
+              }
+            }
+            break;
+          case "finish":
+            output.text = text;
+            output.reasoning = reasoning;
+            output.toolCalls = toolCalls;
+            output.finishReason = chunk.finishReason;
+            output.usage = chunk.usage;
+            if (object !== void 0) {
+              output.object = object;
+            }
+            span.log({
+              output: await processOutput(output),
+              metrics: extractTokenMetrics(output)
+            });
+            span.end();
+            break;
+        }
+        controller.enqueue(chunk);
+      }
+    });
+    return {
+      ...result,
+      stream: result.stream.pipeThrough(transformStream)
+    };
+  };
+  return new Proxy(resolvedModel, {
+    get(target, prop, receiver) {
+      if (prop === "_braintrustWrapped") {
+        return true;
+      }
+      if (prop === "doGenerate") {
+        return wrappedDoGenerate;
+      }
+      if (prop === "doStream" && originalDoStream) {
+        return wrappedDoStream;
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+};
+var wrapGenerateText = (generateText, options = {}, aiSDK) => {
+  return makeGenerateTextWrapper("generateText", options, generateText, aiSDK);
+};
+var wrapGenerateObject = (generateObject, options = {}, aiSDK) => {
   return async function wrappedGenerateObject(params) {
     return traced(
       async (span) => {
         const result = await generateObject({
           ...params,
+          model: wrapModel(params.model, aiSDK),
           tools: wrapTools(params.tools)
         });
+        const output = await processOutput(result, options.denyOutputPaths);
         span.log({
-          output: processOutput(result, options.denyOutputPaths),
+          output,
           metrics: extractTokenMetrics(result)
         });
         return result;
@@ -9146,10 +9364,10 @@ var wrapGenerateObject = (generateObject, options = {}) => {
     );
   };
 };
-var wrapStreamText = (streamText, options = {}) => {
-  return function wrappedStreamText(params) {
+var makeStreamTextWrapper = (name, options, streamText, aiSDK) => {
+  const wrapper = function(params) {
     const span = startSpan({
-      name: "streamText",
+      name,
       spanAttributes: {
         type: "llm" /* LLM */
       },
@@ -9171,6 +9389,7 @@ var wrapStreamText = (streamText, options = {}) => {
         span,
         () => streamText({
           ...params,
+          model: wrapModel(params.model, aiSDK),
           tools: wrapTools(params.tools),
           onChunk: (chunk) => {
             if (!receivedFirst) {
@@ -9181,10 +9400,10 @@ var wrapStreamText = (streamText, options = {}) => {
                 }
               });
             }
-            _optionalChain([params, 'access', _125 => _125.onChunk, 'optionalCall', _126 => _126(chunk)]);
+            _optionalChain([params, 'access', _137 => _137.onChunk, 'optionalCall', _138 => _138(chunk)]);
           },
           onFinish: async (event) => {
-            _optionalChain([params, 'access', _127 => _127.onFinish, 'optionalCall', _128 => _128(event)]);
+            _optionalChain([params, 'access', _139 => _139.onFinish, 'optionalCall', _140 => _140(event)]);
             span.log({
               output: await processOutput(event, options.denyOutputPaths),
               metrics: extractTokenMetrics(event)
@@ -9192,7 +9411,7 @@ var wrapStreamText = (streamText, options = {}) => {
             span.end();
           },
           onError: async (err) => {
-            _optionalChain([params, 'access', _129 => _129.onError, 'optionalCall', _130 => _130(err)]);
+            _optionalChain([params, 'access', _141 => _141.onError, 'optionalCall', _142 => _142(err)]);
             span.log({
               error: serializeError(err)
             });
@@ -9237,8 +9456,13 @@ var wrapStreamText = (streamText, options = {}) => {
       throw error2;
     }
   };
+  Object.defineProperty(wrapper, "name", { value: name, writable: false });
+  return wrapper;
 };
-var wrapStreamObject = (streamObject, options = {}) => {
+var wrapStreamText = (streamText, options = {}, aiSDK) => {
+  return makeStreamTextWrapper("streamText", options, streamText, aiSDK);
+};
+var wrapStreamObject = (streamObject, options = {}, aiSDK) => {
   return function wrappedStreamObject(params) {
     const span = startSpan({
       name: "streamObject",
@@ -9263,6 +9487,7 @@ var wrapStreamObject = (streamObject, options = {}) => {
         span,
         () => streamObject({
           ...params,
+          model: wrapModel(params.model, aiSDK),
           tools: wrapTools(params.tools),
           onChunk: (chunk) => {
             if (!receivedFirst) {
@@ -9273,10 +9498,10 @@ var wrapStreamObject = (streamObject, options = {}) => {
                 }
               });
             }
-            _optionalChain([params, 'access', _131 => _131.onChunk, 'optionalCall', _132 => _132(chunk)]);
+            _optionalChain([params, 'access', _143 => _143.onChunk, 'optionalCall', _144 => _144(chunk)]);
           },
           onFinish: async (event) => {
-            _optionalChain([params, 'access', _133 => _133.onFinish, 'optionalCall', _134 => _134(event)]);
+            _optionalChain([params, 'access', _145 => _145.onFinish, 'optionalCall', _146 => _146(event)]);
             span.log({
               output: await processOutput(event, options.denyOutputPaths),
               metrics: extractTokenMetrics(event)
@@ -9284,7 +9509,7 @@ var wrapStreamObject = (streamObject, options = {}) => {
             span.end();
           },
           onError: async (err) => {
-            _optionalChain([params, 'access', _135 => _135.onError, 'optionalCall', _136 => _136(err)]);
+            _optionalChain([params, 'access', _147 => _147.onError, 'optionalCall', _148 => _148(err)]);
             span.log({
               error: serializeError(err)
             });
@@ -9421,7 +9646,7 @@ var serializeError = (error2) => {
   return String(error2);
 };
 var serializeModel = (model) => {
-  return typeof model === "string" ? model : _optionalChain([model, 'optionalAccess', _137 => _137.modelId]);
+  return typeof model === "string" ? model : _optionalChain([model, 'optionalAccess', _149 => _149.modelId]);
 };
 var isZodSchema = (value) => {
   return value != null && typeof value === "object" && "_def" in value && typeof value._def === "object";
@@ -9470,8 +9695,12 @@ var processInputAttachments2 = (input) => {
   if (input.messages && Array.isArray(input.messages)) {
     processed.messages = input.messages.map(processMessage);
   }
-  if (input.prompt && typeof input.prompt === "object" && !Array.isArray(input.prompt)) {
-    processed.prompt = processPromptContent(input.prompt);
+  if (input.prompt && typeof input.prompt === "object") {
+    if (Array.isArray(input.prompt)) {
+      processed.prompt = input.prompt.map(processMessage);
+    } else {
+      processed.prompt = processPromptContent(input.prompt);
+    }
   }
   if (input.tools) {
     processed.tools = processTools(input.tools);
@@ -9564,7 +9793,7 @@ var convertImageToAttachment = (image, explicitMimeType) => {
   try {
     if (typeof image === "string" && image.startsWith("data:")) {
       const [mimeTypeSection, base64Data] = image.split(",");
-      const mimeType = _optionalChain([mimeTypeSection, 'access', _138 => _138.match, 'call', _139 => _139(/data:(.*?);/), 'optionalAccess', _140 => _140[1]]);
+      const mimeType = _optionalChain([mimeTypeSection, 'access', _150 => _150.match, 'call', _151 => _151(/data:(.*?);/), 'optionalAccess', _152 => _152[1]]);
       if (mimeType && base64Data) {
         const blob = convertDataToBlob(base64Data, mimeType);
         if (blob) {
@@ -9641,6 +9870,7 @@ var extractGetterValues = (obj) => {
   const getterValues = {};
   const getterNames = [
     "text",
+    "object",
     "finishReason",
     "usage",
     "toolCalls",
@@ -9722,7 +9952,7 @@ var convertFileToAttachment = (file, index) => {
 };
 function extractTokenMetrics(result) {
   const metrics = {};
-  const usage = _optionalChain([result, 'optionalAccess', _141 => _141.usage]);
+  const usage = _optionalChain([result, 'optionalAccess', _153 => _153.usage]);
   if (!usage) {
     return metrics;
   }
@@ -9842,7 +10072,7 @@ var omit = (obj, paths) => {
 // src/wrappers/ai-sdk/deprecated/wrapAISDKModel.ts
 function wrapAISDKModel(model) {
   const m = model;
-  if (_optionalChain([m, 'optionalAccess', _142 => _142.specificationVersion]) === "v1" && typeof _optionalChain([m, 'optionalAccess', _143 => _143.provider]) === "string" && typeof _optionalChain([m, 'optionalAccess', _144 => _144.modelId]) === "string") {
+  if (_optionalChain([m, 'optionalAccess', _154 => _154.specificationVersion]) === "v1" && typeof _optionalChain([m, 'optionalAccess', _155 => _155.provider]) === "string" && typeof _optionalChain([m, 'optionalAccess', _156 => _156.modelId]) === "string") {
     return new BraintrustLanguageModelWrapper(m);
   } else {
     console.warn("Unsupported AI SDK model. Not wrapping.");
@@ -9899,10 +10129,10 @@ var BraintrustLanguageModelWrapper = class {
         metrics: {
           time_to_first_token: getCurrentUnixTimestamp() - startTime,
           tokens: !isEmpty(ret.usage) ? ret.usage.promptTokens + ret.usage.completionTokens : void 0,
-          prompt_tokens: _optionalChain([ret, 'access', _145 => _145.usage, 'optionalAccess', _146 => _146.promptTokens]),
-          completion_tokens: _optionalChain([ret, 'access', _147 => _147.usage, 'optionalAccess', _148 => _148.completionTokens]),
+          prompt_tokens: _optionalChain([ret, 'access', _157 => _157.usage, 'optionalAccess', _158 => _158.promptTokens]),
+          completion_tokens: _optionalChain([ret, 'access', _159 => _159.usage, 'optionalAccess', _160 => _160.completionTokens]),
           cached: parseCachedHeader(
-            _nullishCoalesce(_optionalChain([ret, 'access', _149 => _149.rawResponse, 'optionalAccess', _150 => _150.headers, 'optionalAccess', _151 => _151[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _152 => _152.rawResponse, 'optionalAccess', _153 => _153.headers, 'optionalAccess', _154 => _154[LEGACY_CACHED_HEADER]])))
+            _nullishCoalesce(_optionalChain([ret, 'access', _161 => _161.rawResponse, 'optionalAccess', _162 => _162.headers, 'optionalAccess', _163 => _163[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _164 => _164.rawResponse, 'optionalAccess', _165 => _165.headers, 'optionalAccess', _166 => _166[LEGACY_CACHED_HEADER]])))
           )
         }
       });
@@ -9994,10 +10224,10 @@ var BraintrustLanguageModelWrapper = class {
                 metrics: {
                   time_to_first_token,
                   tokens: !isEmpty(usage) ? usage.promptTokens + usage.completionTokens : void 0,
-                  prompt_tokens: _optionalChain([usage, 'optionalAccess', _155 => _155.promptTokens]),
-                  completion_tokens: _optionalChain([usage, 'optionalAccess', _156 => _156.completionTokens]),
+                  prompt_tokens: _optionalChain([usage, 'optionalAccess', _167 => _167.promptTokens]),
+                  completion_tokens: _optionalChain([usage, 'optionalAccess', _168 => _168.completionTokens]),
                   cached: parseCachedHeader(
-                    _nullishCoalesce(_optionalChain([ret, 'access', _157 => _157.rawResponse, 'optionalAccess', _158 => _158.headers, 'optionalAccess', _159 => _159[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _160 => _160.rawResponse, 'optionalAccess', _161 => _161.headers, 'optionalAccess', _162 => _162[LEGACY_CACHED_HEADER]])))
+                    _nullishCoalesce(_optionalChain([ret, 'access', _169 => _169.rawResponse, 'optionalAccess', _170 => _170.headers, 'optionalAccess', _171 => _171[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _172 => _172.rawResponse, 'optionalAccess', _173 => _173.headers, 'optionalAccess', _174 => _174[LEGACY_CACHED_HEADER]])))
                   )
                 }
               });
@@ -10041,7 +10271,7 @@ function postProcessPrompt(prompt) {
         return [
           {
             role: "assistant",
-            content: _optionalChain([textPart, 'optionalAccess', _163 => _163.text]),
+            content: _optionalChain([textPart, 'optionalAccess', _175 => _175.text]),
             ...toolCallParts.length > 0 ? {
               tool_calls: toolCallParts.map((part) => ({
                 id: part.toolCallId,
@@ -10134,23 +10364,23 @@ function extractAnthropicCacheTokens(cacheReadTokens = 0, cacheCreationTokens =
 
 // src/wrappers/ai-sdk/deprecated/BraintrustMiddleware.ts
 function detectProviderFromResult(result) {
-  if (!_optionalChain([result, 'optionalAccess', _164 => _164.providerMetadata])) {
+  if (!_optionalChain([result, 'optionalAccess', _176 => _176.providerMetadata])) {
     return void 0;
   }
   const keys = Object.keys(result.providerMetadata);
-  return _optionalChain([keys, 'optionalAccess', _165 => _165.at, 'call', _166 => _166(0)]);
+  return _optionalChain([keys, 'optionalAccess', _177 => _177.at, 'call', _178 => _178(0)]);
 }
 function extractModelFromResult(result) {
-  if (_optionalChain([result, 'optionalAccess', _167 => _167.response, 'optionalAccess', _168 => _168.modelId])) {
+  if (_optionalChain([result, 'optionalAccess', _179 => _179.response, 'optionalAccess', _180 => _180.modelId])) {
     return result.response.modelId;
   }
-  if (_optionalChain([result, 'optionalAccess', _169 => _169.request, 'optionalAccess', _170 => _170.body, 'optionalAccess', _171 => _171.model])) {
+  if (_optionalChain([result, 'optionalAccess', _181 => _181.request, 'optionalAccess', _182 => _182.body, 'optionalAccess', _183 => _183.model])) {
     return result.request.body.model;
   }
   return void 0;
 }
 function extractModelFromWrapGenerateCallback(model) {
-  return _optionalChain([model, 'optionalAccess', _172 => _172.modelId]);
+  return _optionalChain([model, 'optionalAccess', _184 => _184.modelId]);
 }
 function camelToSnake(str) {
   return str.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
@@ -10195,7 +10425,7 @@ function normalizeUsageMetrics(usage, provider, providerMetadata) {
     metrics.prompt_cached_tokens = cachedInputTokens;
   }
   if (provider === "anthropic") {
-    const anthropicMetadata = _optionalChain([providerMetadata, 'optionalAccess', _173 => _173.anthropic]);
+    const anthropicMetadata = _optionalChain([providerMetadata, 'optionalAccess', _185 => _185.anthropic]);
     if (anthropicMetadata) {
       const cacheReadTokens = getNumberProperty(anthropicMetadata.usage, "cache_read_input_tokens") || 0;
       const cacheCreationTokens = getNumberProperty(
@@ -10221,7 +10451,7 @@ function buildAssistantOutputWithToolCalls(result, toolCalls) {
     {
       index: 0,
       logprobs: null,
-      finish_reason: _nullishCoalesce(normalizeFinishReason(_optionalChain([result, 'optionalAccess', _174 => _174.finishReason])), () => ( (toolCalls.length ? "tool_calls" : void 0))),
+      finish_reason: _nullishCoalesce(normalizeFinishReason(_optionalChain([result, 'optionalAccess', _186 => _186.finishReason])), () => ( (toolCalls.length ? "tool_calls" : void 0))),
       message: {
         role: "assistant",
         tool_calls: toolCalls.length > 0 ? toolCalls : void 0
@@ -10234,7 +10464,7 @@ function extractToolCallsFromSteps(steps) {
   if (!Array.isArray(steps)) return toolCalls;
   let idx = 0;
   for (const step of steps) {
-    const blocks = _optionalChain([step, 'optionalAccess', _175 => _175.content]);
+    const blocks = _optionalChain([step, 'optionalAccess', _187 => _187.content]);
     if (!Array.isArray(blocks)) continue;
     for (const block of blocks) {
       if (block && typeof block === "object" && block.type === "tool-call") {
@@ -10257,7 +10487,7 @@ function extractToolCallsFromBlocks(blocks) {
   return extractToolCallsFromSteps([{ content: blocks }]);
 }
 function extractInput(params) {
-  return _nullishCoalesce(_nullishCoalesce(_optionalChain([params, 'optionalAccess', _176 => _176.prompt]), () => ( _optionalChain([params, 'optionalAccess', _177 => _177.messages]))), () => ( _optionalChain([params, 'optionalAccess', _178 => _178.system])));
+  return _nullishCoalesce(_nullishCoalesce(_optionalChain([params, 'optionalAccess', _188 => _188.prompt]), () => ( _optionalChain([params, 'optionalAccess', _189 => _189.messages]))), () => ( _optionalChain([params, 'optionalAccess', _190 => _190.system])));
 }
 var V2_EXCLUDE_KEYS = /* @__PURE__ */ new Set([
   "prompt",
@@ -10281,16 +10511,16 @@ function BraintrustMiddleware(config = {}) {
       const rawInput = extractInput(params);
       const processedInput = processInputAttachments(rawInput);
       const spanArgs = {
-        name: _optionalChain([config, 'access', _179 => _179.spanInfo, 'optionalAccess', _180 => _180.name]) || "ai-sdk.doGenerate",
+        name: _optionalChain([config, 'access', _191 => _191.spanInfo, 'optionalAccess', _192 => _192.name]) || "ai-sdk.doGenerate",
         spanAttributes: {
           type: "llm" /* LLM */,
-          ..._optionalChain([config, 'access', _181 => _181.spanInfo, 'optionalAccess', _182 => _182.spanAttributes]) || {}
+          ..._optionalChain([config, 'access', _193 => _193.spanInfo, 'optionalAccess', _194 => _194.spanAttributes]) || {}
         },
         event: {
           input: processedInput,
           metadata: {
             ...extractModelParameters(params, V2_EXCLUDE_KEYS),
-            ..._optionalChain([config, 'access', _183 => _183.spanInfo, 'optionalAccess', _184 => _184.metadata]) || {}
+            ..._optionalChain([config, 'access', _195 => _195.spanInfo, 'optionalAccess', _196 => _196.metadata]) || {}
           }
         }
       };
@@ -10316,12 +10546,12 @@ function BraintrustMiddleware(config = {}) {
             metadata.model = modelId;
           }
         }
-        let toolCalls = extractToolCallsFromSteps(_optionalChain([result, 'optionalAccess', _185 => _185.steps]));
+        let toolCalls = extractToolCallsFromSteps(_optionalChain([result, 'optionalAccess', _197 => _197.steps]));
         if (!toolCalls || toolCalls.length === 0) {
-          toolCalls = extractToolCallsFromBlocks(_optionalChain([result, 'optionalAccess', _186 => _186.content]));
+          toolCalls = extractToolCallsFromBlocks(_optionalChain([result, 'optionalAccess', _198 => _198.content]));
         }
         span.log({
-          output: toolCalls.length > 0 ? buildAssistantOutputWithToolCalls(result, toolCalls) : _optionalChain([result, 'optionalAccess', _187 => _187.content]),
+          output: toolCalls.length > 0 ? buildAssistantOutputWithToolCalls(result, toolCalls) : _optionalChain([result, 'optionalAccess', _199 => _199.content]),
           metadata,
           metrics: normalizeUsageMetrics(
             result.usage,
@@ -10343,16 +10573,16 @@ function BraintrustMiddleware(config = {}) {
       const rawInput = extractInput(params);
       const processedInput = processInputAttachments(rawInput);
       const spanArgs = {
-        name: _optionalChain([config, 'access', _188 => _188.spanInfo, 'optionalAccess', _189 => _189.name]) || "ai-sdk.doStream",
+        name: _optionalChain([config, 'access', _200 => _200.spanInfo, 'optionalAccess', _201 => _201.name]) || "ai-sdk.doStream",
         spanAttributes: {
           type: "llm" /* LLM */,
-          ..._optionalChain([config, 'access', _190 => _190.spanInfo, 'optionalAccess', _191 => _191.spanAttributes]) || {}
+          ..._optionalChain([config, 'access', _202 => _202.spanInfo, 'optionalAccess', _203 => _203.spanAttributes]) || {}
         },
         event: {
           input: processedInput,
           metadata: {
             ...extractModelParameters(params, V2_EXCLUDE_KEYS),
-            ..._optionalChain([config, 'access', _192 => _192.spanInfo, 'optionalAccess', _193 => _193.metadata]) || {}
+            ..._optionalChain([config, 'access', _204 => _204.spanInfo, 'optionalAccess', _205 => _205.metadata]) || {}
           }
         }
       };
@@ -10624,9 +10854,9 @@ function streamNextProxy(stream, sspan) {
       return result;
     }
     const item = result.value;
-    switch (_optionalChain([item, 'optionalAccess', _194 => _194.type])) {
+    switch (_optionalChain([item, 'optionalAccess', _206 => _206.type])) {
       case "message_start":
-        const msg = _optionalChain([item, 'optionalAccess', _195 => _195.message]);
+        const msg = _optionalChain([item, 'optionalAccess', _207 => _207.message]);
         if (msg) {
           const event = parseEventFromMessage(msg);
           totals = { ...totals, ...event.metrics };
@@ -10639,20 +10869,20 @@ function streamNextProxy(stream, sspan) {
         }
         break;
       case "content_block_delta":
-        if (_optionalChain([item, 'access', _196 => _196.delta, 'optionalAccess', _197 => _197.type]) === "text_delta") {
-          const text = _optionalChain([item, 'optionalAccess', _198 => _198.delta, 'optionalAccess', _199 => _199.text]);
+        if (_optionalChain([item, 'access', _208 => _208.delta, 'optionalAccess', _209 => _209.type]) === "text_delta") {
+          const text = _optionalChain([item, 'optionalAccess', _210 => _210.delta, 'optionalAccess', _211 => _211.text]);
           if (text) {
             deltas.push(text);
           }
         }
         break;
       case "message_delta":
-        const usage = _optionalChain([item, 'optionalAccess', _200 => _200.usage]);
+        const usage = _optionalChain([item, 'optionalAccess', _212 => _212.usage]);
         if (usage) {
           const metrics = parseMetricsFromUsage2(usage);
           totals = { ...totals, ...metrics };
         }
-        const delta = _optionalChain([item, 'optionalAccess', _201 => _201.delta]);
+        const delta = _optionalChain([item, 'optionalAccess', _213 => _213.delta]);
         if (delta) {
           metadata = { ...metadata, ...delta };
         }
@@ -10665,7 +10895,7 @@ function streamNextProxy(stream, sspan) {
 }
 function parseEventFromMessage(message) {
   const output = message ? { role: message.role, content: message.content } : null;
-  const metrics = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _202 => _202.usage]));
+  const metrics = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _214 => _214.usage]));
   const metas = ["stop_reason", "stop_sequence"];
   const metadata = {};
   for (const m of metas) {
@@ -10812,7 +11042,7 @@ function wrapClaudeAgentQuery(queryFn, defaultThis) {
           finalResults.push(finalMessageContent);
         }
         const lastMessage = currentMessages[currentMessages.length - 1];
-        if (_optionalChain([lastMessage, 'optionalAccess', _203 => _203.message, 'optionalAccess', _204 => _204.usage])) {
+        if (_optionalChain([lastMessage, 'optionalAccess', _215 => _215.message, 'optionalAccess', _216 => _216.usage])) {
           const outputTokens = getNumberProperty2(lastMessage.message.usage, "output_tokens") || 0;
           accumulatedOutputTokens += outputTokens;
         }
@@ -10827,20 +11057,20 @@ function wrapClaudeAgentQuery(queryFn, defaultThis) {
           );
           for await (const message of generator) {
             const currentTime = getCurrentUnixTimestamp();
-            const messageId = _optionalChain([message, 'access', _205 => _205.message, 'optionalAccess', _206 => _206.id]);
+            const messageId = _optionalChain([message, 'access', _217 => _217.message, 'optionalAccess', _218 => _218.id]);
             if (messageId && messageId !== currentMessageId) {
               await createLLMSpan();
               currentMessageId = messageId;
               currentMessageStartTime = currentTime;
             }
-            if (message.type === "assistant" && _optionalChain([message, 'access', _207 => _207.message, 'optionalAccess', _208 => _208.usage])) {
+            if (message.type === "assistant" && _optionalChain([message, 'access', _219 => _219.message, 'optionalAccess', _220 => _220.usage])) {
               currentMessages.push(message);
             }
             if (message.type === "result" && message.usage) {
               finalUsageMetrics = _extractUsageFromMessage(message);
               if (currentMessages.length > 0 && finalUsageMetrics.completion_tokens !== void 0) {
                 const lastMessage = currentMessages[currentMessages.length - 1];
-                if (_optionalChain([lastMessage, 'optionalAccess', _209 => _209.message, 'optionalAccess', _210 => _210.usage])) {
+                if (_optionalChain([lastMessage, 'optionalAccess', _221 => _221.message, 'optionalAccess', _222 => _222.usage])) {
                   const adjustedTokens = finalUsageMetrics.completion_tokens - accumulatedOutputTokens;
                   if (adjustedTokens >= 0) {
                     lastMessage.message.usage.output_tokens = adjustedTokens;
@@ -10921,7 +11151,7 @@ function _extractUsageFromMessage(message) {
   const metrics = {};
   let usage;
   if (message.type === "assistant") {
-    usage = _optionalChain([message, 'access', _211 => _211.message, 'optionalAccess', _212 => _212.usage]);
+    usage = _optionalChain([message, 'access', _223 => _223.message, 'optionalAccess', _224 => _224.usage]);
   } else if (message.type === "result") {
     usage = message.usage;
   }
@@ -10953,14 +11183,14 @@ function _extractUsageFromMessage(message) {
 async function _createLLMSpanForMessages(messages, prompt, conversationHistory, options, startTime, parentSpan) {
   if (messages.length === 0) return void 0;
   const lastMessage = messages[messages.length - 1];
-  if (lastMessage.type !== "assistant" || !_optionalChain([lastMessage, 'access', _213 => _213.message, 'optionalAccess', _214 => _214.usage])) {
+  if (lastMessage.type !== "assistant" || !_optionalChain([lastMessage, 'access', _225 => _225.message, 'optionalAccess', _226 => _226.usage])) {
     return void 0;
   }
   const model = lastMessage.message.model || options.model;
   const usage = _extractUsageFromMessage(lastMessage);
   const input = _buildLLMInput(prompt, conversationHistory);
   const outputs = messages.map(
-    (m) => _optionalChain([m, 'access', _215 => _215.message, 'optionalAccess', _216 => _216.content]) && _optionalChain([m, 'access', _217 => _217.message, 'optionalAccess', _218 => _218.role]) ? { content: m.message.content, role: m.message.role } : void 0
+    (m) => _optionalChain([m, 'access', _227 => _227.message, 'optionalAccess', _228 => _228.content]) && _optionalChain([m, 'access', _229 => _229.message, 'optionalAccess', _230 => _230.role]) ? { content: m.message.content, role: m.message.role } : void 0
   ).filter((c) => c !== void 0);
   await traced(
     (llmSpan) => {
@@ -10980,7 +11210,7 @@ async function _createLLMSpanForMessages(messages, prompt, conversationHistory,
       parent: parentSpan
     }
   );
-  return _optionalChain([lastMessage, 'access', _219 => _219.message, 'optionalAccess', _220 => _220.content]) && _optionalChain([lastMessage, 'access', _221 => _221.message, 'optionalAccess', _222 => _222.role]) ? { content: lastMessage.message.content, role: lastMessage.message.role } : void 0;
+  return _optionalChain([lastMessage, 'access', _231 => _231.message, 'optionalAccess', _232 => _232.content]) && _optionalChain([lastMessage, 'access', _233 => _233.message, 'optionalAccess', _234 => _234.role]) ? { content: lastMessage.message.content, role: lastMessage.message.role } : void 0;
 }
 function wrapClaudeAgentSDK(sdk) {
   const cache = /* @__PURE__ */ new Map();
@@ -11270,7 +11500,7 @@ function serializePart(part) {
   return part;
 }
 function serializeTools(params) {
-  if (!_optionalChain([params, 'access', _223 => _223.config, 'optionalAccess', _224 => _224.tools])) {
+  if (!_optionalChain([params, 'access', _235 => _235.config, 'optionalAccess', _236 => _236.tools])) {
     return null;
   }
   try {
@@ -11353,7 +11583,7 @@ function aggregateGenerateContentChunks(chunks, start, firstTokenTime) {
     }
     if (chunk.candidates && Array.isArray(chunk.candidates)) {
       for (const candidate of chunk.candidates) {
-        if (_optionalChain([candidate, 'access', _225 => _225.content, 'optionalAccess', _226 => _226.parts])) {
+        if (_optionalChain([candidate, 'access', _237 => _237.content, 'optionalAccess', _238 => _238.parts])) {
           for (const part of candidate.content.parts) {
             if (part.text !== void 0) {
               if (part.thought) {
@@ -11384,7 +11614,7 @@ function aggregateGenerateContentChunks(chunks, start, firstTokenTime) {
     parts.push({ text });
   }
   parts.push(...otherParts);
-  if (parts.length > 0 && _optionalChain([lastResponse, 'optionalAccess', _227 => _227.candidates])) {
+  if (parts.length > 0 && _optionalChain([lastResponse, 'optionalAccess', _239 => _239.candidates])) {
     const candidates = [];
     for (const candidate of lastResponse.candidates) {
       const candidateDict = {
@@ -12769,6 +12999,7 @@ var waterfall$1 = awaitify(waterfall);
 
 // src/framework.ts
 var _chalk = require('chalk'); var _chalk2 = _interopRequireDefault(_chalk);
+var _termilink = require('termi-link');
 var _boxen = require('boxen'); var _boxen2 = _interopRequireDefault(_boxen);
 var _pluralize = require('pluralize'); var _pluralize2 = _interopRequireDefault(_pluralize);
 var _clitable3 = require('cli-table3'); var _clitable32 = _interopRequireDefault(_clitable3);
@@ -13541,10 +13772,10 @@ async function runEvaluatorInternal(experiment, evaluator, progressReporter, fil
                 span,
                 parameters: _nullishCoalesce(parameters, () => ( {})),
                 reportProgress: (event) => {
-                  _optionalChain([stream, 'optionalCall', _228 => _228({
+                  _optionalChain([stream, 'optionalCall', _240 => _240({
                     ...event,
                     id: rootSpan.id,
-                    origin: _optionalChain([baseEvent, 'access', _229 => _229.event, 'optionalAccess', _230 => _230.origin]),
+                    origin: _optionalChain([baseEvent, 'access', _241 => _241.event, 'optionalAccess', _242 => _242.origin]),
                     name: evaluator.evalName,
                     object_type: "task"
                   })]);
@@ -13699,7 +13930,7 @@ async function runEvaluatorInternal(experiment, evaluator, progressReporter, fil
             metadata,
             scores: mergedScores,
             error: error2,
-            origin: _optionalChain([baseEvent, 'access', _231 => _231.event, 'optionalAccess', _232 => _232.origin])
+            origin: _optionalChain([baseEvent, 'access', _243 => _243.event, 'optionalAccess', _244 => _244.origin])
           });
         }
       };
@@ -13732,7 +13963,7 @@ async function runEvaluatorInternal(experiment, evaluator, progressReporter, fil
           break;
         }
         scheduledTrials++;
-        _optionalChain([progressReporter, 'access', _233 => _233.setTotal, 'optionalCall', _234 => _234(evaluator.evalName, scheduledTrials)]);
+        _optionalChain([progressReporter, 'access', _245 => _245.setTotal, 'optionalCall', _246 => _246(evaluator.evalName, scheduledTrials)]);
         q.push({ datum, trialIndex });
       }
     }
@@ -13994,7 +14225,11 @@ function formatExperimentSummary(summary) {
     tableParts.push(combinedTable.toString());
   }
   const content = [comparisonLine, ...tableParts].filter(Boolean).join("\n");
-  const footer = summary.experimentUrl ? `See results at ${summary.experimentUrl}` : "";
+  const footer = summary.experimentUrl ? _termilink.terminalLink.call(void 0, 
+    `View results for ${summary.experimentName}`,
+    summary.experimentUrl,
+    { fallback: () => `See results at ${summary.experimentUrl}` }
+  ) : "";
   const boxContent = [content, footer].filter(Boolean).join("\n\n");
   return "\n" + _boxen2.default.call(void 0, boxContent, {
     title: _chalk2.default.gray("Experiment summary"),
@@ -14158,4 +14393,4 @@ var index_default = exports_node_exports;
 
 
 
-exports.Attachment = Attachment; exports.BaseAttachment = BaseAttachment; exports.BaseExperiment = BaseExperiment; exports.BraintrustMiddleware = BraintrustMiddleware; exports.BraintrustState = BraintrustState; exports.BraintrustStream = BraintrustStream; exports.CodeFunction = CodeFunction; exports.CodePrompt = CodePrompt; exports.ContextManager = ContextManager; exports.Dataset = Dataset2; exports.ERR_PERMALINK = ERR_PERMALINK; exports.Eval = Eval; exports.EvalResultWithSummary = EvalResultWithSummary; exports.Experiment = Experiment2; exports.ExternalAttachment = ExternalAttachment; exports.FailedHTTPResponse = FailedHTTPResponse; exports.IDGenerator = IDGenerator; exports.INTERNAL_BTQL_LIMIT = INTERNAL_BTQL_LIMIT; exports.JSONAttachment = JSONAttachment; exports.LEGACY_CACHED_HEADER = LEGACY_CACHED_HEADER; exports.LazyValue = LazyValue; exports.Logger = Logger; exports.LoginInvalidOrgError = LoginInvalidOrgError; exports.NOOP_SPAN = NOOP_SPAN; exports.NOOP_SPAN_PERMALINK = NOOP_SPAN_PERMALINK; exports.NoopSpan = NoopSpan; exports.Project = Project2; exports.ProjectNameIdMap = ProjectNameIdMap; exports.Prompt = Prompt2; exports.PromptBuilder = PromptBuilder; exports.ReadonlyAttachment = ReadonlyAttachment; exports.ReadonlyExperiment = ReadonlyExperiment; exports.Reporter = Reporter; exports.ScorerBuilder = ScorerBuilder; exports.SpanImpl = SpanImpl; exports.TestBackgroundLogger = TestBackgroundLogger; exports.ToolBuilder = ToolBuilder; exports.UUIDGenerator = UUIDGenerator; exports.X_CACHED_HEADER = X_CACHED_HEADER; exports._exportsForTestingOnly = _exportsForTestingOnly; exports._internalGetGlobalState = _internalGetGlobalState; exports._internalSetInitialState = _internalSetInitialState; exports.braintrustStreamChunkSchema = braintrustStreamChunkSchema; exports.buildLocalSummary = buildLocalSummary; exports.createFinalValuePassThroughStream = createFinalValuePassThroughStream; exports.currentExperiment = currentExperiment; exports.currentLogger = currentLogger; exports.currentSpan = currentSpan; exports.deepCopyEvent = deepCopyEvent; exports.default = index_default; exports.defaultErrorScoreHandler = defaultErrorScoreHandler; exports.deserializePlainStringAsJSON = deserializePlainStringAsJSON; exports.devNullWritableStream = devNullWritableStream; exports.evaluatorDefinitionSchema = evaluatorDefinitionSchema; exports.evaluatorDefinitionsSchema = evaluatorDefinitionsSchema; exports.flush = flush; exports.getContextManager = getContextManager; exports.getIdGenerator = getIdGenerator; exports.getPromptVersions = getPromptVersions; exports.getSpanParentObject = getSpanParentObject; exports.graph = graph_framework_exports; exports.init = init; exports.initDataset = initDataset; exports.initExperiment = initExperiment; exports.initFunction = initFunction; exports.initLogger = initLogger; exports.invoke = invoke; exports.loadPrompt = loadPrompt; exports.log = log; exports.logError = logError; exports.login = login; exports.loginToState = loginToState; exports.newId = newId; exports.parseCachedHeader = parseCachedHeader; exports.permalink = permalink; exports.projects = projects; exports.promptContentsSchema = promptContentsSchema; exports.promptDefinitionSchema = promptDefinitionSchema; exports.promptDefinitionToPromptData = promptDefinitionToPromptData; exports.promptDefinitionWithToolsSchema = promptDefinitionWithToolsSchema; exports.renderMessage = renderMessage; exports.renderPromptParams = renderPromptParams; exports.reportFailures = reportFailures; exports.runEvaluator = runEvaluator; exports.setFetch = setFetch; exports.setMaskingFunction = setMaskingFunction; exports.spanComponentsToObjectId = spanComponentsToObjectId; exports.startSpan = startSpan; exports.summarize = summarize; exports.toolFunctionDefinitionSchema = ToolFunctionDefinition; exports.traceable = traceable; exports.traced = traced; exports.updateSpan = updateSpan; exports.withCurrent = withCurrent; exports.withDataset = withDataset; exports.withExperiment = withExperiment; exports.withLogger = withLogger; exports.withParent = withParent; exports.wrapAISDK = wrapAISDK; exports.wrapAISDKModel = wrapAISDKModel; exports.wrapAnthropic = wrapAnthropic; exports.wrapClaudeAgentSDK = wrapClaudeAgentSDK; exports.wrapGoogleGenAI = wrapGoogleGenAI; exports.wrapMastraAgent = wrapMastraAgent; exports.wrapOpenAI = wrapOpenAI; exports.wrapOpenAIv4 = wrapOpenAIv4; exports.wrapTraced = wrapTraced;
+exports.Attachment = Attachment; exports.BaseAttachment = BaseAttachment; exports.BaseExperiment = BaseExperiment; exports.BraintrustMiddleware = BraintrustMiddleware; exports.BraintrustState = BraintrustState; exports.BraintrustStream = BraintrustStream; exports.CodeFunction = CodeFunction; exports.CodePrompt = CodePrompt; exports.ContextManager = ContextManager; exports.DEFAULT_FETCH_BATCH_SIZE = DEFAULT_FETCH_BATCH_SIZE; exports.Dataset = Dataset2; exports.ERR_PERMALINK = ERR_PERMALINK; exports.Eval = Eval; exports.EvalResultWithSummary = EvalResultWithSummary; exports.Experiment = Experiment2; exports.ExternalAttachment = ExternalAttachment; exports.FailedHTTPResponse = FailedHTTPResponse; exports.IDGenerator = IDGenerator; exports.JSONAttachment = JSONAttachment; exports.LEGACY_CACHED_HEADER = LEGACY_CACHED_HEADER; exports.LazyValue = LazyValue; exports.Logger = Logger; exports.LoginInvalidOrgError = LoginInvalidOrgError; exports.NOOP_SPAN = NOOP_SPAN; exports.NOOP_SPAN_PERMALINK = NOOP_SPAN_PERMALINK; exports.NoopSpan = NoopSpan; exports.Project = Project2; exports.ProjectNameIdMap = ProjectNameIdMap; exports.Prompt = Prompt2; exports.PromptBuilder = PromptBuilder; exports.ReadonlyAttachment = ReadonlyAttachment; exports.ReadonlyExperiment = ReadonlyExperiment; exports.Reporter = Reporter; exports.ScorerBuilder = ScorerBuilder; exports.SpanImpl = SpanImpl; exports.TestBackgroundLogger = TestBackgroundLogger; exports.ToolBuilder = ToolBuilder; exports.UUIDGenerator = UUIDGenerator; exports.X_CACHED_HEADER = X_CACHED_HEADER; exports._exportsForTestingOnly = _exportsForTestingOnly; exports._internalGetGlobalState = _internalGetGlobalState; exports._internalSetInitialState = _internalSetInitialState; exports.braintrustStreamChunkSchema = braintrustStreamChunkSchema; exports.buildLocalSummary = buildLocalSummary; exports.createFinalValuePassThroughStream = createFinalValuePassThroughStream; exports.currentExperiment = currentExperiment; exports.currentLogger = currentLogger; exports.currentSpan = currentSpan; exports.deepCopyEvent = deepCopyEvent; exports.default = index_default; exports.defaultErrorScoreHandler = defaultErrorScoreHandler; exports.deserializePlainStringAsJSON = deserializePlainStringAsJSON; exports.devNullWritableStream = devNullWritableStream; exports.evaluatorDefinitionSchema = evaluatorDefinitionSchema; exports.evaluatorDefinitionsSchema = evaluatorDefinitionsSchema; exports.flush = flush; exports.getContextManager = getContextManager; exports.getIdGenerator = getIdGenerator; exports.getPromptVersions = getPromptVersions; exports.getSpanParentObject = getSpanParentObject; exports.graph = graph_framework_exports; exports.init = init; exports.initDataset = initDataset; exports.initExperiment = initExperiment; exports.initFunction = initFunction; exports.initLogger = initLogger; exports.invoke = invoke; exports.loadPrompt = loadPrompt; exports.log = log; exports.logError = logError; exports.login = login; exports.loginToState = loginToState; exports.newId = newId; exports.parseCachedHeader = parseCachedHeader; exports.permalink = permalink; exports.projects = projects; exports.promptContentsSchema = promptContentsSchema; exports.promptDefinitionSchema = promptDefinitionSchema; exports.promptDefinitionToPromptData = promptDefinitionToPromptData; exports.promptDefinitionWithToolsSchema = promptDefinitionWithToolsSchema; exports.renderMessage = renderMessage; exports.renderPromptParams = renderPromptParams; exports.reportFailures = reportFailures; exports.runEvaluator = runEvaluator; exports.setFetch = setFetch; exports.setMaskingFunction = setMaskingFunction; exports.spanComponentsToObjectId = spanComponentsToObjectId; exports.startSpan = startSpan; exports.summarize = summarize; exports.toolFunctionDefinitionSchema = ToolFunctionDefinition; exports.traceable = traceable; exports.traced = traced; exports.updateSpan = updateSpan; exports.withCurrent = withCurrent; exports.withDataset = withDataset; exports.withExperiment = withExperiment; exports.withLogger = withLogger; exports.withParent = withParent; exports.wrapAISDK = wrapAISDK; exports.wrapAISDKModel = wrapAISDKModel; exports.wrapAnthropic = wrapAnthropic; exports.wrapClaudeAgentSDK = wrapClaudeAgentSDK; exports.wrapGoogleGenAI = wrapGoogleGenAI; exports.wrapMastraAgent = wrapMastraAgent; exports.wrapOpenAI = wrapOpenAI; exports.wrapOpenAIv4 = wrapOpenAIv4; exports.wrapTraced = wrapTraced;