Skip to content

Commit dedaebe

Browse files
committed
added bedrock agent id cfa and sqs queue url
1 parent 46afde1 commit dedaebe

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

aws-distro-opentelemetry-node-autoinstrumentation/src/aws-metric-attribute-generator.ts

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -420,15 +420,21 @@ export class AwsMetricAttributeGenerator implements MetricAttributeGenerator {
420420
span.attributes[AWS_ATTRIBUTE_KEYS.AWS_SQS_QUEUE_NAME]
421421
);
422422
} else if (AwsSpanProcessingUtil.isKeyPresent(span, AWS_ATTRIBUTE_KEYS.AWS_SQS_QUEUE_URL)) {
423-
remoteResourceType = NORMALIZED_SQS_SERVICE_NAME + '::Queue';
424-
remoteResourceIdentifier = SqsUrlParser.getQueueName(
425-
AwsMetricAttributeGenerator.escapeDelimiters(span.attributes[AWS_ATTRIBUTE_KEYS.AWS_SQS_QUEUE_URL])
423+
const sqsQueueUrl = AwsMetricAttributeGenerator.escapeDelimiters(
424+
span.attributes[AWS_ATTRIBUTE_KEYS.AWS_SQS_QUEUE_URL]
426425
);
426+
427+
remoteResourceType = NORMALIZED_SQS_SERVICE_NAME + '::Queue';
428+
remoteResourceIdentifier = SqsUrlParser.getQueueName(sqsQueueUrl);
429+
cloudFormationIdentifier = sqsQueueUrl;
427430
} else if (AwsSpanProcessingUtil.isKeyPresent(span, AWS_ATTRIBUTE_KEYS.AWS_BEDROCK_AGENT_ID)) {
428431
remoteResourceType = NORMALIZED_BEDROCK_SERVICE_NAME + '::Agent';
429432
remoteResourceIdentifier = AwsMetricAttributeGenerator.escapeDelimiters(
430433
span.attributes[AWS_ATTRIBUTE_KEYS.AWS_BEDROCK_AGENT_ID]
431434
);
435+
cloudFormationIdentifier = `${AwsMetricAttributeGenerator.escapeDelimiters(
436+
span.attributes[AWS_ATTRIBUTE_KEYS.AWS_BEDROCK_KNOWLEDGE_BASE_ID]
437+
)}|${remoteResourceIdentifier}`;
432438
} else if (AwsSpanProcessingUtil.isKeyPresent(span, AWS_ATTRIBUTE_KEYS.AWS_BEDROCK_DATA_SOURCE_ID)) {
433439
remoteResourceType = NORMALIZED_BEDROCK_SERVICE_NAME + '::DataSource';
434440
remoteResourceIdentifier = AwsMetricAttributeGenerator.escapeDelimiters(

0 commit comments

Comments
 (0)