Add tx to unit embedding test

Author: amychisholm03

test/unit/llm-events/openai/embedding.test.js

@@ -92,13 +92,14 @@ test('should set error to true', (t, end) => {
   }
 
   const api = helper.getAgentApi()
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     api.startSegment('fakeSegment', false, () => {
       const segment = agent.tracer.getSegment()
       segment.end()
       const embeddingEvent = new LlmEmbedding({
         agent,
         segment,
+        transaction: tx,
         request: req,
         response: res,
         error: true
@@ -123,11 +124,12 @@ test('respects record_content by not recording content when set to false', (t, e
   const api = helper.getAgentApi()
   api.setLlmTokenCountCallback(cb)
 
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     const segment = agent.tracer.getSegment()
     const embeddingEvent = new LlmEmbedding({
       agent,
       segment,
+      transaction: tx,
       request: req,
       response: res
     })
@@ -150,11 +152,12 @@ test('respects record_content by recording content when true', (t, end) => {
 
   const api = helper.getAgentApi()
   api.setLlmTokenCountCallback(cb)
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     const segment = agent.tracer.getSegment()
     const embeddingEvent = new LlmEmbedding({
       agent,
       segment,
+      transaction: tx,
       request: req,
       response: res
     })
@@ -176,11 +179,12 @@ test('does not calculate tokens when no content exists', (t, end) => {
 
   const api = helper.getAgentApi()
   api.setLlmTokenCountCallback(cb)
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     const segment = agent.tracer.getSegment()
     const embeddingEvent = new LlmEmbedding({
       agent,
       segment,
+      transaction: tx,
       request: req,
       response: res
     })
@@ -197,13 +201,14 @@ test('assigns total_tokens from response', (t, end) => {
     model: 'gpt-3.5-turbo-0613'
   }
 
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     const segment = agent.tracer.getSegment()
     const embeddingEvent = new LlmEmbedding({
       agent,
       segment,
       request: req,
-      response: res
+      response: res,
+      transaction: tx
     })
     assert.equal(embeddingEvent['response.usage.total_tokens'], 30)
     end()