fix: API Key Logged in Debug Output

Author: PROFeNoM

packages/dd-trace/src/llmobs/plugins/openai/constants.js

@@ -0,0 +1,16 @@
+'use strict'
+
+const INPUT_TYPE_IMAGE = 'input_image'
+const INPUT_TYPE_FILE = 'input_file'
+const INPUT_TYPE_TEXT = 'input_text'
+
+const IMAGE_FALLBACK = '[image]'
+const FILE_FALLBACK = '[file]'
+
+module.exports = {
+  INPUT_TYPE_IMAGE,
+  INPUT_TYPE_FILE,
+  INPUT_TYPE_TEXT,
+  IMAGE_FALLBACK,
+  FILE_FALLBACK
+}

packages/dd-trace/src/llmobs/writers/base.js

@@ -86,6 +86,13 @@ class BaseLLMObsWriter {
     return `${apiKey}:${site}`
   }
 
+  _getMaskedRoutingKey (routing) {
+    const apiKey = routing?.apiKey || this._config.apiKey || ''
+    const site = routing?.site || this._config.site || ''
+    const maskedKey = apiKey ? `****${apiKey.slice(-4)}` : ''
+    return `${maskedKey}:${site}`
+  }
+
   _getOrCreateBuffer (routingKey, routing) {
     if (!this._buffers.has(routingKey)) {
       this._buffers.set(routingKey, {
@@ -122,7 +129,7 @@ class BaseLLMObsWriter {
       return
     }
 
-    for (const [routingKey, buffer] of this._buffers) {
+    for (const [, buffer] of this._buffers) {
       if (buffer.events.length === 0) continue
 
       const events = buffer.events
@@ -133,7 +140,7 @@ class BaseLLMObsWriter {
       const options = this._getOptions(buffer.routing)
       const url = this._getUrlForRouting(buffer.routing)
 
-      log.debug('Encoded LLMObs payload for %s: %s', routingKey, payload)
+      log.debug('Encoded LLMObs payload for %s: %s', this._getMaskedRoutingKey(buffer.routing), payload)
 
       request(payload, options, (err, resp, code) => {
         parseResponseAndLog(err, code, events.length, url, this._eventType)