Skip to content

Commit 0a93834

Browse files
committed
feat: create new enum for bedrock agent runtime requests
1 parent b6abeed commit 0a93834

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

instrumentation/aws-sdk/aws-sdk-2.2/library/src/main/java/io/opentelemetry/instrumentation/awssdk/v2_2/AwsSdkRequest.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCK;
99
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCKAGENTOPERATION;
10+
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCKAGENTRUNTIMEOPERATION;
1011
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCKDATASOURCEOPERATION;
1112
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCKKNOWLEDGEBASEOPERATION;
1213
import static io.opentelemetry.instrumentation.awssdk.v2_2.AwsSdkRequestType.BEDROCKRUNTIME;
@@ -42,7 +43,7 @@ enum AwsSdkRequest {
4243
SqsRequest(SQS, "SqsRequest"),
4344
KinesisRequest(KINESIS, "KinesisRequest"),
4445
BedrockRequest(BEDROCK, "BedrockRequest"),
45-
BedrockAgentRuntimeRequest(BEDROCKAGENTOPERATION, "BedrockAgentRuntimeRequest"),
46+
BedrockAgentRuntimeRequest(BEDROCKAGENTRUNTIMEOPERATION, "BedrockAgentRuntimeRequest"),
4647
BedrockRuntimeRequest(BEDROCKRUNTIME, "BedrockRuntimeRequest"),
4748
// BedrockAgent API based requests. We only support operations that are related to
4849
// Agent/DataSources/KnowledgeBases

instrumentation/aws-sdk/aws-sdk-2.2/library/src/main/java/io/opentelemetry/instrumentation/awssdk/v2_2/AwsSdkRequestType.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,8 @@ enum AwsSdkRequestType {
3939
request(AWS_GUARDRAIL_ID.getKey(), "guardrailIdentifier"),
4040
response(AWS_GUARDRAIL_ARN.getKey(), "guardrailArn")),
4141
BEDROCKAGENTOPERATION(
42+
request(AWS_AGENT_ID.getKey(), "agentId"), response(AWS_AGENT_ID.getKey(), "agentId")),
43+
BEDROCKAGENTRUNTIMEOPERATION(
4244
request(AWS_AGENT_ID.getKey(), "agentId"),
4345
response(AWS_AGENT_ID.getKey(), "agentId"),
4446
request(AWS_KNOWLEDGE_BASE_ID.getKey(), "knowledgeBaseId"),

0 commit comments

Comments
 (0)