braintrust 3.9.0 → 3.11.0

package/dist/auto-instrumentations/bundler/rollup.cjs

@@ -404,6 +404,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({
@@ -929,6 +972,71 @@ var claudeAgentSDKConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/cursor-sdk-channels.ts
+var cursorSDKChannels = defineChannels("@cursor/sdk", {
+  create: channel({
+    channelName: "Agent.create",
+    kind: "async"
+  }),
+  resume: channel({
+    channelName: "Agent.resume",
+    kind: "async"
+  }),
+  prompt: channel({
+    channelName: "Agent.prompt",
+    kind: "async"
+  }),
+  send: channel({
+    channelName: "agent.send",
+    kind: "async"
+  })
+});
+
+// src/auto-instrumentations/configs/cursor-sdk.ts
+var cursorSDKVersionRange = ">=1.0.7 <2.0.0";
+var cursorSDKEntrypoints = ["dist/esm/index.js", "dist/cjs/index.js"];
+var cursorSDKConfigs = cursorSDKEntrypoints.flatMap((filePath) => [
+  {
+    channelName: cursorSDKChannels.create.channelName,
+    module: {
+      name: "@cursor/sdk",
+      versionRange: cursorSDKVersionRange,
+      filePath
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cursorSDKChannels.resume.channelName,
+    module: {
+      name: "@cursor/sdk",
+      versionRange: cursorSDKVersionRange,
+      filePath
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "resume",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: cursorSDKChannels.prompt.channelName,
+    module: {
+      name: "@cursor/sdk",
+      versionRange: cursorSDKVersionRange,
+      filePath
+    },
+    functionQuery: {
+      className: "Agent",
+      methodName: "prompt",
+      kind: "Async"
+    }
+  }
+]);
+
 // src/instrumentation/plugins/google-genai-channels.ts
 var googleGenAIChannels = defineChannels("@google/genai", {
   generateContent: channel({
@@ -1425,6 +1533,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"
@@ -1523,6 +1647,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: {
@@ -1664,6 +1892,32 @@ var cohereConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: cohereChannels.chat.channelName,
+    module: {
+      name: "cohere-ai",
+      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: {
+      className: "V2Client",
+      methodName: "chat",
+      kind: "Async"
+    }
+  },
   {
     channelName: cohereChannels.chat.channelName,
     module: {
@@ -1703,6 +1957,32 @@ var cohereConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: cohereChannels.chatStream.channelName,
+    module: {
+      name: "cohere-ai",
+      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: {
+      className: "V2Client",
+      methodName: "chatStream",
+      kind: "Async"
+    }
+  },
   {
     channelName: cohereChannels.chatStream.channelName,
     module: {
@@ -1742,6 +2022,32 @@ var cohereConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: cohereChannels.embed.channelName,
+    module: {
+      name: "cohere-ai",
+      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: {
+      className: "V2Client",
+      methodName: "embed",
+      kind: "Async"
+    }
+  },
   {
     channelName: cohereChannels.embed.channelName,
     module: {
@@ -1781,6 +2087,32 @@ var cohereConfigs = [
       kind: "Async"
     }
   },
+  {
+    channelName: cohereChannels.rerank.channelName,
+    module: {
+      name: "cohere-ai",
+      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: {
+      className: "V2Client",
+      methodName: "rerank",
+      kind: "Async"
+    }
+  },
   {
     channelName: cohereChannels.rerank.channelName,
     module: {
@@ -1809,6 +2141,347 @@ var cohereConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/groq-channels.ts
+var groqChannels = defineChannels("groq-sdk", {
+  chatCompletionsCreate: channel({
+    channelName: "chat.completions.create",
+    kind: "async"
+  }),
+  embeddingsCreate: channel(
+    {
+      channelName: "embeddings.create",
+      kind: "async"
+    }
+  )
+});
+
+// src/auto-instrumentations/configs/groq.ts
+var groqConfigs = [
+  {
+    channelName: groqChannels.chatCompletionsCreate.channelName,
+    module: {
+      name: "groq-sdk",
+      versionRange: ">=1.0.0",
+      filePath: "resources/chat/completions.mjs"
+    },
+    functionQuery: {
+      className: "Completions",
+      methodName: "create",
+      kind: "Async"
+    }
+  },
+  {
+    channelName: groqChannels.embeddingsCreate.channelName,
+    module: {
+      name: "groq-sdk",
+      versionRange: ">=1.0.0",
+      filePath: "resources/embeddings.mjs"
+    },
+    functionQuery: {
+      className: "Embeddings",
+      methodName: "create",
+      kind: "Async"
+    }
+  }
+];
+
+// 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/bundler/plugin.ts
 function getModuleVersion(basedir) {
   try {
@@ -1824,15 +2497,20 @@ function getModuleVersion(basedir) {
 var unplugin = (0, import_unplugin.createUnplugin)((options = {}) => {
   const allInstrumentations = [
     ...openaiConfigs,
+    ...openAICodexConfigs,
     ...anthropicConfigs,
     ...aiSDKConfigs,
     ...claudeAgentSDKConfigs,
+    ...cursorSDKConfigs,
     ...googleGenAIConfigs,
     ...huggingFaceConfigs,
     ...openRouterConfigs,
     ...openRouterAgentConfigs,
     ...mistralConfigs,
     ...cohereConfigs,
+    ...groqConfigs,
+    ...genkitConfigs,
+    ...gitHubCopilotConfigs,
     ...options.instrumentations || []
   ];
   const dcModule = options.browser === false ? void 0 : "dc-browser";
@@ -1841,6 +2519,9 @@ var unplugin = (0, import_unplugin.createUnplugin)((options = {}) => {
     name: "code-transformer",
     enforce: "pre",
     transform(code, id) {
+      if (!id) {
+        return null;
+      }
       const filePath = id.startsWith("file:") ? (0, import_url.fileURLToPath)(id) : id;
       const ext = (0, import_path.extname)(filePath);
       let isModule = ext === ".mjs" || ext === ".ts" || ext === ".tsx";
@@ -1872,8 +2553,12 @@ var unplugin = (0, import_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) {