wip

DataDog · Oct 16, 2024 · 9d68a0c · 9d68a0c
1 parent ad28001
commit 9d68a0c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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', () => {