Skip to content

Commit 141c154

Browse files
author
elasticsearchmachine
committed
[CI] Auto commit changes from spotless
1 parent a06fd59 commit 141c154

File tree

7 files changed

+58
-64
lines changed

7 files changed

+58
-64
lines changed

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/AmazonBedrockService.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434
import org.elasticsearch.xpack.inference.chunking.ChunkingSettingsBuilder;
3535
import org.elasticsearch.xpack.inference.chunking.EmbeddingRequestChunker;
3636
import org.elasticsearch.xpack.inference.common.amazon.AwsSecretSettings;
37-
import org.elasticsearch.xpack.inference.external.action.SenderExecutableAction;
3837
import org.elasticsearch.xpack.inference.external.http.sender.EmbeddingsInput;
3938
import org.elasticsearch.xpack.inference.external.http.sender.HttpRequestSender;
4039
import org.elasticsearch.xpack.inference.external.http.sender.InferenceInputs;
@@ -82,7 +81,10 @@ public class AmazonBedrockService extends SenderService {
8281
private final Sender amazonBedrockSender;
8382

8483
private static final EnumSet<TaskType> supportedTaskTypes = EnumSet.of(
85-
TaskType.TEXT_EMBEDDING, TaskType.COMPLETION, TaskType.CHAT_COMPLETION);
84+
TaskType.TEXT_EMBEDDING,
85+
TaskType.COMPLETION,
86+
TaskType.CHAT_COMPLETION
87+
);
8688

8789
private static final EnumSet<InputType> VALID_INPUT_TYPE_VALUES = EnumSet.of(
8890
InputType.INGEST,

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/completion/AmazonBedrockChatCompletionModel.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ public static AmazonBedrockChatCompletionModel of(AmazonBedrockChatCompletionMod
3434
return new AmazonBedrockChatCompletionModel(completionModel, taskSettingsToUse);
3535
}
3636

37-
3837
public static AmazonBedrockChatCompletionModel of(AmazonBedrockChatCompletionModel model, UnifiedCompletionRequest request) {
3938
if (request.model() == null) {
4039
return model;

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/request/completion/AmazonBedrockConverseUtils.java

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,13 @@
77

88
package org.elasticsearch.xpack.inference.services.amazonbedrock.request.completion;
99

10-
import org.elasticsearch.inference.UnifiedCompletionRequest;
11-
1210
import software.amazon.awssdk.services.bedrockruntime.model.ContentBlock;
1311
import software.amazon.awssdk.services.bedrockruntime.model.InferenceConfiguration;
1412
import software.amazon.awssdk.services.bedrockruntime.model.Message;
1513

1614
import org.elasticsearch.core.Nullable;
1715
import org.elasticsearch.core.Strings;
16+
import org.elasticsearch.inference.UnifiedCompletionRequest;
1817

1918
import java.util.List;
2019
import java.util.Optional;
@@ -32,10 +31,12 @@ public static List<Message> getConverseMessageList(List<String> texts) {
3231

3332
public static List<Message> getUnifiedConverseMessageList(List<UnifiedCompletionRequest.Message> messages) {
3433
return messages.stream()
35-
.map(message -> Message.builder().role(message.role())
36-
.content(ContentBlock.builder()
37-
.text(message.content().toString())
38-
.build()).build())
34+
.map(
35+
message -> Message.builder()
36+
.role(message.role())
37+
.content(ContentBlock.builder().text(message.content().toString()).build())
38+
.build()
39+
)
3940
.toList();
4041
}
4142

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/request/completion/AmazonBedrockUnifiedChatCompletionEntityFactory.java

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -14,18 +14,23 @@
1414

1515
public class AmazonBedrockUnifiedChatCompletionEntityFactory {
1616
public static AmazonBedrockUnifiedConverseRequestEntity createEntity(
17-
AmazonBedrockChatCompletionModel model, UnifiedCompletionRequest request) {
17+
AmazonBedrockChatCompletionModel model,
18+
UnifiedCompletionRequest request
19+
) {
1820
Objects.requireNonNull(model);
1921
Objects.requireNonNull(request);
2022
var serviceSettings = model.getServiceSettings();
2123

22-
var messages = request.messages().stream()
23-
.map(message -> new UnifiedCompletionRequest.Message(
24-
message.content(),
25-
toBedrockRole(message.role()),
26-
message.toolCallId(),
27-
message.toolCalls()
28-
))
24+
var messages = request.messages()
25+
.stream()
26+
.map(
27+
message -> new UnifiedCompletionRequest.Message(
28+
message.content(),
29+
toBedrockRole(message.role()),
30+
message.toolCallId(),
31+
message.toolCalls()
32+
)
33+
)
2934
.toList();
3035

3136
switch (serviceSettings.provider()) {

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/request/completion/AmazonBedrockUnifiedChatCompletionRequest.java

Lines changed: 26 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -7,15 +7,6 @@
77

88
package org.elasticsearch.xpack.inference.services.amazonbedrock.request.completion;
99

10-
import org.elasticsearch.core.Nullable;
11-
import org.elasticsearch.core.TimeValue;
12-
import org.elasticsearch.inference.InferenceServiceResults;
13-
import org.elasticsearch.inference.TaskType;
14-
import org.elasticsearch.xpack.inference.services.amazonbedrock.client.AmazonBedrockBaseClient;
15-
import org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionModel;
16-
import org.elasticsearch.xpack.inference.services.amazonbedrock.request.AmazonBedrockRequest;
17-
import org.elasticsearch.xpack.inference.services.amazonbedrock.response.completion.AmazonBedrockChatCompletionResponseListener;
18-
1910
import software.amazon.awssdk.core.document.Document;
2011
import software.amazon.awssdk.services.bedrockruntime.model.ConverseStreamRequest;
2112
import software.amazon.awssdk.services.bedrockruntime.model.SpecificToolChoice;
@@ -25,6 +16,15 @@
2516
import software.amazon.awssdk.services.bedrockruntime.model.ToolInputSchema;
2617
import software.amazon.awssdk.services.bedrockruntime.model.ToolSpecification;
2718

19+
import org.elasticsearch.core.Nullable;
20+
import org.elasticsearch.core.TimeValue;
21+
import org.elasticsearch.inference.InferenceServiceResults;
22+
import org.elasticsearch.inference.TaskType;
23+
import org.elasticsearch.xpack.inference.services.amazonbedrock.client.AmazonBedrockBaseClient;
24+
import org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionModel;
25+
import org.elasticsearch.xpack.inference.services.amazonbedrock.request.AmazonBedrockRequest;
26+
import org.elasticsearch.xpack.inference.services.amazonbedrock.response.completion.AmazonBedrockChatCompletionResponseListener;
27+
2828
import java.util.Objects;
2929
import java.util.concurrent.Flow;
3030

@@ -57,24 +57,22 @@ public Flow.Publisher<? extends InferenceServiceResults.Result> executeStreamCha
5757

5858
if (requestEntity.tools() != null) {
5959
requestEntity.tools().forEach(tool -> {
60-
converseStreamRequest
61-
.toolConfig(ToolConfiguration.builder()
62-
.tools(Tool.builder()
63-
.toolSpec(ToolSpecification.builder()
64-
.name(tool.function().name())
65-
.description(tool.function().description())
66-
.inputSchema(
67-
ToolInputSchema.builder()
68-
.json(Document.fromString(""))
69-
.build())
70-
.build())
71-
.build())
72-
.toolChoice(ToolChoice.builder()
73-
.tool(SpecificToolChoice.builder()
74-
.name("")
75-
.build())
76-
.build())
77-
.build());
60+
converseStreamRequest.toolConfig(
61+
ToolConfiguration.builder()
62+
.tools(
63+
Tool.builder()
64+
.toolSpec(
65+
ToolSpecification.builder()
66+
.name(tool.function().name())
67+
.description(tool.function().description())
68+
.inputSchema(ToolInputSchema.builder().json(Document.fromString("")).build())
69+
.build()
70+
)
71+
.build()
72+
)
73+
.toolChoice(ToolChoice.builder().tool(SpecificToolChoice.builder().name("").build()).build())
74+
.build()
75+
);
7876
});
7977
}
8078

@@ -84,8 +82,7 @@ public Flow.Publisher<? extends InferenceServiceResults.Result> executeStreamCha
8482
}
8583

8684
@Override
87-
protected void executeRequest(AmazonBedrockBaseClient client) {
88-
}
85+
protected void executeRequest(AmazonBedrockBaseClient client) {}
8986

9087
@Override
9188
public TaskType taskType() {

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/amazonbedrock/request/completion/AmazonBedrockUnifiedConverseRequestEntity.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
import java.util.List;
1414

15-
public record AmazonBedrockUnifiedConverseRequestEntity (
15+
public record AmazonBedrockUnifiedConverseRequestEntity(
1616
List<UnifiedCompletionRequest.Message> messages,
1717
@Nullable String model,
1818
@Nullable Long maxCompletionTokens,
@@ -22,4 +22,3 @@ public record AmazonBedrockUnifiedConverseRequestEntity (
2222
@Nullable List<UnifiedCompletionRequest.Tool> tools,
2323
@Nullable Float topP
2424
) {}
25-

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/amazonbedrock/AmazonBedrockServiceTests.java

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -7,18 +7,6 @@
77

88
package org.elasticsearch.xpack.inference.services.amazonbedrock;
99

10-
import org.elasticsearch.inference.UnifiedCompletionRequest;
11-
import org.elasticsearch.test.http.MockResponse;
12-
13-
import org.elasticsearch.test.http.MockWebServer;
14-
import org.elasticsearch.xpack.core.inference.results.UnifiedChatCompletionException;
15-
import org.elasticsearch.xpack.inference.external.http.HttpClientManager;
16-
import org.elasticsearch.xpack.inference.external.http.sender.HttpRequestSenderTests;
17-
import org.elasticsearch.xpack.inference.logging.ThrottlerManager;
18-
import org.elasticsearch.xpack.inference.services.InferenceEventsAssertion;
19-
20-
import org.elasticsearch.xpack.inference.services.amazonbedrock.client.AmazonBedrockRequestSenderTests;
21-
2210
import software.amazon.awssdk.services.bedrockruntime.model.BedrockRuntimeException;
2311

2412
import org.elasticsearch.ElasticsearchException;
@@ -43,7 +31,10 @@
4331
import org.elasticsearch.inference.ModelSecrets;
4432
import org.elasticsearch.inference.SimilarityMeasure;
4533
import org.elasticsearch.inference.TaskType;
34+
import org.elasticsearch.inference.UnifiedCompletionRequest;
4635
import org.elasticsearch.test.ESTestCase;
36+
import org.elasticsearch.test.http.MockResponse;
37+
import org.elasticsearch.test.http.MockWebServer;
4738
import org.elasticsearch.threadpool.ThreadPool;
4839
import org.elasticsearch.xcontent.ToXContent;
4940
import org.elasticsearch.xcontent.XContentType;
@@ -53,10 +44,14 @@
5344
import org.elasticsearch.xpack.core.inference.results.TextEmbeddingFloatResults;
5445
import org.elasticsearch.xpack.inference.Utils;
5546
import org.elasticsearch.xpack.inference.common.amazon.AwsSecretSettings;
47+
import org.elasticsearch.xpack.inference.external.http.HttpClientManager;
5648
import org.elasticsearch.xpack.inference.external.http.sender.HttpRequestSender;
5749
import org.elasticsearch.xpack.inference.external.http.sender.Sender;
50+
import org.elasticsearch.xpack.inference.logging.ThrottlerManager;
51+
import org.elasticsearch.xpack.inference.services.InferenceEventsAssertion;
5852
import org.elasticsearch.xpack.inference.services.ServiceComponentsTests;
5953
import org.elasticsearch.xpack.inference.services.amazonbedrock.client.AmazonBedrockMockRequestSender;
54+
import org.elasticsearch.xpack.inference.services.amazonbedrock.client.AmazonBedrockRequestSenderTests;
6055
import org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionModel;
6156
import org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionModelTests;
6257
import org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionServiceSettings;
@@ -89,19 +84,15 @@
8984
import static org.elasticsearch.xpack.inference.chunking.ChunkingSettingsTests.createRandomChunkingSettings;
9085
import static org.elasticsearch.xpack.inference.chunking.ChunkingSettingsTests.createRandomChunkingSettingsMap;
9186
import static org.elasticsearch.xpack.inference.common.amazon.AwsSecretSettingsTests.getAmazonBedrockSecretSettingsMap;
92-
import static org.elasticsearch.xpack.inference.external.http.Utils.getUrl;
9387
import static org.elasticsearch.xpack.inference.services.ServiceComponentsTests.createWithEmptySettings;
9488
import static org.elasticsearch.xpack.inference.services.amazonbedrock.AmazonBedrockProviderCapabilities.getProviderDefaultSimilarityMeasure;
95-
import static org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionModelTests.createModel;
9689
import static org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionServiceSettingsTests.createChatCompletionRequestSettingsMap;
9790
import static org.elasticsearch.xpack.inference.services.amazonbedrock.completion.AmazonBedrockChatCompletionTaskSettingsTests.getChatCompletionTaskSettingsMap;
9891
import static org.elasticsearch.xpack.inference.services.amazonbedrock.embeddings.AmazonBedrockEmbeddingsServiceSettingsTests.createEmbeddingsRequestSettingsMap;
9992
import static org.hamcrest.CoreMatchers.is;
10093
import static org.hamcrest.Matchers.containsString;
101-
import static org.hamcrest.Matchers.equalTo;
10294
import static org.hamcrest.Matchers.hasSize;
10395
import static org.hamcrest.Matchers.instanceOf;
104-
import static org.hamcrest.Matchers.isA;
10596
import static org.mockito.Mockito.mock;
10697
import static org.mockito.Mockito.times;
10798
import static org.mockito.Mockito.verify;

0 commit comments

Comments
 (0)