Skip to content

Commit 71b265a

Browse files
author
Max Hniebergall
committed
Revert "remove elser ml node task settings which were unused"
This reverts commit bee840d. # Conflicts: # x-pack/plugin/inference/src/internalClusterTest/java/org/elasticsearch/xpack/inference/integration/ModelRegistryIT.java # x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/services/elasticsearch/ElserInternalServiceTests.java
1 parent 272452c commit 71b265a

File tree

7 files changed

+661
-5
lines changed

7 files changed

+661
-5
lines changed

x-pack/plugin/inference/src/internalClusterTest/java/org/elasticsearch/xpack/inference/integration/ModelRegistryIT.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@
2828
import org.elasticsearch.xpack.inference.InferencePlugin;
2929
import org.elasticsearch.xpack.inference.registry.ModelRegistry;
3030
import org.elasticsearch.xpack.inference.services.elasticsearch.ElserInternalServiceSettingsTests;
31+
import org.elasticsearch.xpack.inference.services.elasticsearch.ElserInternalServiceTests;
32+
import org.elasticsearch.xpack.inference.services.elasticsearch.ElserMlNodeTaskSettingsTests;
3133
import org.elasticsearch.xpack.inference.services.elser.ElserInternalModel;
3234
import org.elasticsearch.xpack.inference.services.elser.ElserInternalService;
3335
import org.junit.Before;

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/InferenceNamedWriteablesProvider.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
import org.elasticsearch.xpack.inference.services.elasticsearch.ElasticsearchInternalServiceSettings;
6464
import org.elasticsearch.xpack.inference.services.elasticsearch.MultilingualE5SmallInternalServiceSettings;
6565
import org.elasticsearch.xpack.inference.services.elser.ElserInternalServiceSettings;
66+
import org.elasticsearch.xpack.inference.services.elser.ElserMlNodeTaskSettings;
6667
import org.elasticsearch.xpack.inference.services.googleaistudio.completion.GoogleAiStudioCompletionServiceSettings;
6768
import org.elasticsearch.xpack.inference.services.googleaistudio.embeddings.GoogleAiStudioEmbeddingsServiceSettings;
6869
import org.elasticsearch.xpack.inference.services.googlevertexai.GoogleVertexAiSecretSettings;

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

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,15 +10,22 @@
1010
import org.elasticsearch.ResourceNotFoundException;
1111
import org.elasticsearch.action.ActionListener;
1212
import org.elasticsearch.inference.Model;
13-
import org.elasticsearch.inference.TaskSettings;
1413
import org.elasticsearch.inference.TaskType;
1514
import org.elasticsearch.xpack.core.ml.action.CreateTrainedModelAssignmentAction;
1615
import org.elasticsearch.xpack.core.ml.utils.ExceptionsHelper;
16+
import org.elasticsearch.xpack.inference.services.elser.ElserInternalServiceSettings;
17+
import org.elasticsearch.xpack.inference.services.elser.ElserMlNodeTaskSettings;
1718

1819
public class ElserInternalModel extends ElasticsearchInternalModel {
1920

20-
public ElserInternalModel(String inferenceEntityId, TaskType taskType, String service, ElserInternalServiceSettings serviceSettings) {
21-
super(inferenceEntityId, taskType, service, serviceSettings);
21+
public ElserInternalModel(
22+
String inferenceEntityId,
23+
TaskType taskType,
24+
String service,
25+
ElserInternalServiceSettings serviceSettings,
26+
ElserMlNodeTaskSettings taskSettings
27+
) {
28+
super(inferenceEntityId, taskType, service, serviceSettings, taskSettings);
2229
}
2330

2431
@Override
@@ -27,8 +34,8 @@ public ElserInternalServiceSettings getServiceSettings() {
2734
}
2835

2936
@Override
30-
public TaskSettings getTaskSettings() {
31-
return super.getTaskSettings();
37+
public ElserMlNodeTaskSettings getTaskSettings() {
38+
return (ElserMlNodeTaskSettings) super.getTaskSettings();
3239
}
3340

3441
@Override
Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
/*
2+
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
3+
* or more contributor license agreements. Licensed under the Elastic License
4+
* 2.0; you may not use this file except in compliance with the Elastic License
5+
* 2.0.
6+
*/
7+
8+
package org.elasticsearch.xpack.inference.services.elser;
9+
10+
import org.elasticsearch.TransportVersion;
11+
import org.elasticsearch.TransportVersions;
12+
import org.elasticsearch.common.io.stream.StreamInput;
13+
import org.elasticsearch.common.io.stream.StreamOutput;
14+
import org.elasticsearch.inference.TaskSettings;
15+
import org.elasticsearch.xcontent.XContentBuilder;
16+
17+
import java.io.IOException;
18+
import java.util.Objects;
19+
20+
public class ElserMlNodeTaskSettings implements TaskSettings {
21+
22+
public static final String NAME = "elser_mlnode_task_settings";
23+
24+
public static ElserMlNodeTaskSettings DEFAULT = new ElserMlNodeTaskSettings();
25+
26+
public ElserMlNodeTaskSettings() {}
27+
28+
public ElserMlNodeTaskSettings(StreamInput in) {}
29+
30+
@Override
31+
public XContentBuilder toXContent(XContentBuilder builder, Params params) throws IOException {
32+
builder.startObject();
33+
builder.endObject();
34+
return builder;
35+
}
36+
37+
@Override
38+
public String getWriteableName() {
39+
return NAME;
40+
}
41+
42+
@Override
43+
public TransportVersion getMinimalSupportedVersion() {
44+
return TransportVersions.V_8_11_X;
45+
}
46+
47+
@Override
48+
public void writeTo(StreamOutput out) throws IOException {}
49+
50+
@Override
51+
public boolean equals(Object obj) {
52+
if (this == obj) return true;
53+
if (obj == null || getClass() != obj.getClass()) return false;
54+
return true;
55+
}
56+
57+
@Override
58+
public int hashCode() {
59+
// TODO Class has no members all instances are equivalent
60+
// Return the hash of NAME to make the serialization tests pass
61+
return Objects.hash(NAME);
62+
}
63+
}

x-pack/plugin/inference/src/test/java/org/elasticsearch/xpack/inference/ModelConfigurationsTests.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import org.elasticsearch.inference.TaskType;
1616
import org.elasticsearch.test.AbstractWireSerializingTestCase;
1717
import org.elasticsearch.xpack.inference.services.elasticsearch.ElserInternalServiceSettingsTests;
18+
import org.elasticsearch.xpack.inference.services.elser.ElserMlNodeTaskSettings;
1819

1920
public class ModelConfigurationsTests extends AbstractWireSerializingTestCase<ModelConfigurations> {
2021

0 commit comments

Comments
 (0)