braintrust 3.25.0 → 3.27.0

package/dist/auto-instrumentations/{chunk-JPVCUKTY.mjs → chunk-XEYKUBLY.mjs}

@@ -5,6 +5,9 @@ import {
   bedrockRuntimeConfigs,
   channel,
   claudeAgentSDKConfigs,
+  cloudflareAIChatConfigs,
+  cloudflareAgentsConfigs,
+  cloudflareThinkConfigs,
   cohereConfigs,
   cursorSDKConfigs,
   defineChannels,
@@ -17,21 +20,25 @@ import {
   langSmithConfigs,
   langchainConfigs,
   mistralConfigs,
+  ollamaConfigs,
   openAIAgentsCoreConfigs,
   openAICodexConfigs,
   openRouterAgentConfigs,
   openRouterConfigs,
   openaiConfigs,
   piCodingAgentConfigs
-} from "./chunk-CZ24KNHT.mjs";
+} from "./chunk-ZNHTSSGI.mjs";
 import {
   applySpecialCasePatch,
   create
-} from "./chunk-M6DLIJ2Z.mjs";
+} from "./chunk-I55G56ZL.mjs";
+import {
+  getPackageName
+} from "./chunk-P5YLNB2A.mjs";
 
 // src/auto-instrumentations/bundler/plugin.ts
 import { createUnplugin } from "unplugin";
-import { extname, join, sep } from "path";
+import { extname, isAbsolute, join, sep } from "path";
 import { readFileSync } from "fs";
 import { fileURLToPath } from "url";
 import moduleDetailsFromPath from "module-details-from-path";
@@ -53,6 +60,9 @@ var envIntegrationAliases = {
   strandsagentsdk: "strandsAgentSDK",
   "strands-agent-sdk": "strandsAgentSDK",
   "@strands-agents/sdk": "strandsAgentSDK",
+  agents: "cloudflareAgents",
+  "cloudflare-agents": "cloudflareAgents",
+  cloudflareagents: "cloudflareAgents",
   anthropic: "anthropic",
   aisdk: "aisdk",
   "ai-sdk": "aisdk",
@@ -60,6 +70,10 @@ var envIntegrationAliases = {
   vercel: "vercel",
   claudeagentsdk: "claudeAgentSDK",
   "claude-agent-sdk": "claudeAgentSDK",
+  cloudflareaichat: "cloudflareAIChat",
+  "cloudflare-ai-chat": "cloudflareAIChat",
+  "@cloudflare/ai-chat": "cloudflareAIChat",
+  cloudflarethink: "cloudflareThink",
   cursor: "cursor",
   "cursor-sdk": "cursorSDK",
   cursorsdk: "cursorSDK",
@@ -74,10 +88,13 @@ var envIntegrationAliases = {
   "google-genai": "googleGenAI",
   googlegenai: "googleGenAI",
   huggingface: "huggingface",
+  "@huggingface/transformers": "huggingface",
+  transformers: "huggingface",
   openrouter: "openrouter",
   openrouteragent: "openrouterAgent",
   "openrouter-agent": "openrouterAgent",
   mistral: "mistral",
+  ollama: "ollama",
   googleadk: "googleADK",
   "google-adk": "googleADK",
   cohere: "cohere",
@@ -146,6 +163,83 @@ var flueConfigs = [
   }
 ];
 
+// src/instrumentation/plugins/huggingface-transformers-channels.ts
+var huggingFaceTransformersChannels = defineChannels(
+  "@huggingface/transformers",
+  {
+    pipeline: channel({
+      channelName: "pipeline",
+      kind: "async"
+    }),
+    pipelineCall: channel({
+      channelName: "pipeline.call",
+      kind: "async"
+    })
+  },
+  { instrumentationName: INSTRUMENTATION_NAMES.HUGGINGFACE }
+);
+
+// src/auto-instrumentations/configs/huggingface-transformers.ts
+var moduleFiles = [
+  {
+    filePath: "dist/transformers.mjs",
+    versionRange: ">=3.0.0 <3.4.0"
+  },
+  {
+    filePath: "dist/transformers.cjs",
+    versionRange: ">=3.0.0 <3.4.0"
+  },
+  {
+    filePath: "dist/transformers.js",
+    versionRange: ">=3.0.0 <3.4.0"
+  },
+  {
+    filePath: "dist/transformers.node.mjs",
+    versionRange: ">=3.4.0 <5.0.0"
+  },
+  {
+    filePath: "dist/transformers.node.cjs",
+    versionRange: ">=3.4.0 <5.0.0"
+  },
+  {
+    filePath: "dist/transformers.web.js",
+    versionRange: ">=3.4.0 <5.0.0"
+  }
+];
+var pipelineClasses = [
+  "TextGenerationPipeline",
+  "Text2TextGenerationPipeline",
+  "FeatureExtractionPipeline",
+  "QuestionAnsweringPipeline"
+];
+var huggingFaceTransformersConfigs = moduleFiles.flatMap(({ filePath, versionRange }) => [
+  {
+    channelName: huggingFaceTransformersChannels.pipeline.channelName,
+    module: {
+      name: "@huggingface/transformers",
+      versionRange,
+      filePath
+    },
+    functionQuery: {
+      functionName: "pipeline",
+      kind: "Async"
+    }
+  },
+  ...pipelineClasses.map((className) => ({
+    channelName: huggingFaceTransformersChannels.pipelineCall.channelName,
+    module: {
+      name: "@huggingface/transformers",
+      versionRange,
+      filePath
+    },
+    functionQuery: {
+      className,
+      methodName: "_call",
+      kind: "Async"
+    }
+  }))
+]);
+
 // src/instrumentation/plugins/strands-agent-sdk-channels.ts
 var strandsAgentSDKChannels = defineChannels(
   "@strands-agents/sdk",
@@ -230,6 +324,18 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["claudeAgentSDK"],
     configs: claudeAgentSDKConfigs
   },
+  {
+    integrations: ["cloudflareAIChat"],
+    configs: cloudflareAIChatConfigs
+  },
+  {
+    integrations: ["cloudflareAgents"],
+    configs: cloudflareAgentsConfigs
+  },
+  {
+    integrations: ["cloudflareThink"],
+    configs: cloudflareThinkConfigs
+  },
   { integrations: ["cursor", "cursorSDK"], configs: cursorSDKConfigs },
   {
     integrations: ["openAIAgents"],
@@ -239,7 +345,10 @@ var defaultInstrumentationConfigGroups = [
     integrations: ["google", "googleGenAI"],
     configs: googleGenAIConfigs
   },
-  { integrations: ["huggingface"], configs: huggingFaceConfigs },
+  {
+    integrations: ["huggingface"],
+    configs: [...huggingFaceConfigs, ...huggingFaceTransformersConfigs]
+  },
   {
     integrations: ["langchain", "langgraph"],
     configs: langchainConfigs
@@ -251,6 +360,7 @@ var defaultInstrumentationConfigGroups = [
     configs: openRouterAgentConfigs
   },
   { integrations: ["mistral"], configs: mistralConfigs },
+  { integrations: ["ollama"], configs: ollamaConfigs },
   { integrations: ["googleADK"], configs: googleADKConfigs },
   { integrations: ["cohere"], configs: cohereConfigs },
   { integrations: ["groq"], configs: groqConfigs },
@@ -310,78 +420,86 @@ function getModuleVersion(basedir) {
   }
   return void 0;
 }
-var unplugin = createUnplugin(
-  (options = {}) => {
-    const allInstrumentations = getDefaultInstrumentationConfigs({
-      additionalInstrumentations: options.instrumentations
-    });
-    const dcModule = options.browser === false ? void 0 : "dc-browser";
-    const instrumentationMatcher = create(allInstrumentations, dcModule);
-    return {
-      name: "code-transformer",
-      enforce: "pre",
-      transform(code, id) {
-        if (!id) {
-          return null;
-        }
-        const filePath = id.startsWith("file:") ? fileURLToPath(id) : id;
-        const ext = extname(filePath);
-        let isModule = ext === ".mjs" || ext === ".ts" || ext === ".tsx";
-        if (ext === ".js") {
-          isModule = code.includes("export ") || code.includes("import ");
-        }
-        const normalizedForPlatform = filePath.split("/").join(sep);
-        const moduleDetails = moduleDetailsFromPath(normalizedForPlatform);
-        if (!moduleDetails) {
-          return null;
-        }
-        const moduleName = moduleDetails.name;
-        const normalizedModulePath = moduleDetails.path.replace(/\\/g, "/");
-        const moduleVersion = getModuleVersion(moduleDetails.basedir);
-        if (options.browser !== true) {
-          const patched = applySpecialCasePatch({
-            packageName: moduleName,
-            modulePath: normalizedModulePath,
-            source: code,
-            format: isModule ? "esm" : "cjs"
-          });
-          if (patched !== null) {
-            return { code: patched, map: null };
-          }
-        }
-        if (!moduleVersion) {
-          console.warn(
-            `No 'package.json' version found for module ${moduleName} at ${moduleDetails.basedir}. Skipping transformation.`
-          );
-          return null;
-        }
-        const transformer = instrumentationMatcher.getTransformer(
-          moduleName,
-          moduleVersion,
-          normalizedModulePath
-        );
-        if (!transformer) {
-          return null;
-        }
+var unplugin = createUnplugin((options = {}) => {
+  const browser = options.browser ?? options.useDiagnosticChannelCompatShim ?? false;
+  const allInstrumentations = getDefaultInstrumentationConfigs({
+    additionalInstrumentations: options.instrumentations
+  });
+  const instrumentationMatcher = create(allInstrumentations);
+  return {
+    name: "code-transformer",
+    enforce: "pre",
+    transformInclude(id) {
+      const pathWithoutQuery = id.split("?", 1)[0];
+      if (!pathWithoutQuery) {
+        return false;
+      }
+      if (pathWithoutQuery.startsWith("file:")) {
         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: transformedCode,
-            map: result.map
-          };
-        } catch (error) {
-          console.warn(`Code transformation failed for ${id}: ${error}`);
-          return null;
+          return isAbsolute(fileURLToPath(pathWithoutQuery));
+        } catch {
+          return false;
         }
       }
-    };
-  }
-);
+      return isAbsolute(pathWithoutQuery);
+    },
+    transform(code, id) {
+      if (!id) {
+        return null;
+      }
+      const cleanId = id.replace(/[?#].*$/, "");
+      const filePath = cleanId.startsWith("file:") ? fileURLToPath(cleanId) : cleanId;
+      const ext = extname(filePath);
+      let isModule = ext === ".mjs" || ext === ".ts" || ext === ".tsx";
+      if (ext === ".js") {
+        isModule = code.includes("export ") || code.includes("import ");
+      }
+      const normalizedForPlatform = filePath.split("/").join(sep);
+      const moduleDetails = moduleDetailsFromPath(normalizedForPlatform);
+      if (!moduleDetails) {
+        return null;
+      }
+      const moduleName = getPackageName(moduleDetails.basedir) ?? moduleDetails.name;
+      const normalizedModulePath = moduleDetails.path.replace(/\\/g, "/");
+      const moduleVersion = getModuleVersion(moduleDetails.basedir);
+      const patched = applySpecialCasePatch({
+        packageName: moduleName,
+        modulePath: normalizedModulePath,
+        source: code,
+        format: isModule ? "esm" : "cjs",
+        browser
+      });
+      if (patched !== null) {
+        return { code: patched, map: null };
+      }
+      if (!moduleVersion) {
+        console.warn(
+          `No 'package.json' version found for module ${moduleName} at ${moduleDetails.basedir}. Skipping transformation.`
+        );
+        return null;
+      }
+      const transformer = instrumentationMatcher.getTransformer(
+        moduleName,
+        moduleVersion,
+        normalizedModulePath
+      );
+      if (!transformer) {
+        return null;
+      }
+      try {
+        const moduleType = isModule ? "esm" : "cjs";
+        const result = transformer.transform(code, moduleType);
+        return {
+          code: result.code,
+          map: result.map
+        };
+      } catch (error) {
+        console.warn(`Code transformation failed for ${id}: ${error}`);
+        return null;
+      }
+    }
+  };
+});
 
 export {
   unplugin