braintrust 0.2.0 → 0.2.1

package/dev/dist/index.js

@@ -6677,6 +6677,8 @@ var evalBodySchema = _zod.z.object({
       name: _zod.z.string()
     })
   ).nullish(),
+  experiment_name: _zod.z.string().nullish(),
+  project_id: _zod.z.string().nullish(),
   parent: _typespecs.invokeParent.optional(),
   stream: _zod.z.boolean().optional()
 });
@@ -6759,7 +6761,16 @@ function runDevServer(evaluators, opts) {
     "/eval",
     checkAuthorized,
     asyncHandler(async (req, res) => {
-      const { name, parameters, parent, data, scores, stream } = evalBodySchema.parse(req.body);
+      const {
+        name,
+        parameters,
+        parent,
+        experiment_name,
+        project_id,
+        data,
+        scores,
+        stream
+      } = evalBodySchema.parse(req.body);
       const state = await cachedLogin({ apiKey: _optionalChain([req, 'access', _88 => _88.ctx, 'optionalAccess', _89 => _89.token]) });
       const evaluator = allEvaluators[name];
       if (!evaluator) {
@@ -6818,17 +6829,19 @@ function runDevServer(evaluators, opts) {
               )]), () => ( []))
             ),
             task,
-            state
+            state,
+            experimentName: _nullishCoalesce(experiment_name, () => ( void 0)),
+            projectId: _nullishCoalesce(project_id, () => ( void 0))
           },
           {
             // Avoid printing the bar to the console.
             progress: {
-              start: (name2, total) => {
+              start: () => {
               },
               stop: () => {
                 console.log("Finished running experiment");
               },
-              increment: (name2) => {
+              increment: () => {
               }
             },
             stream: (data2) => {

package/dev/dist/index.mjs

@@ -6677,6 +6677,8 @@ var evalBodySchema = z6.object({
       name: z6.string()
     })
   ).nullish(),
+  experiment_name: z6.string().nullish(),
+  project_id: z6.string().nullish(),
   parent: invokeParent.optional(),
   stream: z6.boolean().optional()
 });
@@ -6759,7 +6761,16 @@ function runDevServer(evaluators, opts) {
     "/eval",
     checkAuthorized,
     asyncHandler(async (req, res) => {
-      const { name, parameters, parent, data, scores, stream } = evalBodySchema.parse(req.body);
+      const {
+        name,
+        parameters,
+        parent,
+        experiment_name,
+        project_id,
+        data,
+        scores,
+        stream
+      } = evalBodySchema.parse(req.body);
       const state = await cachedLogin({ apiKey: req.ctx?.token });
       const evaluator = allEvaluators[name];
       if (!evaluator) {
@@ -6818,17 +6829,19 @@ function runDevServer(evaluators, opts) {
               ) ?? []
             ),
             task,
-            state
+            state,
+            experimentName: experiment_name ?? void 0,
+            projectId: project_id ?? void 0
           },
           {
             // Avoid printing the bar to the console.
             progress: {
-              start: (name2, total) => {
+              start: () => {
               },
               stop: () => {
                 console.log("Finished running experiment");
               },
-              increment: (name2) => {
+              increment: () => {
               }
             },
             stream: (data2) => {

package/dist/browser.d.mts

@@ -1690,6 +1690,7 @@ interface DatasetSummary {
     datasetUrl: string;
     dataSummary: DataSummary | undefined;
 }
+declare function setInitialTestState(): void;
 declare function simulateLoginForTests(): Promise<BraintrustState>;
 declare function simulateLogoutForTests(): BraintrustState;
 declare const _exportsForTestingOnly: {
@@ -1699,6 +1700,7 @@ declare const _exportsForTestingOnly: {
     clearTestBackgroundLogger: typeof clearTestBackgroundLogger;
     simulateLoginForTests: typeof simulateLoginForTests;
     simulateLogoutForTests: typeof simulateLogoutForTests;
+    setInitialTestState: typeof setInitialTestState;
 };
 
 declare const braintrustStreamChunkSchema: z.ZodUnion<[z.ZodObject<{
@@ -2109,7 +2111,7 @@ declare global {
  * not configured, nothing will be traced. If this is not an `OpenAI` object, this function is
  * a no-op.
  *
- * Currently, this only supports the `v4` API.
+ * Currently, this supports both the `v4` and `v5` API.
  *
  * @param openai
  * @returns The wrapped `OpenAI` object.

package/dist/browser.d.ts

@@ -1690,6 +1690,7 @@ interface DatasetSummary {
     datasetUrl: string;
     dataSummary: DataSummary | undefined;
 }
+declare function setInitialTestState(): void;
 declare function simulateLoginForTests(): Promise<BraintrustState>;
 declare function simulateLogoutForTests(): BraintrustState;
 declare const _exportsForTestingOnly: {
@@ -1699,6 +1700,7 @@ declare const _exportsForTestingOnly: {
     clearTestBackgroundLogger: typeof clearTestBackgroundLogger;
     simulateLoginForTests: typeof simulateLoginForTests;
     simulateLogoutForTests: typeof simulateLogoutForTests;
+    setInitialTestState: typeof setInitialTestState;
 };
 
 declare const braintrustStreamChunkSchema: z.ZodUnion<[z.ZodObject<{
@@ -2109,7 +2111,7 @@ declare global {
  * not configured, nothing will be traced. If this is not an `OpenAI` object, this function is
  * a no-op.
  *
- * Currently, this only supports the `v4` API.
+ * Currently, this supports both the `v4` and `v5` API.
  *
  * @param openai
  * @returns The wrapped `OpenAI` object.

package/dist/browser.js

@@ -4563,6 +4563,11 @@ var Prompt = (_class13 = class _Prompt {
   }
 }, _class13);
 var TEST_API_KEY = "___TEST_API_KEY__THIS_IS_NOT_REAL___";
+function setInitialTestState() {
+  if (!_internalGetGlobalState()) {
+    _internalSetInitialState();
+  }
+}
 async function simulateLoginForTests() {
   return await login({
     apiKey: TEST_API_KEY,
@@ -4580,7 +4585,8 @@ var _exportsForTestingOnly = {
   useTestBackgroundLogger,
   clearTestBackgroundLogger,
   simulateLoginForTests,
-  simulateLogoutForTests
+  simulateLogoutForTests,
+  setInitialTestState
 };
 
 // src/browser-config.ts
@@ -5030,10 +5036,17 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  const completionProxy = createEndpointProxy(
-    openai.chat.completions,
-    wrapChatCompletion
-  );
+  const completionProxy = new Proxy(openai.chat.completions, {
+    get(target, name, receiver) {
+      const baseVal = Reflect.get(target, name, receiver);
+      if (name === "create") {
+        return wrapChatCompletion(baseVal.bind(target));
+      } else if (name === "parse") {
+        return wrapBetaChatCompletionParse(baseVal.bind(target));
+      }
+      return baseVal;
+    }
+  });
   const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
@@ -5427,6 +5440,8 @@ var evalBodySchema = _zod.z.object({
       name: _zod.z.string()
     })
   ).nullish(),
+  experiment_name: _zod.z.string().nullish(),
+  project_id: _zod.z.string().nullish(),
   parent: _typespecs.invokeParent.optional(),
   stream: _zod.z.boolean().optional()
 });

package/dist/browser.mjs

@@ -4563,6 +4563,11 @@ var Prompt = class _Prompt {
   }
 };
 var TEST_API_KEY = "___TEST_API_KEY__THIS_IS_NOT_REAL___";
+function setInitialTestState() {
+  if (!_internalGetGlobalState()) {
+    _internalSetInitialState();
+  }
+}
 async function simulateLoginForTests() {
   return await login({
     apiKey: TEST_API_KEY,
@@ -4580,7 +4585,8 @@ var _exportsForTestingOnly = {
   useTestBackgroundLogger,
   clearTestBackgroundLogger,
   simulateLoginForTests,
-  simulateLogoutForTests
+  simulateLogoutForTests,
+  setInitialTestState
 };
 
 // src/browser-config.ts
@@ -5030,10 +5036,17 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  const completionProxy = createEndpointProxy(
-    openai.chat.completions,
-    wrapChatCompletion
-  );
+  const completionProxy = new Proxy(openai.chat.completions, {
+    get(target, name, receiver) {
+      const baseVal = Reflect.get(target, name, receiver);
+      if (name === "create") {
+        return wrapChatCompletion(baseVal.bind(target));
+      } else if (name === "parse") {
+        return wrapBetaChatCompletionParse(baseVal.bind(target));
+      }
+      return baseVal;
+    }
+  });
   const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
@@ -5427,6 +5440,8 @@ var evalBodySchema = z3.object({
       name: z3.string()
     })
   ).nullish(),
+  experiment_name: z3.string().nullish(),
+  project_id: z3.string().nullish(),
   parent: invokeParent.optional(),
   stream: z3.boolean().optional()
 });

package/dist/cli.js

@@ -1232,7 +1232,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "0.2.0",
+      version: "0.2.1",
       description: "SDK for integrating Braintrust",
       repository: {
         type: "git",
@@ -1283,11 +1283,13 @@ var require_package = __commonJS({
         docs: "npx typedoc --options typedoc.json src/index.ts",
         prepublishOnly: "../../scripts/node_prepublish_sdk.py",
         postpublish: "../../scripts/node_postpublish_sdk.py",
-        test: "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts",
+        test: "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts  --exclude src/wrappers/ai-sdk-v1.test.ts --exclude src/wrappers/ai-sdk-v2.test.ts",
         "test:anthropic": "vitest run src/wrappers/anthropic.test.ts",
         "test:openai": "vitest run src/wrappers/oai.test.ts",
         "test:otel": "vitest run src/otel.test.ts",
-        "test:otel-no-deps": "vitest run src/otel-no-deps.test.ts --reporter=verbose"
+        "test:otel-no-deps": "vitest run src/otel-no-deps.test.ts --reporter=verbose",
+        "test:ai-sdk-v1": "vitest run src/wrappers/ai-sdk-v1.test.ts",
+        "test:ai-sdk-v2": "vitest run src/wrappers/ai-sdk-v2.test.ts src/wrappers/ai-sdk-v1.test.ts"
       },
       author: "",
       license: "MIT",
@@ -8875,6 +8877,8 @@ var evalBodySchema = import_zod8.z.object({
       name: import_zod8.z.string()
     })
   ).nullish(),
+  experiment_name: import_zod8.z.string().nullish(),
+  project_id: import_zod8.z.string().nullish(),
   parent: import_typespecs6.invokeParent.optional(),
   stream: import_zod8.z.boolean().optional()
 });
@@ -8957,7 +8961,16 @@ function runDevServer(evaluators, opts) {
     "/eval",
     checkAuthorized,
     asyncHandler(async (req, res) => {
-      const { name, parameters, parent, data, scores, stream } = evalBodySchema.parse(req.body);
+      const {
+        name,
+        parameters,
+        parent,
+        experiment_name,
+        project_id,
+        data,
+        scores,
+        stream
+      } = evalBodySchema.parse(req.body);
       const state = await cachedLogin({ apiKey: req.ctx?.token });
       const evaluator = allEvaluators[name];
       if (!evaluator) {
@@ -9016,17 +9029,19 @@ function runDevServer(evaluators, opts) {
               ) ?? []
             ),
             task,
-            state
+            state,
+            experimentName: experiment_name ?? void 0,
+            projectId: project_id ?? void 0
           },
           {
             // Avoid printing the bar to the console.
             progress: {
-              start: (name2, total) => {
+              start: () => {
               },
               stop: () => {
                 console.log("Finished running experiment");
               },
-              increment: (name2) => {
+              increment: () => {
               }
             },
             stream: (data2) => {

package/dist/index.d.mts

@@ -1691,6 +1691,7 @@ interface DatasetSummary {
     datasetUrl: string;
     dataSummary: DataSummary | undefined;
 }
+declare function setInitialTestState(): void;
 declare function simulateLoginForTests(): Promise<BraintrustState>;
 declare function simulateLogoutForTests(): BraintrustState;
 declare const _exportsForTestingOnly: {
@@ -1700,6 +1701,7 @@ declare const _exportsForTestingOnly: {
     clearTestBackgroundLogger: typeof clearTestBackgroundLogger;
     simulateLoginForTests: typeof simulateLoginForTests;
     simulateLogoutForTests: typeof simulateLogoutForTests;
+    setInitialTestState: typeof setInitialTestState;
 };
 
 declare const braintrustStreamChunkSchema: z.ZodUnion<[z.ZodObject<{
@@ -7134,7 +7136,7 @@ declare global {
  * not configured, nothing will be traced. If this is not an `OpenAI` object, this function is
  * a no-op.
  *
- * Currently, this only supports the `v4` API.
+ * Currently, this supports both the `v4` and `v5` API.
  *
  * @param openai
  * @returns The wrapped `OpenAI` object.
@@ -7145,6 +7147,50 @@ declare const LEGACY_CACHED_HEADER = "x-cached";
 declare const X_CACHED_HEADER = "x-bt-cached";
 declare function parseCachedHeader(value: string | null | undefined): number | undefined;
 
+interface LanguageModelV2Middleware<TModel = any, TCallOptions = any> {
+    wrapGenerate?: (options: {
+        doGenerate: () => any;
+        doStream: () => any;
+        params: TCallOptions;
+        model: TModel;
+    }) => Promise<any>;
+    wrapStream?: (options: {
+        doGenerate: () => any;
+        doStream: () => any;
+        params: TCallOptions;
+        model: TModel;
+    }) => Promise<any>;
+}
+/**
+ * Configuration options for the AI SDK middleware
+ */
+interface MiddlewareConfig {
+    /** Enable debug logging */
+    debug?: boolean;
+    /** Name identifier for the middleware instance */
+    name?: string;
+}
+/**
+ * Creates a Braintrust middleware for AI SDK v2 that automatically traces
+ * generateText and streamText calls with comprehensive metadata and metrics.
+ *
+ * @param config - Configuration options for the middleware
+ * @returns A middleware object compatible with AI SDK v2's wrapLanguageModel
+ *
+ * @example
+ * ```typescript
+ * import { wrapLanguageModel } from "ai";
+ * import { openai } from "@ai-sdk/openai";
+ * import { BraintrustMiddleware } from "braintrust";
+ *
+ * const model = wrapLanguageModel({
+ *   model: openai("gpt-4"),
+ *   middleware: BraintrustMiddleware({ debug: true, name: "MyMiddleware" })
+ * });
+ * ```
+ */
+declare function BraintrustMiddleware(config?: MiddlewareConfig): LanguageModelV2Middleware<any, any>;
+
 /**
  * Wrap an ai-sdk model (created with `.chat()`, `.completion()`, etc.) to add tracing. If Braintrust is
  * not configured, this is a no-op
@@ -7406,6 +7452,7 @@ declare const braintrust_BaseExperiment: typeof BaseExperiment;
 type braintrust_BaseMetadata = BaseMetadata;
 type braintrust_BraintrustExporter = BraintrustExporter;
 declare const braintrust_BraintrustExporter: typeof BraintrustExporter;
+declare const braintrust_BraintrustMiddleware: typeof BraintrustMiddleware;
 type braintrust_BraintrustSpanProcessor = BraintrustSpanProcessor;
 declare const braintrust_BraintrustSpanProcessor: typeof BraintrustSpanProcessor;
 type braintrust_BraintrustState = BraintrustState;
@@ -7564,7 +7611,7 @@ declare const braintrust_wrapOpenAI: typeof wrapOpenAI;
 declare const braintrust_wrapOpenAIv4: typeof wrapOpenAIv4;
 declare const braintrust_wrapTraced: typeof wrapTraced;
 declare namespace braintrust {
-  export { braintrust_AISpanProcessor as AISpanProcessor, type braintrust_AnyDataset as AnyDataset, braintrust_Attachment as Attachment, type braintrust_AttachmentParams as AttachmentParams, type braintrust_BackgroundLoggerOpts as BackgroundLoggerOpts, braintrust_BaseAttachment as BaseAttachment, braintrust_BaseExperiment as BaseExperiment, type braintrust_BaseMetadata as BaseMetadata, braintrust_BraintrustExporter as BraintrustExporter, braintrust_BraintrustSpanProcessor as BraintrustSpanProcessor, braintrust_BraintrustState as BraintrustState, braintrust_BraintrustStream as BraintrustStream, type braintrust_BraintrustStreamChunk as BraintrustStreamChunk, type braintrust_ChatPrompt as ChatPrompt, braintrust_CodeFunction as CodeFunction, type braintrust_CodeOpts as CodeOpts, braintrust_CodePrompt as CodePrompt, type braintrust_CompiledPrompt as CompiledPrompt, type braintrust_CompiledPromptParams as CompiledPromptParams, type braintrust_CompletionPrompt as CompletionPrompt, type braintrust_CreateProjectOpts as CreateProjectOpts, type braintrust_DataSummary as DataSummary, braintrust_Dataset as Dataset, type braintrust_DatasetSummary as DatasetSummary, type braintrust_DefaultMetadataType as DefaultMetadataType, type braintrust_DefaultPromptArgs as DefaultPromptArgs, braintrust_ERR_PERMALINK as ERR_PERMALINK, type braintrust_EndSpanArgs as EndSpanArgs, braintrust_Eval as Eval, type braintrust_EvalCase as EvalCase, type braintrust_EvalHooks as EvalHooks, type braintrust_EvalResult as EvalResult, braintrust_EvalResultWithSummary as EvalResultWithSummary, type braintrust_EvalScorer as EvalScorer, type braintrust_EvalScorerArgs as EvalScorerArgs, type braintrust_EvalTask as EvalTask, type braintrust_Evaluator as Evaluator, type braintrust_EvaluatorDef as EvaluatorDef, type braintrust_EvaluatorFile as EvaluatorFile, braintrust_Experiment as Experiment, type braintrust_ExperimentSummary as ExperimentSummary, type braintrust_Exportable as Exportable, braintrust_ExternalAttachment as ExternalAttachment, type braintrust_ExternalAttachmentParams as ExternalAttachmentParams, braintrust_FailedHTTPResponse as FailedHTTPResponse, type braintrust_FullInitOptions as FullInitOptions, type braintrust_FullLoginOptions as FullLoginOptions, type braintrust_FunctionEvent as FunctionEvent, braintrust_INTERNAL_BTQL_LIMIT as INTERNAL_BTQL_LIMIT, type braintrust_InitOptions as InitOptions, type braintrust_InvokeFunctionArgs as InvokeFunctionArgs, type braintrust_InvokeReturn as InvokeReturn, braintrust_LEGACY_CACHED_HEADER as LEGACY_CACHED_HEADER, braintrust_LazyValue as LazyValue, type braintrust_LogOptions as LogOptions, braintrust_Logger as Logger, type braintrust_LoginOptions as LoginOptions, type braintrust_MetricSummary as MetricSummary, braintrust_NOOP_SPAN as NOOP_SPAN, braintrust_NOOP_SPAN_PERMALINK as NOOP_SPAN_PERMALINK, braintrust_NoopSpan as NoopSpan, type braintrust_ObjectMetadata as ObjectMetadata, braintrust_Project as Project, braintrust_ProjectNameIdMap as ProjectNameIdMap, type braintrust_PromiseUnless as PromiseUnless, braintrust_Prompt as Prompt, braintrust_PromptBuilder as PromptBuilder, type braintrust_PromptContents as PromptContents, type braintrust_PromptDefinition as PromptDefinition, type braintrust_PromptDefinitionWithTools as PromptDefinitionWithTools, type braintrust_PromptOpts as PromptOpts, type braintrust_PromptRowWithId as PromptRowWithId, braintrust_ReadonlyAttachment as ReadonlyAttachment, braintrust_ReadonlyExperiment as ReadonlyExperiment, braintrust_Reporter as Reporter, type braintrust_ReporterBody as ReporterBody, type braintrust_ScoreSummary as ScoreSummary, braintrust_ScorerBuilder as ScorerBuilder, type braintrust_ScorerOpts as ScorerOpts, type braintrust_SerializedBraintrustState as SerializedBraintrustState, type braintrust_SetCurrentArg as SetCurrentArg, type Span$1 as Span, type braintrust_SpanContext as SpanContext, braintrust_SpanImpl as SpanImpl, type braintrust_StartSpanArgs as StartSpanArgs, braintrust_TestBackgroundLogger as TestBackgroundLogger, braintrust_ToolBuilder as ToolBuilder, type braintrust_WithTransactionId as WithTransactionId, braintrust_X_CACHED_HEADER as X_CACHED_HEADER, braintrust__exportsForTestingOnly as _exportsForTestingOnly, braintrust__internalGetGlobalState as _internalGetGlobalState, braintrust__internalSetInitialState as _internalSetInitialState, braintrust_braintrustStreamChunkSchema as braintrustStreamChunkSchema, braintrust_buildLocalSummary as buildLocalSummary, braintrust_createFinalValuePassThroughStream as createFinalValuePassThroughStream, braintrust_currentExperiment as currentExperiment, braintrust_currentLogger as currentLogger, braintrust_currentSpan as currentSpan, braintrust_defaultErrorScoreHandler as defaultErrorScoreHandler, braintrust_deserializePlainStringAsJSON as deserializePlainStringAsJSON, braintrust_devNullWritableStream as devNullWritableStream, braintrust_flush as flush, braintrust_getSpanParentObject as getSpanParentObject, graphFramework as graph, braintrust_init as init, braintrust_initDataset as initDataset, braintrust_initExperiment as initExperiment, braintrust_initFunction as initFunction, braintrust_initLogger as initLogger, braintrust_invoke as invoke, braintrust_loadPrompt as loadPrompt, braintrust_log as log, braintrust_logError as logError, braintrust_login as login, braintrust_loginToState as loginToState, braintrust_newId as newId, braintrust_parseCachedHeader as parseCachedHeader, braintrust_permalink as permalink, braintrust_projects as projects, braintrust_promptContentsSchema as promptContentsSchema, braintrust_promptDefinitionSchema as promptDefinitionSchema, braintrust_promptDefinitionToPromptData as promptDefinitionToPromptData, braintrust_promptDefinitionWithToolsSchema as promptDefinitionWithToolsSchema, braintrust_renderMessage as renderMessage, braintrust_renderPromptParams as renderPromptParams, braintrust_reportFailures as reportFailures, braintrust_runEvaluator as runEvaluator, braintrust_setFetch as setFetch, braintrust_spanComponentsToObjectId as spanComponentsToObjectId, braintrust_startSpan as startSpan, braintrust_summarize as summarize, braintrust_toolFunctionDefinitionSchema as toolFunctionDefinitionSchema, braintrust_traceable as traceable, braintrust_traced as traced, braintrust_updateSpan as updateSpan, braintrust_withCurrent as withCurrent, braintrust_withDataset as withDataset, braintrust_withExperiment as withExperiment, braintrust_withLogger as withLogger, braintrust_withParent as withParent, braintrust_wrapAISDKModel as wrapAISDKModel, braintrust_wrapAnthropic as wrapAnthropic, braintrust_wrapOpenAI as wrapOpenAI, braintrust_wrapOpenAIv4 as wrapOpenAIv4, braintrust_wrapTraced as wrapTraced };
+  export { braintrust_AISpanProcessor as AISpanProcessor, type braintrust_AnyDataset as AnyDataset, braintrust_Attachment as Attachment, type braintrust_AttachmentParams as AttachmentParams, type braintrust_BackgroundLoggerOpts as BackgroundLoggerOpts, braintrust_BaseAttachment as BaseAttachment, braintrust_BaseExperiment as BaseExperiment, type braintrust_BaseMetadata as BaseMetadata, braintrust_BraintrustExporter as BraintrustExporter, braintrust_BraintrustMiddleware as BraintrustMiddleware, braintrust_BraintrustSpanProcessor as BraintrustSpanProcessor, braintrust_BraintrustState as BraintrustState, braintrust_BraintrustStream as BraintrustStream, type braintrust_BraintrustStreamChunk as BraintrustStreamChunk, type braintrust_ChatPrompt as ChatPrompt, braintrust_CodeFunction as CodeFunction, type braintrust_CodeOpts as CodeOpts, braintrust_CodePrompt as CodePrompt, type braintrust_CompiledPrompt as CompiledPrompt, type braintrust_CompiledPromptParams as CompiledPromptParams, type braintrust_CompletionPrompt as CompletionPrompt, type braintrust_CreateProjectOpts as CreateProjectOpts, type braintrust_DataSummary as DataSummary, braintrust_Dataset as Dataset, type braintrust_DatasetSummary as DatasetSummary, type braintrust_DefaultMetadataType as DefaultMetadataType, type braintrust_DefaultPromptArgs as DefaultPromptArgs, braintrust_ERR_PERMALINK as ERR_PERMALINK, type braintrust_EndSpanArgs as EndSpanArgs, braintrust_Eval as Eval, type braintrust_EvalCase as EvalCase, type braintrust_EvalHooks as EvalHooks, type braintrust_EvalResult as EvalResult, braintrust_EvalResultWithSummary as EvalResultWithSummary, type braintrust_EvalScorer as EvalScorer, type braintrust_EvalScorerArgs as EvalScorerArgs, type braintrust_EvalTask as EvalTask, type braintrust_Evaluator as Evaluator, type braintrust_EvaluatorDef as EvaluatorDef, type braintrust_EvaluatorFile as EvaluatorFile, braintrust_Experiment as Experiment, type braintrust_ExperimentSummary as ExperimentSummary, type braintrust_Exportable as Exportable, braintrust_ExternalAttachment as ExternalAttachment, type braintrust_ExternalAttachmentParams as ExternalAttachmentParams, braintrust_FailedHTTPResponse as FailedHTTPResponse, type braintrust_FullInitOptions as FullInitOptions, type braintrust_FullLoginOptions as FullLoginOptions, type braintrust_FunctionEvent as FunctionEvent, braintrust_INTERNAL_BTQL_LIMIT as INTERNAL_BTQL_LIMIT, type braintrust_InitOptions as InitOptions, type braintrust_InvokeFunctionArgs as InvokeFunctionArgs, type braintrust_InvokeReturn as InvokeReturn, braintrust_LEGACY_CACHED_HEADER as LEGACY_CACHED_HEADER, braintrust_LazyValue as LazyValue, type braintrust_LogOptions as LogOptions, braintrust_Logger as Logger, type braintrust_LoginOptions as LoginOptions, type braintrust_MetricSummary as MetricSummary, braintrust_NOOP_SPAN as NOOP_SPAN, braintrust_NOOP_SPAN_PERMALINK as NOOP_SPAN_PERMALINK, braintrust_NoopSpan as NoopSpan, type braintrust_ObjectMetadata as ObjectMetadata, braintrust_Project as Project, braintrust_ProjectNameIdMap as ProjectNameIdMap, type braintrust_PromiseUnless as PromiseUnless, braintrust_Prompt as Prompt, braintrust_PromptBuilder as PromptBuilder, type braintrust_PromptContents as PromptContents, type braintrust_PromptDefinition as PromptDefinition, type braintrust_PromptDefinitionWithTools as PromptDefinitionWithTools, type braintrust_PromptOpts as PromptOpts, type braintrust_PromptRowWithId as PromptRowWithId, braintrust_ReadonlyAttachment as ReadonlyAttachment, braintrust_ReadonlyExperiment as ReadonlyExperiment, braintrust_Reporter as Reporter, type braintrust_ReporterBody as ReporterBody, type braintrust_ScoreSummary as ScoreSummary, braintrust_ScorerBuilder as ScorerBuilder, type braintrust_ScorerOpts as ScorerOpts, type braintrust_SerializedBraintrustState as SerializedBraintrustState, type braintrust_SetCurrentArg as SetCurrentArg, type Span$1 as Span, type braintrust_SpanContext as SpanContext, braintrust_SpanImpl as SpanImpl, type braintrust_StartSpanArgs as StartSpanArgs, braintrust_TestBackgroundLogger as TestBackgroundLogger, braintrust_ToolBuilder as ToolBuilder, type braintrust_WithTransactionId as WithTransactionId, braintrust_X_CACHED_HEADER as X_CACHED_HEADER, braintrust__exportsForTestingOnly as _exportsForTestingOnly, braintrust__internalGetGlobalState as _internalGetGlobalState, braintrust__internalSetInitialState as _internalSetInitialState, braintrust_braintrustStreamChunkSchema as braintrustStreamChunkSchema, braintrust_buildLocalSummary as buildLocalSummary, braintrust_createFinalValuePassThroughStream as createFinalValuePassThroughStream, braintrust_currentExperiment as currentExperiment, braintrust_currentLogger as currentLogger, braintrust_currentSpan as currentSpan, braintrust_defaultErrorScoreHandler as defaultErrorScoreHandler, braintrust_deserializePlainStringAsJSON as deserializePlainStringAsJSON, braintrust_devNullWritableStream as devNullWritableStream, braintrust_flush as flush, braintrust_getSpanParentObject as getSpanParentObject, graphFramework as graph, braintrust_init as init, braintrust_initDataset as initDataset, braintrust_initExperiment as initExperiment, braintrust_initFunction as initFunction, braintrust_initLogger as initLogger, braintrust_invoke as invoke, braintrust_loadPrompt as loadPrompt, braintrust_log as log, braintrust_logError as logError, braintrust_login as login, braintrust_loginToState as loginToState, braintrust_newId as newId, braintrust_parseCachedHeader as parseCachedHeader, braintrust_permalink as permalink, braintrust_projects as projects, braintrust_promptContentsSchema as promptContentsSchema, braintrust_promptDefinitionSchema as promptDefinitionSchema, braintrust_promptDefinitionToPromptData as promptDefinitionToPromptData, braintrust_promptDefinitionWithToolsSchema as promptDefinitionWithToolsSchema, braintrust_renderMessage as renderMessage, braintrust_renderPromptParams as renderPromptParams, braintrust_reportFailures as reportFailures, braintrust_runEvaluator as runEvaluator, braintrust_setFetch as setFetch, braintrust_spanComponentsToObjectId as spanComponentsToObjectId, braintrust_startSpan as startSpan, braintrust_summarize as summarize, braintrust_toolFunctionDefinitionSchema as toolFunctionDefinitionSchema, braintrust_traceable as traceable, braintrust_traced as traced, braintrust_updateSpan as updateSpan, braintrust_withCurrent as withCurrent, braintrust_withDataset as withDataset, braintrust_withExperiment as withExperiment, braintrust_withLogger as withLogger, braintrust_withParent as withParent, braintrust_wrapAISDKModel as wrapAISDKModel, braintrust_wrapAnthropic as wrapAnthropic, braintrust_wrapOpenAI as wrapOpenAI, braintrust_wrapOpenAIv4 as wrapOpenAIv4, braintrust_wrapTraced as wrapTraced };
 }
 
 type EvaluatorManifest = Record<string, EvaluatorDef<unknown, unknown, unknown, BaseMetadata>>;
@@ -14792,4 +14839,4 @@ type EvaluatorDefinitions = z.infer<typeof evaluatorDefinitionsSchema>;
  * @module braintrust
  */
 
-export { AISpanProcessor, type AnyDataset, Attachment, type AttachmentParams, type BackgroundLoggerOpts, BaseAttachment, BaseExperiment, type BaseMetadata, BraintrustExporter, BraintrustSpanProcessor, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, CodeFunction, type CodeOpts, CodePrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type CreateProjectOpts, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, ERR_PERMALINK, type EndSpanArgs, Eval, type EvalCase, type EvalHooks, type EvalParameterSerializedSchema, type EvalParameters, type EvalResult, EvalResultWithSummary, type EvalScorer, type EvalScorerArgs, type EvalTask, type Evaluator, type EvaluatorDef, type EvaluatorDefinition, type EvaluatorDefinitions, type EvaluatorFile, type EvaluatorManifest, Experiment, type ExperimentSummary, type Exportable, ExternalAttachment, type ExternalAttachmentParams, FailedHTTPResponse, type FullInitOptions, type FullLoginOptions, type FunctionEvent, INTERNAL_BTQL_LIMIT, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, LazyValue, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NOOP_SPAN_PERMALINK, NoopSpan, type ObjectMetadata, Project, ProjectNameIdMap, type PromiseUnless, Prompt, PromptBuilder, type PromptContents, type PromptDefinition, type PromptDefinitionWithTools, type PromptOpts, type PromptRowWithId, ReadonlyAttachment, ReadonlyExperiment, Reporter, type ReporterBody, type ScoreSummary, ScorerBuilder, type ScorerOpts, type SerializedBraintrustState, type SetCurrentArg, type Span$1 as Span, type SpanContext, SpanImpl, type StartSpanArgs, TestBackgroundLogger, ToolBuilder, type WithTransactionId, X_CACHED_HEADER, _exportsForTestingOnly, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, buildLocalSummary, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, braintrust as default, defaultErrorScoreHandler, deserializePlainStringAsJSON, devNullWritableStream, evaluatorDefinitionSchema, evaluatorDefinitionsSchema, flush, getSpanParentObject, graphFramework as graph, init, initDataset, initExperiment, initFunction, initLogger, invoke, loadPrompt, log, logError, login, loginToState, newId, parseCachedHeader, permalink, projects, promptContentsSchema, promptDefinitionSchema, promptDefinitionToPromptData, promptDefinitionWithToolsSchema, renderMessage, renderPromptParams, reportFailures, runEvaluator, setFetch, spanComponentsToObjectId, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, withParent, wrapAISDKModel, wrapAnthropic, wrapOpenAI, wrapOpenAIv4, wrapTraced };
+export { AISpanProcessor, type AnyDataset, Attachment, type AttachmentParams, type BackgroundLoggerOpts, BaseAttachment, BaseExperiment, type BaseMetadata, BraintrustExporter, BraintrustMiddleware, BraintrustSpanProcessor, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, CodeFunction, type CodeOpts, CodePrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type CreateProjectOpts, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, ERR_PERMALINK, type EndSpanArgs, Eval, type EvalCase, type EvalHooks, type EvalParameterSerializedSchema, type EvalParameters, type EvalResult, EvalResultWithSummary, type EvalScorer, type EvalScorerArgs, type EvalTask, type Evaluator, type EvaluatorDef, type EvaluatorDefinition, type EvaluatorDefinitions, type EvaluatorFile, type EvaluatorManifest, Experiment, type ExperimentSummary, type Exportable, ExternalAttachment, type ExternalAttachmentParams, FailedHTTPResponse, type FullInitOptions, type FullLoginOptions, type FunctionEvent, INTERNAL_BTQL_LIMIT, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, LazyValue, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NOOP_SPAN_PERMALINK, NoopSpan, type ObjectMetadata, Project, ProjectNameIdMap, type PromiseUnless, Prompt, PromptBuilder, type PromptContents, type PromptDefinition, type PromptDefinitionWithTools, type PromptOpts, type PromptRowWithId, ReadonlyAttachment, ReadonlyExperiment, Reporter, type ReporterBody, type ScoreSummary, ScorerBuilder, type ScorerOpts, type SerializedBraintrustState, type SetCurrentArg, type Span$1 as Span, type SpanContext, SpanImpl, type StartSpanArgs, TestBackgroundLogger, ToolBuilder, type WithTransactionId, X_CACHED_HEADER, _exportsForTestingOnly, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, buildLocalSummary, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, braintrust as default, defaultErrorScoreHandler, deserializePlainStringAsJSON, devNullWritableStream, evaluatorDefinitionSchema, evaluatorDefinitionsSchema, flush, getSpanParentObject, graphFramework as graph, init, initDataset, initExperiment, initFunction, initLogger, invoke, loadPrompt, log, logError, login, loginToState, newId, parseCachedHeader, permalink, projects, promptContentsSchema, promptDefinitionSchema, promptDefinitionToPromptData, promptDefinitionWithToolsSchema, renderMessage, renderPromptParams, reportFailures, runEvaluator, setFetch, spanComponentsToObjectId, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, withParent, wrapAISDKModel, wrapAnthropic, wrapOpenAI, wrapOpenAIv4, wrapTraced };

package/dist/index.d.ts

@@ -1691,6 +1691,7 @@ interface DatasetSummary {
     datasetUrl: string;
     dataSummary: DataSummary | undefined;
 }
+declare function setInitialTestState(): void;
 declare function simulateLoginForTests(): Promise<BraintrustState>;
 declare function simulateLogoutForTests(): BraintrustState;
 declare const _exportsForTestingOnly: {
@@ -1700,6 +1701,7 @@ declare const _exportsForTestingOnly: {
     clearTestBackgroundLogger: typeof clearTestBackgroundLogger;
     simulateLoginForTests: typeof simulateLoginForTests;
     simulateLogoutForTests: typeof simulateLogoutForTests;
+    setInitialTestState: typeof setInitialTestState;
 };
 
 declare const braintrustStreamChunkSchema: z.ZodUnion<[z.ZodObject<{
@@ -7134,7 +7136,7 @@ declare global {
  * not configured, nothing will be traced. If this is not an `OpenAI` object, this function is
  * a no-op.
  *
- * Currently, this only supports the `v4` API.
+ * Currently, this supports both the `v4` and `v5` API.
  *
  * @param openai
  * @returns The wrapped `OpenAI` object.
@@ -7145,6 +7147,50 @@ declare const LEGACY_CACHED_HEADER = "x-cached";
 declare const X_CACHED_HEADER = "x-bt-cached";
 declare function parseCachedHeader(value: string | null | undefined): number | undefined;
 
+interface LanguageModelV2Middleware<TModel = any, TCallOptions = any> {
+    wrapGenerate?: (options: {
+        doGenerate: () => any;
+        doStream: () => any;
+        params: TCallOptions;
+        model: TModel;
+    }) => Promise<any>;
+    wrapStream?: (options: {
+        doGenerate: () => any;
+        doStream: () => any;
+        params: TCallOptions;
+        model: TModel;
+    }) => Promise<any>;
+}
+/**
+ * Configuration options for the AI SDK middleware
+ */
+interface MiddlewareConfig {
+    /** Enable debug logging */
+    debug?: boolean;
+    /** Name identifier for the middleware instance */
+    name?: string;
+}
+/**
+ * Creates a Braintrust middleware for AI SDK v2 that automatically traces
+ * generateText and streamText calls with comprehensive metadata and metrics.
+ *
+ * @param config - Configuration options for the middleware
+ * @returns A middleware object compatible with AI SDK v2's wrapLanguageModel
+ *
+ * @example
+ * ```typescript
+ * import { wrapLanguageModel } from "ai";
+ * import { openai } from "@ai-sdk/openai";
+ * import { BraintrustMiddleware } from "braintrust";
+ *
+ * const model = wrapLanguageModel({
+ *   model: openai("gpt-4"),
+ *   middleware: BraintrustMiddleware({ debug: true, name: "MyMiddleware" })
+ * });
+ * ```
+ */
+declare function BraintrustMiddleware(config?: MiddlewareConfig): LanguageModelV2Middleware<any, any>;
+
 /**
  * Wrap an ai-sdk model (created with `.chat()`, `.completion()`, etc.) to add tracing. If Braintrust is
  * not configured, this is a no-op
@@ -7406,6 +7452,7 @@ declare const braintrust_BaseExperiment: typeof BaseExperiment;
 type braintrust_BaseMetadata = BaseMetadata;
 type braintrust_BraintrustExporter = BraintrustExporter;
 declare const braintrust_BraintrustExporter: typeof BraintrustExporter;
+declare const braintrust_BraintrustMiddleware: typeof BraintrustMiddleware;
 type braintrust_BraintrustSpanProcessor = BraintrustSpanProcessor;
 declare const braintrust_BraintrustSpanProcessor: typeof BraintrustSpanProcessor;
 type braintrust_BraintrustState = BraintrustState;
@@ -7564,7 +7611,7 @@ declare const braintrust_wrapOpenAI: typeof wrapOpenAI;
 declare const braintrust_wrapOpenAIv4: typeof wrapOpenAIv4;
 declare const braintrust_wrapTraced: typeof wrapTraced;
 declare namespace braintrust {
-  export { braintrust_AISpanProcessor as AISpanProcessor, type braintrust_AnyDataset as AnyDataset, braintrust_Attachment as Attachment, type braintrust_AttachmentParams as AttachmentParams, type braintrust_BackgroundLoggerOpts as BackgroundLoggerOpts, braintrust_BaseAttachment as BaseAttachment, braintrust_BaseExperiment as BaseExperiment, type braintrust_BaseMetadata as BaseMetadata, braintrust_BraintrustExporter as BraintrustExporter, braintrust_BraintrustSpanProcessor as BraintrustSpanProcessor, braintrust_BraintrustState as BraintrustState, braintrust_BraintrustStream as BraintrustStream, type braintrust_BraintrustStreamChunk as BraintrustStreamChunk, type braintrust_ChatPrompt as ChatPrompt, braintrust_CodeFunction as CodeFunction, type braintrust_CodeOpts as CodeOpts, braintrust_CodePrompt as CodePrompt, type braintrust_CompiledPrompt as CompiledPrompt, type braintrust_CompiledPromptParams as CompiledPromptParams, type braintrust_CompletionPrompt as CompletionPrompt, type braintrust_CreateProjectOpts as CreateProjectOpts, type braintrust_DataSummary as DataSummary, braintrust_Dataset as Dataset, type braintrust_DatasetSummary as DatasetSummary, type braintrust_DefaultMetadataType as DefaultMetadataType, type braintrust_DefaultPromptArgs as DefaultPromptArgs, braintrust_ERR_PERMALINK as ERR_PERMALINK, type braintrust_EndSpanArgs as EndSpanArgs, braintrust_Eval as Eval, type braintrust_EvalCase as EvalCase, type braintrust_EvalHooks as EvalHooks, type braintrust_EvalResult as EvalResult, braintrust_EvalResultWithSummary as EvalResultWithSummary, type braintrust_EvalScorer as EvalScorer, type braintrust_EvalScorerArgs as EvalScorerArgs, type braintrust_EvalTask as EvalTask, type braintrust_Evaluator as Evaluator, type braintrust_EvaluatorDef as EvaluatorDef, type braintrust_EvaluatorFile as EvaluatorFile, braintrust_Experiment as Experiment, type braintrust_ExperimentSummary as ExperimentSummary, type braintrust_Exportable as Exportable, braintrust_ExternalAttachment as ExternalAttachment, type braintrust_ExternalAttachmentParams as ExternalAttachmentParams, braintrust_FailedHTTPResponse as FailedHTTPResponse, type braintrust_FullInitOptions as FullInitOptions, type braintrust_FullLoginOptions as FullLoginOptions, type braintrust_FunctionEvent as FunctionEvent, braintrust_INTERNAL_BTQL_LIMIT as INTERNAL_BTQL_LIMIT, type braintrust_InitOptions as InitOptions, type braintrust_InvokeFunctionArgs as InvokeFunctionArgs, type braintrust_InvokeReturn as InvokeReturn, braintrust_LEGACY_CACHED_HEADER as LEGACY_CACHED_HEADER, braintrust_LazyValue as LazyValue, type braintrust_LogOptions as LogOptions, braintrust_Logger as Logger, type braintrust_LoginOptions as LoginOptions, type braintrust_MetricSummary as MetricSummary, braintrust_NOOP_SPAN as NOOP_SPAN, braintrust_NOOP_SPAN_PERMALINK as NOOP_SPAN_PERMALINK, braintrust_NoopSpan as NoopSpan, type braintrust_ObjectMetadata as ObjectMetadata, braintrust_Project as Project, braintrust_ProjectNameIdMap as ProjectNameIdMap, type braintrust_PromiseUnless as PromiseUnless, braintrust_Prompt as Prompt, braintrust_PromptBuilder as PromptBuilder, type braintrust_PromptContents as PromptContents, type braintrust_PromptDefinition as PromptDefinition, type braintrust_PromptDefinitionWithTools as PromptDefinitionWithTools, type braintrust_PromptOpts as PromptOpts, type braintrust_PromptRowWithId as PromptRowWithId, braintrust_ReadonlyAttachment as ReadonlyAttachment, braintrust_ReadonlyExperiment as ReadonlyExperiment, braintrust_Reporter as Reporter, type braintrust_ReporterBody as ReporterBody, type braintrust_ScoreSummary as ScoreSummary, braintrust_ScorerBuilder as ScorerBuilder, type braintrust_ScorerOpts as ScorerOpts, type braintrust_SerializedBraintrustState as SerializedBraintrustState, type braintrust_SetCurrentArg as SetCurrentArg, type Span$1 as Span, type braintrust_SpanContext as SpanContext, braintrust_SpanImpl as SpanImpl, type braintrust_StartSpanArgs as StartSpanArgs, braintrust_TestBackgroundLogger as TestBackgroundLogger, braintrust_ToolBuilder as ToolBuilder, type braintrust_WithTransactionId as WithTransactionId, braintrust_X_CACHED_HEADER as X_CACHED_HEADER, braintrust__exportsForTestingOnly as _exportsForTestingOnly, braintrust__internalGetGlobalState as _internalGetGlobalState, braintrust__internalSetInitialState as _internalSetInitialState, braintrust_braintrustStreamChunkSchema as braintrustStreamChunkSchema, braintrust_buildLocalSummary as buildLocalSummary, braintrust_createFinalValuePassThroughStream as createFinalValuePassThroughStream, braintrust_currentExperiment as currentExperiment, braintrust_currentLogger as currentLogger, braintrust_currentSpan as currentSpan, braintrust_defaultErrorScoreHandler as defaultErrorScoreHandler, braintrust_deserializePlainStringAsJSON as deserializePlainStringAsJSON, braintrust_devNullWritableStream as devNullWritableStream, braintrust_flush as flush, braintrust_getSpanParentObject as getSpanParentObject, graphFramework as graph, braintrust_init as init, braintrust_initDataset as initDataset, braintrust_initExperiment as initExperiment, braintrust_initFunction as initFunction, braintrust_initLogger as initLogger, braintrust_invoke as invoke, braintrust_loadPrompt as loadPrompt, braintrust_log as log, braintrust_logError as logError, braintrust_login as login, braintrust_loginToState as loginToState, braintrust_newId as newId, braintrust_parseCachedHeader as parseCachedHeader, braintrust_permalink as permalink, braintrust_projects as projects, braintrust_promptContentsSchema as promptContentsSchema, braintrust_promptDefinitionSchema as promptDefinitionSchema, braintrust_promptDefinitionToPromptData as promptDefinitionToPromptData, braintrust_promptDefinitionWithToolsSchema as promptDefinitionWithToolsSchema, braintrust_renderMessage as renderMessage, braintrust_renderPromptParams as renderPromptParams, braintrust_reportFailures as reportFailures, braintrust_runEvaluator as runEvaluator, braintrust_setFetch as setFetch, braintrust_spanComponentsToObjectId as spanComponentsToObjectId, braintrust_startSpan as startSpan, braintrust_summarize as summarize, braintrust_toolFunctionDefinitionSchema as toolFunctionDefinitionSchema, braintrust_traceable as traceable, braintrust_traced as traced, braintrust_updateSpan as updateSpan, braintrust_withCurrent as withCurrent, braintrust_withDataset as withDataset, braintrust_withExperiment as withExperiment, braintrust_withLogger as withLogger, braintrust_withParent as withParent, braintrust_wrapAISDKModel as wrapAISDKModel, braintrust_wrapAnthropic as wrapAnthropic, braintrust_wrapOpenAI as wrapOpenAI, braintrust_wrapOpenAIv4 as wrapOpenAIv4, braintrust_wrapTraced as wrapTraced };
+  export { braintrust_AISpanProcessor as AISpanProcessor, type braintrust_AnyDataset as AnyDataset, braintrust_Attachment as Attachment, type braintrust_AttachmentParams as AttachmentParams, type braintrust_BackgroundLoggerOpts as BackgroundLoggerOpts, braintrust_BaseAttachment as BaseAttachment, braintrust_BaseExperiment as BaseExperiment, type braintrust_BaseMetadata as BaseMetadata, braintrust_BraintrustExporter as BraintrustExporter, braintrust_BraintrustMiddleware as BraintrustMiddleware, braintrust_BraintrustSpanProcessor as BraintrustSpanProcessor, braintrust_BraintrustState as BraintrustState, braintrust_BraintrustStream as BraintrustStream, type braintrust_BraintrustStreamChunk as BraintrustStreamChunk, type braintrust_ChatPrompt as ChatPrompt, braintrust_CodeFunction as CodeFunction, type braintrust_CodeOpts as CodeOpts, braintrust_CodePrompt as CodePrompt, type braintrust_CompiledPrompt as CompiledPrompt, type braintrust_CompiledPromptParams as CompiledPromptParams, type braintrust_CompletionPrompt as CompletionPrompt, type braintrust_CreateProjectOpts as CreateProjectOpts, type braintrust_DataSummary as DataSummary, braintrust_Dataset as Dataset, type braintrust_DatasetSummary as DatasetSummary, type braintrust_DefaultMetadataType as DefaultMetadataType, type braintrust_DefaultPromptArgs as DefaultPromptArgs, braintrust_ERR_PERMALINK as ERR_PERMALINK, type braintrust_EndSpanArgs as EndSpanArgs, braintrust_Eval as Eval, type braintrust_EvalCase as EvalCase, type braintrust_EvalHooks as EvalHooks, type braintrust_EvalResult as EvalResult, braintrust_EvalResultWithSummary as EvalResultWithSummary, type braintrust_EvalScorer as EvalScorer, type braintrust_EvalScorerArgs as EvalScorerArgs, type braintrust_EvalTask as EvalTask, type braintrust_Evaluator as Evaluator, type braintrust_EvaluatorDef as EvaluatorDef, type braintrust_EvaluatorFile as EvaluatorFile, braintrust_Experiment as Experiment, type braintrust_ExperimentSummary as ExperimentSummary, type braintrust_Exportable as Exportable, braintrust_ExternalAttachment as ExternalAttachment, type braintrust_ExternalAttachmentParams as ExternalAttachmentParams, braintrust_FailedHTTPResponse as FailedHTTPResponse, type braintrust_FullInitOptions as FullInitOptions, type braintrust_FullLoginOptions as FullLoginOptions, type braintrust_FunctionEvent as FunctionEvent, braintrust_INTERNAL_BTQL_LIMIT as INTERNAL_BTQL_LIMIT, type braintrust_InitOptions as InitOptions, type braintrust_InvokeFunctionArgs as InvokeFunctionArgs, type braintrust_InvokeReturn as InvokeReturn, braintrust_LEGACY_CACHED_HEADER as LEGACY_CACHED_HEADER, braintrust_LazyValue as LazyValue, type braintrust_LogOptions as LogOptions, braintrust_Logger as Logger, type braintrust_LoginOptions as LoginOptions, type braintrust_MetricSummary as MetricSummary, braintrust_NOOP_SPAN as NOOP_SPAN, braintrust_NOOP_SPAN_PERMALINK as NOOP_SPAN_PERMALINK, braintrust_NoopSpan as NoopSpan, type braintrust_ObjectMetadata as ObjectMetadata, braintrust_Project as Project, braintrust_ProjectNameIdMap as ProjectNameIdMap, type braintrust_PromiseUnless as PromiseUnless, braintrust_Prompt as Prompt, braintrust_PromptBuilder as PromptBuilder, type braintrust_PromptContents as PromptContents, type braintrust_PromptDefinition as PromptDefinition, type braintrust_PromptDefinitionWithTools as PromptDefinitionWithTools, type braintrust_PromptOpts as PromptOpts, type braintrust_PromptRowWithId as PromptRowWithId, braintrust_ReadonlyAttachment as ReadonlyAttachment, braintrust_ReadonlyExperiment as ReadonlyExperiment, braintrust_Reporter as Reporter, type braintrust_ReporterBody as ReporterBody, type braintrust_ScoreSummary as ScoreSummary, braintrust_ScorerBuilder as ScorerBuilder, type braintrust_ScorerOpts as ScorerOpts, type braintrust_SerializedBraintrustState as SerializedBraintrustState, type braintrust_SetCurrentArg as SetCurrentArg, type Span$1 as Span, type braintrust_SpanContext as SpanContext, braintrust_SpanImpl as SpanImpl, type braintrust_StartSpanArgs as StartSpanArgs, braintrust_TestBackgroundLogger as TestBackgroundLogger, braintrust_ToolBuilder as ToolBuilder, type braintrust_WithTransactionId as WithTransactionId, braintrust_X_CACHED_HEADER as X_CACHED_HEADER, braintrust__exportsForTestingOnly as _exportsForTestingOnly, braintrust__internalGetGlobalState as _internalGetGlobalState, braintrust__internalSetInitialState as _internalSetInitialState, braintrust_braintrustStreamChunkSchema as braintrustStreamChunkSchema, braintrust_buildLocalSummary as buildLocalSummary, braintrust_createFinalValuePassThroughStream as createFinalValuePassThroughStream, braintrust_currentExperiment as currentExperiment, braintrust_currentLogger as currentLogger, braintrust_currentSpan as currentSpan, braintrust_defaultErrorScoreHandler as defaultErrorScoreHandler, braintrust_deserializePlainStringAsJSON as deserializePlainStringAsJSON, braintrust_devNullWritableStream as devNullWritableStream, braintrust_flush as flush, braintrust_getSpanParentObject as getSpanParentObject, graphFramework as graph, braintrust_init as init, braintrust_initDataset as initDataset, braintrust_initExperiment as initExperiment, braintrust_initFunction as initFunction, braintrust_initLogger as initLogger, braintrust_invoke as invoke, braintrust_loadPrompt as loadPrompt, braintrust_log as log, braintrust_logError as logError, braintrust_login as login, braintrust_loginToState as loginToState, braintrust_newId as newId, braintrust_parseCachedHeader as parseCachedHeader, braintrust_permalink as permalink, braintrust_projects as projects, braintrust_promptContentsSchema as promptContentsSchema, braintrust_promptDefinitionSchema as promptDefinitionSchema, braintrust_promptDefinitionToPromptData as promptDefinitionToPromptData, braintrust_promptDefinitionWithToolsSchema as promptDefinitionWithToolsSchema, braintrust_renderMessage as renderMessage, braintrust_renderPromptParams as renderPromptParams, braintrust_reportFailures as reportFailures, braintrust_runEvaluator as runEvaluator, braintrust_setFetch as setFetch, braintrust_spanComponentsToObjectId as spanComponentsToObjectId, braintrust_startSpan as startSpan, braintrust_summarize as summarize, braintrust_toolFunctionDefinitionSchema as toolFunctionDefinitionSchema, braintrust_traceable as traceable, braintrust_traced as traced, braintrust_updateSpan as updateSpan, braintrust_withCurrent as withCurrent, braintrust_withDataset as withDataset, braintrust_withExperiment as withExperiment, braintrust_withLogger as withLogger, braintrust_withParent as withParent, braintrust_wrapAISDKModel as wrapAISDKModel, braintrust_wrapAnthropic as wrapAnthropic, braintrust_wrapOpenAI as wrapOpenAI, braintrust_wrapOpenAIv4 as wrapOpenAIv4, braintrust_wrapTraced as wrapTraced };
 }
 
 type EvaluatorManifest = Record<string, EvaluatorDef<unknown, unknown, unknown, BaseMetadata>>;
@@ -14792,4 +14839,4 @@ type EvaluatorDefinitions = z.infer<typeof evaluatorDefinitionsSchema>;
  * @module braintrust
  */
 
-export { AISpanProcessor, type AnyDataset, Attachment, type AttachmentParams, type BackgroundLoggerOpts, BaseAttachment, BaseExperiment, type BaseMetadata, BraintrustExporter, BraintrustSpanProcessor, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, CodeFunction, type CodeOpts, CodePrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type CreateProjectOpts, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, ERR_PERMALINK, type EndSpanArgs, Eval, type EvalCase, type EvalHooks, type EvalParameterSerializedSchema, type EvalParameters, type EvalResult, EvalResultWithSummary, type EvalScorer, type EvalScorerArgs, type EvalTask, type Evaluator, type EvaluatorDef, type EvaluatorDefinition, type EvaluatorDefinitions, type EvaluatorFile, type EvaluatorManifest, Experiment, type ExperimentSummary, type Exportable, ExternalAttachment, type ExternalAttachmentParams, FailedHTTPResponse, type FullInitOptions, type FullLoginOptions, type FunctionEvent, INTERNAL_BTQL_LIMIT, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, LazyValue, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NOOP_SPAN_PERMALINK, NoopSpan, type ObjectMetadata, Project, ProjectNameIdMap, type PromiseUnless, Prompt, PromptBuilder, type PromptContents, type PromptDefinition, type PromptDefinitionWithTools, type PromptOpts, type PromptRowWithId, ReadonlyAttachment, ReadonlyExperiment, Reporter, type ReporterBody, type ScoreSummary, ScorerBuilder, type ScorerOpts, type SerializedBraintrustState, type SetCurrentArg, type Span$1 as Span, type SpanContext, SpanImpl, type StartSpanArgs, TestBackgroundLogger, ToolBuilder, type WithTransactionId, X_CACHED_HEADER, _exportsForTestingOnly, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, buildLocalSummary, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, braintrust as default, defaultErrorScoreHandler, deserializePlainStringAsJSON, devNullWritableStream, evaluatorDefinitionSchema, evaluatorDefinitionsSchema, flush, getSpanParentObject, graphFramework as graph, init, initDataset, initExperiment, initFunction, initLogger, invoke, loadPrompt, log, logError, login, loginToState, newId, parseCachedHeader, permalink, projects, promptContentsSchema, promptDefinitionSchema, promptDefinitionToPromptData, promptDefinitionWithToolsSchema, renderMessage, renderPromptParams, reportFailures, runEvaluator, setFetch, spanComponentsToObjectId, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, withParent, wrapAISDKModel, wrapAnthropic, wrapOpenAI, wrapOpenAIv4, wrapTraced };
+export { AISpanProcessor, type AnyDataset, Attachment, type AttachmentParams, type BackgroundLoggerOpts, BaseAttachment, BaseExperiment, type BaseMetadata, BraintrustExporter, BraintrustMiddleware, BraintrustSpanProcessor, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, CodeFunction, type CodeOpts, CodePrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type CreateProjectOpts, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, ERR_PERMALINK, type EndSpanArgs, Eval, type EvalCase, type EvalHooks, type EvalParameterSerializedSchema, type EvalParameters, type EvalResult, EvalResultWithSummary, type EvalScorer, type EvalScorerArgs, type EvalTask, type Evaluator, type EvaluatorDef, type EvaluatorDefinition, type EvaluatorDefinitions, type EvaluatorFile, type EvaluatorManifest, Experiment, type ExperimentSummary, type Exportable, ExternalAttachment, type ExternalAttachmentParams, FailedHTTPResponse, type FullInitOptions, type FullLoginOptions, type FunctionEvent, INTERNAL_BTQL_LIMIT, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, LazyValue, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NOOP_SPAN_PERMALINK, NoopSpan, type ObjectMetadata, Project, ProjectNameIdMap, type PromiseUnless, Prompt, PromptBuilder, type PromptContents, type PromptDefinition, type PromptDefinitionWithTools, type PromptOpts, type PromptRowWithId, ReadonlyAttachment, ReadonlyExperiment, Reporter, type ReporterBody, type ScoreSummary, ScorerBuilder, type ScorerOpts, type SerializedBraintrustState, type SetCurrentArg, type Span$1 as Span, type SpanContext, SpanImpl, type StartSpanArgs, TestBackgroundLogger, ToolBuilder, type WithTransactionId, X_CACHED_HEADER, _exportsForTestingOnly, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, buildLocalSummary, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, braintrust as default, defaultErrorScoreHandler, deserializePlainStringAsJSON, devNullWritableStream, evaluatorDefinitionSchema, evaluatorDefinitionsSchema, flush, getSpanParentObject, graphFramework as graph, init, initDataset, initExperiment, initFunction, initLogger, invoke, loadPrompt, log, logError, login, loginToState, newId, parseCachedHeader, permalink, projects, promptContentsSchema, promptDefinitionSchema, promptDefinitionToPromptData, promptDefinitionWithToolsSchema, renderMessage, renderPromptParams, reportFailures, runEvaluator, setFetch, spanComponentsToObjectId, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, withParent, wrapAISDKModel, wrapAnthropic, wrapOpenAI, wrapOpenAIv4, wrapTraced };

package/dist/index.js

@@ -7979,6 +7979,11 @@ var Prompt = (_class23 = class _Prompt {
   }
 }, _class23);
 var TEST_API_KEY = "___TEST_API_KEY__THIS_IS_NOT_REAL___";
+function setInitialTestState() {
+  if (!_internalGetGlobalState()) {
+    _internalSetInitialState();
+  }
+}
 async function simulateLoginForTests() {
   return await login({
     apiKey: TEST_API_KEY,
@@ -7996,7 +8001,8 @@ var _exportsForTestingOnly = {
   useTestBackgroundLogger,
   clearTestBackgroundLogger,
   simulateLoginForTests,
-  simulateLogoutForTests
+  simulateLogoutForTests,
+  setInitialTestState
 };
 
 // src/node.ts
@@ -8035,6 +8041,7 @@ _chunkCDBUTZMHjs.__export.call(void 0, exports_node_exports, {
   BaseAttachment: () => BaseAttachment,
   BaseExperiment: () => BaseExperiment,
   BraintrustExporter: () => BraintrustExporter,
+  BraintrustMiddleware: () => BraintrustMiddleware,
   BraintrustSpanProcessor: () => BraintrustSpanProcessor,
   BraintrustState: () => BraintrustState,
   BraintrustStream: () => BraintrustStream,
@@ -10841,10 +10848,17 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  const completionProxy = createEndpointProxy(
-    openai.chat.completions,
-    wrapChatCompletion
-  );
+  const completionProxy = new Proxy(openai.chat.completions, {
+    get(target, name, receiver) {
+      const baseVal = Reflect.get(target, name, receiver);
+      if (name === "create") {
+        return wrapChatCompletion(baseVal.bind(target));
+      } else if (name === "parse") {
+        return wrapBetaChatCompletionParse(baseVal.bind(target));
+      }
+      return baseVal;
+    }
+  });
   const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
@@ -11220,10 +11234,221 @@ var WrapperStream = class {
   }
 };
 
-// src/wrappers/ai-sdk.ts
+// src/wrappers/ai-sdk-v2.ts
+
+function detectProviderFromResult(result) {
+  if (!_optionalChain([result, 'optionalAccess', _176 => _176.providerMetadata])) {
+    return void 0;
+  }
+  const keys = Object.keys(result.providerMetadata);
+  return _optionalChain([keys, 'optionalAccess', _177 => _177.at, 'call', _178 => _178(0)]);
+}
+function extractModelFromResult(result) {
+  if (_optionalChain([result, 'optionalAccess', _179 => _179.response, 'optionalAccess', _180 => _180.modelId])) {
+    return result.response.modelId;
+  }
+  if (_optionalChain([result, 'optionalAccess', _181 => _181.request, 'optionalAccess', _182 => _182.body, 'optionalAccess', _183 => _183.model])) {
+    return result.request.body.model;
+  }
+  return void 0;
+}
+function camelToSnake(str) {
+  return str.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
+}
+function extractModelParameters(params) {
+  const modelParams = {};
+  const excludeKeys = /* @__PURE__ */ new Set([
+    "prompt",
+    // Already captured as input
+    "system",
+    // Already captured as input
+    "messages",
+    // Already captured as input
+    "model",
+    // Already captured in metadata.model
+    "providerOptions"
+    // Internal AI SDK configuration
+  ]);
+  for (const [key, value] of Object.entries(params)) {
+    if (value !== void 0 && !excludeKeys.has(key)) {
+      const snakeKey = camelToSnake(key);
+      modelParams[snakeKey] = value;
+    }
+  }
+  return modelParams;
+}
+function getNumberProperty(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;
+}
+function normalizeUsageMetrics(usage) {
+  const metrics = {};
+  const inputTokens = getNumberProperty(usage, "inputTokens");
+  if (inputTokens !== void 0) {
+    metrics.prompt_tokens = inputTokens;
+  }
+  const outputTokens = getNumberProperty(usage, "outputTokens");
+  if (outputTokens !== void 0) {
+    metrics.completion_tokens = outputTokens;
+  }
+  const totalTokens = getNumberProperty(usage, "totalTokens");
+  if (totalTokens !== void 0) {
+    metrics.tokens = totalTokens;
+  }
+  const reasoningTokens = getNumberProperty(usage, "reasoningTokens");
+  if (reasoningTokens !== void 0) {
+    metrics.completion_reasoning_tokens = reasoningTokens;
+  }
+  const cachedInputTokens = getNumberProperty(usage, "cachedInputTokens");
+  if (cachedInputTokens !== void 0) {
+    metrics.prompt_cached_tokens = cachedInputTokens;
+  }
+  return metrics;
+}
+function BraintrustMiddleware(config = {}) {
+  return {
+    wrapGenerate: async ({ doGenerate, params }) => {
+      const spanArgs = {
+        name: "ai-sdk.generateText",
+        spanAttributes: {
+          type: _core.SpanTypeAttribute.LLM
+        },
+        event: {
+          input: params.prompt,
+          metadata: {
+            ...extractModelParameters(params)
+          }
+        }
+      };
+      const span = startSpan(spanArgs);
+      try {
+        const result = await doGenerate();
+        const metadata = {};
+        const provider = detectProviderFromResult(result);
+        if (provider !== void 0) {
+          metadata.provider = provider;
+        }
+        if (result.finishReason !== void 0) {
+          metadata.finish_reason = result.finishReason;
+        }
+        const model = extractModelFromResult(result);
+        if (model !== void 0) {
+          metadata.model = model;
+        }
+        span.log({
+          output: result.content,
+          metadata,
+          metrics: normalizeUsageMetrics(result.usage)
+        });
+        return result;
+      } catch (error2) {
+        span.log({
+          error: error2 instanceof Error ? error2.message : String(error2)
+        });
+        throw error2;
+      } finally {
+        span.end();
+      }
+    },
+    wrapStream: async ({ doStream, params }) => {
+      const spanArgs = {
+        name: "ai-sdk.streamText",
+        spanAttributes: {
+          type: _core.SpanTypeAttribute.LLM
+        },
+        event: {
+          input: params.prompt,
+          metadata: {
+            ...extractModelParameters(params)
+          }
+        }
+      };
+      const span = startSpan(spanArgs);
+      try {
+        const { stream, ...rest } = await doStream();
+        const textChunks = [];
+        let finalUsage = {};
+        let finalFinishReason = void 0;
+        let providerMetadata = {};
+        const transformStream = new TransformStream({
+          // eslint-disable-next-line @typescript-eslint/no-explicit-any
+          transform(chunk, controller) {
+            try {
+              if (chunk.type === "text-delta" && chunk.delta) {
+                textChunks.push(chunk.delta);
+              }
+              if (chunk.type === "finish") {
+                finalFinishReason = chunk.finishReason;
+                finalUsage = chunk.usage || {};
+                providerMetadata = chunk.providerMetadata || {};
+              }
+              controller.enqueue(chunk);
+            } catch (error2) {
+              span.log({
+                error: error2 instanceof Error ? error2.message : String(error2)
+              });
+              span.end();
+              controller.error(error2);
+            }
+          },
+          flush() {
+            try {
+              const generatedText = textChunks.join("");
+              const output = generatedText ? [{ type: "text", text: generatedText }] : [];
+              const resultForDetection = {
+                providerMetadata,
+                response: rest.response,
+                ...rest
+              };
+              const metadata = {};
+              const provider = detectProviderFromResult(resultForDetection);
+              if (provider !== void 0) {
+                metadata.provider = provider;
+              }
+              if (finalFinishReason !== void 0) {
+                metadata.finish_reason = finalFinishReason;
+              }
+              const model = extractModelFromResult(resultForDetection);
+              if (model !== void 0) {
+                metadata.model = model;
+              }
+              span.log({
+                output,
+                metadata,
+                metrics: normalizeUsageMetrics(finalUsage)
+              });
+              span.end();
+            } catch (error2) {
+              span.log({
+                error: error2 instanceof Error ? error2.message : String(error2)
+              });
+              span.end();
+              throw error2;
+            }
+          }
+        });
+        return {
+          stream: stream.pipeThrough(transformStream),
+          ...rest
+        };
+      } catch (error2) {
+        span.log({
+          error: error2 instanceof Error ? error2.message : String(error2)
+        });
+        span.end();
+        throw error2;
+      }
+    }
+  };
+}
+
+// src/wrappers/ai-sdk-v1.ts
 function wrapAISDKModel(model) {
   const m = model;
-  if (_optionalChain([m, 'optionalAccess', _176 => _176.specificationVersion]) === "v1" && typeof _optionalChain([m, 'optionalAccess', _177 => _177.provider]) === "string" && typeof _optionalChain([m, 'optionalAccess', _178 => _178.modelId]) === "string") {
+  if (_optionalChain([m, 'optionalAccess', _184 => _184.specificationVersion]) === "v1" && typeof _optionalChain([m, 'optionalAccess', _185 => _185.provider]) === "string" && typeof _optionalChain([m, 'optionalAccess', _186 => _186.modelId]) === "string") {
     return new BraintrustLanguageModelWrapper(m);
   } else {
     console.warn("Unsupported AI SDK model. Not wrapping.");
@@ -11279,10 +11504,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', _179 => _179.usage, 'optionalAccess', _180 => _180.promptTokens]),
-          completion_tokens: _optionalChain([ret, 'access', _181 => _181.usage, 'optionalAccess', _182 => _182.completionTokens]),
+          prompt_tokens: _optionalChain([ret, 'access', _187 => _187.usage, 'optionalAccess', _188 => _188.promptTokens]),
+          completion_tokens: _optionalChain([ret, 'access', _189 => _189.usage, 'optionalAccess', _190 => _190.completionTokens]),
           cached: parseCachedHeader(
-            _nullishCoalesce(_optionalChain([ret, 'access', _183 => _183.rawResponse, 'optionalAccess', _184 => _184.headers, 'optionalAccess', _185 => _185[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _186 => _186.rawResponse, 'optionalAccess', _187 => _187.headers, 'optionalAccess', _188 => _188[LEGACY_CACHED_HEADER]])))
+            _nullishCoalesce(_optionalChain([ret, 'access', _191 => _191.rawResponse, 'optionalAccess', _192 => _192.headers, 'optionalAccess', _193 => _193[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _194 => _194.rawResponse, 'optionalAccess', _195 => _195.headers, 'optionalAccess', _196 => _196[LEGACY_CACHED_HEADER]])))
           )
         }
       });
@@ -11374,10 +11599,10 @@ var BraintrustLanguageModelWrapper = class {
                 metrics: {
                   time_to_first_token,
                   tokens: !isEmpty(usage) ? usage.promptTokens + usage.completionTokens : void 0,
-                  prompt_tokens: _optionalChain([usage, 'optionalAccess', _189 => _189.promptTokens]),
-                  completion_tokens: _optionalChain([usage, 'optionalAccess', _190 => _190.completionTokens]),
+                  prompt_tokens: _optionalChain([usage, 'optionalAccess', _197 => _197.promptTokens]),
+                  completion_tokens: _optionalChain([usage, 'optionalAccess', _198 => _198.completionTokens]),
                   cached: parseCachedHeader(
-                    _nullishCoalesce(_optionalChain([ret, 'access', _191 => _191.rawResponse, 'optionalAccess', _192 => _192.headers, 'optionalAccess', _193 => _193[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _194 => _194.rawResponse, 'optionalAccess', _195 => _195.headers, 'optionalAccess', _196 => _196[LEGACY_CACHED_HEADER]])))
+                    _nullishCoalesce(_optionalChain([ret, 'access', _199 => _199.rawResponse, 'optionalAccess', _200 => _200.headers, 'optionalAccess', _201 => _201[X_CACHED_HEADER]]), () => ( _optionalChain([ret, 'access', _202 => _202.rawResponse, 'optionalAccess', _203 => _203.headers, 'optionalAccess', _204 => _204[LEGACY_CACHED_HEADER]])))
                   )
                 }
               });
@@ -11424,7 +11649,7 @@ function postProcessPrompt(prompt) {
         return [
           {
             role: "assistant",
-            content: _optionalChain([textPart, 'optionalAccess', _197 => _197.text]),
+            content: _optionalChain([textPart, 'optionalAccess', _205 => _205.text]),
             ...toolCallParts.length > 0 ? {
               tool_calls: toolCallParts.map((part) => ({
                 id: part.toolCallId,
@@ -11649,9 +11874,9 @@ function streamNextProxy(stream, sspan) {
       return result;
     }
     const item = result.value;
-    switch (_optionalChain([item, 'optionalAccess', _198 => _198.type])) {
+    switch (_optionalChain([item, 'optionalAccess', _206 => _206.type])) {
       case "message_start":
-        const msg = _optionalChain([item, 'optionalAccess', _199 => _199.message]);
+        const msg = _optionalChain([item, 'optionalAccess', _207 => _207.message]);
         if (msg) {
           const event = parseEventFromMessage(msg);
           totals = { ...totals, ...event.metrics };
@@ -11659,20 +11884,20 @@ function streamNextProxy(stream, sspan) {
         }
         break;
       case "content_block_delta":
-        if (_optionalChain([item, 'access', _200 => _200.delta, 'optionalAccess', _201 => _201.type]) === "text_delta") {
-          const text = _optionalChain([item, 'optionalAccess', _202 => _202.delta, 'optionalAccess', _203 => _203.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', _204 => _204.usage]);
+        const usage = _optionalChain([item, 'optionalAccess', _212 => _212.usage]);
         if (usage) {
           const metrics = parseMetricsFromUsage2(usage);
           totals = { ...totals, ...metrics };
         }
-        const delta = _optionalChain([item, 'optionalAccess', _205 => _205.delta]);
+        const delta = _optionalChain([item, 'optionalAccess', _213 => _213.delta]);
         if (delta) {
           metadata = { ...metadata, ...delta };
         }
@@ -11684,8 +11909,8 @@ function streamNextProxy(stream, sspan) {
   };
 }
 function parseEventFromMessage(message) {
-  const output = _optionalChain([message, 'optionalAccess', _206 => _206.content]) || null;
-  const metrics = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _207 => _207.usage]));
+  const output = _optionalChain([message, 'optionalAccess', _214 => _214.content]) || null;
+  const metrics = parseMetricsFromUsage2(_optionalChain([message, 'optionalAccess', _215 => _215.usage]));
   const metas = ["stop_reason", "stop_sequence"];
   const metadata = {};
   for (const m of metas) {
@@ -11976,6 +12201,8 @@ var evalBodySchema = _zod.z.object({
       name: _zod.z.string()
     })
   ).nullish(),
+  experiment_name: _zod.z.string().nullish(),
+  project_id: _zod.z.string().nullish(),
   parent: _typespecs.invokeParent.optional(),
   stream: _zod.z.boolean().optional()
 });
@@ -12103,4 +12330,5 @@ var index_default = exports_node_exports;
 
 
 
-exports.AISpanProcessor = AISpanProcessor; exports.Attachment = Attachment; exports.BaseAttachment = BaseAttachment; exports.BaseExperiment = BaseExperiment; exports.BraintrustExporter = BraintrustExporter; exports.BraintrustSpanProcessor = BraintrustSpanProcessor; exports.BraintrustState = BraintrustState; exports.BraintrustStream = BraintrustStream; exports.CodeFunction = CodeFunction; exports.CodePrompt = CodePrompt; exports.Dataset = Dataset; exports.ERR_PERMALINK = ERR_PERMALINK; exports.Eval = Eval; exports.EvalResultWithSummary = EvalResultWithSummary; exports.Experiment = Experiment; exports.ExternalAttachment = ExternalAttachment; exports.FailedHTTPResponse = FailedHTTPResponse; exports.INTERNAL_BTQL_LIMIT = INTERNAL_BTQL_LIMIT; exports.LEGACY_CACHED_HEADER = LEGACY_CACHED_HEADER; exports.LazyValue = LazyValue; exports.Logger = Logger; exports.NOOP_SPAN = NOOP_SPAN; exports.NOOP_SPAN_PERMALINK = NOOP_SPAN_PERMALINK; exports.NoopSpan = NoopSpan; exports.Project = Project; exports.ProjectNameIdMap = ProjectNameIdMap; exports.Prompt = Prompt; 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.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.default = index_default; exports.defaultErrorScoreHandler = defaultErrorScoreHandler; exports.deserializePlainStringAsJSON = deserializePlainStringAsJSON; exports.devNullWritableStream = devNullWritableStream; exports.evaluatorDefinitionSchema = evaluatorDefinitionSchema; exports.evaluatorDefinitionsSchema = evaluatorDefinitionsSchema; exports.flush = flush; 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.spanComponentsToObjectId = spanComponentsToObjectId; exports.startSpan = startSpan; exports.summarize = summarize; exports.toolFunctionDefinitionSchema = _typespecs.toolFunctionDefinitionSchema; 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.wrapAISDKModel = wrapAISDKModel; exports.wrapAnthropic = wrapAnthropic; exports.wrapOpenAI = wrapOpenAI; exports.wrapOpenAIv4 = wrapOpenAIv4; exports.wrapTraced = wrapTraced;
+
+exports.AISpanProcessor = AISpanProcessor; exports.Attachment = Attachment; exports.BaseAttachment = BaseAttachment; exports.BaseExperiment = BaseExperiment; exports.BraintrustExporter = BraintrustExporter; exports.BraintrustMiddleware = BraintrustMiddleware; exports.BraintrustSpanProcessor = BraintrustSpanProcessor; exports.BraintrustState = BraintrustState; exports.BraintrustStream = BraintrustStream; exports.CodeFunction = CodeFunction; exports.CodePrompt = CodePrompt; exports.Dataset = Dataset; exports.ERR_PERMALINK = ERR_PERMALINK; exports.Eval = Eval; exports.EvalResultWithSummary = EvalResultWithSummary; exports.Experiment = Experiment; exports.ExternalAttachment = ExternalAttachment; exports.FailedHTTPResponse = FailedHTTPResponse; exports.INTERNAL_BTQL_LIMIT = INTERNAL_BTQL_LIMIT; exports.LEGACY_CACHED_HEADER = LEGACY_CACHED_HEADER; exports.LazyValue = LazyValue; exports.Logger = Logger; exports.NOOP_SPAN = NOOP_SPAN; exports.NOOP_SPAN_PERMALINK = NOOP_SPAN_PERMALINK; exports.NoopSpan = NoopSpan; exports.Project = Project; exports.ProjectNameIdMap = ProjectNameIdMap; exports.Prompt = Prompt; 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.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.default = index_default; exports.defaultErrorScoreHandler = defaultErrorScoreHandler; exports.deserializePlainStringAsJSON = deserializePlainStringAsJSON; exports.devNullWritableStream = devNullWritableStream; exports.evaluatorDefinitionSchema = evaluatorDefinitionSchema; exports.evaluatorDefinitionsSchema = evaluatorDefinitionsSchema; exports.flush = flush; 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.spanComponentsToObjectId = spanComponentsToObjectId; exports.startSpan = startSpan; exports.summarize = summarize; exports.toolFunctionDefinitionSchema = _typespecs.toolFunctionDefinitionSchema; 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.wrapAISDKModel = wrapAISDKModel; exports.wrapAnthropic = wrapAnthropic; exports.wrapOpenAI = wrapOpenAI; exports.wrapOpenAIv4 = wrapOpenAIv4; exports.wrapTraced = wrapTraced;

package/dist/index.mjs

@@ -7979,6 +7979,11 @@ var Prompt = class _Prompt {
   }
 };
 var TEST_API_KEY = "___TEST_API_KEY__THIS_IS_NOT_REAL___";
+function setInitialTestState() {
+  if (!_internalGetGlobalState()) {
+    _internalSetInitialState();
+  }
+}
 async function simulateLoginForTests() {
   return await login({
     apiKey: TEST_API_KEY,
@@ -7996,7 +8001,8 @@ var _exportsForTestingOnly = {
   useTestBackgroundLogger,
   clearTestBackgroundLogger,
   simulateLoginForTests,
-  simulateLogoutForTests
+  simulateLogoutForTests,
+  setInitialTestState
 };
 
 // src/node.ts
@@ -8035,6 +8041,7 @@ __export(exports_node_exports, {
   BaseAttachment: () => BaseAttachment,
   BaseExperiment: () => BaseExperiment,
   BraintrustExporter: () => BraintrustExporter,
+  BraintrustMiddleware: () => BraintrustMiddleware,
   BraintrustSpanProcessor: () => BraintrustSpanProcessor,
   BraintrustState: () => BraintrustState,
   BraintrustStream: () => BraintrustStream,
@@ -10841,10 +10848,17 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  const completionProxy = createEndpointProxy(
-    openai.chat.completions,
-    wrapChatCompletion
-  );
+  const completionProxy = new Proxy(openai.chat.completions, {
+    get(target, name, receiver) {
+      const baseVal = Reflect.get(target, name, receiver);
+      if (name === "create") {
+        return wrapChatCompletion(baseVal.bind(target));
+      } else if (name === "parse") {
+        return wrapBetaChatCompletionParse(baseVal.bind(target));
+      }
+      return baseVal;
+    }
+  });
   const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
@@ -11220,7 +11234,218 @@ var WrapperStream = class {
   }
 };
 
-// src/wrappers/ai-sdk.ts
+// src/wrappers/ai-sdk-v2.ts
+import { SpanTypeAttribute as SpanTypeAttribute4 } from "@braintrust/core";
+function detectProviderFromResult(result) {
+  if (!result?.providerMetadata) {
+    return void 0;
+  }
+  const keys = Object.keys(result.providerMetadata);
+  return keys?.at(0);
+}
+function extractModelFromResult(result) {
+  if (result?.response?.modelId) {
+    return result.response.modelId;
+  }
+  if (result?.request?.body?.model) {
+    return result.request.body.model;
+  }
+  return void 0;
+}
+function camelToSnake(str) {
+  return str.replace(/[A-Z]/g, (letter) => `_${letter.toLowerCase()}`);
+}
+function extractModelParameters(params) {
+  const modelParams = {};
+  const excludeKeys = /* @__PURE__ */ new Set([
+    "prompt",
+    // Already captured as input
+    "system",
+    // Already captured as input
+    "messages",
+    // Already captured as input
+    "model",
+    // Already captured in metadata.model
+    "providerOptions"
+    // Internal AI SDK configuration
+  ]);
+  for (const [key, value] of Object.entries(params)) {
+    if (value !== void 0 && !excludeKeys.has(key)) {
+      const snakeKey = camelToSnake(key);
+      modelParams[snakeKey] = value;
+    }
+  }
+  return modelParams;
+}
+function getNumberProperty(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;
+}
+function normalizeUsageMetrics(usage) {
+  const metrics = {};
+  const inputTokens = getNumberProperty(usage, "inputTokens");
+  if (inputTokens !== void 0) {
+    metrics.prompt_tokens = inputTokens;
+  }
+  const outputTokens = getNumberProperty(usage, "outputTokens");
+  if (outputTokens !== void 0) {
+    metrics.completion_tokens = outputTokens;
+  }
+  const totalTokens = getNumberProperty(usage, "totalTokens");
+  if (totalTokens !== void 0) {
+    metrics.tokens = totalTokens;
+  }
+  const reasoningTokens = getNumberProperty(usage, "reasoningTokens");
+  if (reasoningTokens !== void 0) {
+    metrics.completion_reasoning_tokens = reasoningTokens;
+  }
+  const cachedInputTokens = getNumberProperty(usage, "cachedInputTokens");
+  if (cachedInputTokens !== void 0) {
+    metrics.prompt_cached_tokens = cachedInputTokens;
+  }
+  return metrics;
+}
+function BraintrustMiddleware(config = {}) {
+  return {
+    wrapGenerate: async ({ doGenerate, params }) => {
+      const spanArgs = {
+        name: "ai-sdk.generateText",
+        spanAttributes: {
+          type: SpanTypeAttribute4.LLM
+        },
+        event: {
+          input: params.prompt,
+          metadata: {
+            ...extractModelParameters(params)
+          }
+        }
+      };
+      const span = startSpan(spanArgs);
+      try {
+        const result = await doGenerate();
+        const metadata = {};
+        const provider = detectProviderFromResult(result);
+        if (provider !== void 0) {
+          metadata.provider = provider;
+        }
+        if (result.finishReason !== void 0) {
+          metadata.finish_reason = result.finishReason;
+        }
+        const model = extractModelFromResult(result);
+        if (model !== void 0) {
+          metadata.model = model;
+        }
+        span.log({
+          output: result.content,
+          metadata,
+          metrics: normalizeUsageMetrics(result.usage)
+        });
+        return result;
+      } catch (error2) {
+        span.log({
+          error: error2 instanceof Error ? error2.message : String(error2)
+        });
+        throw error2;
+      } finally {
+        span.end();
+      }
+    },
+    wrapStream: async ({ doStream, params }) => {
+      const spanArgs = {
+        name: "ai-sdk.streamText",
+        spanAttributes: {
+          type: SpanTypeAttribute4.LLM
+        },
+        event: {
+          input: params.prompt,
+          metadata: {
+            ...extractModelParameters(params)
+          }
+        }
+      };
+      const span = startSpan(spanArgs);
+      try {
+        const { stream, ...rest } = await doStream();
+        const textChunks = [];
+        let finalUsage = {};
+        let finalFinishReason = void 0;
+        let providerMetadata = {};
+        const transformStream = new TransformStream({
+          // eslint-disable-next-line @typescript-eslint/no-explicit-any
+          transform(chunk, controller) {
+            try {
+              if (chunk.type === "text-delta" && chunk.delta) {
+                textChunks.push(chunk.delta);
+              }
+              if (chunk.type === "finish") {
+                finalFinishReason = chunk.finishReason;
+                finalUsage = chunk.usage || {};
+                providerMetadata = chunk.providerMetadata || {};
+              }
+              controller.enqueue(chunk);
+            } catch (error2) {
+              span.log({
+                error: error2 instanceof Error ? error2.message : String(error2)
+              });
+              span.end();
+              controller.error(error2);
+            }
+          },
+          flush() {
+            try {
+              const generatedText = textChunks.join("");
+              const output = generatedText ? [{ type: "text", text: generatedText }] : [];
+              const resultForDetection = {
+                providerMetadata,
+                response: rest.response,
+                ...rest
+              };
+              const metadata = {};
+              const provider = detectProviderFromResult(resultForDetection);
+              if (provider !== void 0) {
+                metadata.provider = provider;
+              }
+              if (finalFinishReason !== void 0) {
+                metadata.finish_reason = finalFinishReason;
+              }
+              const model = extractModelFromResult(resultForDetection);
+              if (model !== void 0) {
+                metadata.model = model;
+              }
+              span.log({
+                output,
+                metadata,
+                metrics: normalizeUsageMetrics(finalUsage)
+              });
+              span.end();
+            } catch (error2) {
+              span.log({
+                error: error2 instanceof Error ? error2.message : String(error2)
+              });
+              span.end();
+              throw error2;
+            }
+          }
+        });
+        return {
+          stream: stream.pipeThrough(transformStream),
+          ...rest
+        };
+      } catch (error2) {
+        span.log({
+          error: error2 instanceof Error ? error2.message : String(error2)
+        });
+        span.end();
+        throw error2;
+      }
+    }
+  };
+}
+
+// src/wrappers/ai-sdk-v1.ts
 function wrapAISDKModel(model) {
   const m = model;
   if (m?.specificationVersion === "v1" && typeof m?.provider === "string" && typeof m?.modelId === "string") {
@@ -11484,7 +11709,7 @@ function postProcessOutput(text, toolCalls, finishReason) {
 }
 
 // src/wrappers/anthropic.ts
-import { SpanTypeAttribute as SpanTypeAttribute4 } from "@braintrust/core";
+import { SpanTypeAttribute as SpanTypeAttribute5 } from "@braintrust/core";
 function wrapAnthropic(anthropic) {
   const au = anthropic;
   if (au && typeof au === "object" && "messages" in au && typeof au.messages === "object" && au.messages && "create" in au.messages) {
@@ -11542,7 +11767,7 @@ function createProxy(create) {
       const spanArgs = {
         name: "anthropic.messages.create",
         spanAttributes: {
-          type: SpanTypeAttribute4.LLM
+          type: SpanTypeAttribute5.LLM
         },
         event: {
           input,
@@ -11976,6 +12201,8 @@ var evalBodySchema = z5.object({
       name: z5.string()
     })
   ).nullish(),
+  experiment_name: z5.string().nullish(),
+  project_id: z5.string().nullish(),
   parent: invokeParent.optional(),
   stream: z5.boolean().optional()
 });
@@ -12013,6 +12240,7 @@ export {
   BaseAttachment,
   BaseExperiment,
   BraintrustExporter,
+  BraintrustMiddleware,
   BraintrustSpanProcessor,
   BraintrustState,
   BraintrustStream,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "braintrust",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "SDK for integrating Braintrust",
   "repository": {
     "type": "git",
@@ -51,11 +51,13 @@
     "docs": "npx typedoc --options typedoc.json src/index.ts",
     "prepublishOnly": "../../scripts/node_prepublish_sdk.py",
     "postpublish": "../../scripts/node_postpublish_sdk.py",
-    "test": "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts",
+    "test": "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts  --exclude src/wrappers/ai-sdk-v1.test.ts --exclude src/wrappers/ai-sdk-v2.test.ts",
     "test:anthropic": "vitest run src/wrappers/anthropic.test.ts",
     "test:openai": "vitest run src/wrappers/oai.test.ts",
     "test:otel": "vitest run src/otel.test.ts",
-    "test:otel-no-deps": "vitest run src/otel-no-deps.test.ts --reporter=verbose"
+    "test:otel-no-deps": "vitest run src/otel-no-deps.test.ts --reporter=verbose",
+    "test:ai-sdk-v1": "vitest run src/wrappers/ai-sdk-v1.test.ts",
+    "test:ai-sdk-v2": "vitest run src/wrappers/ai-sdk-v2.test.ts src/wrappers/ai-sdk-v1.test.ts"
   },
   "author": "",
   "license": "MIT",