Skip to content

Commit fb8f41e

Browse files
committed
Review fixes for aws-sdk-2.2:library
Automated code review of instrumentation/aws-sdk/aws-sdk-2.2/library.
1 parent 5f63164 commit fb8f41e

2 files changed

Lines changed: 11 additions & 15 deletions

File tree

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,8 @@ public AwsSdkTelemetryBuilder setRecordIndividualHttpError(boolean recordIndivid
9898
* #setUseConfiguredPropagatorForMessaging(boolean)}
9999
*/
100100
@CanIgnoreReturnValue
101-
AwsSdkTelemetryBuilder setUseXrayPropagator(boolean useMessagingPropagator) {
102-
this.useXrayPropagator = useMessagingPropagator;
101+
AwsSdkTelemetryBuilder setUseXrayPropagator(boolean useXrayPropagator) {
102+
this.useXrayPropagator = useXrayPropagator;
103103
return this;
104104
}
105105

instrumentation/aws-sdk/aws-sdk-2.2/library/src/testBedrockRuntime/java/io/opentelemetry/instrumentation/awssdk/v2_2/internal/Aws2BedrockRuntimeTest.java

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@
3333
import io.opentelemetry.instrumentation.testing.junit.LibraryInstrumentationExtension;
3434
import java.util.ArrayList;
3535
import java.util.List;
36-
import java.util.concurrent.ExecutionException;
3736
import java.util.function.Consumer;
3837
import org.junit.jupiter.api.BeforeAll;
3938
import org.junit.jupiter.api.Test;
@@ -403,8 +402,7 @@ void testConverseToolCallNoMessageContent() {
403402
}
404403

405404
@Test
406-
void testConverseToolCallStreamNoMessageContent()
407-
throws InterruptedException, ExecutionException {
405+
void testConverseToolCallStreamNoMessageContent() {
408406
BedrockRuntimeAsyncClientBuilder builder = BedrockRuntimeAsyncClient.builder();
409407
AwsSdkTelemetry telemetry =
410408
AwsSdkTelemetry.builder(testing.getOpenTelemetry())
@@ -472,7 +470,7 @@ void testConverseToolCallStreamNoMessageContent()
472470
.toolConfig(currentWeatherToolConfig())
473471
.build(),
474472
responseHandler)
475-
.get();
473+
.join();
476474

477475
if (currentToolArgs.length() > 0 && !responseChunksTools.isEmpty()) {
478476
JsonNode node = JsonNode.parser().parse(currentToolArgs.toString());
@@ -650,7 +648,7 @@ void testConverseToolCallStreamNoMessageContent()
650648
.toolConfig(currentWeatherToolConfig())
651649
.build(),
652650
responseHandler1)
653-
.get();
651+
.join();
654652

655653
assertThat(String.join("", responseChunks))
656654
.contains(
@@ -1191,8 +1189,7 @@ void testInvokeModelToolCallAmazonNovaNoMessageContent() {
11911189
}
11921190

11931191
@Test
1194-
void testInvokeModelWithResponseStreamToolCallAmazonNovaNoMessageContent()
1195-
throws InterruptedException, ExecutionException {
1192+
void testInvokeModelWithResponseStreamToolCallAmazonNovaNoMessageContent() {
11961193
BedrockRuntimeAsyncClientBuilder builder = BedrockRuntimeAsyncClient.builder();
11971194
AwsSdkTelemetry telemetry =
11981195
AwsSdkTelemetry.builder(testing.getOpenTelemetry())
@@ -1337,7 +1334,7 @@ public void accept(PayloadPart chunk) {
13371334
.build())
13381335
.build();
13391336

1340-
client.invokeModelWithResponseStream(request0, responseHandler0).get();
1337+
client.invokeModelWithResponseStream(request0, responseHandler0).join();
13411338

13421339
String seattleToolUseId0 = "";
13431340
String sanFranciscoToolUseId0 = "";
@@ -1557,7 +1554,7 @@ public void accept(PayloadPart chunk) {
15571554
.build())
15581555
.build();
15591556

1560-
client.invokeModelWithResponseStream(request1, responseHandler1).get();
1557+
client.invokeModelWithResponseStream(request1, responseHandler1).join();
15611558

15621559
assertThat(text.toString())
15631560
.contains(
@@ -2027,8 +2024,7 @@ void testInvokeModelToolCallAnthropicClaudeNoMessageContent() {
20272024
}
20282025

20292026
@Test
2030-
void testInvokeModelWithResponseStreamToolCallAnthropicClaudeNoMessageContent()
2031-
throws InterruptedException, ExecutionException {
2027+
void testInvokeModelWithResponseStreamToolCallAnthropicClaudeNoMessageContent() {
20322028
BedrockRuntimeAsyncClientBuilder builder = BedrockRuntimeAsyncClient.builder();
20332029
AwsSdkTelemetry telemetry =
20342030
AwsSdkTelemetry.builder(testing.getOpenTelemetry())
@@ -2176,7 +2172,7 @@ public void accept(PayloadPart chunk) {
21762172
.build())
21772173
.build();
21782174

2179-
client.invokeModelWithResponseStream(request0, responseHandler0).get();
2175+
client.invokeModelWithResponseStream(request0, responseHandler0).join();
21802176

21812177
String seattleToolUseId0 = "";
21822178
String sanFranciscoToolUseId0 = "";
@@ -2370,7 +2366,7 @@ public void accept(PayloadPart chunk) {
23702366
.build())
23712367
.build();
23722368

2373-
client.invokeModelWithResponseStream(request1, responseHandler1).get();
2369+
client.invokeModelWithResponseStream(request1, responseHandler1).join();
23742370

23752371
assertThat(text.toString())
23762372
.contains(

0 commit comments

Comments
 (0)