braintrust 0.4.9 → 0.4.10

package/dist/index.js

@@ -1504,7 +1504,7 @@ var init_util = __esm({
 
 // src/generated_types.ts
 
-var AclObjectType, Permission, Acl, AISecret, ResponseFormatJsonSchema, ResponseFormatNullish, AnyModelParams, ApiKey, AsyncScoringState, AsyncScoringControl, BraintrustAttachmentReference, ExternalAttachmentReference, AttachmentReference, UploadStatus, AttachmentStatus, BraintrustModelParams, CallEvent, ChatCompletionContentPartTextWithTitle, ChatCompletionContentPartImageWithTitle, ChatCompletionContentPart, ChatCompletionContentPartText, ChatCompletionMessageToolCall, ChatCompletionMessageReasoning, ChatCompletionMessageParam, ChatCompletionOpenAIMessageParam, ChatCompletionTool, CodeBundle, Dataset, ObjectReferenceNullish, DatasetEvent, EnvVar, RepoInfo, Experiment, SpanType, SpanAttributes, ExperimentEvent, ExtendedSavedFunctionId, PromptBlockDataNullish, ModelParams, PromptOptionsNullish, PromptParserNullish, SavedFunctionId, PromptDataNullish, FunctionTypeEnumNullish, FunctionIdRef, PromptBlockData, GraphNode, GraphEdge, GraphData, FunctionData, Function, FunctionFormat, PromptData, FunctionTypeEnum, FunctionId, FunctionObjectType, FunctionOutputType, GitMetadataSettings, Group, IfExists, InvokeParent, StreamingMode, InvokeFunction, MessageRole, ObjectReference, OnlineScoreConfig, Organization, ProjectSettings, Project, RetentionObjectType, ProjectAutomation, ProjectLogsEvent, ProjectScoreType, ProjectScoreCategory, ProjectScoreCategories, ProjectScoreConfig, ProjectScore, ProjectTag, Prompt, PromptOptions, PromptSessionEvent, ResponseFormat, Role, RunEval, ServiceToken, SpanIFrame, SSEConsoleEventData, SSEProgressEventData, ToolFunctionDefinition, User, ViewDataSearch, ViewData, ViewOptions, View;
+var AclObjectType, Permission, Acl, AISecret, ResponseFormatJsonSchema, ResponseFormatNullish, AnyModelParams, ApiKey, AsyncScoringState, AsyncScoringControl, BraintrustAttachmentReference, ExternalAttachmentReference, AttachmentReference, UploadStatus, AttachmentStatus, BraintrustModelParams, CallEvent, ChatCompletionContentPartTextWithTitle, ChatCompletionContentPartImageWithTitle, ChatCompletionContentPartFileFile, ChatCompletionContentPartFileWithTitle, ChatCompletionContentPart, ChatCompletionContentPartText, ChatCompletionMessageToolCall, ChatCompletionMessageReasoning, ChatCompletionMessageParam, ChatCompletionOpenAIMessageParam, ChatCompletionTool, CodeBundle, Dataset, ObjectReferenceNullish, DatasetEvent, EnvVar, RepoInfo, Experiment, SpanType, SpanAttributes, ExperimentEvent, ExtendedSavedFunctionId, PromptBlockDataNullish, ModelParams, PromptOptionsNullish, PromptParserNullish, SavedFunctionId, PromptDataNullish, FunctionTypeEnumNullish, FunctionIdRef, PromptBlockData, GraphNode, GraphEdge, GraphData, FunctionData, Function, FunctionFormat, PromptData, FunctionTypeEnum, FunctionId, FunctionObjectType, FunctionOutputType, GitMetadataSettings, Group, IfExists, InvokeParent, StreamingMode, InvokeFunction, MessageRole, ObjectReference, OnlineScoreConfig, Organization, ProjectSettings, Project, RetentionObjectType, ProjectAutomation, ProjectLogsEvent, ProjectScoreType, ProjectScoreCategory, ProjectScoreCategories, ProjectScoreConfig, ProjectScore, ProjectTag, Prompt, PromptOptions, PromptSessionEvent, ResponseFormat, Role, RunEval, ServiceToken, SpanIFrame, SSEConsoleEventData, SSEProgressEventData, ToolFunctionDefinition, User, ViewDataSearch, ViewData, ViewOptions, View;
 var init_generated_types = __esm({
   "src/generated_types.ts"() {
     "use strict";
@@ -1595,10 +1595,12 @@ var init_generated_types = __esm({
       ]).optional(),
       n: _v3.z.number().optional(),
       stop: _v3.z.array(_v3.z.string()).optional(),
-      reasoning_effort: _v3.z.enum(["minimal", "low", "medium", "high"]).optional(),
+      reasoning_effort: _v3.z.enum(["none", "minimal", "low", "medium", "high"]).optional(),
       verbosity: _v3.z.enum(["low", "medium", "high"]).optional(),
       top_k: _v3.z.number().optional(),
       stop_sequences: _v3.z.array(_v3.z.string()).optional(),
+      reasoning_enabled: _v3.z.boolean().optional(),
+      reasoning_budget: _v3.z.number().optional(),
       max_tokens_to_sample: _v3.z.number().optional(),
       maxOutputTokens: _v3.z.number().optional(),
       topP: _v3.z.number().optional(),
@@ -1654,7 +1656,11 @@ var init_generated_types = __esm({
       upload_status: UploadStatus,
       error_message: _v3.z.string().optional()
     });
-    BraintrustModelParams = _v3.z.object({ use_cache: _v3.z.boolean() }).partial();
+    BraintrustModelParams = _v3.z.object({
+      use_cache: _v3.z.boolean(),
+      reasoning_enabled: _v3.z.boolean(),
+      reasoning_budget: _v3.z.number()
+    }).partial();
     CallEvent = _v3.z.union([
       _v3.z.object({
         id: _v3.z.string().optional(),
@@ -1709,9 +1715,15 @@ var init_generated_types = __esm({
       }),
       type: _v3.z.literal("image_url")
     });
+    ChatCompletionContentPartFileFile = _v3.z.object({ file_data: _v3.z.string(), filename: _v3.z.string(), file_id: _v3.z.string() }).partial();
+    ChatCompletionContentPartFileWithTitle = _v3.z.object({
+      file: ChatCompletionContentPartFileFile,
+      type: _v3.z.literal("file")
+    });
     ChatCompletionContentPart = _v3.z.union([
       ChatCompletionContentPartTextWithTitle,
-      ChatCompletionContentPartImageWithTitle
+      ChatCompletionContentPartImageWithTitle,
+      ChatCompletionContentPartFileWithTitle
     ]);
     ChatCompletionContentPartText = _v3.z.object({
       text: _v3.z.string().default(""),
@@ -1808,7 +1820,7 @@ var init_generated_types = __esm({
     });
     CodeBundle = _v3.z.object({
       runtime_context: _v3.z.object({
-        runtime: _v3.z.enum(["node", "python"]),
+        runtime: _v3.z.enum(["node", "python", "browser"]),
         version: _v3.z.string()
       }),
       location: _v3.z.union([
@@ -1869,7 +1881,9 @@ var init_generated_types = __esm({
       span_id: _v3.z.string(),
       root_span_id: _v3.z.string(),
       is_root: _v3.z.union([_v3.z.boolean(), _v3.z.null()]).optional(),
-      origin: ObjectReferenceNullish.optional()
+      origin: ObjectReferenceNullish.optional(),
+      comments: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional(),
+      audit_data: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional()
     });
     EnvVar = _v3.z.object({
       id: _v3.z.string().uuid(),
@@ -1949,7 +1963,9 @@ var init_generated_types = __esm({
       root_span_id: _v3.z.string(),
       span_attributes: SpanAttributes.optional(),
       is_root: _v3.z.union([_v3.z.boolean(), _v3.z.null()]).optional(),
-      origin: ObjectReferenceNullish.optional()
+      origin: ObjectReferenceNullish.optional(),
+      comments: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional(),
+      audit_data: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional()
     });
     ExtendedSavedFunctionId = _v3.z.union([
       _v3.z.object({ type: _v3.z.literal("function"), id: _v3.z.string() }),
@@ -1961,17 +1977,19 @@ var init_generated_types = __esm({
       })
     ]);
     PromptBlockDataNullish = _v3.z.union([
-      _v3.z.object({ type: _v3.z.literal("completion"), content: _v3.z.string() }),
       _v3.z.object({
         type: _v3.z.literal("chat"),
         messages: _v3.z.array(ChatCompletionMessageParam),
         tools: _v3.z.string().optional()
       }),
+      _v3.z.object({ type: _v3.z.literal("completion"), content: _v3.z.string() }),
       _v3.z.null()
     ]);
     ModelParams = _v3.z.union([
       _v3.z.object({
         use_cache: _v3.z.boolean(),
+        reasoning_enabled: _v3.z.boolean(),
+        reasoning_budget: _v3.z.number(),
         temperature: _v3.z.number(),
         top_p: _v3.z.number(),
         max_tokens: _v3.z.number(),
@@ -1995,11 +2013,13 @@ var init_generated_types = __esm({
         ]),
         n: _v3.z.number(),
         stop: _v3.z.array(_v3.z.string()),
-        reasoning_effort: _v3.z.enum(["minimal", "low", "medium", "high"]),
+        reasoning_effort: _v3.z.enum(["none", "minimal", "low", "medium", "high"]),
         verbosity: _v3.z.enum(["low", "medium", "high"])
       }).partial().passthrough(),
       _v3.z.object({
         use_cache: _v3.z.boolean().optional(),
+        reasoning_enabled: _v3.z.boolean().optional(),
+        reasoning_budget: _v3.z.number().optional(),
         max_tokens: _v3.z.number(),
         temperature: _v3.z.number(),
         top_p: _v3.z.number().optional(),
@@ -2009,6 +2029,8 @@ var init_generated_types = __esm({
       }).passthrough(),
       _v3.z.object({
         use_cache: _v3.z.boolean(),
+        reasoning_enabled: _v3.z.boolean(),
+        reasoning_budget: _v3.z.number(),
         temperature: _v3.z.number(),
         maxOutputTokens: _v3.z.number(),
         topP: _v3.z.number(),
@@ -2016,10 +2038,16 @@ var init_generated_types = __esm({
       }).partial().passthrough(),
       _v3.z.object({
         use_cache: _v3.z.boolean(),
+        reasoning_enabled: _v3.z.boolean(),
+        reasoning_budget: _v3.z.number(),
         temperature: _v3.z.number(),
         topK: _v3.z.number()
       }).partial().passthrough(),
-      _v3.z.object({ use_cache: _v3.z.boolean() }).partial().passthrough()
+      _v3.z.object({
+        use_cache: _v3.z.boolean(),
+        reasoning_enabled: _v3.z.boolean(),
+        reasoning_budget: _v3.z.number()
+      }).partial().passthrough()
     ]);
     PromptOptionsNullish = _v3.z.union([
       _v3.z.object({ model: _v3.z.string(), params: ModelParams, position: _v3.z.string() }).partial(),
@@ -2055,17 +2083,17 @@ var init_generated_types = __esm({
       _v3.z.null()
     ]);
     FunctionTypeEnumNullish = _v3.z.union([
-      _v3.z.enum(["llm", "scorer", "task", "tool"]),
+      _v3.z.enum(["llm", "scorer", "task", "tool", "custom_view"]),
       _v3.z.null()
     ]);
     FunctionIdRef = _v3.z.object({}).partial().passthrough();
     PromptBlockData = _v3.z.union([
-      _v3.z.object({ type: _v3.z.literal("completion"), content: _v3.z.string() }),
       _v3.z.object({
         type: _v3.z.literal("chat"),
         messages: _v3.z.array(ChatCompletionMessageParam),
         tools: _v3.z.string().optional()
-      })
+      }),
+      _v3.z.object({ type: _v3.z.literal("completion"), content: _v3.z.string() })
     ]);
     GraphNode = _v3.z.union([
       _v3.z.object({
@@ -2133,7 +2161,7 @@ var init_generated_types = __esm({
           _v3.z.object({
             type: _v3.z.literal("inline"),
             runtime_context: _v3.z.object({
-              runtime: _v3.z.enum(["node", "python"]),
+              runtime: _v3.z.enum(["node", "python", "browser"]),
               version: _v3.z.string()
             }),
             code: _v3.z.string()
@@ -2192,7 +2220,13 @@ var init_generated_types = __esm({
         _v3.z.null()
       ])
     }).partial();
-    FunctionTypeEnum = _v3.z.enum(["llm", "scorer", "task", "tool"]);
+    FunctionTypeEnum = _v3.z.enum([
+      "llm",
+      "scorer",
+      "task",
+      "tool",
+      "custom_view"
+    ]);
     FunctionId = _v3.z.union([
       _v3.z.object({ function_id: _v3.z.string(), version: _v3.z.string().optional() }),
       _v3.z.object({
@@ -2208,7 +2242,7 @@ var init_generated_types = __esm({
       }),
       _v3.z.object({
         inline_context: _v3.z.object({
-          runtime: _v3.z.enum(["node", "python"]),
+          runtime: _v3.z.enum(["node", "python", "browser"]),
           version: _v3.z.string()
         }),
         code: _v3.z.string(),
@@ -2231,7 +2265,8 @@ var init_generated_types = __esm({
       "tool",
       "scorer",
       "task",
-      "agent"
+      "agent",
+      "custom_view"
     ]);
     FunctionOutputType = _v3.z.enum(["completion", "score", "any"]);
     GitMetadataSettings = _v3.z.object({
@@ -2359,7 +2394,8 @@ var init_generated_types = __esm({
             })
           ),
           _v3.z.null()
-        ])
+        ]),
+        disable_realtime_queries: _v3.z.union([_v3.z.boolean(), _v3.z.null()])
       }).partial(),
       _v3.z.null()
     ]);
@@ -2447,7 +2483,10 @@ var init_generated_types = __esm({
       root_span_id: _v3.z.string(),
       is_root: _v3.z.union([_v3.z.boolean(), _v3.z.null()]).optional(),
       span_attributes: SpanAttributes.optional(),
-      origin: ObjectReferenceNullish.optional()
+      origin: ObjectReferenceNullish.optional(),
+      comments: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional(),
+      audit_data: _v3.z.union([_v3.z.array(_v3.z.unknown()), _v3.z.null()]).optional(),
+      _async_scoring_state: _v3.z.unknown().optional()
     });
     ProjectScoreType = _v3.z.enum([
       "slider",
@@ -2662,7 +2701,7 @@ var init_generated_types = __esm({
       _v3.z.null()
     ]);
     ViewData = _v3.z.union([
-      _v3.z.object({ search: ViewDataSearch }).partial(),
+      _v3.z.object({ search: ViewDataSearch, custom_charts: _v3.z.unknown() }).partial(),
       _v3.z.null()
     ]);
     ViewOptions = _v3.z.union([
@@ -2728,7 +2767,8 @@ var init_generated_types = __esm({
           _v3.z.string(),
           _v3.z.object({ from: _v3.z.string(), to: _v3.z.string() }),
           _v3.z.null()
-        ])
+        ]),
+        queryShape: _v3.z.union([_v3.z.enum(["traces", "spans"]), _v3.z.null()])
       }).partial(),
       _v3.z.null()
     ]);
@@ -2749,7 +2789,8 @@ var init_generated_types = __esm({
         "scorers",
         "logs",
         "agents",
-        "monitor"
+        "monitor",
+        "for_review"
       ]),
       name: _v3.z.string(),
       created: _v3.z.union([_v3.z.string(), _v3.z.null()]).optional(),
@@ -6771,6 +6812,19 @@ function renderMessage(render, message) {
                 url: render(c.image_url.url)
               }
             };
+          case "file":
+            return {
+              ...c,
+              file: {
+                file_data: render(c.file.file_data || ""),
+                ...c.file.file_id && {
+                  file_id: render(c.file.file_id)
+                },
+                ...c.file.filename && {
+                  filename: render(c.file.filename)
+                }
+              }
+            };
           default:
             const _exhaustiveCheck = c;
             return _exhaustiveCheck;
@@ -8529,6 +8583,7 @@ Error: ${errorText}`;
                 },
                 use_columnstore: false,
                 brainstore_realtime: true,
+                query_source: `js_sdk_object_fetcher_${this.objectType}`,
                 ...this.pinnedVersion !== void 0 ? {
                   version: this.pinnedVersion
                 } : {}
@@ -14852,250 +14907,6 @@ function BraintrustMiddleware(config = {}) {
   };
 }
 
-// src/wrappers/mastra/mastra.ts
-init_logger();
-var aiSDKFormatWarning = false;
-function wrapMastraAgent(agent, options) {
-  const prefix = _nullishCoalesce(_nullishCoalesce(_nullishCoalesce(_optionalChain([options, 'optionalAccess', _226 => _226.name]), () => ( _optionalChain([options, 'optionalAccess', _227 => _227.span_name]))), () => ( agent.name)), () => ( "Agent"));
-  if (!hasAllMethods(agent)) {
-    return agent;
-  }
-  if (agent.tools) {
-    agent.__setTools(wrapTools2(agent.tools));
-  }
-  return new Proxy(agent, {
-    get(target, prop, receiver) {
-      const value = Reflect.get(target, prop, receiver);
-      if (prop === "generate" && typeof value === "function") {
-        return wrapGenerate(value, target, prefix);
-      }
-      if (prop === "stream" && typeof value === "function") {
-        return wrapStream(value, target, prefix);
-      }
-      if (typeof value === "function") {
-        return value.bind(target);
-      }
-      return value;
-    }
-  });
-}
-function hasAllMethods(a) {
-  return typeof a.generate === "function" && typeof a.stream === "function";
-}
-function wrapGenerate(original, target, prefix) {
-  return function(...args) {
-    const input = args[0];
-    return traced(
-      async (span) => {
-        const result = await original.apply(target, args);
-        const provider = detectProviderFromResult2(result);
-        const model = extractModelFromResult2(result);
-        const finishReason = normalizeFinishReason2(_optionalChain([result, 'optionalAccess', _228 => _228.finishReason]));
-        const metrics2 = _optionalChain([result, 'optionalAccess', _229 => _229.usage]) ? normalizeUsageMetrics2(
-          result.usage,
-          provider,
-          result.providerMetadata
-        ) : {};
-        span.log({
-          input,
-          output: result,
-          metadata: {
-            agent_name: _nullishCoalesce(target.name, () => ( prefix)),
-            ...provider ? { provider } : {},
-            ...model ? { model } : {},
-            ...finishReason ? { finish_reason: finishReason } : {}
-          },
-          metrics: metrics2
-        });
-        return result;
-      },
-      {
-        name: `${prefix}.generate`
-      }
-    );
-  };
-}
-function wrapStream(original, target, prefix) {
-  return function(...args) {
-    const input = args[0];
-    const span = startSpan({
-      name: `${prefix}.stream`,
-      event: {
-        input,
-        metadata: {
-          agent_name: _nullishCoalesce(target.name, () => ( prefix))
-        }
-      }
-    });
-    const baseOpts = typeof args[1] === "object" && args[1] !== null ? args[1] : {};
-    if (baseOpts.format && baseOpts.format !== "aisdk" && !aiSDKFormatWarning) {
-      aiSDKFormatWarning = true;
-      console.warn(
-        `Braintrust Mastra wrapper: For best compatibility, use { format: 'aisdk' } (AI SDK v5) instead of format: '${baseOpts.format}'. See https://mastra.ai/en/docs/frameworks/agentic-uis/ai-sdk for more details.`
-      );
-    }
-    const wrappedOpts = {
-      ...baseOpts,
-      format: baseOpts.format || "aisdk"
-      // Default to AI SDK v5 format if not specified
-    };
-    const userOnChunk = _optionalChain([baseOpts, 'optionalAccess', _230 => _230.onChunk]);
-    const userOnFinish = _optionalChain([baseOpts, 'optionalAccess', _231 => _231.onFinish]);
-    const userOnError = _optionalChain([baseOpts, 'optionalAccess', _232 => _232.onError]);
-    const startTime = Date.now();
-    let receivedFirst = false;
-    wrappedOpts.onChunk = (chunk) => {
-      try {
-        _optionalChain([userOnChunk, 'optionalCall', _233 => _233(chunk)]);
-      } finally {
-        if (!receivedFirst) {
-          receivedFirst = true;
-          span.log({
-            metrics: { time_to_first_token: (Date.now() - startTime) / 1e3 }
-          });
-        }
-      }
-    };
-    wrappedOpts.onFinish = async (event) => {
-      try {
-        await _optionalChain([userOnFinish, 'optionalCall', _234 => _234(event)]);
-      } finally {
-        const e = event;
-        const provider = detectProviderFromResult2(e);
-        const model = extractModelFromResult2(e);
-        const finishReason = normalizeFinishReason2(_optionalChain([e, 'optionalAccess', _235 => _235.finishReason]));
-        const metrics2 = _optionalChain([e, 'optionalAccess', _236 => _236.usage]) ? normalizeUsageMetrics2(e.usage, provider, e.providerMetadata) : {};
-        span.log({
-          output: _nullishCoalesce(_nullishCoalesce(e.text, () => ( e.content)), () => ( e)),
-          metadata: {
-            agent_name: _nullishCoalesce(target.name, () => ( prefix)),
-            ...provider ? { provider } : {},
-            ...model ? { model } : {},
-            ...finishReason ? { finish_reason: finishReason } : {}
-          },
-          metrics: metrics2
-        });
-        span.end();
-      }
-    };
-    wrappedOpts.onError = async (err) => {
-      try {
-        await _optionalChain([userOnError, 'optionalCall', _237 => _237(err)]);
-      } finally {
-        logError(span, err);
-        span.end();
-      }
-    };
-    return withCurrent(
-      span,
-      () => original.apply(target, [args[0], wrappedOpts, ...args.slice(2)])
-    );
-  };
-}
-function detectProviderFromResult2(result) {
-  if (!_optionalChain([result, 'optionalAccess', _238 => _238.providerMetadata])) {
-    return void 0;
-  }
-  const keys = Object.keys(result.providerMetadata);
-  return _optionalChain([keys, 'optionalAccess', _239 => _239.at, 'call', _240 => _240(0)]);
-}
-function extractModelFromResult2(result) {
-  if (_optionalChain([result, 'optionalAccess', _241 => _241.response, 'optionalAccess', _242 => _242.modelId])) {
-    return result.response.modelId;
-  }
-  if (_optionalChain([result, 'optionalAccess', _243 => _243.request, 'optionalAccess', _244 => _244.body, 'optionalAccess', _245 => _245.model])) {
-    return result.request.body.model;
-  }
-  return void 0;
-}
-function normalizeFinishReason2(reason) {
-  if (typeof reason !== "string") return void 0;
-  return reason.replace(/-/g, "_");
-}
-function normalizeUsageMetrics2(usage, provider, providerMetadata) {
-  const metrics2 = {};
-  const inputTokens = getNumberProperty2(usage, "inputTokens");
-  if (inputTokens !== void 0) {
-    metrics2.prompt_tokens = inputTokens;
-  }
-  const outputTokens = getNumberProperty2(usage, "outputTokens");
-  if (outputTokens !== void 0) {
-    metrics2.completion_tokens = outputTokens;
-  }
-  const totalTokens = getNumberProperty2(usage, "totalTokens");
-  if (totalTokens !== void 0) {
-    metrics2.tokens = totalTokens;
-  }
-  const reasoningTokens = getNumberProperty2(usage, "reasoningTokens");
-  if (reasoningTokens !== void 0) {
-    metrics2.completion_reasoning_tokens = reasoningTokens;
-  }
-  const cachedInputTokens = getNumberProperty2(usage, "cachedInputTokens");
-  if (cachedInputTokens !== void 0) {
-    metrics2.prompt_cached_tokens = cachedInputTokens;
-  }
-  if (provider === "anthropic") {
-    const anthropicMetadata = _optionalChain([providerMetadata, 'optionalAccess', _246 => _246.anthropic]);
-    if (anthropicMetadata) {
-      const cacheReadTokens = getNumberProperty2(anthropicMetadata.usage, "cache_read_input_tokens") || 0;
-      const cacheCreationTokens = getNumberProperty2(
-        anthropicMetadata.usage,
-        "cache_creation_input_tokens"
-      ) || 0;
-      const cacheTokens = extractAnthropicCacheTokens(
-        cacheReadTokens,
-        cacheCreationTokens
-      );
-      Object.assign(metrics2, cacheTokens);
-      Object.assign(metrics2, finalizeAnthropicTokens(metrics2));
-    }
-  }
-  return metrics2;
-}
-function wrapTools2(tools) {
-  if (!tools) return tools;
-  const inferName = (tool, fallback2) => tool && (tool.name || tool.toolName || tool.id) || fallback2;
-  if (Array.isArray(tools)) {
-    const arr = tools;
-    const out = arr.map((tool, idx) => {
-      if (tool != null && typeof tool === "object" && "execute" in tool && typeof tool.execute === "function") {
-        const name = inferName(tool, `tool[${idx}]`);
-        return {
-          ...tool,
-          execute: wrapTraced(tool.execute.bind(tool), {
-            name,
-            type: "tool"
-          })
-        };
-      }
-      return tool;
-    });
-    return out;
-  }
-  const wrappedTools = {};
-  for (const [key, tool] of Object.entries(tools)) {
-    if (tool != null && typeof tool === "object" && "execute" in tool && typeof tool.execute === "function") {
-      wrappedTools[key] = {
-        ...tool,
-        execute: wrapTraced(tool.execute.bind(tool), {
-          name: key,
-          type: "tool"
-        })
-      };
-    } else {
-      wrappedTools[key] = tool;
-    }
-  }
-  return wrappedTools;
-}
-function getNumberProperty2(obj, key) {
-  if (!obj || typeof obj !== "object" || !(key in obj)) {
-    return void 0;
-  }
-  const value = Reflect.get(obj, key);
-  return typeof value === "number" ? value : void 0;
-}
-
 // src/wrappers/anthropic.ts
 init_logger();
 init_util();
@@ -15171,10 +14982,14 @@ function createProxy(create) {
       const apiPromise = Reflect.apply(target, thisArg, argArray);
       const onThen = function(msgOrStream) {
         if (!args["stream"]) {
+          const ttft = getCurrentUnixTimestamp() - sspan.startTime;
           const event = parseEventFromMessage(msgOrStream);
           span.log({
             ...event,
-            metrics: event.metrics ? finalizeAnthropicTokens(event.metrics) : void 0
+            metrics: event.metrics ? finalizeAnthropicTokens({
+              ...event.metrics,
+              time_to_first_token: ttft
+            }) : { time_to_first_token: ttft }
           });
           span.end();
           return msgOrStream;
@@ -15266,9 +15081,9 @@ function streamNextProxy(stream, sspan) {
       return result;
     }
     const item = result.value;
-    switch (_optionalChain([item, 'optionalAccess', _247 => _247.type])) {
+    switch (_optionalChain([item, 'optionalAccess', _226 => _226.type])) {
       case "message_start":
-        const msg = _optionalChain([item, 'optionalAccess', _248 => _248.message]);
+        const msg = _optionalChain([item, 'optionalAccess', _227 => _227.message]);
         if (msg) {
           const event = parseEventFromMessage(msg);
           totals = { ...totals, ...event.metrics };
@@ -15281,20 +15096,20 @@ function streamNextProxy(stream, sspan) {
         }
         break;
       case "content_block_delta":
-        if (_optionalChain([item, 'access', _249 => _249.delta, 'optionalAccess', _250 => _250.type]) === "text_delta") {
-          const text = _optionalChain([item, 'optionalAccess', _251 => _251.delta, 'optionalAccess', _252 => _252.text]);
+        if (_optionalChain([item, 'access', _228 => _228.delta, 'optionalAccess', _229 => _229.type]) === "text_delta") {
+          const text = _optionalChain([item, 'optionalAccess', _230 => _230.delta, 'optionalAccess', _231 => _231.text]);
           if (text) {
             deltas.push(text);
           }
         }
         break;
       case "message_delta":
-        const usage = _optionalChain([item, 'optionalAccess', _253 => _253.usage]);
+        const usage = _optionalChain([item, 'optionalAccess', _232 => _232.usage]);
         if (usage) {
           const metrics2 = parseMetricsFromUsage2(usage);
           totals = { ...totals, ...metrics2 };
         }
-        const delta = _optionalChain([item, 'optionalAccess', _254 => _254.delta]);
+        const delta = _optionalChain([item, 'optionalAccess', _233 => _233.delta]);
         if (delta) {
           metadata = { ...metadata, ...delta };
         }
@@ -15307,7 +15122,7 @@ function streamNextProxy(stream, sspan) {
 }
 function parseEventFromMessage(message) {
   const output = message ? { role: message.role, content: message.content } : null;
-  const metrics2 = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _255 => _255.usage]));
+  const metrics2 = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _234 => _234.usage]));
   const metas = ["stop_reason", "stop_sequence"];
   const metadata = {};
   for (const m of metas) {
@@ -15390,6 +15205,11 @@ function coalesceInput(messages, system) {
   return input;
 }
 
+// src/wrappers/mastra.ts
+function wrapMastraAgent(agent, _options) {
+  return agent;
+}
+
 // src/wrappers/claude-agent-sdk/claude-agent-sdk.ts
 init_logger();
 init_util2();
@@ -15452,8 +15272,8 @@ function wrapClaudeAgentQuery(queryFn, defaultThis) {
           finalResults.push(finalMessageContent);
         }
         const lastMessage = currentMessages[currentMessages.length - 1];
-        if (_optionalChain([lastMessage, 'optionalAccess', _256 => _256.message, 'optionalAccess', _257 => _257.usage])) {
-          const outputTokens = getNumberProperty3(lastMessage.message.usage, "output_tokens") || 0;
+        if (_optionalChain([lastMessage, 'optionalAccess', _235 => _235.message, 'optionalAccess', _236 => _236.usage])) {
+          const outputTokens = getNumberProperty2(lastMessage.message.usage, "output_tokens") || 0;
           accumulatedOutputTokens += outputTokens;
         }
         currentMessages.length = 0;
@@ -15467,20 +15287,20 @@ function wrapClaudeAgentQuery(queryFn, defaultThis) {
           );
           for await (const message of generator) {
             const currentTime = getCurrentUnixTimestamp();
-            const messageId = _optionalChain([message, 'access', _258 => _258.message, 'optionalAccess', _259 => _259.id]);
+            const messageId = _optionalChain([message, 'access', _237 => _237.message, 'optionalAccess', _238 => _238.id]);
             if (messageId && messageId !== currentMessageId) {
               await createLLMSpan();
               currentMessageId = messageId;
               currentMessageStartTime = currentTime;
             }
-            if (message.type === "assistant" && _optionalChain([message, 'access', _260 => _260.message, 'optionalAccess', _261 => _261.usage])) {
+            if (message.type === "assistant" && _optionalChain([message, 'access', _239 => _239.message, 'optionalAccess', _240 => _240.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', _262 => _262.message, 'optionalAccess', _263 => _263.usage])) {
+                if (_optionalChain([lastMessage, 'optionalAccess', _241 => _241.message, 'optionalAccess', _242 => _242.usage])) {
                   const adjustedTokens = finalUsageMetrics.completion_tokens - accumulatedOutputTokens;
                   if (adjustedTokens >= 0) {
                     lastMessage.message.usage.output_tokens = adjustedTokens;
@@ -15561,23 +15381,23 @@ function _extractUsageFromMessage(message) {
   const metrics2 = {};
   let usage;
   if (message.type === "assistant") {
-    usage = _optionalChain([message, 'access', _264 => _264.message, 'optionalAccess', _265 => _265.usage]);
+    usage = _optionalChain([message, 'access', _243 => _243.message, 'optionalAccess', _244 => _244.usage]);
   } else if (message.type === "result") {
     usage = message.usage;
   }
   if (!usage || typeof usage !== "object") {
     return metrics2;
   }
-  const inputTokens = getNumberProperty3(usage, "input_tokens");
+  const inputTokens = getNumberProperty2(usage, "input_tokens");
   if (inputTokens !== void 0) {
     metrics2.prompt_tokens = inputTokens;
   }
-  const outputTokens = getNumberProperty3(usage, "output_tokens");
+  const outputTokens = getNumberProperty2(usage, "output_tokens");
   if (outputTokens !== void 0) {
     metrics2.completion_tokens = outputTokens;
   }
-  const cacheReadTokens = getNumberProperty3(usage, "cache_read_input_tokens") || 0;
-  const cacheCreationTokens = getNumberProperty3(usage, "cache_creation_input_tokens") || 0;
+  const cacheReadTokens = getNumberProperty2(usage, "cache_read_input_tokens") || 0;
+  const cacheCreationTokens = getNumberProperty2(usage, "cache_creation_input_tokens") || 0;
   if (cacheReadTokens > 0 || cacheCreationTokens > 0) {
     const cacheTokens = extractAnthropicCacheTokens(
       cacheReadTokens,
@@ -15593,14 +15413,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', _266 => _266.message, 'optionalAccess', _267 => _267.usage])) {
+  if (lastMessage.type !== "assistant" || !_optionalChain([lastMessage, 'access', _245 => _245.message, 'optionalAccess', _246 => _246.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', _268 => _268.message, 'optionalAccess', _269 => _269.content]) && _optionalChain([m, 'access', _270 => _270.message, 'optionalAccess', _271 => _271.role]) ? { content: m.message.content, role: m.message.role } : void 0
+    (m) => _optionalChain([m, 'access', _247 => _247.message, 'optionalAccess', _248 => _248.content]) && _optionalChain([m, 'access', _249 => _249.message, 'optionalAccess', _250 => _250.role]) ? { content: m.message.content, role: m.message.role } : void 0
   ).filter((c) => c !== void 0);
   await traced(
     (llmSpan) => {
@@ -15620,7 +15440,7 @@ async function _createLLMSpanForMessages(messages, prompt, conversationHistory,
       parent: parentSpan
     }
   );
-  return _optionalChain([lastMessage, 'access', _272 => _272.message, 'optionalAccess', _273 => _273.content]) && _optionalChain([lastMessage, 'access', _274 => _274.message, 'optionalAccess', _275 => _275.role]) ? { content: lastMessage.message.content, role: lastMessage.message.role } : void 0;
+  return _optionalChain([lastMessage, 'access', _251 => _251.message, 'optionalAccess', _252 => _252.content]) && _optionalChain([lastMessage, 'access', _253 => _253.message, 'optionalAccess', _254 => _254.role]) ? { content: lastMessage.message.content, role: lastMessage.message.role } : void 0;
 }
 function wrapClaudeAgentSDK(sdk) {
   const cache = /* @__PURE__ */ new Map();
@@ -15668,7 +15488,7 @@ function wrapClaudeAgentSDK(sdk) {
     }
   });
 }
-function getNumberProperty3(obj, key) {
+function getNumberProperty2(obj, key) {
   if (!obj || typeof obj !== "object" || !(key in obj)) {
     return void 0;
   }
@@ -15913,7 +15733,7 @@ function serializePart(part) {
   return part;
 }
 function serializeTools(params) {
-  if (!_optionalChain([params, 'access', _276 => _276.config, 'optionalAccess', _277 => _277.tools])) {
+  if (!_optionalChain([params, 'access', _255 => _255.config, 'optionalAccess', _256 => _256.tools])) {
     return null;
   }
   try {
@@ -15996,7 +15816,7 @@ function aggregateGenerateContentChunks(chunks, start, firstTokenTime) {
     }
     if (chunk.candidates && Array.isArray(chunk.candidates)) {
       for (const candidate of chunk.candidates) {
-        if (_optionalChain([candidate, 'access', _278 => _278.content, 'optionalAccess', _279 => _279.parts])) {
+        if (_optionalChain([candidate, 'access', _257 => _257.content, 'optionalAccess', _258 => _258.parts])) {
           for (const part of candidate.content.parts) {
             if (part.text !== void 0) {
               if (part.thought) {
@@ -16027,7 +15847,7 @@ function aggregateGenerateContentChunks(chunks, start, firstTokenTime) {
     parts.push({ text });
   }
   parts.push(...otherParts);
-  if (parts.length > 0 && _optionalChain([lastResponse, 'optionalAccess', _280 => _280.candidates])) {
+  if (parts.length > 0 && _optionalChain([lastResponse, 'optionalAccess', _259 => _259.candidates])) {
     const candidates = [];
     for (const candidate of lastResponse.candidates) {
       const candidateDict = {
@@ -16256,8 +16076,8 @@ var BraintrustSpanProcessor = class _BraintrustSpanProcessor {
                   span.instrumentationScope = span.instrumentationLibrary;
                 }
                 if (!span.parentSpanContext && span.parentSpanId) {
-                  const spanContext = _optionalChain([span, 'access', _281 => _281.spanContext, 'optionalCall', _282 => _282()]);
-                  if (_optionalChain([spanContext, 'optionalAccess', _283 => _283.traceId])) {
+                  const spanContext = _optionalChain([span, 'access', _260 => _260.spanContext, 'optionalCall', _261 => _261()]);
+                  if (_optionalChain([spanContext, 'optionalAccess', _262 => _262.traceId])) {
                     span.parentSpanContext = {
                       spanId: span.parentSpanId,
                       traceId: spanContext.traceId
@@ -16300,7 +16120,7 @@ var BraintrustSpanProcessor = class _BraintrustSpanProcessor {
       let parentValue;
       if (otelApi && otelApi.context) {
         const currentContext = otelApi.context.active();
-        const contextValue = _optionalChain([currentContext, 'access', _284 => _284.getValue, 'optionalCall', _285 => _285("braintrust.parent")]);
+        const contextValue = _optionalChain([currentContext, 'access', _263 => _263.getValue, 'optionalCall', _264 => _264("braintrust.parent")]);
         if (typeof contextValue === "string") {
           parentValue = contextValue;
         }
@@ -16364,7 +16184,7 @@ function otelContextFromSpanExport(exportStr) {
     traceId: traceIdHex,
     spanId: spanIdHex,
     isRemote: true,
-    traceFlags: _nullishCoalesce(_optionalChain([otelApi, 'access', _286 => _286.TraceFlags, 'optionalAccess', _287 => _287.SAMPLED]), () => ( 1))
+    traceFlags: _nullishCoalesce(_optionalChain([otelApi, 'access', _265 => _265.TraceFlags, 'optionalAccess', _266 => _266.SAMPLED]), () => ( 1))
     // SAMPLED flag
   };
   const nonRecordingSpan = otelTrace2.wrapSpanContext(spanContext);
@@ -16469,7 +16289,7 @@ var BraintrustExporter = (_class21 = class _BraintrustExporter {
 function addParentToBaggage(parent, ctx) {
   if (!OTEL_AVAILABLE2 || !otelApi) {
     console.error("OpenTelemetry not available");
-    return ctx || _optionalChain([otelApi, 'optionalAccess', _288 => _288.context, 'access', _289 => _289.active, 'call', _290 => _290()]);
+    return ctx || _optionalChain([otelApi, 'optionalAccess', _267 => _267.context, 'access', _268 => _268.active, 'call', _269 => _269()]);
   }
   try {
     const propagation2 = otelApi.propagation;
@@ -16530,7 +16350,7 @@ function parentFromHeaders(headers) {
       return void 0;
     }
     const baggage = propagation2.getBaggage(ctx);
-    const braintrustParent = _optionalChain([baggage, 'optionalAccess', _291 => _291.getEntry, 'call', _292 => _292("braintrust.parent"), 'optionalAccess', _293 => _293.value]);
+    const braintrustParent = _optionalChain([baggage, 'optionalAccess', _270 => _270.getEntry, 'call', _271 => _271("braintrust.parent"), 'optionalAccess', _272 => _272.value]);
     if (!braintrustParent) {
       console.warn(
         "parentFromHeaders: braintrust.parent not found in OTEL baggage. Cannot create Braintrust parent without project information. Ensure the OTEL span sets braintrust.parent in baggage before exporting headers."