amend

Author: amychisholm03

lib/instrumentation/aws-sdk/v3/bedrock.js

@@ -215,10 +215,9 @@ function recordEmbeddingMessage({
     agent,
     segment,
     transaction,
-    requestModel: bedrockCommand?.modelId,
+    bedrockCommand,
     requestInput: prompt.content,
-    requestId: bedrockResponse?.requestId,
-    totalTokenCount: bedrockResponse?.totalTokenCount,
+    bedrockResponse,
     error: err !== null
   }))
 

test/versioned/aws-sdk-v3/bedrock-embeddings.test.js

@@ -93,8 +93,7 @@ test.afterEach(afterEach)
       const embedding = events.filter(([{ type }]) => type === 'LlmEmbedding')[0]
       const [segment] = tx.trace.getChildren(tx.trace.root.id)
       const expectedEmbedding = {
-        id: /[a-f0-9]{36}/,
-        appName: 'New Relic for Node.js tests',
+        id: /[a-f0-9]{32}/,
         request_id: 'eda0760a-c3f0-4fc1-9a1e-75559d642866',
         trace_id: tx.traceId,
         span_id: segment.id,
@@ -137,8 +136,7 @@ test.afterEach(afterEach)
       const embedding = events.filter(([{ type }]) => type === 'LlmEmbedding')[0]
       const [segment] = tx.trace.getChildren(tx.trace.root.id)
       const expectedEmbedding = {
-        id: /[a-f0-9]{36}/,
-        appName: 'New Relic for Node.js tests',
+        id: /[a-f0-9]{32}/,
         request_id: 'eda0760a-c3f0-4fc1-9a1e-75559d642866',
         trace_id: tx.traceId,
         span_id: segment.id,
@@ -184,7 +182,7 @@ test.afterEach(afterEach)
           'http.statusCode': 400,
           'error.message': expectedMsg,
           'error.code': expectedType,
-          embedding_id: /[a-f0-9]{36}/
+          embedding_id: /[a-f0-9]{32}/
         },
         agentAttributes: {
           spanId: /\w+/
@@ -202,8 +200,7 @@ test.afterEach(afterEach)
       const embedding = events.filter(([{ type }]) => type === 'LlmEmbedding')[0]
       const [segment] = tx.trace.getChildren(tx.trace.root.id)
       const expectedEmbedding = {
-        id: /[a-f0-9]{36}/,
-        appName: 'New Relic for Node.js tests',
+        id: /[a-f0-9]{32}/,
         request_id: '743dd35b-744b-4ddf-b5c6-c0f3de2e3142',
         trace_id: tx.traceId,
         span_id: segment.id,