fix skipping of openai/embeddings

newrelic · Jul 9, 2024 · ee7c15a · ee7c15a
1 parent a5afa18
commit ee7c15a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/instrumentation/openai.js b/lib/instrumentation/openai.js
@@ -212,6 +212,7 @@ function instrumentStream({ agent, shim, request, response, segment }) {
   })
 }
 
+/* eslint-disable sonarjs/cognitive-complexity */
 module.exports = function initialize(agent, openai, moduleName, shim) {
   if (shouldSkipInstrumentation(agent.config, shim)) {
     shim.logger.debug(
@@ -297,6 +298,11 @@ module.exports = function initialize(agent, openai, moduleName, shim) {
     openai.Embeddings.prototype,
     'create',
     function wrapEmbeddingCreate(shim, embeddingCreate, name, args) {
+      if (shouldSkipInstrumentation(agent.config, shim) === true) {
+        shim.logger.debug('skipping sending of ai data')
+        return
+      }
+
       const [request] = args
       return new RecorderSpec({
         name: OPENAI.EMBEDDING,