diff --git a/packages/dd-trace/src/llmobs/index.js b/packages/dd-trace/src/llmobs/index.js index 1d75d0dd89..b0c63639c0 100644 --- a/packages/dd-trace/src/llmobs/index.js +++ b/packages/dd-trace/src/llmobs/index.js @@ -7,6 +7,7 @@ const { storage } = require('../../../datadog-core') const { channel } = require('dc-polyfill') const injectCh = channel('dd-trace:span:inject') +// this is where we will enable and disable LLMObs plugins function enable (config) { injectCh.subscribe(handleLLMObsParentIdInjection) } diff --git a/packages/dd-trace/test/llmobs/sdk/integration.spec.js b/packages/dd-trace/test/llmobs/sdk/integration.spec.js index ce415e6475..a4cebf82c9 100644 --- a/packages/dd-trace/test/llmobs/sdk/integration.spec.js +++ b/packages/dd-trace/test/llmobs/sdk/integration.spec.js @@ -60,6 +60,9 @@ describe('end to end sdk integration tests', () => { EvalMetricsWriter.prototype.append.resetHistory() process.removeAllListeners('beforeExit') + + llmobs.disable() + llmobs.enable({ mlApp: 'test', apiKey: 'test' }) }) after(() => { @@ -242,10 +245,6 @@ describe('end to end sdk integration tests', () => { ] check(expected, llmobsSpans) - - // restore original mlApp - llmobs.disable() - llmobs.enable({ mlApp: 'test' }) }) it('submits evaluations', () => {