dd-trace 5.90.0 → 5.92.0

package/packages/datadog-plugin-graphql/src/resolve.js

@@ -10,13 +10,13 @@ class GraphQLResolvePlugin extends TracingPlugin {
   static operation = 'resolve'
 
   start (fieldCtx) {
-    const { info, rootCtx, args } = fieldCtx
+    const { info, rootCtx, args, path: pathAsArray, pathString } = fieldCtx
 
-    const path = getPath(info, this.config)
+    const path = getPath(this.config, pathAsArray)
 
     // we need to get the parent span to the field if it exists for correct span parenting
     // of nested fields
-    const parentField = getParentField(rootCtx, pathToArray(info && info.path))
+    const parentField = getParentField(rootCtx, pathString)
     const childOf = parentField?.ctx?.currentStore?.span
 
     fieldCtx.parent = parentField
@@ -76,9 +76,9 @@ class GraphQLResolvePlugin extends TracingPlugin {
     super(...args)
 
     this.addTraceSub('updateField', (ctx) => {
-      const { field, info, error } = ctx
+      const { field, error, path: pathAsArray } = ctx
 
-      const path = getPath(info, this.config)
+      const path = getPath(this.config, pathAsArray)
 
       if (!shouldInstrument(this.config, path)) return
 
@@ -118,28 +118,14 @@ function shouldInstrument (config, path) {
   return config.depth < 0 || config.depth >= depth
 }
 
-function getPath (info, config) {
-  const responsePathAsArray = config.collapse
-    ? withCollapse(pathToArray)
-    : pathToArray
-  return responsePathAsArray(info && info.path)
+function getPath (config, pathAsArray) {
+  return config.collapse
+    ? withCollapse(pathAsArray)
+    : pathAsArray
 }
 
-function pathToArray (path) {
-  const flattened = []
-  let curr = path
-  while (curr) {
-    flattened.push(curr.key)
-    curr = curr.prev
-  }
-  return flattened.reverse()
-}
-
-function withCollapse (responsePathAsArray) {
-  return function () {
-    return responsePathAsArray.apply(this, arguments)
-      .map(segment => typeof segment === 'number' ? '*' : segment)
-  }
+function withCollapse (pathAsArray) {
+  return pathAsArray.map(segment => typeof segment === 'number' ? '*' : segment)
 }
 
 function getResolverInfo (info, args) {
@@ -173,19 +159,20 @@ function getResolverInfo (info, args) {
   return resolverInfo
 }
 
-function getParentField (parentCtx, path) {
-  for (let i = path.length - 1; i > 0; i--) {
-    const field = getField(parentCtx, path.slice(0, i))
-    if (field) {
-      return field
-    }
+function getParentField (parentCtx, pathToString) {
+  let current = pathToString
+
+  while (current) {
+    const lastJoin = current.lastIndexOf('.')
+    if (lastJoin === -1) break
+
+    current = current.slice(0, lastJoin)
+    const field = parentCtx.fields[current]
+
+    if (field) return field
   }
 
   return null
 }
 
-function getField (parentCtx, path) {
-  return parentCtx.fields[path.join('.')]
-}
-
 module.exports = GraphQLResolvePlugin

package/packages/datadog-plugin-http/src/client.js

@@ -211,7 +211,7 @@ function getHeaders (config) {
     if (typeof header === 'string') {
       const separatorIndex = header.indexOf(':')
       result.push(separatorIndex === -1
-        ? [header, undefined]
+        ? [header.toLowerCase(), undefined]
         : [
             header.slice(0, separatorIndex).toLowerCase(),
             header.slice(separatorIndex + 1),

package/packages/datadog-plugin-langgraph/src/index.js

@@ -0,0 +1,24 @@
+'use strict'
+
+const CompositePlugin = require('../../dd-trace/src/plugins/composite')
+const langgraphLLMObsPlugins = require('../../dd-trace/src/llmobs/plugins/langgraph')
+const streamPlugin = require('./stream')
+
+const plugins = {}
+
+// CRITICAL: LLMObs plugins MUST come first
+for (const Plugin of langgraphLLMObsPlugins) {
+  plugins[Plugin.id] = Plugin
+}
+
+// Tracing plugins second
+for (const Plugin of streamPlugin) {
+  plugins[Plugin.id] = Plugin
+}
+
+class LanggraphPlugin extends CompositePlugin {
+  static id = 'langgraph'
+  static plugins = plugins
+}
+
+module.exports = LanggraphPlugin

package/packages/datadog-plugin-langgraph/src/stream.js

@@ -0,0 +1,41 @@
+'use strict'
+
+const TracingPlugin = require('../../dd-trace/src/plugins/tracing')
+const { spanHasError } = require('../../dd-trace/src/llmobs/util')
+
+// We are only tracing Pregel.stream because Pregel.invoke calls stream internally resulting in
+// a graph with spans that look redundant.
+class PregelStreamPlugin extends TracingPlugin {
+  static id = 'langgraph_pregel_stream'
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_stream'
+
+  bindStart (ctx) {
+    this.startSpan('LangGraph', {
+      service: this.config.service,
+      kind: 'internal',
+      component: 'langgraph',
+    }, ctx)
+    return ctx.currentStore
+  }
+}
+class NextStreamPlugin extends TracingPlugin {
+  static id = 'langgraph_stream_next'
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_stream_next'
+
+  bindStart (ctx) {
+    return ctx.currentStore
+  }
+
+  asyncEnd (ctx) {
+    const span = ctx.currentStore?.span
+    if (!span) return
+    if (ctx.result.done === true || spanHasError(span)) {
+      span.finish()
+    }
+  }
+}
+
+module.exports = [
+  PregelStreamPlugin,
+  NextStreamPlugin,
+]

package/packages/dd-trace/src/config/defaults.js

@@ -2,6 +2,7 @@
 
 const pkg = require('../pkg')
 const { isFalse, isTrue } = require('../util')
+const { DD_MAJOR } = require('../../../../version')
 const { getEnvironmentVariable: getEnv } = require('./helper')
 
 const {
@@ -106,6 +107,7 @@ const defaultsWithoutSupportedConfigurationEntry = {
   isGCPFunction: false,
   instrumentationSource: 'manual',
   isServiceUserProvided: false,
+  isServiceNameInferred: true,
   lookup: undefined,
   plugins: true,
 }
@@ -116,6 +118,7 @@ const defaultsWithoutSupportedConfigurationEntry = {
 // TODO: These entries should be removed. They are off by default
 // because they rely on other configs.
 const defaultsWithConditionalRuntimeBehavior = {
+  startupLogs: DD_MAJOR >= 6,
   isGitUploadEnabled: false,
   isImpactedTestsEnabled: false,
   isIntelligentTestRunnerEnabled: false,

package/packages/dd-trace/src/config/index.js

@@ -49,6 +49,8 @@ const VALID_PROPAGATION_BEHAVIOR_EXTRACT = new Set(['continue', 'restart', 'igno
 const VALID_LOG_LEVELS = new Set(['debug', 'info', 'warn', 'error'])
 const DEFAULT_OTLP_PORT = 4318
 const RUNTIME_ID = uuid()
+// eslint-disable-next-line eslint-rules/eslint-process-env -- internal propagation, not user config
+const ROOT_SESSION_ID = process.env.DD_ROOT_JS_SESSION_ID || RUNTIME_ID
 const NAMING_VERSIONS = new Set(['v0', 'v1'])
 const DEFAULT_NAMING_VERSION = 'v0'
 
@@ -145,6 +147,8 @@ class Config {
       'runtime-id': RUNTIME_ID,
     })
 
+    this.rootSessionId = ROOT_SESSION_ID
+
     if (this.isCiVisibility) {
       tagger.add(this.tags, {
         [ORIGIN_KEY]: 'ciapp-test',
@@ -722,8 +726,10 @@ class Config {
     // Priority:
     // DD_SERVICE > tags.service > OTEL_SERVICE_NAME > NX_TASK_TARGET_PROJECT (if DD_ENABLE_NX_SERVICE_NAME) > default
     let serviceName = DD_SERVICE || tags.service || OTEL_SERVICE_NAME
+    let isServiceNameInferred
     if (!serviceName && NX_TASK_TARGET_PROJECT) {
       if (isTrue(DD_ENABLE_NX_SERVICE_NAME)) {
+        isServiceNameInferred = true
         serviceName = NX_TASK_TARGET_PROJECT
       } else if (DD_MAJOR < 6) {
         // Warn about v6 behavior change for Nx projects
@@ -734,6 +740,7 @@ class Config {
       }
     }
     setString(target, 'service', serviceName)
+    if (serviceName) setBoolean(target, 'isServiceNameInferred', isServiceNameInferred ?? false)
     if (DD_SERVICE_MAPPING) {
       target.serviceMapping = Object.fromEntries(
         DD_SERVICE_MAPPING.split(',').map(x => x.trim().split(':'))
@@ -1004,7 +1011,11 @@ class Config {
     setUnit(opts, 'sampleRate', options.sampleRate ?? options.ingestion.sampleRate)
     opts['sampler.rateLimit'] = maybeInt(options.rateLimit ?? options.ingestion.rateLimit)
     setSamplingRule(opts, 'sampler.rules', options.samplingRules)
-    setString(opts, 'service', options.service || tags.service)
+    const optService = options.service || tags.service
+    setString(opts, 'service', optService)
+    if (optService) {
+      setBoolean(opts, 'isServiceNameInferred', false)
+    }
     opts.serviceMapping = options.serviceMapping
     setString(opts, 'site', options.site)
     if (options.spanAttributeSchema) {
@@ -1118,6 +1129,8 @@ class Config {
       setBoolean(calc, 'reportHostname', true)
       // Clear sampling rules - server-side sampling handles this
       calc['sampler.rules'] = []
+      // Agentless intake only accepts 64-bit trace IDs; disable 128-bit generation
+      setBoolean(calc, 'traceId128BitGenerationEnabled', false)
     }
 
     if (this.#isCiVisibility()) {

package/packages/dd-trace/src/config/supported-configurations.json

@@ -2995,6 +2995,13 @@
         "default": "true"
       }
     ],
+    "DD_TRACE_LANGGRAPH_ENABLED": [
+      {
+        "implementation": "C",
+        "type": "boolean",
+        "default": "true"
+      }
+    ],
     "DD_TRACE_LDAPJS_ENABLED": [
       {
         "implementation": "A",

package/packages/dd-trace/src/constants.js

@@ -23,6 +23,7 @@ module.exports = {
   SPAN_SAMPLING_MAX_PER_SECOND: '_dd.span_sampling.max_per_second',
   DATADOG_LAMBDA_EXTENSION_PATH: '/opt/extensions/datadog-agent',
   DECISION_MAKER_KEY: '_dd.p.dm',
+  SAMPLING_KNUTH_RATE: '_dd.p.ksr',
   PROCESS_ID: 'process_id',
   ERROR_TYPE: 'error.type',
   ERROR_MESSAGE: 'error.message',

package/packages/dd-trace/src/crashtracking/crashtracker.js

@@ -74,7 +74,7 @@ class Crashtracker {
         timeout_ms: 3000,
       },
       timeout: { secs: 5, nanos: 0 },
-      demangle_names: false,
+      demangle_names: true,
       signals: [],
       resolve_frames: resolveMode,
     }

package/packages/dd-trace/src/debugger/devtools_client/config.js

@@ -2,8 +2,11 @@
 
 const { workerData: { config: parentConfig, parentThreadId, configPort } } = require('node:worker_threads')
 const { getAgentUrl } = require('../../agent/url')
+const processTags = require('../../process-tags')
 const log = require('./log')
 
+processTags.initialize()
+
 const config = module.exports = {
   ...parentConfig,
   parentThreadId,

package/packages/dd-trace/src/dogstatsd.js

@@ -26,6 +26,7 @@ class DogStatsDClient {
   constructor (options = {}) {
     if (options.metricsProxyUrl) {
       this._httpOptions = {
+        method: 'POST',
         url: options.metricsProxyUrl.toString(),
         path: '/dogstatsd/v2/proxy',
       }

package/packages/dd-trace/src/encode/agentless-json.js

@@ -16,6 +16,9 @@ const SOFT_LIMIT = 8 * 1024 * 1024 // 8MB
 function formatSpan (span, isFirstSpan) {
   span = normalizeSpan(truncateSpan(span, false))
 
+  // Remove _dd.p.tid (the upper 64 bits of a 128-bit trace ID) since trace_id is truncated to lower 64 bits
+  delete span.meta['_dd.p.tid']
+
   if (span.span_events) {
     span.meta.events = JSON.stringify(span.span_events)
     delete span.span_events
@@ -45,7 +48,7 @@ function formatSpan (span, isFirstSpan) {
  */
 function spanToJSON (span) {
   const result = {
-    trace_id: span.trace_id.toString(16).toLowerCase(),
+    trace_id: span.trace_id.toString(16).toLowerCase().slice(-16),
     span_id: span.span_id.toString(16).toLowerCase(),
     parent_id: span.parent_id.toString(16).toLowerCase(),
     name: span.name,

package/packages/dd-trace/src/llmobs/plugins/langchain/handlers/chain.js

@@ -15,6 +15,17 @@ class LangChainLLMObsChainHandler extends LangChainLLMObsHandler {
     // chain spans will always be workflows
     this._tagger.tagTextIO(span, input, output)
   }
+
+  getName ({ span, instance }) {
+    const firstCallable = instance?.first
+
+    if (firstCallable?.constructor?.name === 'ChannelWrite') return
+
+    const firstCallableIsLangGraph = firstCallable?.lc_namespace?.includes('langgraph')
+    const firstCallableName = firstCallable?.name
+
+    return firstCallableIsLangGraph ? firstCallableName : super.getName({ span })
+  }
 }
 
 module.exports = LangChainLLMObsChainHandler

package/packages/dd-trace/src/llmobs/plugins/langchain/index.js

@@ -54,6 +54,8 @@ class BaseLangChainLLMObsPlugin extends LLMObsPlugin {
     const handler = this._handlers[ctx.type]
     const name = handler?.getName({ span, instance })
 
+    if (name == null) return
+
     return {
       modelProvider,
       modelName,

package/packages/dd-trace/src/llmobs/plugins/langgraph/index.js

@@ -0,0 +1,114 @@
+'use strict'
+
+const LLMObsPlugin = require('../base')
+const { spanHasError } = require('../../util')
+
+const streamDataMap = new WeakMap()
+
+function formatIO (data) {
+  if (data == null) return ''
+
+  if (typeof data === 'string' || typeof data === 'number' || typeof data === 'boolean') {
+    return data
+  }
+
+  if (data.constructor?.name === 'Object') {
+    const formatted = {}
+    for (const [key, value] of Object.entries(data)) {
+      formatted[key] = formatIO(value)
+    }
+    return formatted
+  }
+
+  if (Array.isArray(data)) {
+    return data.map(item => formatIO(item))
+  }
+
+  try {
+    return JSON.stringify(data)
+  } catch {
+    return String(data)
+  }
+}
+
+class PregelStreamLLMObsPlugin extends LLMObsPlugin {
+  static id = 'llmobs_langgraph_pregel_stream'
+  static integration = 'langgraph'
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_stream'
+
+  getLLMObsSpanRegisterOptions (ctx) {
+    const name = ctx.self.name || 'LangGraph'
+
+    const enabled = this._tracerConfig.llmobs.enabled
+    if (!enabled) return
+
+    const span = ctx.currentStore?.span
+    if (!span) return
+    streamDataMap.set(span, {
+      streamInputs: ctx.arguments?.[0],
+      chunks: [],
+    })
+
+    return {
+      kind: 'workflow',
+      name,
+    }
+  }
+
+  asyncEnd () {}
+}
+
+class NextStreamLLMObsPlugin extends LLMObsPlugin {
+  static id = 'llmobs_langgraph_next_stream'
+  static prefix = 'tracing:orchestrion:@langchain/langgraph:Pregel_stream_next'
+
+  start () {} // no-op: span was already registered by PregelStreamLLMObsPlugin
+
+  end () {} // no-op: context restore is handled by PregelStreamLLMObsPlugin
+
+  error (ctx) {
+    const span = ctx.currentStore?.span
+    if (!span) return
+
+    this.#tagAndCleanup(span, true)
+  }
+
+  setLLMObsTags (ctx) {
+    const span = ctx.currentStore?.span
+    if (!span) return
+
+    // Accumulate chunks until done
+    if (ctx.result?.value && !ctx.result.done) {
+      const streamData = streamDataMap.get(span)
+      if (streamData) {
+        streamData.chunks.push(ctx.result.value)
+      }
+      return
+    }
+
+    // Tag on last chunk
+    if (ctx.result?.done) {
+      const hasError = ctx.error || spanHasError(span)
+      this.#tagAndCleanup(span, hasError)
+    }
+  }
+
+  #tagAndCleanup (span, hasError) {
+    const streamData = streamDataMap.get(span)
+    if (!streamData) return
+
+    const { streamInputs: inputs, chunks } = streamData
+    const input = inputs == null ? undefined : formatIO(inputs)
+    const lastChunk = chunks.length > 0 ? chunks[chunks.length - 1] : undefined
+    const output = !hasError && lastChunk != null ? formatIO(lastChunk) : undefined
+
+    this._tagger.tagTextIO(span, input, output)
+
+    streamDataMap.delete(span)
+  }
+}
+
+module.exports = [
+  PregelStreamLLMObsPlugin,
+  NextStreamLLMObsPlugin,
+]

package/packages/dd-trace/src/plugins/ci_plugin.js

@@ -125,7 +125,7 @@ module.exports = class CiPlugin extends Plugin {
     this._pendingRequestErrorTags = []
 
     this.addSub(`ci:${this.constructor.id}:library-configuration`, (ctx) => {
-      const { onDone, isParallel, frameworkVersion } = ctx
+      const { onDone, frameworkVersion } = ctx
       ctx.currentStore = storage('legacy').getStore()
 
       if (!this.tracer._exporter || !this.tracer._exporter.getLibraryConfiguration) {
@@ -143,7 +143,7 @@ module.exports = class CiPlugin extends Plugin {
           ? getSessionRequestErrorTags(this.testSessionSpan)
           : Object.fromEntries(this._pendingRequestErrorTags.map(({ tag, value }) => [tag, value]))
 
-        const libraryCapabilitiesTags = getLibraryCapabilitiesTags(this.constructor.id, isParallel, frameworkVersion)
+        const libraryCapabilitiesTags = getLibraryCapabilitiesTags(this.constructor.id, frameworkVersion)
         const metadataTags = {
           test: {
             ...libraryCapabilitiesTags,

package/packages/dd-trace/src/plugins/index.js

@@ -31,6 +31,7 @@ const plugins = {
   get '@redis/client' () { return require('../../../datadog-plugin-redis/src') },
   get '@smithy/smithy-client' () { return require('../../../datadog-plugin-aws-sdk/src') },
   get '@vitest/runner' () { return require('../../../datadog-plugin-vitest/src') },
+  get '@langchain/langgraph' () { return require('../../../datadog-plugin-langgraph/src') },
   get aerospike () { return require('../../../datadog-plugin-aerospike/src') },
   get ai () { return require('../../../datadog-plugin-ai/src') },
   get amqp10 () { return require('../../../datadog-plugin-amqp10/src') },

package/packages/dd-trace/src/plugins/util/test.js

@@ -150,7 +150,6 @@ const DD_CAPABILITIES_FAILED_TEST_REPLAY = '_dd.library_capabilities.failed_test
 const DD_CI_LIBRARY_CONFIGURATION_ERROR = '_dd.ci.library_configuration_error'
 
 const UNSUPPORTED_TIA_FRAMEWORKS = new Set(['playwright', 'vitest'])
-const UNSUPPORTED_TIA_FRAMEWORKS_PARALLEL_MODE = new Set(['cucumber', 'mocha'])
 const MINIMUM_FRAMEWORK_VERSION_FOR_EFD = {
   playwright: '>=1.38.0',
 }
@@ -170,7 +169,6 @@ const MINIMUM_FRAMEWORK_VERSION_FOR_FAILED_TEST_REPLAY = {
   playwright: '>=1.38.0',
 }
 
-const UNSUPPORTED_ATTEMPT_TO_FIX_FRAMEWORKS_PARALLEL_MODE = new Set(['mocha'])
 const NOT_SUPPORTED_GRANULARITY_IMPACTED_TESTS_FRAMEWORKS = new Set(['mocha', 'playwright', 'vitest'])
 
 const TEST_LEVEL_EVENT_TYPES = [
@@ -987,9 +985,8 @@ function getFormattedError (error, repositoryRoot) {
   return newError
 }
 
-function isTiaSupported (testFramework, isParallel) {
-  return !(UNSUPPORTED_TIA_FRAMEWORKS.has(testFramework) ||
-           (isParallel && UNSUPPORTED_TIA_FRAMEWORKS_PARALLEL_MODE.has(testFramework)))
+function isTiaSupported (testFramework) {
+  return !UNSUPPORTED_TIA_FRAMEWORKS.has(testFramework)
 }
 
 function isEarlyFlakeDetectionSupported (testFramework, frameworkVersion) {
@@ -1016,12 +1013,12 @@ function isDisableSupported (testFramework, frameworkVersion) {
     : true
 }
 
-function isAttemptToFixSupported (testFramework, isParallel, frameworkVersion) {
+function isAttemptToFixSupported (testFramework, frameworkVersion) {
   if (testFramework === 'playwright') {
     return satisfies(frameworkVersion, MINIMUM_FRAMEWORK_VERSION_FOR_ATTEMPT_TO_FIX[testFramework])
   }
 
-  return !(isParallel && UNSUPPORTED_ATTEMPT_TO_FIX_FRAMEWORKS_PARALLEL_MODE.has(testFramework))
+  return true
 }
 
 function isFailedTestReplaySupported (testFramework, frameworkVersion) {
@@ -1030,9 +1027,9 @@ function isFailedTestReplaySupported (testFramework, frameworkVersion) {
     : true
 }
 
-function getLibraryCapabilitiesTags (testFramework, isParallel, frameworkVersion) {
+function getLibraryCapabilitiesTags (testFramework, frameworkVersion) {
   return {
-    [DD_CAPABILITIES_TEST_IMPACT_ANALYSIS]: isTiaSupported(testFramework, isParallel)
+    [DD_CAPABILITIES_TEST_IMPACT_ANALYSIS]: isTiaSupported(testFramework)
       ? '1'
       : undefined,
     [DD_CAPABILITIES_EARLY_FLAKE_DETECTION]: isEarlyFlakeDetectionSupported(testFramework, frameworkVersion)
@@ -1049,7 +1046,7 @@ function getLibraryCapabilitiesTags (testFramework, isParallel, frameworkVersion
       ? '1'
       : undefined,
     [DD_CAPABILITIES_TEST_MANAGEMENT_ATTEMPT_TO_FIX]:
-      isAttemptToFixSupported(testFramework, isParallel, frameworkVersion)
+      isAttemptToFixSupported(testFramework, frameworkVersion)
         ? '5'
         : undefined,
     [DD_CAPABILITIES_FAILED_TEST_REPLAY]: isFailedTestReplaySupported(testFramework, frameworkVersion)

package/packages/dd-trace/src/priority_sampler.js

@@ -31,10 +31,21 @@ const {
   SAMPLING_LIMIT_DECISION,
   SAMPLING_AGENT_DECISION,
   DECISION_MAKER_KEY,
+  SAMPLING_KNUTH_RATE,
 } = require('./constants')
 
 const DEFAULT_KEY = 'service:,env:'
 
+/**
+ * Formats a sampling rate as a string with up to 6 significant digits and no trailing zeros.
+ *
+ * @param {number} rate
+ * @returns {string}
+ */
+function formatKnuthRate (rate) {
+  return Number(rate.toPrecision(6)).toString()
+}
+
 const defaultSampler = new Sampler(AUTO_KEEP)
 
 /**
@@ -254,6 +265,7 @@ class PrioritySampler {
    */
   #getPriorityByRule (context, rule) {
     context._trace[SAMPLING_RULE_DECISION] = rule.sampleRate
+    context._trace.tags[SAMPLING_KNUTH_RATE] = formatKnuthRate(rule.sampleRate)
     context._sampling.mechanism = SAMPLING_MECHANISM_RULE
     if (rule.provenance === 'customer') context._sampling.mechanism = SAMPLING_MECHANISM_REMOTE_USER
     if (rule.provenance === 'dynamic') context._sampling.mechanism = SAMPLING_MECHANISM_REMOTE_DYNAMIC
@@ -290,9 +302,15 @@ class PrioritySampler {
     // TODO: Change underscored properties to private ones.
     const sampler = this._samplers[key] || this._samplers[DEFAULT_KEY]
 
-    context._trace[SAMPLING_AGENT_DECISION] = sampler.rate()
+    const rate = sampler.rate()
+    context._trace[SAMPLING_AGENT_DECISION] = rate
 
-    context._sampling.mechanism = sampler === defaultSampler ? SAMPLING_MECHANISM_DEFAULT : SAMPLING_MECHANISM_AGENT
+    if (sampler === defaultSampler) {
+      context._sampling.mechanism = SAMPLING_MECHANISM_DEFAULT
+    } else {
+      context._trace.tags[SAMPLING_KNUTH_RATE] = formatKnuthRate(rate)
+      context._sampling.mechanism = SAMPLING_MECHANISM_AGENT
+    }
 
     return sampler.isSampled(context) ? AUTO_KEEP : AUTO_REJECT
   }