dd-trace 5.94.0 → 5.95.0

package/packages/dd-trace/src/llmobs/plugins/anthropic.js

@@ -1,5 +1,6 @@
 'use strict'
 
+const { UNKNOWN_MODEL_PROVIDER } = require('../constants/tags')
 const LLMObsPlugin = require('./base')
 
 const ALLOWED_METADATA_KEYS = new Set([
@@ -108,16 +109,24 @@ class AnthropicLLMObsPlugin extends LLMObsPlugin {
   }
 
   getLLMObsSpanRegisterOptions (ctx) {
-    const { options } = ctx
+    const { options, baseUrl } = ctx
     const { model } = options
+    const modelProvider = this._getModelProvider(baseUrl)
 
     return {
       kind: 'llm',
       modelName: model,
-      modelProvider: 'anthropic',
+      modelProvider,
     }
   }
 
+  _getModelProvider (baseUrl = '') {
+    if (baseUrl.includes('anthropic')) {
+      return 'anthropic'
+    }
+    return UNKNOWN_MODEL_PROVIDER
+  }
+
   setLLMObsTags (ctx) {
     const span = ctx.currentStore?.span
     if (!span) return

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

@@ -5,6 +5,7 @@ const {
   PROMPT_TRACKING_INSTRUMENTATION_METHOD,
   PROMPT_MULTIMODAL,
   INSTRUMENTATION_METHOD_AUTO,
+  UNKNOWN_MODEL_PROVIDER,
 } = require('../../constants/tags')
 const {
   extractChatTemplateFromInstructions,
@@ -90,10 +91,12 @@ class OpenAiLLMObsPlugin extends LLMObsPlugin {
   _getModelProviderAndClient (baseUrl = '') {
     if (baseUrl.includes('azure')) {
       return { modelProvider: 'azure_openai', client: 'AzureOpenAI' }
+    } else if (baseUrl.includes('openai')) {
+      return { modelProvider: 'openai', client: 'OpenAI' }
     } else if (baseUrl.includes('deepseek')) {
       return { modelProvider: 'deepseek', client: 'DeepSeek' }
     }
-    return { modelProvider: 'openai', client: 'OpenAI' }
+    return { modelProvider: UNKNOWN_MODEL_PROVIDER, client: 'OpenAI' }
   }
 
   _extractMetrics (response) {

package/packages/dd-trace/src/llmobs/writers/spans.js

@@ -32,7 +32,7 @@ class LLMObsSpanWriter extends BaseWriter {
     let processedEventSizeBytes = eventSizeBytes
 
     if (shouldTruncate) {
-      logger.warn(`Dropping event input/output because its size (${eventSizeBytes}) exceeds the 1MB event size limit`)
+      logger.warn(`Dropping event input/output because its size (${eventSizeBytes}) exceeds the 5MB event size limit`)
       event = this._truncateSpanEvent(event)
       processedEventSizeBytes = Buffer.byteLength(JSON.stringify(event))
     }

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

@@ -249,7 +249,7 @@ class PrioritySampler {
     }
     const rawPriority = tags[SAMPLING_PRIORITY]
     if (rawPriority !== undefined) {
-      const priority = Number.parseInt(String(rawPriority), 10)
+      const priority = Math.trunc(rawPriority)
 
       if (priority === 1 || priority === 2) {
         return USER_KEEP

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

@@ -8,7 +8,8 @@ class RateLimiter {
    * @param {'second'|'minute'|'hour'|'day'} [interval='second'] - Time window for the limiter.
    */
   constructor (rateLimit, interval = 'second') {
-    this._rateLimit = Number.parseInt(String(rateLimit))
+    // TODO: Change rateLimit to integers. Right now these are sometimes strings, sometimes numbers.
+    this._rateLimit = Math.trunc(rateLimit)
     // The limiter constructor accepts a token count number and an interval string
     this._limiter = new limiter.RateLimiter(this._rateLimit, interval)
     this._tokensRequested = 0

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

@@ -1,6 +1,6 @@
 'use strict'
 
-const log = require('./log')
+// TODO: Rename and move file. This is a general purpose helper for adding tags to a carrier.
 
 function addNonEmpty (carrier, key, value) {
   if (key !== '') {
@@ -11,47 +11,43 @@ function addNonEmpty (carrier, key, value) {
 function add (carrier, keyValuePairs) {
   if (!carrier) return
 
-  try {
-    if (typeof keyValuePairs === 'string') {
-      let valueStart = 0
-      let keyStart = 0
-
-      for (let i = 0; i < keyValuePairs.length; i++) {
-        const char = keyValuePairs[i]
-
-        if (char === ':') {
-          if (valueStart === 0) {
-            valueStart = i
-          }
-        } else if (char === ',') {
-          valueStart ||= i
-          addNonEmpty(
-            carrier,
-            keyValuePairs.slice(keyStart, valueStart).trim(),
-            keyValuePairs.slice(valueStart + 1, i).trim()
-          )
-          keyStart = i + 1
-          valueStart = 0
-        }
-      }
+  if (typeof keyValuePairs === 'string') {
+    let valueStart = 0
+    let keyStart = 0
 
-      if (keyValuePairs.at(-1) !== ',') {
-        valueStart ||= keyValuePairs.length
+    for (let i = 0; i < keyValuePairs.length; i++) {
+      const char = keyValuePairs[i]
+
+      if (char === ':') {
+        if (valueStart === 0) {
+          valueStart = i
+        }
+      } else if (char === ',') {
+        valueStart ||= i
         addNonEmpty(
           carrier,
           keyValuePairs.slice(keyStart, valueStart).trim(),
-          keyValuePairs.slice(valueStart + 1).trim()
+          keyValuePairs.slice(valueStart + 1, i).trim()
         )
+        keyStart = i + 1
+        valueStart = 0
       }
-    } else if (Array.isArray(keyValuePairs)) {
-      for (const tags of keyValuePairs) {
-        add(carrier, tags)
-      }
-    } else {
-      Object.assign(carrier, keyValuePairs)
     }
-  } catch (error) {
-    log.error('Error adding tags', error)
+
+    if (keyValuePairs.at(-1) !== ',') {
+      valueStart ||= keyValuePairs.length
+      addNonEmpty(
+        carrier,
+        keyValuePairs.slice(keyStart, valueStart).trim(),
+        keyValuePairs.slice(valueStart + 1).trim()
+      )
+    }
+  } else if (Array.isArray(keyValuePairs)) {
+    for (const tags of keyValuePairs) {
+      add(carrier, tags)
+    }
+  } else {
+    Object.assign(carrier, keyValuePairs)
   }
 }
 

package/vendor/dist/@apm-js-collab/code-transformer/LICENSE

@@ -0,0 +1,28 @@
+
+Copyright (c) 2009-2011, Mozilla Foundation and contributors
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the names of the Mozilla Foundation nor the names of project
+  contributors may be used to endorse or promote products derived from this
+  software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.