braintrust 3.10.0 → 3.11.0

package/dist/auto-instrumentations/{chunk-GZNXBBPU.mjs → chunk-DIV5TO4S.mjs}

@@ -360,6 +360,49 @@ var openaiConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/openai-codex-channels.ts
+var openAICodexChannels = defineChannels("@openai/codex-sdk", {
+  run: channel({
+    channelName: "Thread.run",
+    kind: "async"
+  }),
+  runStreamed: channel({
+    channelName: "Thread.runStreamed",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/openai-codex.ts
+var openAICodexVersionRange = ">=0.128.0 <1.0.0";
+var openAICodexConfigs = [
+  {
+    channelName: openAICodexChannels.run.channelName,
+    module: {
+      name: "@openai/codex-sdk",
+      versionRange: openAICodexVersionRange,
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      className: "Thread",
+      methodName: "run",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: openAICodexChannels.runStreamed.channelName,
+    module: {
+      name: "@openai/codex-sdk",
+      versionRange: openAICodexVersionRange,
+      filePath: "dist/index.js"
+    },
+    functionQuery: {
+      className: "Thread",
+      methodName: "runStreamed",
+      kind: "Async"
+    }
+  }
+];
+
 // src/instrumentation/plugins/anthropic-channels.ts
 var anthropicChannels = defineChannels("@anthropic-ai/sdk", {
   messagesCreate: channel({
@@ -1446,6 +1489,22 @@ var mistralChannels = defineChannels("@mistralai/mistralai", {
     channelName: "embeddings.create",
     kind: "async"
   }),
+  classifiersModerate: channel({
+    channelName: "classifiers.moderate",
+    kind: "async"
+  }),
+  classifiersModerateChat: channel({
+    channelName: "classifiers.moderateChat",
+    kind: "async"
+  }),
+  classifiersClassify: channel({
+    channelName: "classifiers.classify",
+    kind: "async"
+  }),
+  classifiersClassifyChat: channel({
+    channelName: "classifiers.classifyChat",
+    kind: "async"
+  }),
   fimComplete: channel({
     channelName: "fim.complete",
     kind: "async"
@@ -1544,6 +1603,110 @@ var mistralConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: mistralChannels.classifiersModerate.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "moderate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersModerate.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "moderate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersModerateChat.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.0.0 <2.0.0",
+      filePath: "sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "moderateChat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersModerateChat.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "moderateChat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersClassify.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.10.0 <2.0.0",
+      filePath: "sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "classify",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersClassify.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "classify",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersClassifyChat.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=1.10.0 <2.0.0",
+      filePath: "sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "classifyChat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: mistralChannels.classifiersClassifyChat.channelName,
+    module: {
+      name: "@mistralai/mistralai",
+      versionRange: ">=2.0.0 <3.0.0",
+      filePath: "esm/sdk/classifiers.js"
+    },
+    functionQuery: {
+      className: "Classifiers",
+      methodName: "classifyChat",
+      kind: "Async"
+    }
+  },
   {
     channelName: mistralChannels.fimComplete.channelName,
     module: {
@@ -1689,7 +1852,20 @@ var cohereConfigs = [
     channelName: cohereChannels.chat.channelName,
     module: {
       name: "cohere-ai",
-      versionRange: ">=7.20.0 <8.0.0",
+      versionRange: ">=7.0.0 <7.21.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2",
+      methodName: "chat",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chat.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.21.0 <8.0.0",
       filePath: "api/resources/v2/client/Client.js"
     },
     functionQuery: {
@@ -1741,7 +1917,20 @@ var cohereConfigs = [
     channelName: cohereChannels.chatStream.channelName,
     module: {
       name: "cohere-ai",
-      versionRange: ">=7.20.0 <8.0.0",
+      versionRange: ">=7.0.0 <7.21.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2",
+      methodName: "chatStream",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.chatStream.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.21.0 <8.0.0",
       filePath: "api/resources/v2/client/Client.js"
     },
     functionQuery: {
@@ -1793,7 +1982,20 @@ var cohereConfigs = [
     channelName: cohereChannels.embed.channelName,
     module: {
       name: "cohere-ai",
-      versionRange: ">=7.20.0 <8.0.0",
+      versionRange: ">=7.0.0 <7.21.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.embed.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.21.0 <8.0.0",
       filePath: "api/resources/v2/client/Client.js"
     },
     functionQuery: {
@@ -1845,7 +2047,20 @@ var cohereConfigs = [
     channelName: cohereChannels.rerank.channelName,
     module: {
       name: "cohere-ai",
-      versionRange: ">=7.20.0 <8.0.0",
+      versionRange: ">=7.0.0 <7.21.0",
+      filePath: "api/resources/v2/client/Client.js"
+    },
+    functionQuery: {
+      className: "V2",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cohereChannels.rerank.channelName,
+    module: {
+      name: "cohere-ai",
+      versionRange: ">=7.21.0 <8.0.0",
       filePath: "api/resources/v2/client/Client.js"
     },
     functionQuery: {
@@ -1926,10 +2141,308 @@ var groqConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/genkit-channels.ts
+var genkitChannels = defineChannels("@genkit-ai/ai", {
+  generate: channel({
+    channelName: "generate",
+    kind: "async"
+  }),
+  generateStream: channel({
+    channelName: "generateStream",
+    kind: "sync-stream"
+  }),
+  embed: channel({
+    channelName: "embed",
+    kind: "async"
+  }),
+  embedMany: channel({
+    channelName: "embedMany",
+    kind: "async"
+  }),
+  actionRun: channel({
+    channelName: "action.run",
+    kind: "async"
+  }),
+  actionStream: channel({
+    channelName: "action.stream",
+    kind: "sync-stream"
+  })
+});
+var genkitCoreChannels = defineChannels("@genkit-ai/core", {
+  actionSpan: channel({
+    channelName: "action.span",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/genkit.ts
+var genkitVersionRange = ">=1.0.0 <2.0.0";
+var genkitConfigs = [
+  {
+    channelName: genkitChannels.generate.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.mjs"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.generate.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.js"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "generate",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.generateStream.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.mjs"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "generateStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: genkitChannels.generateStream.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.js"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "generateStream",
+      kind: "Sync"
+    }
+  },
+  {
+    channelName: genkitChannels.embed.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.mjs"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.embed.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.js"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.embedMany.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.mjs"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "embedMany",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.embedMany.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.js"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "embedMany",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.actionRun.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.mjs"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "run",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitChannels.actionRun.channelName,
+    module: {
+      name: "@genkit-ai/ai",
+      versionRange: genkitVersionRange,
+      filePath: "lib/genkit-ai.js"
+    },
+    functionQuery: {
+      className: "GenkitAI",
+      methodName: "run",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitCoreChannels.actionSpan.channelName,
+    module: {
+      name: "@genkit-ai/core",
+      versionRange: genkitVersionRange,
+      filePath: "lib/tracing/instrumentation.mjs"
+    },
+    functionQuery: {
+      functionName: "runInNewSpan",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: genkitCoreChannels.actionSpan.channelName,
+    module: {
+      name: "@genkit-ai/core",
+      versionRange: genkitVersionRange,
+      filePath: "lib/tracing/instrumentation.js"
+    },
+    functionQuery: {
+      functionName: "runInNewSpan",
+      kind: "Async"
+    }
+  }
+];
+
+// src/instrumentation/plugins/github-copilot-channels.ts
+var gitHubCopilotChannels = defineChannels("@github/copilot-sdk", {
+  createSession: channel({
+    channelName: "client.createSession",
+    kind: "async"
+  }),
+  resumeSession: channel({
+    channelName: "client.resumeSession",
+    kind: "async"
+  }),
+  sendAndWait: channel({
+    channelName: "session.sendAndWait",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/github-copilot.ts
+var gitHubCopilotConfigs = [
+  // ESM: CopilotClient.createSession
+  {
+    channelName: gitHubCopilotChannels.createSession.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/client.js"
+    },
+    functionQuery: {
+      className: "CopilotClient",
+      methodName: "createSession",
+      kind: "Async"
+    }
+  },
+  // CJS: CopilotClient.createSession
+  {
+    channelName: gitHubCopilotChannels.createSession.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/cjs/client.js"
+    },
+    functionQuery: {
+      className: "CopilotClient",
+      methodName: "createSession",
+      kind: "Async"
+    }
+  },
+  // ESM: CopilotClient.resumeSession
+  {
+    channelName: gitHubCopilotChannels.resumeSession.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/client.js"
+    },
+    functionQuery: {
+      className: "CopilotClient",
+      methodName: "resumeSession",
+      kind: "Async"
+    }
+  },
+  // CJS: CopilotClient.resumeSession
+  {
+    channelName: gitHubCopilotChannels.resumeSession.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/cjs/client.js"
+    },
+    functionQuery: {
+      className: "CopilotClient",
+      methodName: "resumeSession",
+      kind: "Async"
+    }
+  },
+  // ESM: CopilotSession.sendAndWait
+  {
+    channelName: gitHubCopilotChannels.sendAndWait.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/session.js"
+    },
+    functionQuery: {
+      className: "CopilotSession",
+      methodName: "sendAndWait",
+      kind: "Async"
+    }
+  },
+  // CJS: CopilotSession.sendAndWait
+  {
+    channelName: gitHubCopilotChannels.sendAndWait.channelName,
+    module: {
+      name: "@github/copilot-sdk",
+      versionRange: ">=0.3.0",
+      filePath: "dist/cjs/session.js"
+    },
+    functionQuery: {
+      className: "CopilotSession",
+      methodName: "sendAndWait",
+      kind: "Async"
+    }
+  }
+];
+
 export {
   channel,
   defineChannels,
   openaiConfigs,
+  openAICodexConfigs,
   anthropicConfigs,
   aiSDKConfigs,
   claudeAgentSDKConfigs,
@@ -1940,5 +2453,7 @@ export {
   openRouterConfigs,
   mistralConfigs,
   cohereConfigs,
-  groqConfigs
+  groqConfigs,
+  genkitConfigs,
+  gitHubCopilotConfigs
 };

package/dist/auto-instrumentations/{chunk-XWEQQOQH.mjs → chunk-G6ZWXGZB.mjs}

@@ -4,14 +4,17 @@ import {
   claudeAgentSDKConfigs,
   cohereConfigs,
   cursorSDKConfigs,
+  genkitConfigs,
+  gitHubCopilotConfigs,
   googleGenAIConfigs,
   groqConfigs,
   huggingFaceConfigs,
   mistralConfigs,
+  openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs
-} from "./chunk-GZNXBBPU.mjs";
+} from "./chunk-DIV5TO4S.mjs";
 
 // src/auto-instrumentations/bundler/plugin.ts
 import { createUnplugin } from "unplugin";
@@ -36,6 +39,7 @@ function getModuleVersion(basedir) {
 var unplugin = createUnplugin((options = {}) => {
   const allInstrumentations = [
     ...openaiConfigs,
+    ...openAICodexConfigs,
     ...anthropicConfigs,
     ...aiSDKConfigs,
     ...claudeAgentSDKConfigs,
@@ -47,6 +51,8 @@ var unplugin = createUnplugin((options = {}) => {
     ...mistralConfigs,
     ...cohereConfigs,
     ...groqConfigs,
+    ...genkitConfigs,
+    ...gitHubCopilotConfigs,
     ...options.instrumentations || []
   ];
   const dcModule = options.browser === false ? void 0 : "dc-browser";
@@ -89,8 +95,12 @@ var unplugin = createUnplugin((options = {}) => {
       try {
         const moduleType = isModule ? "esm" : "cjs";
         const result = transformer.transform(code, moduleType);
+        const transformedCode = result.code.replace(
+          /const \{tracingChannel: ([A-Za-z_$][\w$]*)\} = ([A-Za-z_$][\w$]*);/g,
+          "const $1 = $2.tracingChannel;"
+        );
         return {
-          code: result.code,
+          code: transformedCode,
           map: result.map
         };
       } catch (error) {

package/dist/auto-instrumentations/{chunk-ITP7RAUY.mjs → chunk-MWZXZQUO.mjs}

@@ -53,7 +53,29 @@ function getPackageName(baseDir) {
   return void 0;
 }
 
+// src/auto-instrumentations/loader/openai-api-promise-patch.ts
+var OPENAI_API_PROMISE_PATCH = `
+;(function __btPatchAPIPromise() {
+  if (typeof APIPromise === "undefined" || APIPromise.prototype.__btParsePatched) return;
+  APIPromise.prototype.__btParsePatched = true;
+  var _origThen = APIPromise.prototype.then;
+  APIPromise.prototype.then = function __btThen(onfulfilled, onrejected) {
+    if (!this.__btParseWrapped && Object.prototype.hasOwnProperty.call(this, "parseResponse")) {
+      this.__btParseWrapped = true;
+      var _origParse = this.parseResponse;
+      var _cached;
+      this.parseResponse = function() {
+        if (!_cached) _cached = _origParse.apply(this, arguments);
+        return _cached;
+      };
+    }
+    return _origThen.call(this, onfulfilled, onrejected);
+  };
+})();
+`;
+
 export {
   getPackageVersion,
-  getPackageName
+  getPackageName,
+  OPENAI_API_PROMISE_PATCH
 };