braintrust 3.10.0 → 3.11.0

package/dist/auto-instrumentations/hook.mjs

@@ -1,7 +1,8 @@
 import {
+  OPENAI_API_PROMISE_PATCH,
   getPackageName,
   getPackageVersion
-} from "./chunk-ITP7RAUY.mjs";
+} from "./chunk-MWZXZQUO.mjs";
 
 // src/auto-instrumentations/hook.mts
 import { register } from "module";
@@ -368,6 +369,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({
@@ -1454,6 +1498,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"
@@ -1552,6 +1612,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: {
@@ -1941,7 +2105,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: {
@@ -1993,7 +2170,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: {
@@ -2045,7 +2235,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: {
@@ -2097,7 +2300,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: {
@@ -2178,6 +2394,303 @@ 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"
+    }
+  }
+];
+
 // src/auto-instrumentations/loader/cjs-patch.ts
 import {
   create
@@ -2215,8 +2728,12 @@ var ModulePatch = class {
         if (!self.packages.has(packageName)) {
           return self.originalCompile.apply(this, args);
         }
-        const version = getPackageVersion(resolvedModule.basedir);
         const normalizedModulePath = resolvedModule.path.replace(/\\/g, "/");
+        if (packageName === "openai" && normalizedModulePath.includes("api-promise")) {
+          args[0] = content + OPENAI_API_PROMISE_PATCH;
+          return self.originalCompile.apply(this, args);
+        }
+        const version = getPackageVersion(resolvedModule.basedir);
         const transformer = self.instrumentator.getTransformer(
           packageName,
           version,
@@ -2272,7 +2789,11 @@ function patchTracingChannel(tracingChannelFn) {
   }
   if (TracingChannel.prototype.tracePromise) {
     TracingChannel.prototype.tracePromise = function(fn, context = {}, thisArg, ...args) {
-      const { start, end, asyncStart, asyncEnd, error } = this;
+      const start = this.start;
+      const end = this.end;
+      const asyncStart = this.asyncStart;
+      const asyncEnd = this.asyncEnd;
+      const error = this.error;
       function publishRejected(err) {
         context.error = err;
         error?.publish(context);
@@ -2356,6 +2877,13 @@ function isDisabled(disabled, ...names) {
 var disabledIntegrations = readDisabledIntegrations();
 var allConfigs = [
   ...isDisabled(disabledIntegrations, "openai") ? [] : openaiConfigs,
+  ...isDisabled(
+    disabledIntegrations,
+    "openai-codex",
+    "openai-codex-sdk",
+    "codex",
+    "codex-sdk"
+  ) ? [] : openAICodexConfigs,
   ...isDisabled(disabledIntegrations, "anthropic") ? [] : anthropicConfigs,
   ...isDisabled(disabledIntegrations, "aisdk", "ai-sdk", "vercel-ai") ? [] : aiSDKConfigs,
   ...isDisabled(disabledIntegrations, "claudeagentsdk", "claude-agent-sdk") ? [] : claudeAgentSDKConfigs,
@@ -2367,7 +2895,14 @@ var allConfigs = [
   ...isDisabled(disabledIntegrations, "mistral") ? [] : mistralConfigs,
   ...isDisabled(disabledIntegrations, "googleadk", "google-adk") ? [] : googleADKConfigs,
   ...isDisabled(disabledIntegrations, "cohere") ? [] : cohereConfigs,
-  ...isDisabled(disabledIntegrations, "groq", "groq-sdk") ? [] : groqConfigs
+  ...isDisabled(disabledIntegrations, "groq", "groq-sdk") ? [] : groqConfigs,
+  ...isDisabled(disabledIntegrations, "genkit", "firebase-genkit") ? [] : genkitConfigs,
+  ...isDisabled(
+    disabledIntegrations,
+    "githubcopilot",
+    "github-copilot",
+    "copilot-sdk"
+  ) ? [] : gitHubCopilotConfigs
 ];
 register("./loader/esm-hook.mjs", {
   parentURL: import.meta.url,