amend

Author: amychisholm03

test/unit/llm-events/google-genai/embedding.test.js

@@ -94,12 +94,13 @@ 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 = api.shim.getActiveSegment()
       const embeddingEvent = new LlmEmbedding({
         agent,
         segment,
+        transaction: tx,
         request: req,
         response: res,
         error: true
@@ -119,11 +120,12 @@ test('respects record_content', (t, end) => {
   agent.config.ai_monitoring.record_content.enabled = false
 
   const api = helper.getAgentApi()
-  helper.runInTransaction(agent, () => {
+  helper.runInTransaction(agent, (tx) => {
     const segment = api.shim.getActiveSegment()
     const embeddingEvent = new LlmEmbedding({
       agent,
       segment,
+      transaction: tx,
       request: req,
       response: res
     })