Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,12 @@ public void visitLineNumber(int line, Label start) {
lineNumber = line;
}

@Override
public void visitLabel(Label label) {
// asm uses many debug labels that we do not want to consider
// so we ignore labels so they do not become part of the instructions list
}

@Override
public void visitMethodInsn(int opcode, String owner, String name, String descriptor, boolean isInterface) {
if (owner.equals(TRANSPORT_VERSION_SET_CLASS) && name.equals(TRANSPORT_VERSION_SET_METHOD_NAME)) {
Expand Down
7 changes: 0 additions & 7 deletions server/src/main/java/org/elasticsearch/TransportVersions.java
Original file line number Diff line number Diff line change
Expand Up @@ -214,13 +214,6 @@ static TransportVersion def(int id) {
public static final TransportVersion ML_INFERENCE_HUGGING_FACE_CHAT_COMPLETION_ADDED_8_19 = def(8_841_0_31);
public static final TransportVersion FIELD_CAPS_ADD_CLUSTER_ALIAS = def(8_841_0_32);
public static final TransportVersion INCLUDE_INDEX_MODE_IN_GET_DATA_STREAM_BACKPORT_8_19 = def(8_841_0_33);
public static final TransportVersion ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME_8_19 = def(8_841_0_34);
public static final TransportVersion RERANKER_FAILURES_ALLOWED_8_19 = def(8_841_0_35);
public static final TransportVersion ML_INFERENCE_HUGGING_FACE_RERANK_ADDED_8_19 = def(8_841_0_36);
public static final TransportVersion ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION_8_19 = def(8_841_0_37);
public static final TransportVersion ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED_8_19 = def(8_841_0_38);
public static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED_8_19 = def(8_841_0_39);
public static final TransportVersion IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19 = def(8_841_0_40);

/*
* STOP! READ THIS FIRST! No, really,
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9077000,8841034
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9087000,8841040
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9084000,8841039
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9080000,8841036
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9082000,8841037
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9083000,8841038
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
9013000,8841035
Original file line number Diff line number Diff line change
Expand Up @@ -467,7 +467,7 @@ public void testTransportVersionsLocked() {
"TransportVersions.java is locked. Generate transport versions with TransportVersion.fromName "
+ "and generateTransportVersion gradle task",
versions.get(versions.size() - 1).id(),
equalTo(8_841_0_40)
equalTo(8_841_0_33)
);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -295,6 +295,10 @@ public static class Status implements Operator.Status {
Status::new
);

private static final TransportVersion ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME = TransportVersion.fromName(
"esql_hash_operator_status_output_time"
);

/**
* Nanoseconds this operator has spent hashing grouping keys.
*/
Expand Down Expand Up @@ -348,7 +352,7 @@ protected Status(StreamInput in) throws IOException {
rowsReceived = 0;
rowsEmitted = 0;
}
if (in.getTransportVersion().onOrAfter(TransportVersions.ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME_8_19)) {
if (in.getTransportVersion().supports(ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME)) {
emitNanos = in.readVLong();
} else {
emitNanos = 0;
Expand All @@ -365,7 +369,7 @@ public void writeTo(StreamOutput out) throws IOException {
out.writeVLong(rowsReceived);
out.writeVLong(rowsEmitted);
}
if (out.getTransportVersion().onOrAfter(TransportVersions.ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME_8_19)) {
if (out.getTransportVersion().supports(ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME)) {
out.writeVLong(emitNanos);
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
*/
package org.elasticsearch.xpack.idp.action;

import org.elasticsearch.TransportVersions;
import org.elasticsearch.TransportVersion;
import org.elasticsearch.action.ActionRequestValidationException;
import org.elasticsearch.action.LegacyActionRequest;
import org.elasticsearch.common.Strings;
Expand All @@ -21,6 +21,8 @@

public class SamlInitiateSingleSignOnRequest extends LegacyActionRequest {

private static final TransportVersion IDP_CUSTOM_SAML_ATTRIBUTES = TransportVersion.fromName("idp_custom_saml_attributes");

private String spEntityId;
private String assertionConsumerService;
private SamlAuthenticationState samlAuthenticationState;
Expand All @@ -31,7 +33,7 @@ public SamlInitiateSingleSignOnRequest(StreamInput in) throws IOException {
spEntityId = in.readString();
assertionConsumerService = in.readString();
samlAuthenticationState = in.readOptionalWriteable(SamlAuthenticationState::new);
if (in.getTransportVersion().onOrAfter(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19)) {
if (in.getTransportVersion().supports(IDP_CUSTOM_SAML_ATTRIBUTES)) {
attributes = in.readOptionalWriteable(SamlInitiateSingleSignOnAttributes::new);
}
}
Expand Down Expand Up @@ -99,7 +101,7 @@ public void writeTo(StreamOutput out) throws IOException {
out.writeString(spEntityId);
out.writeString(assertionConsumerService);
out.writeOptionalWriteable(samlAuthenticationState);
if (out.getTransportVersion().onOrAfter(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19)) {
if (out.getTransportVersion().supports(IDP_CUSTOM_SAML_ATTRIBUTES)) {
out.writeOptionalWriteable(attributes);
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
package org.elasticsearch.xpack.idp.action;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.action.ActionRequestValidationException;
import org.elasticsearch.common.io.stream.BytesStreamOutput;
import org.elasticsearch.common.io.stream.StreamInput;
Expand All @@ -27,6 +26,8 @@

public class SamlInitiateSingleSignOnRequestTests extends ESTestCase {

private static final TransportVersion IDP_CUSTOM_SAML_ATTRIBUTES = TransportVersion.fromName("idp_custom_saml_attributes");

public void testSerializationCurrentVersion() throws Exception {
final SamlInitiateSingleSignOnRequest request = new SamlInitiateSingleSignOnRequest();
request.setSpEntityId("https://kibana_url");
Expand All @@ -43,11 +44,7 @@ public void testSerializationCurrentVersion() throws Exception {
final BytesStreamOutput out = new BytesStreamOutput();
if (randomBoolean()) {
out.setTransportVersion(
TransportVersionUtils.randomVersionBetween(
random(),
TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19,
TransportVersion.current()
)
TransportVersionUtils.randomVersionBetween(random(), IDP_CUSTOM_SAML_ATTRIBUTES, TransportVersion.current())
);
}
request.writeTo(out);
Expand Down Expand Up @@ -83,7 +80,7 @@ public void testSerializationOldTransportVersion() throws Exception {
TransportVersionUtils.randomVersionBetween(
random(),
TransportVersion.minimumCompatible(),
TransportVersionUtils.getPreviousVersion(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19)
TransportVersionUtils.getPreviousVersion(IDP_CUSTOM_SAML_ATTRIBUTES)
)
);
request.writeTo(out);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,8 @@ public class TextSimilarityRankBuilder extends RankBuilder {
License.OperationMode.ENTERPRISE
);

private static final TransportVersion RERANKER_FAILURES_ALLOWED = TransportVersion.fromName("reranker_failures_allowed");

private final String inferenceId;
private final String inferenceText;
private final String field;
Expand Down Expand Up @@ -72,7 +74,7 @@ public TextSimilarityRankBuilder(StreamInput in) throws IOException {
this.inferenceText = in.readString();
this.field = in.readString();
this.minScore = in.readOptionalFloat();
if (in.getTransportVersion().onOrAfter(TransportVersions.RERANKER_FAILURES_ALLOWED_8_19)) {
if (in.getTransportVersion().supports(RERANKER_FAILURES_ALLOWED)) {
this.failuresAllowed = in.readBoolean();
} else {
this.failuresAllowed = false;
Expand All @@ -96,7 +98,7 @@ public void doWriteTo(StreamOutput out) throws IOException {
out.writeString(inferenceText);
out.writeString(field);
out.writeOptionalFloat(minScore);
if (out.getTransportVersion().onOrAfter(TransportVersions.RERANKER_FAILURES_ALLOWED_8_19)) {
if (out.getTransportVersion().supports(RERANKER_FAILURES_ALLOWED)) {
out.writeBoolean(failuresAllowed);
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.custom;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand All @@ -30,6 +29,8 @@ public class CustomSecretSettings implements SecretSettings {
public static final String NAME = "custom_secret_settings";
public static final String SECRET_PARAMETERS = "secret_parameters";

private static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED = TransportVersion.fromName("inference_custom_service_added");

public static CustomSecretSettings fromMap(@Nullable Map<String, Object> map) {
if (map == null) {
return null;
Expand Down Expand Up @@ -90,7 +91,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@

import org.elasticsearch.ElasticsearchStatusException;
import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.action.ActionListener;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.util.LazyInitializable;
Expand Down Expand Up @@ -68,6 +67,8 @@ public class CustomService extends SenderService {
public static final String NAME = "custom";
private static final String SERVICE_NAME = "Custom";

private static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED = TransportVersion.fromName("inference_custom_service_added");

private static final EnumSet<TaskType> supportedTaskTypes = EnumSet.of(
TaskType.TEXT_EMBEDDING,
TaskType.SPARSE_EMBEDDING,
Expand Down Expand Up @@ -353,7 +354,7 @@ private static CustomServiceSettings getCustomServiceSettings(CustomModel custom

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.custom;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.Strings;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
Expand Down Expand Up @@ -66,6 +65,7 @@ public class CustomServiceSettings extends FilteredXContentObject implements Ser
private static final String RESPONSE_SCOPE = String.join(".", ModelConfigurations.SERVICE_SETTINGS, RESPONSE);
private static final int DEFAULT_EMBEDDING_BATCH_SIZE = 10;

private static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED = TransportVersion.fromName("inference_custom_service_added");
private static final TransportVersion ML_INFERENCE_CUSTOM_SERVICE_REMOVE_ERROR_PARSING = TransportVersion.fromName(
"ml_inference_custom_service_remove_error_parsing"
);
Expand Down Expand Up @@ -401,7 +401,7 @@ public ToXContentObject getFilteredXContentObject() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.custom;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand All @@ -33,6 +32,8 @@ public class CustomTaskSettings implements TaskSettings {

static final CustomTaskSettings EMPTY_SETTINGS = new CustomTaskSettings(new HashMap<>());

private static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED = TransportVersion.fromName("inference_custom_service_added");

public static CustomTaskSettings fromMap(Map<String, Object> map) {
ValidationException validationException = new ValidationException();
if (map == null || map.isEmpty()) {
Expand Down Expand Up @@ -100,7 +101,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.googlevertexai.completion;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand Down Expand Up @@ -39,6 +38,10 @@ public class GoogleVertexAiChatCompletionServiceSettings extends FilteredXConten

public static final String NAME = "google_vertex_ai_chatcompletion_service_settings";

private static final TransportVersion ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED = TransportVersion.fromName(
"ml_inference_vertexai_chatcompletion_added"
);

private final String location;
private final String modelId;
private final String projectId;
Expand Down Expand Up @@ -118,7 +121,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED_8_19;
return ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.huggingface.rerank;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand Down Expand Up @@ -39,6 +38,10 @@ public class HuggingFaceRerankServiceSettings extends FilteredXContentObject

private static final RateLimitSettings DEFAULT_RATE_LIMIT_SETTINGS = new RateLimitSettings(3000);

private static final TransportVersion ML_INFERENCE_HUGGING_FACE_RERANK_ADDED = TransportVersion.fromName(
"ml_inference_sagemaker_chat_completion"
);

public static HuggingFaceRerankServiceSettings fromMap(Map<String, Object> map, ConfigurationParseContext context) {
ValidationException validationException = new ValidationException();
var uri = extractUri(map, URL, validationException);
Expand Down Expand Up @@ -115,7 +118,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.ML_INFERENCE_HUGGING_FACE_RERANK_ADDED_8_19;
return ML_INFERENCE_HUGGING_FACE_RERANK_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.huggingface.rerank;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand All @@ -33,6 +32,10 @@ public class HuggingFaceRerankTaskSettings implements TaskSettings {

static final HuggingFaceRerankTaskSettings EMPTY_SETTINGS = new HuggingFaceRerankTaskSettings(null, null);

private static final TransportVersion ML_INFERENCE_HUGGING_FACE_RERANK_ADDED = TransportVersion.fromName(
"ml_inference_hugging_face_rerank_added"
);

public static HuggingFaceRerankTaskSettings fromMap(Map<String, Object> map) {
ValidationException validationException = new ValidationException();

Expand Down Expand Up @@ -118,7 +121,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.ML_INFERENCE_HUGGING_FACE_RERANK_ADDED_8_19;
return ML_INFERENCE_HUGGING_FACE_RERANK_ADDED;
}

@Override
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
package org.elasticsearch.xpack.inference.services.sagemaker.schema.openai;

import org.elasticsearch.TransportVersion;
import org.elasticsearch.TransportVersions;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;
Expand All @@ -26,6 +25,10 @@ record SageMakerOpenAiTaskSettings(@Nullable String user) implements SageMakerSt
static final String NAME = "sagemaker_openai_task_settings";
private static final String USER_FIELD = "user";

public static final TransportVersion ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION = TransportVersion.fromName(
"ml_inference_sagemaker_chat_completion"
);

SageMakerOpenAiTaskSettings(StreamInput in) throws IOException {
this(in.readOptionalString());
}
Expand All @@ -37,7 +40,7 @@ public String getWriteableName() {

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION_8_19;
return ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION;
}

@Override
Expand Down