dd-trace 5.70.0 → 5.72.0

package/packages/datadog-instrumentations/src/jest.js

@@ -46,7 +46,7 @@ const skippableSuitesCh = channel('ci:jest:test-suite:skippable')
 const libraryConfigurationCh = channel('ci:jest:library-configuration')
 const knownTestsCh = channel('ci:jest:known-tests')
 const testManagementTestsCh = channel('ci:jest:test-management-tests')
-const impactedTestsCh = channel('ci:jest:modified-tests')
+const modifiedFilesCh = channel('ci:jest:modified-files')
 
 const itrSkippedSuitesCh = channel('ci:jest:itr:skipped-suites')
 
@@ -78,7 +78,7 @@ let isTestManagementTestsEnabled = false
 let testManagementTests = {}
 let testManagementAttemptToFixRetries = 0
 let isImpactedTestsEnabled = false
-let modifiedTests = {}
+let modifiedFiles = {}
 
 const testContexts = new WeakMap()
 const originalTestFns = new WeakMap()
@@ -89,6 +89,7 @@ const attemptToFixRetriedTestsStatuses = new Map()
 const wrappedWorkers = new WeakSet()
 const testSuiteMockedFiles = new Map()
 const testsToBeRetried = new Set()
+const testSuiteAbsolutePathsWithFastCheck = new Set()
 
 const BREAKPOINT_HIT_GRACE_PERIOD_MS = 200
 
@@ -197,10 +198,8 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
 
       if (this.isImpactedTestsEnabled) {
         try {
-          const hasImpactedTests = Object.keys(modifiedTests).length > 0
-          this.modifiedTestsForThisSuite = hasImpactedTests
-            ? this.getModifiedTestForThisSuite(modifiedTests)
-            : this.getModifiedTestForThisSuite(this.testEnvironmentOptions._ddModifiedTests)
+          const hasImpactedTests = Object.keys(modifiedFiles).length > 0
+          this.modifiedFiles = hasImpactedTests ? modifiedFiles : this.testEnvironmentOptions._ddModifiedFiles
         } catch (e) {
           log.error('Error parsing impacted tests', e)
           this.isImpactedTestsEnabled = false
@@ -290,19 +289,6 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
       return result
     }
 
-    getModifiedTestForThisSuite (modifiedTests) {
-      if (this.modifiedTestsForThisSuite) {
-        return this.modifiedTestsForThisSuite
-      }
-      let modifiedTestsForThisSuite = modifiedTests
-      // If jest is using workers, modified tests are serialized to json.
-      // If jest runs in band, they are not.
-      if (typeof modifiedTestsForThisSuite === 'string') {
-        modifiedTestsForThisSuite = JSON.parse(modifiedTestsForThisSuite)
-      }
-      return modifiedTestsForThisSuite
-    }
-
     // Generic function to handle test retries
     retryTest ({
       jestEvent,
@@ -319,9 +305,13 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
       }
     }
 
+    getShouldStripSeedFromTestName () {
+      return testSuiteAbsolutePathsWithFastCheck.has(this.testSuiteAbsolutePath)
+    }
+
     // At the `add_test` event we don't have the test object yet, so we can't use it
     getTestNameFromAddTestEvent (event, state) {
-      const describeSuffix = getJestTestName(state.currentDescribeBlock)
+      const describeSuffix = getJestTestName(state.currentDescribeBlock, this.getShouldStripSeedFromTestName())
       return describeSuffix ? `${describeSuffix} ${event.testName}` : event.testName
     }
 
@@ -344,7 +334,7 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
         })
       }
       if (event.name === 'test_start') {
-        const testName = getJestTestName(event.test)
+        const testName = getJestTestName(event.test, this.getShouldStripSeedFromTestName())
         if (testsToBeRetried.has(testName)) {
           // This is needed because we're trying tests with the same name
           this.resetSnapshotState()
@@ -378,7 +368,7 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
             this.testSourceFile,
             testStartLine,
             testEndLine,
-            this.modifiedTestsForThisSuite,
+            this.modifiedFiles,
             'jest'
           )
         }
@@ -465,7 +455,7 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
             this.testSourceFile,
             testStartLine,
             testEndLine,
-            this.modifiedTestsForThisSuite,
+            this.modifiedFiles,
             'jest'
           )
           if (isModified && !retriedTestsToNumAttempts.has(testFullName) && this.isEarlyFlakeDetectionEnabled) {
@@ -506,7 +496,7 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
         let failedAllTests = false
         let isAttemptToFix = false
         if (this.isTestManagementTestsEnabled) {
-          const testName = getJestTestName(event.test)
+          const testName = getJestTestName(event.test, this.getShouldStripSeedFromTestName())
           isAttemptToFix = this.testManagementTestsForThisSuite?.attemptToFix?.includes(testName)
           if (isAttemptToFix) {
             if (attemptToFixRetriedTestsStatuses.has(testName)) {
@@ -534,7 +524,7 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
         let isEfdRetry = false
         // We'll store the test statuses of the retries
         if (this.isKnownTestsEnabled) {
-          const testName = getJestTestName(event.test)
+          const testName = getJestTestName(event.test, this.getShouldStripSeedFromTestName())
           const isNewTest = retriedTestsToNumAttempts.has(testName)
           if (isNewTest) {
             if (newTestsTestStatuses.has(testName)) {
@@ -594,16 +584,17 @@ function getWrappedEnvironment (BaseEnvironment, jestVersion) {
         }
       }
       if (event.name === 'test_skip' || event.name === 'test_todo') {
+        const testName = getJestTestName(event.test, this.getShouldStripSeedFromTestName())
         testSkippedCh.publish({
           test: {
-            name: getJestTestName(event.test),
+            name: testName,
             suite: this.testSuite,
             testSourceFile: this.testSourceFile,
             displayName: this.displayName,
             frameworkVersion: jestVersion,
             testStartLine: getTestLineStart(event.test.asyncError, this.testSuite)
           },
-          isDisabled: this.testManagementTestsForThisSuite?.disabled?.includes(getJestTestName(event.test))
+          isDisabled: this.testManagementTestsForThisSuite?.disabled?.includes(testName)
         })
       }
     }
@@ -831,12 +822,12 @@ function getCliWrapper (isNewJestVersion) {
           onDone = resolve
         })
 
-        impactedTestsCh.publish({ onDone })
+        modifiedFilesCh.publish({ onDone })
 
         try {
-          const { err, modifiedTests: receivedModifiedTests } = await impactedTestsPromise
+          const { err, modifiedFiles: receivedModifiedFiles } = await impactedTestsPromise
           if (!err) {
-            modifiedTests = receivedModifiedTests
+            modifiedFiles = receivedModifiedFiles
           }
         } catch (err) {
           log.error('Jest impacted tests error', err)
@@ -1237,7 +1228,7 @@ addHook({
       _ddIsTestManagementTestsEnabled,
       _ddTestManagementTests,
       _ddTestManagementAttemptToFixRetries,
-      _ddModifiedTests,
+      _ddModifiedFiles,
       ...restOfTestEnvironmentOptions
     } = testEnvironmentOptions
 
@@ -1315,6 +1306,10 @@ addHook({
       // To bypass jest's own require engine
       return this._requireCoreModule(moduleName)
     }
+    // This means that `@fast-check/jest` is used in the test file.
+    if (moduleName === '@fast-check/jest') {
+      testSuiteAbsolutePathsWithFastCheck.add(this._testPath)
+    }
     return requireModuleOrMock.apply(this, arguments)
   })
 
@@ -1365,17 +1360,15 @@ function sendWrapper (send) {
 
       const suiteTestManagementTests = testManagementTests?.jest?.suites?.[testSuite]?.tests || {}
 
-      const suiteModifiedTests = Object.keys(modifiedTests).length > 0
-        ? modifiedTests
-        : {}
-
       args[0].config = {
         ...config,
         testEnvironmentOptions: {
           ...config.testEnvironmentOptions,
           _ddKnownTests: suiteKnownTests,
           _ddTestManagementTests: suiteTestManagementTests,
-          _ddModifiedTests: suiteModifiedTests
+          // TODO: figure out if we can reduce the size of the modified files object
+          // Can we use `testSuite` (it'd have to be relative to repository root though)
+          _ddModifiedFiles: modifiedFiles
         }
       }
     }

package/packages/datadog-instrumentations/src/mocha/main.js

@@ -68,7 +68,7 @@ const skippableSuitesCh = channel('ci:mocha:test-suite:skippable')
 const mochaGlobalRunCh = channel('ci:mocha:global:run')
 
 const testManagementTestsCh = channel('ci:mocha:test-management-tests')
-const impactedTestsCh = channel('ci:mocha:modified-tests')
+const modifiedFilesCh = channel('ci:mocha:modified-files')
 const workerReportTraceCh = channel('ci:mocha:worker-report:trace')
 const testSessionStartCh = channel('ci:mocha:session:start')
 const testSessionFinishCh = channel('ci:mocha:session:finish')
@@ -233,12 +233,12 @@ function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFini
     })
   }
 
-  const onReceivedImpactedTests = ({ err, modifiedTests: receivedModifiedTests }) => {
+  const onReceivedImpactedTests = ({ err, modifiedFiles: receivedModifiedFiles }) => {
     if (err) {
-      config.modifiedTests = []
+      config.modifiedFiles = []
       config.isImpactedTestsEnabled = false
     } else {
-      config.modifiedTests = receivedModifiedTests
+      config.modifiedFiles = receivedModifiedFiles
     }
     if (config.isSuitesSkippingEnabled) {
       ctx.onDone = onReceivedSkippableSuites
@@ -260,7 +260,7 @@ function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFini
     }
     if (config.isImpactedTestsEnabled) {
       ctx.onDone = onReceivedImpactedTests
-      impactedTestsCh.runStores(ctx, () => {})
+      modifiedFilesCh.runStores(ctx, () => {})
     } else if (config.isSuitesSkippingEnabled) {
       ctx.onDone = onReceivedSkippableSuites
       skippableSuitesCh.runStores(ctx, () => {})
@@ -284,7 +284,7 @@ function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFini
       testManagementTestsCh.runStores(ctx, () => {})
     } if (config.isImpactedTestsEnabled) {
       ctx.onDone = onReceivedImpactedTests
-      impactedTestsCh.runStores(ctx, () => {})
+      modifiedFilesCh.runStores(ctx, () => {})
     } else if (config.isSuitesSkippingEnabled) {
       ctx.onDone = onReceivedSkippableSuites
       skippableSuitesCh.runStores(ctx, () => {})
@@ -321,7 +321,7 @@ function getExecutionConfiguration (runner, isParallel, frameworkVersion, onFini
       testManagementTestsCh.runStores(ctx, () => {})
     } else if (config.isImpactedTestsEnabled) {
       ctx.onDone = onReceivedImpactedTests
-      impactedTestsCh.runStores(ctx, () => {})
+      modifiedFilesCh.runStores(ctx, () => {})
     } else if (config.isSuitesSkippingEnabled) {
       ctx.onDone = onReceivedSkippableSuites
       skippableSuitesCh.runStores(ctx, () => {})
@@ -696,9 +696,8 @@ addHook({
     }
 
     if (config.isImpactedTestsEnabled) {
-      const testSuiteImpactedTests = config.modifiedTests || {}
       newWorkerArgs._ddIsImpactedTestsEnabled = true
-      newWorkerArgs._ddModifiedTests = testSuiteImpactedTests
+      newWorkerArgs._ddModifiedFiles = config.modifiedFiles || {}
     }
 
     // We pass the known tests for the test file to the worker

package/packages/datadog-instrumentations/src/mocha/utils.js

@@ -446,7 +446,7 @@ function getRunTestsWrapper (runTests, config) {
     if (config.isImpactedTestsEnabled) {
       suite.tests.forEach((test) => {
         isModifiedCh.publish({
-          modifiedTests: config.modifiedTests,
+          modifiedFiles: config.modifiedFiles,
           file: suite.file,
           onDone: (isModified) => {
             if (isModified) {

package/packages/datadog-instrumentations/src/mocha/worker.js

@@ -36,9 +36,9 @@ addHook({
     }
     if (this.options._ddIsImpactedTestsEnabled) {
       config.isImpactedTestsEnabled = true
-      config.modifiedTests = this.options._ddModifiedTests
+      config.modifiedFiles = this.options._ddModifiedFiles
       delete this.options._ddIsImpactedTestsEnabled
-      delete this.options._ddModifiedTests
+      delete this.options._ddModifiedFiles
     }
     if (this.options._ddIsTestManagementTestsEnabled) {
       config.isTestManagementTestsEnabled = true

package/packages/datadog-instrumentations/src/pg.js

@@ -20,7 +20,7 @@ addHook({ name: 'pg', versions: ['>=8.0.3'], file: 'lib/native/client.js' }, Cli
   return Client
 })
 
-addHook({ name: 'pg', versions: ['>=8.0.3 <8.15.0', '>=8.15.0 <9'], file: 'lib/client.js' }, Client => {
+addHook({ name: 'pg', versions: ['>=8.0.3'], file: 'lib/client.js' }, Client => {
   shimmer.wrap(Client.prototype, 'query', query => wrapQuery(query))
   return Client
 })

package/packages/datadog-instrumentations/src/playwright.js

@@ -21,7 +21,7 @@ const testSessionFinishCh = channel('ci:playwright:session:finish')
 const libraryConfigurationCh = channel('ci:playwright:library-configuration')
 const knownTestsCh = channel('ci:playwright:known-tests')
 const testManagementTestsCh = channel('ci:playwright:test-management-tests')
-const impactedTestsCh = channel('ci:playwright:modified-tests')
+const modifiedFilesCh = channel('ci:playwright:modified-files')
 const isModifiedCh = channel('ci:playwright:test:is-modified')
 
 const testSuiteStartCh = channel('ci:playwright:test-suite:start')
@@ -59,7 +59,7 @@ let isTestManagementTestsEnabled = false
 let testManagementAttemptToFixRetries = 0
 let testManagementTests = {}
 let isImpactedTestsEnabled = false
-let modifiedTests = {}
+let modifiedFiles = {}
 const quarantinedOrDisabledTestsAttemptToFix = []
 let quarantinedButNotAttemptToFixFqns = new Set()
 let rootDir = ''
@@ -593,11 +593,11 @@ function runAllTestsWrapper (runAllTests, playwrightVersion) {
 
     if (isImpactedTestsEnabled && satisfies(playwrightVersion, MINIMUM_SUPPORTED_VERSION_RANGE_EFD)) {
       try {
-        const { err, modifiedTests: receivedModifiedTests } = await getChannelPromise(impactedTestsCh)
+        const { err, modifiedFiles: receivedModifiedFiles } = await getChannelPromise(modifiedFilesCh)
         if (err) {
           isImpactedTestsEnabled = false
         } else {
-          modifiedTests = receivedModifiedTests
+          modifiedFiles = receivedModifiedFiles
         }
       } catch (err) {
         isImpactedTestsEnabled = false
@@ -819,7 +819,7 @@ addHook({
       await Promise.all(allTests.map(async (test) => {
         const { isModified } = await getChannelPromise(isModifiedCh, {
           filePath: test._requireFile,
-          modifiedTests
+          modifiedFiles
         })
         if (isModified) {
           test._ddIsModified = true

package/packages/datadog-instrumentations/src/vitest.js

@@ -32,7 +32,7 @@ const libraryConfigurationCh = channel('ci:vitest:library-configuration')
 const knownTestsCh = channel('ci:vitest:known-tests')
 const isEarlyFlakeDetectionFaultyCh = channel('ci:vitest:is-early-flake-detection-faulty')
 const testManagementTestsCh = channel('ci:vitest:test-management-tests')
-const impactedTestsCh = channel('ci:vitest:modified-tests')
+const modifiedFilesCh = channel('ci:vitest:modified-files')
 
 const workerReportTraceCh = channel('ci:vitest:worker-report:trace')
 const workerReportLogsCh = channel('ci:vitest:worker-report:logs')
@@ -92,7 +92,7 @@ function getProvidedContext () {
       _ddTestManagementTests: testManagementTests,
       _ddIsFlakyTestRetriesEnabled: isFlakyTestRetriesEnabled,
       _ddIsImpactedTestsEnabled: isImpactedTestsEnabled,
-      _ddModifiedTests: modifiedTests
+      _ddModifiedFiles: modifiedFiles
     } = globalThis.__vitest_worker__.providedContext
 
     return {
@@ -106,7 +106,7 @@ function getProvidedContext () {
       testManagementTests,
       isFlakyTestRetriesEnabled,
       isImpactedTestsEnabled,
-      modifiedTests
+      modifiedFiles
     }
   } catch {
     log.error('Vitest workers could not parse provided context, so some features will not work.')
@@ -121,7 +121,7 @@ function getProvidedContext () {
       testManagementTests: {},
       isFlakyTestRetriesEnabled: false,
       isImpactedTestsEnabled: false,
-      modifiedTests: {}
+      modifiedFiles: {}
     }
   }
 }
@@ -316,14 +316,14 @@ function getSortWrapper (sort, frameworkVersion) {
     }
 
     if (isImpactedTestsEnabled) {
-      const { err, modifiedTests } = await getChannelPromise(impactedTestsCh)
+      const { err, modifiedFiles } = await getChannelPromise(modifiedFilesCh)
       if (err) {
         log.error('Could not get modified tests.')
       } else {
         try {
           const workspaceProject = this.ctx.getCoreWorkspaceProject()
           workspaceProject._provided._ddIsImpactedTestsEnabled = isImpactedTestsEnabled
-          workspaceProject._provided._ddModifiedTests = modifiedTests
+          workspaceProject._provided._ddModifiedFiles = modifiedFiles
         } catch {
           log.warn('Could not send modified tests to workers so Impacted Tests will not work.')
         }
@@ -464,7 +464,7 @@ addHook({
       testManagementAttemptToFixRetries,
       testManagementTests,
       isImpactedTestsEnabled,
-      modifiedTests
+      modifiedFiles
     } = getProvidedContext()
 
     if (isTestManagementTestsEnabled) {
@@ -499,7 +499,7 @@ addHook({
 
     if (isImpactedTestsEnabled) {
       isModifiedCh.publish({
-        modifiedTests,
+        modifiedFiles,
         testSuiteAbsolutePath: task.file.filepath,
         onDone: (isImpacted) => {
           if (isImpacted) {

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

@@ -0,0 +1,17 @@
+'use strict'
+
+const CompositePlugin = require('../../dd-trace/src/plugins/composite')
+const AnthropicTracingPlugin = require('./tracing')
+const AnthropicLLMObsPlugin = require('../../dd-trace/src/llmobs/plugins/anthropic')
+
+class AnthropicPlugin extends CompositePlugin {
+  static id = 'anthropic'
+  static get plugins () {
+    return {
+      llmobs: AnthropicLLMObsPlugin,
+      tracing: AnthropicTracingPlugin
+    }
+  }
+}
+
+module.exports = AnthropicPlugin

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

@@ -0,0 +1,30 @@
+'use strict'
+
+const TracingPlugin = require('../../dd-trace/src/plugins/tracing')
+
+class AnthropicTracingPlugin extends TracingPlugin {
+  static id = 'anthropic'
+  static operation = 'request'
+  static system = 'anthropic'
+  static prefix = 'tracing:apm:anthropic:request'
+
+  bindStart (ctx) {
+    const { resource, options } = ctx
+
+    this.startSpan('anthropic.request', {
+      meta: {
+        'resource.name': `Messages.${resource}`,
+        'anthropic.request.model': options.model
+      }
+    }, ctx)
+
+    return ctx.currentStore
+  }
+
+  asyncEnd (ctx) {
+    const span = ctx.currentStore?.span
+    span?.finish()
+  }
+}
+
+module.exports = AnthropicTracingPlugin

package/packages/datadog-plugin-aws-sdk/src/services/bedrockruntime/utils.js

@@ -53,10 +53,13 @@ function extractTextAndResponseReasonFromStream (chunks, modelProvider, modelNam
 
     switch (modelProviderUpper) {
       case PROVIDER.AMAZON: {
-        message += body?.outputText
-
-        inputTokens = body?.inputTextTokenCount
-        outputTokens = body?.totalOutputTextTokenCount
+        if (body?.outputText) {
+          message += body?.outputText
+          inputTokens = body?.inputTextTokenCount
+          outputTokens = body?.totalOutputTextTokenCount
+        } else if (body?.contentBlockDelta?.delta?.text) {
+          message += body.contentBlockDelta.delta.text
+        }
 
         break
       }
@@ -155,17 +158,16 @@ class RequestParams {
     stream = '',
     n
   } = {}) {
-    // stringify prompt as it could be a single prompt as well as a list of message objects
-    this.prompt = typeof prompt === 'string' ? prompt : JSON.stringify(prompt) || ''
-    this.temperature = temperature === undefined ? undefined : temperature
-    this.topP = topP === undefined ? undefined : topP
-    this.topK = topK === undefined ? undefined : topK
-    this.maxTokens = maxTokens === undefined ? undefined : maxTokens
+    this.prompt = prompt
+    this.temperature = temperature
+    this.topP = topP
+    this.topK = topK
+    this.maxTokens = maxTokens
     this.stopSequences = stopSequences || []
     this.inputType = inputType || ''
     this.truncate = truncate || ''
     this.stream = stream || ''
-    this.n = n === undefined ? undefined : n
+    this.n = n
   }
 }
 
@@ -234,17 +236,48 @@ function extractRequestParams (params, provider) {
       })
     }
     case PROVIDER.AMAZON: {
+      const prompt = requestBody.inputText
       if (modelId.includes('embed')) {
-        return new RequestParams({ prompt: requestBody.inputText })
+        return new RequestParams({ prompt })
+      } else if (prompt !== undefined) {
+        const textGenerationConfig = requestBody.textGenerationConfig || {}
+        return new RequestParams({
+          prompt,
+          temperature: textGenerationConfig.temperature,
+          topP: textGenerationConfig.topP,
+          maxTokens: textGenerationConfig.maxTokenCount,
+          stopSequences: textGenerationConfig.stopSequences
+        })
+      } else if (Array.isArray(requestBody.messages)) {
+        const inferenceConfig = requestBody.inferenceConfig || {}
+        const messages = []
+        if (Array.isArray(requestBody.system)) {
+          for (const sysMsg of requestBody.system) {
+            messages.push({
+              content: sysMsg.text,
+              role: 'system'
+            })
+          }
+        }
+        for (const message of requestBody.messages) {
+          const textBlocks = message.content?.filter(block => block.text) || []
+          if (textBlocks.length > 0) {
+            messages.push({
+              content: textBlocks.map(block => block.text).join(''),
+              role: message.role
+            })
+          }
+        }
+        return new RequestParams({
+          prompt: messages,
+          temperature: inferenceConfig.temperature,
+          topP: inferenceConfig.topP,
+          maxTokens: inferenceConfig.maxTokens,
+          stopSequences: inferenceConfig.stopSequences
+        })
       }
-      const textGenerationConfig = requestBody.textGenerationConfig || {}
-      return new RequestParams({
-        prompt: requestBody.inputText,
-        temperature: textGenerationConfig.temperature,
-        topP: textGenerationConfig.topP,
-        maxTokens: textGenerationConfig.maxTokenCount,
-        stopSequences: textGenerationConfig.stopSequences
-      })
+
+      return new RequestParams({ prompt })
     }
     case PROVIDER.ANTHROPIC: {
       let prompt = requestBody.prompt
@@ -350,14 +383,27 @@ function extractTextAndResponseReason (response, provider, modelName) {
         if (modelName.includes('embed')) {
           return new Generation({ message: body.embedding })
         }
-        const results = body.results || []
-        if (results.length > 0) {
-          const result = results[0]
+        if (body.results) {
+          const results = body.results || []
+          if (results.length > 0) {
+            const result = results[0]
+            return new Generation({
+              message: result.outputText,
+              finishReason: result.completionReason,
+              inputTokens: body.inputTextTokenCount,
+              outputTokens: result.tokenCount
+            })
+          }
+        } else if (body.output) {
+          const output = body.output || {}
           return new Generation({
-            message: result.outputText,
-            finishReason: result.completionReason,
-            inputTokens: body.inputTextTokenCount,
-            outputTokens: result.tokenCount
+            message: output.message?.content[0]?.text ?? 'Unsupported content type',
+            finishReason: body.stopReason,
+            role: output.message?.role,
+            inputTokens: body.usage?.inputTokens,
+            outputTokens: body.usage?.outputTokens,
+            cacheReadInputTokenCount: body.usage?.cacheReadInputTokenCount,
+            cacheWriteInputTokenCount: body.usage?.cacheWriteInputTokenCount
           })
         }
         break

package/packages/datadog-plugin-azure-event-hubs/src/index.js

@@ -0,0 +1,15 @@
+'use strict'
+
+const ProducerPlugin = require('./producer')
+const CompositePlugin = require('../../dd-trace/src/plugins/composite')
+
+class AzureEventHubsPlugin extends CompositePlugin {
+  static get id () { return 'azure-event-hubs' }
+  static get plugins () {
+    return {
+      producer: ProducerPlugin
+    }
+  }
+}
+
+module.exports = AzureEventHubsPlugin