braintrust 3.24.0 → 3.25.0

package/dist/index.mjs

@@ -139,8 +139,9 @@ import {
   wrapMastraAgent,
   wrapTraced,
   zodToJsonSchema
-} from "./chunk-XE5FS7QY.mjs";
+} from "./chunk-2SANLSWX.mjs";
 import {
+  INSTRUMENTATION_NAMES,
   __export,
   aiSDKChannels,
   anthropicChannels,
@@ -163,8 +164,9 @@ import {
   openRouterAgentChannels,
   openRouterChannels,
   piCodingAgentChannels,
-  strandsAgentSDKChannels
-} from "./chunk-7F6GCRHH.mjs";
+  strandsAgentSDKChannels,
+  withSpanInstrumentationName
+} from "./chunk-ABR2QWDP.mjs";
 
 // src/exports.ts
 var exports_exports = {};
@@ -941,6 +943,12 @@ var wrapAgentClass = (AgentClass, options = {}) => {
           const original = Reflect.get(instanceTarget, prop, instanceTarget);
           if (harnessAgent && typeof original === "function") {
             switch (prop) {
+              case "createSession":
+                return wrapHarnessAgentCreateSession(
+                  // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+                  original,
+                  harnessAgent
+                );
               case "generate":
                 return wrapHarnessAgentGenerate(
                   original,
@@ -990,6 +998,19 @@ var wrapAgentClass = (AgentClass, options = {}) => {
     }
   });
 };
+var wrapHarnessAgentCreateSession = (createSession, instance) => {
+  const wrapper = function(params) {
+    return harnessAgentChannels.createSession.tracePromise(
+      () => params === void 0 ? createSession.call(instance) : createSession.call(instance, params),
+      createAISDKChannelContext(params ?? {}, { self: instance })
+    );
+  };
+  Object.defineProperty(wrapper, "name", {
+    value: "HarnessAgent.createSession",
+    writable: false
+  });
+  return wrapper;
+};
 var wrapHarnessAgentGenerate = (generate, instance, channel, name, options) => makeGenerateTextWrapper(
   channel,
   name,
@@ -1259,12 +1280,17 @@ var BraintrustLanguageModelWrapper = class {
   // For the first cut, do not support custom span_info arguments. We can
   // propagate those via async local storage
   async doGenerate(options) {
-    const span = startSpan({
-      name: "Chat Completion",
-      spanAttributes: {
-        type: "llm"
-      }
-    });
+    const span = startSpan(
+      withSpanInstrumentationName(
+        {
+          name: "Chat Completion",
+          spanAttributes: {
+            type: "llm"
+          }
+        },
+        INSTRUMENTATION_NAMES.AI_SDK
+      )
+    );
     const { prompt, mode, ...rest } = options;
     const startTime = getCurrentUnixTimestamp();
     try {
@@ -1295,12 +1321,17 @@ var BraintrustLanguageModelWrapper = class {
   async doStream(options) {
     const { prompt, mode, ...rest } = options;
     const startTime = getCurrentUnixTimestamp();
-    const span = startSpan({
-      name: "Chat Completion",
-      spanAttributes: {
-        type: "llm"
-      }
-    });
+    const span = startSpan(
+      withSpanInstrumentationName(
+        {
+          name: "Chat Completion",
+          spanAttributes: {
+            type: "llm"
+          }
+        },
+        INSTRUMENTATION_NAMES.AI_SDK
+      )
+    );
     span.log({
       input: postProcessPrompt(prompt),
       metadata: {
@@ -1655,7 +1686,9 @@ function BraintrustMiddleware(config = {}) {
           }
         }
       };
-      const span = startSpan(spanArgs);
+      const span = startSpan(
+        withSpanInstrumentationName(spanArgs, INSTRUMENTATION_NAMES.AI_SDK)
+      );
       try {
         const result = await doGenerate();
         const metadata = {};
@@ -1717,7 +1750,9 @@ function BraintrustMiddleware(config = {}) {
           }
         }
       };
-      const span = startSpan(spanArgs);
+      const span = startSpan(
+        withSpanInstrumentationName(spanArgs, INSTRUMENTATION_NAMES.AI_SDK)
+      );
       try {
         const { stream, ...rest } = await doStream();
         const textChunks = [];
@@ -1924,7 +1959,12 @@ var EveBridge = class {
     };
     const span = withCurrent(
       NOOP_SPAN,
-      () => _internalStartSpanWithInitialMerge({ ...args, startTime })
+      () => _internalStartSpanWithInitialMerge(
+        withSpanInstrumentationName(
+          { ...args, startTime },
+          INSTRUMENTATION_NAMES.EVE
+        )
+      )
     );
     if (typeof rowId !== "string") {
       return span;
@@ -8836,26 +8876,30 @@ async function runEvaluatorInternal(experiment, evaluator, progressReporter, fil
         }
       }
     })();
+    let timeoutId;
+    let abortHandler;
+    const cleanupCancellation = () => {
+      if (timeoutId !== void 0) {
+        clearTimeout(timeoutId);
+        timeoutId = void 0;
+      }
+      if (abortHandler && evaluator.signal) {
+        evaluator.signal.removeEventListener("abort", abortHandler);
+        abortHandler = void 0;
+      }
+    };
     const cancel = async () => {
       await new Promise((_, reject2) => {
         if (cancelled) {
           reject2(new InternalAbortError("Evaluator already cancelled"));
           return;
         }
-        let timeoutId;
-        let abortHandler;
         const rejectOnce = (error) => {
           if (cancelled) {
             return;
           }
           cancelled = true;
-          if (timeoutId) {
-            clearTimeout(timeoutId);
-            timeoutId = void 0;
-          }
-          if (abortHandler && evaluator.signal) {
-            evaluator.signal.removeEventListener("abort", abortHandler);
-          }
+          cleanupCancellation();
           reject2(error);
         };
         if (evaluator.timeout) {
@@ -8895,6 +8939,7 @@ async function runEvaluatorInternal(experiment, evaluator, progressReporter, fil
       }
       throw e;
     } finally {
+      cleanupCancellation();
       if (!collectResults) {
         collectedResults.length = 0;
       }

package/dist/instrumentation/index.d.mts

@@ -7942,7 +7942,12 @@ declare class BraintrustState {
     private _contextManager;
     private _otelFlushCallback;
     spanOriginEnvironment: SpanOriginEnvironment | undefined;
+    private traceContextSigningSecret;
     constructor(loginParams: LoginOptions);
+    /** @internal */
+    _internalSetTraceContextSigningSecret(secret: string | undefined): void;
+    /** @internal */
+    _internalGetTraceContextSigningSecret(): string | undefined;
     resetLoginInfo(): void;
     resetIdGenState(): void;
     [RESET_CONTEXT_MANAGER_STATE](): void;

package/dist/instrumentation/index.d.ts

@@ -7942,7 +7942,12 @@ declare class BraintrustState {
     private _contextManager;
     private _otelFlushCallback;
     spanOriginEnvironment: SpanOriginEnvironment | undefined;
+    private traceContextSigningSecret;
     constructor(loginParams: LoginOptions);
+    /** @internal */
+    _internalSetTraceContextSigningSecret(secret: string | undefined): void;
+    /** @internal */
+    _internalGetTraceContextSigningSecret(): string | undefined;
     resetLoginInfo(): void;
     resetIdGenState(): void;
     [RESET_CONTEXT_MANAGER_STATE](): void;