dd-trace 5.46.0 → 5.48.0

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

@@ -30,7 +30,8 @@ const {
   TEST_MANAGEMENT_IS_DISABLED,
   TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
   TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
-  TEST_HAS_FAILED_ALL_RETRIES
+  TEST_HAS_FAILED_ALL_RETRIES,
+  TEST_RETRY_REASON_TYPES
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const id = require('../../dd-trace/src/id')
@@ -344,7 +345,8 @@ class JestPlugin extends CiPlugin {
       status,
       testStartLine,
       attemptToFixPassed,
-      failedAllTests
+      failedAllTests,
+      isAtrRetry
     }) => {
       const span = storage('legacy').getStore().span
       span.setTag(TEST_STATUS, status)
@@ -357,6 +359,10 @@ class JestPlugin extends CiPlugin {
       if (failedAllTests) {
         span.setTag(TEST_HAS_FAILED_ALL_RETRIES, 'true')
       }
+      if (isAtrRetry) {
+        span.setTag(TEST_IS_RETRY, 'true')
+        span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atr)
+      }
 
       const spanTags = span.context()._tags
       this.telemetry.ciVisEvent(
@@ -447,7 +453,7 @@ class JestPlugin extends CiPlugin {
 
     if (isAttemptToFixRetry) {
       extraTags[TEST_IS_RETRY] = 'true'
-      extraTags[TEST_RETRY_REASON] = 'attempt_to_fix'
+      extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.atf
     }
 
     if (isDisabled) {
@@ -462,12 +468,13 @@ class JestPlugin extends CiPlugin {
       extraTags[TEST_IS_NEW] = 'true'
       if (isEfdRetry) {
         extraTags[TEST_IS_RETRY] = 'true'
-        extraTags[TEST_RETRY_REASON] = 'efd'
+        extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
       }
     }
 
     if (isJestRetry) {
       extraTags[TEST_IS_RETRY] = 'true'
+      extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.ext
     }
 
     return super.startTestSpan(name, suite, this.testSuiteSpan, extraTags)

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

@@ -1,21 +1,27 @@
 'use strict'
 
-const LangChainTracingPlugin = require('./tracing')
-const LangChainLLMObsPlugin = require('../../dd-trace/src/llmobs/plugins/langchain')
+const langChainTracingPlugins = require('./tracing')
+const langChainLLMObsPlugins = require('../../dd-trace/src/llmobs/plugins/langchain')
+// const LangChainLLMObsPlugin = require('../../dd-trace/src/llmobs/plugins/langchain')
 const CompositePlugin = require('../../dd-trace/src/plugins/composite')
 
+const plugins = {}
+
+// ordering here is important - the llm observability plugin must come first
+// so that we can add annotations associated with the span before it finishes.
+// however, because the tracing plugin uses `bindStart` vs the llmobs' `start`,
+// the span is guaranteed to be created in the tracing plugin before the llmobs one is called
+for (const Plugin of langChainLLMObsPlugins) {
+  plugins[Plugin.id] = Plugin
+}
+
+for (const Plugin of langChainTracingPlugins) {
+  plugins[Plugin.id] = Plugin
+}
+
 class LangChainPlugin extends CompositePlugin {
   static get id () { return 'langchain' }
-  static get plugins () {
-    return {
-      // ordering here is important - the llm observability plugin must come first
-      // so that we can add annotations associated with the span before it finishes.
-      // however, because the tracing plugin uses `bindStart` vs the llmobs' `start`,
-      // the span is guaranteed to be created in the tracing plugin before the llmobs one is called
-      llmobs: LangChainLLMObsPlugin,
-      tracing: LangChainTracingPlugin
-    }
-  }
+  static get plugins () { return plugins }
 }
 
 module.exports = LangChainPlugin

package/packages/datadog-plugin-langchain/src/tracing.js

@@ -15,13 +15,10 @@ const LangChainLLMHandler = require('./handlers/language_models/llm')
 const LangChainChainHandler = require('./handlers/chain')
 const LangChainEmbeddingHandler = require('./handlers/embedding')
 
-class LangChainTracingPlugin extends TracingPlugin {
+class BaseLangChainTracingPlugin extends TracingPlugin {
   static get id () { return 'langchain' }
   static get operation () { return 'invoke' }
   static get system () { return 'langchain' }
-  static get prefix () {
-    return 'tracing:apm:langchain:invoke'
-  }
 
   constructor () {
     super(...arguments)
@@ -36,7 +33,15 @@ class LangChainTracingPlugin extends TracingPlugin {
   }
 
   bindStart (ctx) {
-    const { resource, type } = ctx
+    // TODO(bengl): All this renaming is just so we don't have to change the existing handlers
+    ctx.args = ctx.arguments
+    ctx.instance = ctx.self
+    const type = ctx.type = this.constructor.lcType
+
+    // Runnable interfaces have an `lc_namespace` property
+    const ns = ctx.self.lc_namespace || ctx.namespace
+    const resource = ctx.resource = [...ns, ctx.self.constructor.name].join('.')
+
     const handler = this.handlers[type]
 
     const instance = ctx.instance
@@ -85,4 +90,59 @@ class LangChainTracingPlugin extends TracingPlugin {
   }
 }
 
-module.exports = LangChainTracingPlugin
+class RunnableSequenceInvokePlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_rs_invoke' }
+  static get lcType () { return 'chain' }
+  static get prefix () {
+    return 'tracing:orchestrion:@langchain/core:RunnableSequence_invoke'
+  }
+}
+
+class RunnableSequenceBatchPlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_rs_batch' }
+  static get lcType () { return 'chain' }
+  static get prefix () {
+    return 'tracing:orchestrion:@langchain/core:RunnableSequence_batch'
+  }
+}
+
+class BaseChatModelGeneratePlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_chat_model_generate' }
+  static get lcType () { return 'chat_model' }
+  static get prefix () {
+    return 'tracing:orchestrion:@langchain/core:BaseChatModel_generate'
+  }
+}
+
+class BaseLLMGeneratePlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_llm_generate' }
+  static get lcType () { return 'llm' }
+  static get prefix () {
+    return 'tracing:orchestrion:@langchain/core:BaseLLM_generate'
+  }
+}
+
+class EmbeddingsEmbedQueryPlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_embeddings_embed_query' }
+  static get lcType () { return 'embedding' }
+  static get prefix () {
+    return 'tracing:apm:@langchain/core:Embeddings_embedQuery'
+  }
+}
+
+class EmbeddingsEmbedDocumentsPlugin extends BaseLangChainTracingPlugin {
+  static get id () { return 'langchain_embeddings_embed_documents' }
+  static get lcType () { return 'embedding' }
+  static get prefix () {
+    return 'tracing:apm:@langchain/core:Embeddings_embedDocuments'
+  }
+}
+
+module.exports = [
+  RunnableSequenceInvokePlugin,
+  RunnableSequenceBatchPlugin,
+  BaseChatModelGeneratePlugin,
+  BaseLLMGeneratePlugin,
+  EmbeddingsEmbedQueryPlugin,
+  EmbeddingsEmbedDocumentsPlugin
+]

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

@@ -36,7 +36,8 @@ const {
   TEST_MANAGEMENT_IS_DISABLED,
   TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
   TEST_HAS_FAILED_ALL_RETRIES,
-  TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED
+  TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
+  TEST_RETRY_REASON_TYPES
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const {
@@ -207,7 +208,8 @@ class MochaPlugin extends CiPlugin {
       isLastRetry,
       hasFailedAllRetries,
       attemptToFixPassed,
-      isAttemptToFixRetry
+      isAttemptToFixRetry,
+      isAtrRetry
     }) => {
       const store = storage('legacy').getStore()
       const span = store?.span
@@ -216,6 +218,11 @@ class MochaPlugin extends CiPlugin {
         span.setTag(TEST_STATUS, status)
         if (hasBeenRetried) {
           span.setTag(TEST_IS_RETRY, 'true')
+          if (isAtrRetry) {
+            span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atr)
+          } else {
+            span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.ext)
+          }
         }
         if (hasFailedAllRetries) {
           span.setTag(TEST_HAS_FAILED_ALL_RETRIES, 'true')
@@ -225,7 +232,7 @@ class MochaPlugin extends CiPlugin {
         }
         if (isAttemptToFixRetry) {
           span.setTag(TEST_IS_RETRY, 'true')
-          span.setTag(TEST_RETRY_REASON, 'attempt_to_fix')
+          span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atf)
         }
 
         const spanTags = span.context()._tags
@@ -273,13 +280,18 @@ class MochaPlugin extends CiPlugin {
       }
     })
 
-    this.addSub('ci:mocha:test:retry', ({ isFirstAttempt, willBeRetried, err, test }) => {
+    this.addSub('ci:mocha:test:retry', ({ isFirstAttempt, willBeRetried, err, test, isAtrRetry }) => {
       const store = storage('legacy').getStore()
       const span = store?.span
       if (span) {
         span.setTag(TEST_STATUS, 'fail')
         if (!isFirstAttempt) {
           span.setTag(TEST_IS_RETRY, 'true')
+          if (isAtrRetry) {
+            span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atr)
+          } else {
+            span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.ext)
+          }
         }
         if (err) {
           span.setTag('error', err)
@@ -473,7 +485,7 @@ class MochaPlugin extends CiPlugin {
       extraTags[TEST_IS_NEW] = 'true'
       if (isEfdRetry) {
         extraTags[TEST_IS_RETRY] = 'true'
-        extraTags[TEST_RETRY_REASON] = 'efd'
+        extraTags[TEST_RETRY_REASON] = 'early_flake_detection'
       }
     }
 

package/packages/datadog-plugin-mongodb-core/src/index.js

@@ -1,5 +1,6 @@
 'use strict'
 
+const { isTrue } = require('../../dd-trace/src/util')
 const DatabasePlugin = require('../../dd-trace/src/plugins/database')
 const coalesce = require('koalas')
 
@@ -12,9 +13,12 @@ class MongodbCorePlugin extends DatabasePlugin {
 
   configure (config) {
     super.configure(config)
+
+    const heartbeatFromEnv = process.env.DD_TRACE_MONGODB_HEARTBEAT_ENABLED
+
     this.config.heartbeatEnabled = coalesce(
       config.heartbeatEnabled,
-      process.env.DD_TRACE_MONGODB_HEARTBEAT_ENABLED,
+      heartbeatFromEnv && isTrue(heartbeatFromEnv),
       true
     )
   }

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

@@ -33,7 +33,8 @@ const {
   TEST_SUITE_ID,
   TEST_NAME,
   TEST_IS_RUM_ACTIVE,
-  TEST_BROWSER_VERSION
+  TEST_BROWSER_VERSION,
+  TEST_RETRY_REASON_TYPES
 } = require('../../dd-trace/src/plugins/util/test')
 const { RESOURCE_NAME } = require('../../../ext/tags')
 const { COMPONENT } = require('../../dd-trace/src/constants')
@@ -266,6 +267,7 @@ class PlaywrightPlugin extends CiPlugin {
       isAttemptToFixRetry,
       hasFailedAllRetries,
       hasPassedAttemptToFixRetries,
+      isAtrRetry,
       onDone
     }) => {
       const store = storage('legacy').getStore()
@@ -286,11 +288,16 @@ class PlaywrightPlugin extends CiPlugin {
         span.setTag(TEST_IS_NEW, 'true')
         if (isEfdRetry) {
           span.setTag(TEST_IS_RETRY, 'true')
-          span.setTag(TEST_RETRY_REASON, 'efd')
+          span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.efd)
         }
       }
       if (isRetry) {
         span.setTag(TEST_IS_RETRY, 'true')
+        if (isAtrRetry) {
+          span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atr)
+        } else {
+          span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.ext)
+        }
       }
       if (hasFailedAllRetries) {
         span.setTag(TEST_HAS_FAILED_ALL_RETRIES, 'true')
@@ -300,7 +307,7 @@ class PlaywrightPlugin extends CiPlugin {
       }
       if (isAttemptToFixRetry) {
         span.setTag(TEST_IS_RETRY, 'true')
-        span.setTag(TEST_RETRY_REASON, 'attempt_to_fix')
+        span.setTag(TEST_RETRY_REASON, TEST_RETRY_REASON_TYPES.atf)
       }
       if (hasPassedAttemptToFixRetries) {
         span.setTag(TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED, 'true')

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

@@ -25,7 +25,8 @@ const {
   TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
   TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
   TEST_HAS_FAILED_ALL_RETRIES,
-  getLibraryCapabilitiesTags
+  getLibraryCapabilitiesTags,
+  TEST_RETRY_REASON_TYPES
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const {
@@ -104,7 +105,8 @@ class VitestPlugin extends CiPlugin {
       isDisabled,
       mightHitProbe,
       isRetryReasonEfd,
-      isRetryReasonAttemptToFix
+      isRetryReasonAttemptToFix,
+      isRetryReasonAtr
     }) => {
       const testSuite = getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
       const store = storage('legacy').getStore()
@@ -114,19 +116,22 @@ class VitestPlugin extends CiPlugin {
       }
       if (isRetry) {
         extraTags[TEST_IS_RETRY] = 'true'
+        if (isRetryReasonAttemptToFix) {
+          extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.atf
+        } else if (isRetryReasonEfd) {
+          extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
+        } else if (isRetryReasonAtr) {
+          extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.atr
+        } else {
+          extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.ext
+        }
       }
       if (isNew) {
         extraTags[TEST_IS_NEW] = 'true'
       }
-      if (isRetryReasonEfd) {
-        extraTags[TEST_RETRY_REASON] = 'efd'
-      }
       if (isAttemptToFix) {
         extraTags[TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX] = 'true'
       }
-      if (isRetryReasonAttemptToFix) {
-        extraTags[TEST_RETRY_REASON] = 'attempt_to_fix'
-      }
       if (isQuarantined) {
         extraTags[TEST_MANAGEMENT_IS_QUARANTINED] = 'true'
       }

package/packages/datadog-shimmer/src/shimmer.js

@@ -2,9 +2,6 @@
 
 const log = require('../../dd-trace/src/log')
 
-// Use a weak map to avoid polluting the wrapped function/method.
-const unwrappers = new WeakMap()
-
 function copyProperties (original, wrapped) {
   // TODO getPrototypeOf is not fast. Should we instead do this in specific
   // instrumentations where needed?
@@ -27,21 +24,13 @@ function copyProperties (original, wrapped) {
 function wrapFunction (original, wrapper) {
   if (typeof original === 'function') assertNotClass(original)
 
-  let delegate = safeMode
+  const wrapped = safeMode
     ? safeWrapper(original, wrapper)
     : wrapper(original)
 
-  const shim = function shim () {
-    return delegate.apply(this, arguments)
-  }
-
-  unwrappers.set(shim, () => {
-    delegate = original
-  })
-
-  if (typeof original === 'function') copyProperties(original, shim)
+  if (typeof original === 'function') copyProperties(original, wrapped)
 
-  return shim
+  return wrapped
 }
 
 const wrapFn = function (original, delegate) {
@@ -99,8 +88,6 @@ function wrapMethod (target, name, wrapper, noAssert) {
   if (typeof original === 'function') copyProperties(original, wrapped)
 
   if (descriptor) {
-    unwrappers.set(wrapped, () => Object.defineProperty(target, name, descriptor))
-
     if (descriptor.get || descriptor.set) {
       attributes.get = () => wrapped
     } else {
@@ -114,7 +101,6 @@ function wrapMethod (target, name, wrapper, noAssert) {
       })
     }
   } else { // no descriptor means original was on the prototype
-    unwrappers.set(wrapped, () => delete target[name])
     attributes.value = wrapped
     attributes.writable = true
   }
@@ -218,18 +204,6 @@ function wrap (target, name, wrapper) {
     : wrapMethod(target, name, wrapper)
 }
 
-function unwrap (target, name) {
-  if (!target) return target // no target to unwrap
-
-  const unwrapper = unwrappers.get(name ? target[name] : target)
-
-  if (!unwrapper) return target // target is already unwrapped or isn't wrapped
-
-  unwrapper()
-
-  return target
-}
-
 function massWrap (targets, names, wrapper) {
   targets = toArray(targets)
   names = toArray(names)
@@ -241,17 +215,6 @@ function massWrap (targets, names, wrapper) {
   }
 }
 
-function massUnwrap (targets, names) {
-  targets = toArray(targets)
-  names = toArray(names)
-
-  for (const target of targets) {
-    for (const name of names) {
-      unwrap(target, name)
-    }
-  }
-}
-
 function toArray (maybeArray) {
   return Array.isArray(maybeArray) ? maybeArray : [maybeArray]
 }
@@ -294,7 +257,5 @@ module.exports = {
   wrap,
   wrapFunction,
   massWrap,
-  unwrap,
-  massUnwrap,
   setSafe
 }

package/packages/dd-trace/src/appsec/iast/taint-tracking/filter.js

@@ -6,11 +6,11 @@ const isPrivateModule = function (file) {
   return file && file.indexOf(NODE_MODULES) === -1
 }
 
-const isNotLibraryFile = function (file) {
-  return file && file.indexOf('dd-trace-js') === -1 && file.indexOf('dd-trace') === -1
+const isDdTrace = function (file) {
+  return file && (file.indexOf('dd-trace-js') !== -1 || file.indexOf('dd-trace') !== -1)
 }
 
 module.exports = {
   isPrivateModule,
-  isNotLibraryFile
+  isDdTrace
 }

package/packages/dd-trace/src/appsec/iast/taint-tracking/index.js

@@ -1,6 +1,5 @@
 'use strict'
 
-const { enableRewriter, disableRewriter } = require('./rewriter')
 const {
   createTransaction,
   removeTransaction,
@@ -16,7 +15,6 @@ const kafkaContextPlugin = require('../context/kafka-ctx-plugin')
 
 module.exports = {
   enableTaintTracking (config, telemetryVerbosity) {
-    enableRewriter(telemetryVerbosity)
     enableTaintOperations(telemetryVerbosity)
     taintTrackingPlugin.enable(config)
 
@@ -26,7 +24,6 @@ module.exports = {
     setMaxTransactions(config.maxConcurrentRequests)
   },
   disableTaintTracking () {
-    disableRewriter()
     disableTaintOperations()
     taintTrackingPlugin.disable()
 

package/packages/dd-trace/src/appsec/iast/taint-tracking/rewriter-esm.mjs

@@ -3,19 +3,20 @@
 import path from 'path'
 import { URL } from 'url'
 import { getName } from '../telemetry/verbosity.js'
-import { isNotLibraryFile, isPrivateModule } from './filter.js'
+import { isDdTrace, isPrivateModule } from './filter.js'
 import constants from './constants.js'
 
 const currentUrl = new URL(import.meta.url)
 const ddTraceDir = path.join(currentUrl.pathname, '..', '..', '..', '..', '..', '..')
 
-let port, rewriter
+let port, rewriter, iastEnabled
 
 export async function initialize (data) {
   if (rewriter) return Promise.reject(new Error('ALREADY INITIALIZED'))
 
-  const { csiMethods, telemetryVerbosity, chainSourceMap } = data
+  const { csiMethods, telemetryVerbosity, chainSourceMap, orchestrionConfig } = data
   port = data.port
+  iastEnabled = data.iastEnabled
 
   const iastRewriter = await import('@datadog/wasm-js-rewriter')
 
@@ -24,7 +25,8 @@ export async function initialize (data) {
   rewriter = new NonCacheRewriter({
     csiMethods,
     telemetryVerbosity: getName(telemetryVerbosity),
-    chainSourceMap
+    chainSourceMap,
+    orchestrion: orchestrionConfig
   })
 }
 
@@ -35,15 +37,26 @@ export async function load (url, context, nextLoad) {
   if (!result.source) return result
   if (url.includes(ddTraceDir) || url.includes('iitm=true')) return result
 
+  let passes
   try {
-    if (isPrivateModule(url) && isNotLibraryFile(url)) {
-      const rewritten = rewriter.rewrite(result.source.toString(), url)
-
-      if (rewritten?.content) {
-        result.source = rewritten.content || result.source
-        const data = { url, rewritten }
-        port.postMessage({ type: constants.REWRITTEN_MESSAGE, data })
+    if (isDdTrace(url)) {
+      return result
+    }
+    if (isPrivateModule(url)) {
+      // TODO error tracking needs to be added based on config
+      passes = ['error_tracking']
+      if (iastEnabled) {
+        passes.push('iast')
       }
+    } else {
+      passes = ['orchestrion']
+    }
+    const rewritten = rewriter.rewrite(result.source.toString(), url, passes)
+
+    if (rewritten?.content) {
+      result.source = rewritten.content || result.source
+      const data = { url, rewritten }
+      port.postMessage({ type: constants.REWRITTEN_MESSAGE, data })
     }
   } catch (e) {
     const newErrObject = {

package/packages/dd-trace/src/appsec/iast/taint-tracking/rewriter-telemetry.js

@@ -4,39 +4,10 @@ const iastTelemetry = require('../telemetry')
 const { Verbosity } = require('../telemetry/verbosity')
 const { INSTRUMENTED_PROPAGATION } = require('../telemetry/iast-metric')
 
-const telemetryRewriter = {
-  off (content, filename, rewriter) {
-    return rewriter.rewrite(content, filename)
-  },
-
-  information (content, filename, rewriter) {
-    const response = this.off(content, filename, rewriter)
-
-    incrementTelemetry(response.metrics)
-
-    return response
-  }
-}
-
-function getRewriteFunction (rewriter) {
-  switch (iastTelemetry.verbosity) {
-    case Verbosity.OFF:
-      return (content, filename) => telemetryRewriter.off(content, filename, rewriter)
-    default:
-      return (content, filename) => telemetryRewriter.information(content, filename, rewriter)
-  }
-}
-
-function incrementTelemetry (metrics) {
-  if (metrics?.instrumentedPropagation) {
-    INSTRUMENTED_PROPAGATION.inc(undefined, metrics.instrumentedPropagation)
-  }
-}
-
 function incrementTelemetryIfNeeded (metrics) {
-  if (iastTelemetry.verbosity !== Verbosity.OFF) {
-    incrementTelemetry(metrics)
+  if (iastTelemetry.verbosity !== Verbosity.OFF && metrics?.instrumentedPropagation) {
+    INSTRUMENTED_PROPAGATION.inc(undefined, metrics.instrumentedPropagation)
   }
 }
 
-module.exports = { getRewriteFunction, incrementTelemetryIfNeeded }
+module.exports = { incrementTelemetryIfNeeded }