chore(llmobs): update config type to generated types (#7965)

Author: sabrenner

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

@@ -43,9 +43,12 @@ let spanWriter
 /** @type {LLMObsEvalMetricsWriter | null} */
 let evalWriter
 
-/** @type {import('../config')} */
+/** @type {import('../config/config-base')} */
 let globalTracerConfig
 
+/**
+ * @param {@type import('../config/config-base')} config
+ */
 function enable (config) {
   globalTracerConfig = config
 

packages/dd-trace/src/llmobs/sdk.js

@@ -37,7 +37,9 @@ class LLMObs extends NoopLLMObs {
   constructor (tracer, llmobsModule, config) {
     super(tracer)
 
+    /** @type {import('../config/config-base')} */
     this._config = config
+
     this._llmobsModule = llmobsModule
     this._tagger = new LLMObsTagger(config)
   }

packages/dd-trace/src/llmobs/span_processor.js

@@ -49,7 +49,7 @@ class LLMObservabilitySpan {
 }
 
 class LLMObsSpanProcessor {
-  /** @type {import('../config')} */
+  /** @type {import('../config/config-base')} */
   #config
 
   /** @type {((span: LLMObservabilitySpan) => LLMObservabilitySpan | null) | null} */

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

@@ -47,8 +47,11 @@ const { storage } = require('./storage')
 const registry = new WeakMap()
 
 class LLMObsTagger {
+  /** @type {import('../config/config-base')} */
+  #config
+
   constructor (config, softFail = false) {
-    this._config = config
+    this.#config = config
 
     this.softFail = softFail
   }
@@ -72,15 +75,15 @@ class LLMObsTagger {
     integration,
     _decorator,
   } = {}) {
-    if (!this._config.llmobs.enabled) return
+    if (!this.#config.llmobs.enabled) return
     if (!kind) return // do not register it in the map if it doesn't have an llmobs span kind
 
     const spanMlApp =
       mlApp ||
       registry.get(parent)?.[ML_APP] ||
       span.context()._trace.tags[PROPAGATED_ML_APP_KEY] ||
-      this._config.llmobs.mlApp ||
-      this._config.service // this should always have a default
+      this.#config.llmobs.mlApp ||
+      this.#config.service // this should always have a default
 
     if (!spanMlApp) {
       throw new Error(
@@ -624,7 +627,7 @@ class LLMObsTagger {
   }
 
   _register (span) {
-    if (!this._config.llmobs.enabled) return
+    if (!this.#config.llmobs.enabled) return
     if (registry.has(span)) {
       this.#handleFailure(`LLMObs Span "${span._name}" already registered.`)
       return
@@ -634,7 +637,7 @@ class LLMObsTagger {
   }
 
   _setTag (span, key, value) {
-    if (!this._config.llmobs.enabled) return
+    if (!this.#config.llmobs.enabled) return
     if (!registry.has(span)) {
       this.#handleFailure(`Span "${span._name}" must be an LLMObs generated span.`)
       return

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

@@ -45,7 +45,9 @@ class BaseLLMObsWriter {
     /** @type {LLMObsBuffer} */
     this._buffer = new LLMObsBuffer({ events: [], size: 0, isDefault: true })
 
+    /** @type {import('../../config/config-base')} */
     this._config = config
+
     this._endpoint = endpoint
     this._baseEndpoint = endpoint // should not be unset
     this._intake = intake

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

@@ -6,6 +6,9 @@ const telemetry = require('../telemetry')
 const { fetchAgentInfo } = require('../../agent/info')
 const { getAgentUrl } = require('../../agent/url')
 
+/**
+ * @param {import('../../config/config-base')} config
+ */
 function setAgentStrategy (config, setWritersAgentlessValue) {
   const agentlessEnabled = config.llmobs.agentlessEnabled
 

packages/dd-trace/src/plugins/plugin.js

@@ -72,14 +72,16 @@ module.exports = class Plugin {
    * Create a new plugin instance.
    *
    * @param {object} tracer Tracer instance or wrapper containing it under `_tracer`.
-   * @param {object} tracerConfig Global tracer configuration object.
+   * @param {import('../config/config-base')} tracerConfig Global tracer configuration object.
    */
   constructor (tracer, tracerConfig) {
     this._subscriptions = []
     this._bindings = []
     this._enabled = false
     this._tracer = tracer
     this.config = {} // plugin-specific configuration, unset until .configure() is called
+
+    /** @type {import('../config/config-base')} */
     this._tracerConfig = tracerConfig // global tracer configuration
   }