dd-trace 5.80.0 → 5.82.0

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

@@ -4,8 +4,8 @@ const { readFile } = require('fs')
 const { types } = require('util')
 const { join } = require('path')
 const { Worker, MessageChannel, threadId: parentThreadId } = require('worker_threads')
-const getDebuggerConfig = require('./config')
 const log = require('../log')
+const getDebuggerConfig = require('./config')
 
 let worker = null
 let configChannel = null
@@ -89,7 +89,7 @@ function start (config, rc) {
 
     log.error('[debugger] worker thread exited unexpectedly', error)
 
-    // Be nice, clean up now that the worker thread encounted an issue and we can't continue
+    // Be nice, clean up now that the worker thread encountered an issue and we can't continue
     rc.removeProductHandler('LIVE_DEBUGGING')
     worker.removeAllListeners()
     configChannel = null

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

@@ -1,11 +1,12 @@
 'use strict'
 
 const lookup = require('dns').lookup // cache to avoid instrumentation
-const request = require('./exporters/common/request')
 const dgram = require('dgram')
 const isIP = require('net').isIP
-const log = require('./log')
 const { URL, format } = require('url')
+
+const request = require('./exporters/common/request')
+const log = require('./log')
 const Histogram = require('./histogram')
 const defaults = require('./config_defaults')
 

package/packages/dd-trace/src/encode/0.4.js

@@ -1,11 +1,11 @@
 'use strict'
 
-const { truncateSpan, normalizeSpan } = require('./tags-processors')
 const { MsgpackChunk, MsgpackEncoder } = require('../msgpack')
 const log = require('../log')
 const { isTrue } = require('../util')
 const { memoize } = require('../log/utils')
 const { getEnvironmentVariable } = require('../config-helper')
+const { truncateSpan, normalizeSpan } = require('./tags-processors')
 
 const SOFT_LIMIT = 8 * 1024 * 1024 // 8MB
 

package/packages/dd-trace/src/encode/agentless-ci-visibility.js

@@ -1,6 +1,4 @@
 'use strict'
-const { truncateSpan, normalizeSpan } = require('./tags-processors')
-const { AgentEncoder } = require('./0.4')
 const { version: ddTraceVersion } = require('../../../../package.json')
 const { ITR_CORRELATION_ID } = require('../../src/plugins/util/test')
 const id = require('../../src/id')
@@ -9,6 +7,8 @@ const {
   TELEMETRY_ENDPOINT_PAYLOAD_SERIALIZATION_MS,
   TELEMETRY_ENDPOINT_PAYLOAD_EVENTS_COUNT
 } = require('../ci-visibility/telemetry')
+const { AgentEncoder } = require('./0.4')
+const { truncateSpan, normalizeSpan } = require('./tags-processors')
 
 const ENCODING_VERSION = 1
 const ALLOWED_CONTENT_TYPES = new Set(['test_session_end', 'test_module_end', 'test_suite_end', 'test'])

package/packages/dd-trace/src/encode/coverage-ci-visibility.js

@@ -1,5 +1,4 @@
 'use strict'
-const { AgentEncoder } = require('./0.4')
 const { MsgpackChunk } = require('../msgpack')
 
 const {
@@ -8,6 +7,7 @@ const {
   TELEMETRY_ENDPOINT_PAYLOAD_EVENTS_COUNT
 } = require('../ci-visibility/telemetry')
 const FormData = require('../exporters/common/form-data')
+const { AgentEncoder } = require('./0.4')
 
 const COVERAGE_PAYLOAD_VERSION = 2
 const COVERAGE_KEYS_LENGTH = 2

package/packages/dd-trace/src/encode/span-stats.js

@@ -31,7 +31,7 @@ class SpanStatsEncoder extends AgentEncoder {
   }
 
   _encodeStat (bytes, stat) {
-    this._encodeMapPrefix(bytes, 12)
+    this._encodeMapPrefix(bytes, 14)
 
     this._encodeString(bytes, 'Service')
     const service = stat.Service || DEFAULT_SERVICE_NAME
@@ -70,6 +70,12 @@ class SpanStatsEncoder extends AgentEncoder {
 
     this._encodeString(bytes, 'TopLevelHits')
     this._encodeLong(bytes, stat.TopLevelHits)
+
+    this._encodeString(bytes, 'HTTPMethod')
+    this._encodeString(bytes, stat.HTTPMethod)
+
+    this._encodeString(bytes, 'HTTPEndpoint')
+    this._encodeString(bytes, stat.HTTPEndpoint)
   }
 
   _encodeBucket (bytes, bucket) {
@@ -89,7 +95,7 @@ class SpanStatsEncoder extends AgentEncoder {
   }
 
   _encode (bytes, stats) {
-    this._encodeMapPrefix(bytes, 8)
+    this._encodeMapPrefix(bytes, stats.ProcessTags ? 9 : 8)
 
     this._encodeString(bytes, 'Hostname')
     this._encodeString(bytes, stats.Hostname)
@@ -117,6 +123,11 @@ class SpanStatsEncoder extends AgentEncoder {
 
     this._encodeString(bytes, 'Sequence')
     this._encodeLong(bytes, stats.Sequence)
+
+    if (stats.ProcessTags) {
+      this._encodeString(bytes, 'ProcessTags')
+      this._encodeString(bytes, stats.ProcessTags)
+    }
   }
 }
 

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

@@ -1,9 +1,9 @@
 'use strict'
 
-const exporters = require('../../../ext/exporters')
 const fs = require('fs')
-const constants = require('./constants')
+const exporters = require('../../../ext/exporters')
 const { getEnvironmentVariable } = require('../../dd-trace/src/config-helper')
+const constants = require('./constants')
 
 module.exports = function getExporter (name) {
   switch (name) {

package/packages/dd-trace/src/exporters/agent/index.js

@@ -2,8 +2,8 @@
 
 const { URL, format } = require('url')
 const log = require('../../log')
-const Writer = require('./writer')
 const defaults = require('../../config_defaults')
+const Writer = require('./writer')
 
 class AgentExporter {
   #timer

package/packages/dd-trace/src/exporters/common/agent-info-exporter.js

@@ -2,9 +2,9 @@
 
 const { URL, format } = require('url')
 
-const request = require('./request')
 const defaults = require('../../config_defaults')
 const { incrementCountMetric, TELEMETRY_EVENTS_ENQUEUED_FOR_SERIALIZATION } = require('../../ci-visibility/telemetry')
+const request = require('./request')
 
 function fetchAgentInfo (url, callback) {
   request('', {

package/packages/dd-trace/src/exporters/common/request.js

@@ -8,11 +8,11 @@ const http = require('http')
 const https = require('https')
 const zlib = require('zlib')
 
+const { storage } = require('../../../../datadog-core')
+const log = require('../../log')
 const { urlToHttpOptions } = require('./url-to-http-options-polyfill')
 const docker = require('./docker')
 const { httpAgent, httpsAgent } = require('./agents')
-const { storage } = require('../../../../datadog-core')
-const log = require('../../log')
 
 const maxActiveRequests = 8
 

package/packages/dd-trace/src/exporters/common/writer.js

@@ -1,7 +1,7 @@
 'use strict'
 
-const request = require('./request')
 const log = require('../../log')
+const request = require('./request')
 const { safeJSONStringify } = require('./util')
 
 class Writer {

package/packages/dd-trace/src/exporters/span-stats/index.js

@@ -2,8 +2,8 @@
 
 const { URL, format } = require('url')
 
-const { Writer } = require('./writer')
 const defaults = require('../../config_defaults')
+const { Writer } = require('./writer')
 
 class SpanStatsExporter {
   constructor (config) {

package/packages/dd-trace/src/external-logger/src/index.js

@@ -1,8 +1,7 @@
 'use strict'
 
-const tracerLogger = require('../../log')// path to require tracer logger
-
 const https = require('https')
+const tracerLogger = require('../../log')// path to require tracer logger
 
 class ExternalLogger {
   // Note: these attribute names match the corresponding entry in the JSON payload.

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

@@ -2,10 +2,10 @@
 
 const log = require('../log')
 const startupLog = require('../startup-log')
-const FlareFile = require('./file')
 const { LogChannel } = require('../log/channels')
 const request = require('../exporters/common/request')
 const FormData = require('../exporters/common/form-data')
+const FlareFile = require('./file')
 
 const MAX_LOG_SIZE = 12 * 1024 * 1024 // 12MB soft limit
 const TIMEOUT = 20 * 1000 * 60

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

@@ -2,10 +2,11 @@
 
 var path = require('path')
 var Module = require('module')
+
+var nodeVersion = require('../../../../version')
 var isTrue = require('./util').isTrue
 var log = require('./log')
 var telemetry = require('./telemetry')
-var nodeVersion = require('../../../../version')
 
 var NODE_MAJOR = nodeVersion.NODE_MAJOR
 
@@ -14,7 +15,9 @@ function guard (fn) {
   var clobberBailout = false
   var forced = isTrue(process.env.DD_INJECT_FORCE)
   var engines = require('../../../../package.json').engines
-  var minMajor = parseInt(engines.node.replace(/[^0-9]/g, ''))
+  var versions = engines.node.match(/^>=(\d+) <(\d+)$/)
+  var minMajor = versions[1]
+  var nextMajor = versions[2]
   var version = process.versions.node
 
   if (process.env.DD_INJECTION_ENABLED) {
@@ -40,7 +43,7 @@ function guard (fn) {
 
   // If the runtime doesn't match the engines field in package.json, then we
   // should not initialize the tracer.
-  if (!clobberBailout && NODE_MAJOR < minMajor) {
+  if (!clobberBailout && (NODE_MAJOR < minMajor || NODE_MAJOR >= nextMajor)) {
     initBailout = true
     telemetry([
       { name: 'abort', tags: ['reason:incompatible_runtime'] },

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

@@ -1,6 +1,6 @@
 'use strict'
 
-const { DDSketch } = require('@datadog/sketches-js')
+const { DDSketch } = require('../../../vendor/dist/@datadog/sketches-js')
 
 class Histogram {
   constructor () {

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

@@ -14,26 +14,43 @@ let batch = 0
 
 // Internal representation of a trace or span ID.
 class Identifier {
+  /**
+   * @param {string} value
+   * @param {number} [radix]
+   */
   constructor (value, radix = 16) {
     this._buffer = radix === 16
       ? createBuffer(value)
       : fromString(value, radix)
   }
 
+  /**
+   * @param {number} [radix]
+   * @returns {string}
+   */
   toString (radix = 16) {
     return radix === 16
       ? toHexString(this._buffer)
       : toNumberString(this._buffer, radix)
   }
 
+  /**
+   * @returns {bigint}
+   */
   toBigInt () {
     return Buffer.from(this._buffer).readBigUInt64BE(0)
   }
 
+  /**
+   * @returns {number[] | Uint8Array}
+   */
   toBuffer () {
     return this._buffer
   }
 
+  /**
+   * @returns {number[] | Uint8Array}
+   */
   toArray () {
     if (this._buffer.length === 8) {
       return this._buffer
@@ -41,10 +58,17 @@ class Identifier {
     return this._buffer.slice(-8)
   }
 
+  /**
+   * @returns {string}
+   */
   toJSON () {
     return this.toString()
   }
 
+  /**
+   * @param {Identifier} other
+   * @returns {boolean}
+   */
   equals (other) {
     const length = this._buffer.length
     const otherLength = other._buffer.length
@@ -59,6 +83,10 @@ class Identifier {
 }
 
 // Create a buffer, using an optional hexadecimal value if provided.
+/**
+ * @param {string} value
+ * @returns {number[] | Uint8Array}
+ */
 function createBuffer (value) {
   if (value === '0') return zeroId
   if (!value) return pseudoRandom()
@@ -77,6 +105,11 @@ function createBuffer (value) {
 }
 
 // Convert a numerical string to a buffer using the specified radix.
+/**
+ * @param {string} str
+ * @param {number} raddix
+ * @returns {number[]}
+ */
 function fromString (str, raddix) {
   const buffer = new Array(8)
   const len = str.length
@@ -116,6 +149,11 @@ function fromString (str, raddix) {
 }
 
 // Convert a buffer to a numerical string.
+/**
+ * @param {number[] | Uint8Array} buffer
+ * @param {number} [radix]
+ * @returns {string}
+ */
 function toNumberString (buffer, radix) {
   let high = readInt32(buffer, buffer.length - 8)
   let low = readInt32(buffer, buffer.length - 4)
@@ -137,11 +175,18 @@ function toNumberString (buffer, radix) {
 }
 
 // Convert a buffer to a hexadecimal string.
+/**
+ * @param {number[] | Uint8Array} buffer
+ * @returns {string}
+ */
 function toHexString (buffer) {
   return map.call(buffer, pad).join('')
 }
 
 // Simple pseudo-random 64-bit ID generator.
+/**
+ * @returns {number[] | Uint8Array}
+ */
 function pseudoRandom () {
   if (batch === 0) {
     randomFillSync(data)
@@ -164,6 +209,11 @@ function pseudoRandom () {
 }
 
 // Read a buffer to unsigned integer bytes.
+/**
+ * @param {number[] | Uint8Array} buffer
+ * @param {number} offset
+ * @returns {number}
+ */
 function readInt32 (buffer, offset) {
   return (buffer[offset + 0] * 16_777_216) +
     (buffer[offset + 1] << 16) +
@@ -172,6 +222,11 @@ function readInt32 (buffer, offset) {
 }
 
 // Write unsigned integer bytes to a buffer.
+/**
+ * @param {number[] | Uint8Array} buffer
+ * @param {number} value
+ * @param {number} offset
+ */
 function writeUInt32BE (buffer, value, offset) {
   buffer[3 + offset] = value & 255
   value >>= 8
@@ -182,6 +237,11 @@ function writeUInt32BE (buffer, value, offset) {
   buffer[0 + offset] = value & 255
 }
 
+/**
+ * @param {string} [value]
+ * @param {number} [radix]
+ * @returns {Identifier}
+ */
 module.exports = function createIdentifier (value, radix) {
   return new Identifier(value, radix)
 }

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

@@ -1,7 +1,7 @@
 'use strict'
 
-const { isFalse } = require('./util')
 const { getEnvironmentVariable } = require('../../dd-trace/src/config-helper')
+const { isFalse } = require('./util')
 
 // Global `jest` is only present in Jest workers.
 const inJestWorker = typeof jest !== 'undefined'

package/packages/dd-trace/src/lambda/handler.js

@@ -3,8 +3,8 @@
 const log = require('../log')
 const { channel } = require('../../../datadog-instrumentations/src/helpers/instrument')
 const { ERROR_MESSAGE, ERROR_TYPE } = require('../constants')
-const { ImpendingTimeout } = require('./runtime/errors')
 const { getEnvironmentVariable } = require('../config-helper')
+const { ImpendingTimeout } = require('./runtime/errors')
 
 const globalTracer = global._ddtrace
 const tracer = globalTracer._tracer
@@ -21,7 +21,7 @@ let __lambdaTimeout
  * Publishes to the `apm:aws:lambda:timeout` channel when
  * the AWS Lambda run time is about to end.
  *
- * @param {*} context AWS Lambda context object.
+ * @param {object} context AWS Lambda context object.
  */
 function checkTimeout (context) {
   const remainingTimeInMillis = context.getRemainingTimeInMillis()
@@ -63,7 +63,7 @@ function crashFlush () {
 /**
  * Extracts the context from the given Lambda handler arguments.
  *
- * @param {*} args any amount of arguments
+ * @param {unknown[]} args any amount of arguments
  * @returns the context, if extraction was succesful.
  */
 function extractContext (args) {
@@ -80,7 +80,7 @@ function extractContext (args) {
 /**
  * Patches your AWS Lambda handler function to add some tracing support.
  *
- * @param {*} lambdaHandler a Lambda handler function.
+ * @param {Function} lambdaHandler a Lambda handler function.
  */
 exports.datadog = function datadog (lambdaHandler) {
   return (...args) => {

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

@@ -1,7 +1,7 @@
 'use strict'
 
-const { registerLambdaHook } = require('./runtime/ritm')
 const { getEnvironmentVariable } = require('../config-helper')
+const { registerLambdaHook } = require('./runtime/ritm')
 
 /**
  * It is safe to do it this way, since customers will never be expected to disable

package/packages/dd-trace/src/lambda/runtime/patch.js

@@ -2,17 +2,17 @@
 
 const path = require('path')
 
-const { _extractModuleNameAndHandlerPath, _extractModuleRootAndHandler, _getLambdaFilePaths } = require('./ritm')
 const { datadog } = require('../handler')
 const { addHook } = require('../../../../datadog-instrumentations/src/helpers/instrument')
 const shimmer = require('../../../../datadog-shimmer')
 const { getEnvironmentVariable } = require('../../config-helper')
+const { _extractModuleNameAndHandlerPath, _extractModuleRootAndHandler, _getLambdaFilePaths } = require('./ritm')
 
 /**
  * Patches a Datadog Lambda module by calling `patchDatadogLambdaHandler`
  * with the handler name `datadog`.
  *
- * @param {*} datadogLambdaModule node module to be patched.
+ * @param {object} datadogLambdaModule node module to be patched.
  * @returns a Datadog Lambda module with the `datadog` function from
  * `datadog-lambda-js` patched.
  */
@@ -27,7 +27,7 @@ const patchDatadogLambdaModule = (datadogLambdaModule) => {
  * Datadog instrumentation by getting the Lambda handler from its
  * arguments.
  *
- * @param {*} datadogHandler the Datadog Lambda handler to destructure.
+ * @param {Function} datadogHandler the Datadog Lambda handler to destructure.
  * @returns the datadogHandler with its arguments patched.
  */
 function patchDatadogLambdaHandler (datadogHandler) {
@@ -51,7 +51,7 @@ const patchLambdaModule = (handlerPath) => (lambdaModule) => {
 /**
  * Patches a Lambda handler in order to do Datadog instrumentation.
  *
- * @param {*} lambdaHandler the Lambda handler to be patched.
+ * @param {Function} lambdaHandler the Lambda handler to be patched.
  * @returns a function which patches the given Lambda handler.
  */
 function patchLambdaHandler (lambdaHandler) {

package/packages/dd-trace/src/lambda/runtime/ritm.js

@@ -3,7 +3,7 @@
  * Modifications copyright 2022 Datadog, Inc.
  *
  * Some functions are part of aws-lambda-nodejs-runtime-interface-client
- * https://github.com/aws/aws-lambda-nodejs-runtime-interface-client/blob/main/src/utils/UserFunction.ts
+ * https://github.com/aws/aws-lambda-nodejs-runtime-interface-client/blob/v2.1.0/src/utils/UserFunction.ts
  */
 'use strict'
 
@@ -61,7 +61,7 @@ function _extractModuleNameAndHandlerPath (handler) {
 /**
  * Returns all possible paths of the files to be patched when required.
  *
- * @param {*} lambdaStylePath the path comprised of the `LAMBDA_TASK_ROOT`,
+ * @param {string} lambdaStylePath the path comprised of the `LAMBDA_TASK_ROOT`,
  * the root of the module of the Lambda handler, and the module name.
  * @returns the lambdaStylePath with appropiate extensions for the hook.
  */

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

@@ -34,6 +34,13 @@ module.exports = {
   TOTAL_TOKENS_METRIC_KEY: 'total_tokens',
   CACHE_READ_INPUT_TOKENS_METRIC_KEY: 'cache_read_input_tokens',
   CACHE_WRITE_INPUT_TOKENS_METRIC_KEY: 'cache_write_input_tokens',
+  REASONING_OUTPUT_TOKENS_METRIC_KEY: 'reasoning_output_tokens',
 
-  DROPPED_IO_COLLECTION_ERROR: 'dropped_io'
+  DROPPED_IO_COLLECTION_ERROR: 'dropped_io',
+
+  PROMPT_TRACKING_INSTRUMENTATION_METHOD: 'prompt_tracking_instrumentation_method',
+  PROMPT_MULTIMODAL: 'prompt_multimodal',
+  INSTRUMENTATION_METHOD_AUTO: 'auto',
+  INSTRUMENTATION_METHOD_ANNOTATED: 'annotated',
+  INSTRUMENTATION_METHOD_UNKNOWN: 'unknown'
 }

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

@@ -1,5 +1,9 @@
 'use strict'
 
+const util = require('node:util')
+
+const { channel } = require('dc-polyfill')
+
 const log = require('../log')
 const {
   ML_APP,
@@ -7,24 +11,19 @@ const {
   PROPAGATED_PARENT_ID_KEY
 } = require('./constants/tags')
 const { storage } = require('./storage')
-
 const telemetry = require('./telemetry')
 const LLMObsSpanProcessor = require('./span_processor')
+const LLMObsEvalMetricsWriter = require('./writers/evaluations')
+const LLMObsTagger = require('./tagger')
+const LLMObsSpanWriter = require('./writers/spans')
+const { setAgentStrategy } = require('./writers/util')
 
-const { channel } = require('dc-polyfill')
 const spanFinishCh = channel('dd-trace:span:finish')
 const evalMetricAppendCh = channel('llmobs:eval-metric:append')
 const flushCh = channel('llmobs:writers:flush')
 const injectCh = channel('dd-trace:span:inject')
 const registerUserSpanProcessorCh = channel('llmobs:register-processor')
 
-const LLMObsEvalMetricsWriter = require('./writers/evaluations')
-const LLMObsTagger = require('./tagger')
-const LLMObsSpanWriter = require('./writers/spans')
-const { setAgentStrategy } = require('./writers/util')
-
-const util = require('node:util')
-
 /**
  * Setting writers and processor globally when LLMObs is enabled
  * We're setting these in this module instead of on the SDK.

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

@@ -1,10 +1,9 @@
 'use strict'
 
+const { channel } = require('dc-polyfill')
 const BaseLLMObsPlugin = require('../base')
 const { getModelProvider } = require('../../../../../datadog-plugin-ai/src/utils')
 
-const { channel } = require('dc-polyfill')
-
 const toolCreationCh = channel('dd-trace:vercel-ai:tool')
 const setAttributesCh = channel('dd-trace:vercel-ai:span:setAttributes')
 
@@ -21,6 +20,38 @@ const {
   getLlmObsSpanName
 } = require('./util')
 
+/**
+ * @typedef {Record<string, unknown> & { description?: string, id?: string }} AvailableToolArgs
+ */
+
+/**
+ * @typedef {string | number | boolean | null | undefined | string[] | number[] | boolean[]} TagValue
+ * @typedef {Record<string, TagValue>} SpanTags
+ *
+ * @typedef {{ name?: string, description?: string }} ToolForModel
+ *
+ * @typedef {{ type: 'text' | 'reasoning' | 'redacted-reasoning', text?: string, data?: string }} TextPart
+ * @typedef {{ type: 'tool-call', toolName: string, toolCallId: string, args?: unknown, input?: unknown }} ToolCallPart
+ * @typedef {(
+ *   { type: 'tool-result', toolCallId: string, output?: { type: string, value?: unknown }, result?: unknown } &
+ *   Record<string, unknown>
+ * )} ToolResultPart
+ *
+ * @typedef {{
+ *   role: 'system',
+ *   content: string
+ * } | {
+ *   role: 'user',
+ *   content: TextPart[]
+ * } | {
+ *   role: 'assistant',
+ *   content: Array<TextPart | ToolCallPart>
+ * } | {
+ *   role: 'tool',
+ *   content: ToolResultPart[]
+ * }} AiSdkMessage
+ */
+
 const SPAN_NAME_TO_KIND_MAPPING = {
   // embeddings
   embed: 'workflow',
@@ -47,7 +78,7 @@ class VercelAILLMObsPlugin extends BaseLLMObsPlugin {
   /**
    * The available tools within the runtime scope of this integration.
    * This essentially acts as a global registry for all tools made through the Vercel AI SDK.
-   * @type {Set<Record<string, any>>}
+   * @type {Set<AvailableToolArgs>}
    */
   #availableTools
 
@@ -80,7 +111,7 @@ class VercelAILLMObsPlugin extends BaseLLMObsPlugin {
    * We use the tool description as the next best identifier for a tool.
    *
    * @param {string} toolName
-   * @param {string} toolDescription
+   * @param {string | undefined} toolDescription
    * @returns {string | undefined}
    */
   findToolName (toolName, toolDescription) {
@@ -223,7 +254,7 @@ class VercelAILLMObsPlugin extends BaseLLMObsPlugin {
 
   /**
    * @param {import('../../../opentracing/span')} span
-   * @param {Record<string, unknown>} tags
+   * @param {SpanTags} tags
    */
   setLLMOperationTags (span, tags) {
     const toolsForModel = tags['ai.prompt.tools']?.map(getJsonStringValue)
@@ -290,8 +321,8 @@ class VercelAILLMObsPlugin extends BaseLLMObsPlugin {
    * it is possible to have multiple tool call results in a single message that we
    * need to split into multiple messages.
    *
-   * @param {*} message
-   * @param {*} toolsForModel
+   * @param {AiSdkMessage} message
+   * @param {ToolForModel[] | null | undefined} toolsForModel
    * @returns {Array<{role: string, content: string, toolId?: string,
    *   toolCalls?: Array<{arguments: string, name: string, toolId: string, type: string}>}>}
    */