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
14 changes: 0 additions & 14 deletions server/src/main/java/org/elasticsearch/TransportVersions.java
Original file line number Diff line number Diff line change
Expand Up @@ -175,13 +175,6 @@ static TransportVersion def(int id) {
public static final TransportVersion ESQL_DRIVER_TASK_DESCRIPTION_8_19 = def(8_841_0_30);
public static final TransportVersion ML_INFERENCE_HUGGING_FACE_CHAT_COMPLETION_ADDED_8_19 = def(8_841_0_31);
public static final TransportVersion V_8_19_FIELD_CAPS_ADD_CLUSTER_ALIAS = def(8_841_0_32);
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);
public static final TransportVersion INITIAL_ELASTICSEARCH_9_0 = def(9_000_0_00);
public static final TransportVersion REMOVE_SNAPSHOT_FAILURES_90 = def(9_000_0_01);
public static final TransportVersion TRANSPORT_STATS_HANDLING_TIME_REQUIRED_90 = def(9_000_0_02);
Expand All @@ -204,7 +197,6 @@ static TransportVersion def(int id) {
public static final TransportVersion SLM_UNHEALTHY_IF_NO_SNAPSHOT_WITHIN = def(9_010_0_00);
public static final TransportVersion ESQL_SUPPORT_PARTIAL_RESULTS = def(9_011_0_00);
public static final TransportVersion REMOVE_REPOSITORY_CONFLICT_MESSAGE = def(9_012_0_00);
public static final TransportVersion RERANKER_FAILURES_ALLOWED = def(9_013_0_00);
public static final TransportVersion VOYAGE_AI_INTEGRATION_ADDED = def(9_014_0_00);
public static final TransportVersion BYTE_SIZE_VALUE_ALWAYS_USES_BYTES = def(9_015_0_00);
public static final TransportVersion ESQL_SERIALIZE_SOURCE_FUNCTIONS_WARNINGS = def(9_016_0_00);
Expand Down Expand Up @@ -262,15 +254,9 @@ static TransportVersion def(int id) {
public static final TransportVersion INFERENCE_ADD_TIMEOUT_PUT_ENDPOINT = def(9_074_0_00);
public static final TransportVersion ESQL_FIELD_ATTRIBUTE_DROP_TYPE = def(9_075_0_00);
public static final TransportVersion ESQL_TIME_SERIES_SOURCE_STATUS = def(9_076_0_00);
public static final TransportVersion ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME = def(9_077_0_00);
public static final TransportVersion ML_INFERENCE_HUGGING_FACE_CHAT_COMPLETION_ADDED = def(9_078_0_00);
public static final TransportVersion NODES_STATS_SUPPORTS_MULTI_PROJECT = def(9_079_0_00);
public static final TransportVersion ML_INFERENCE_HUGGING_FACE_RERANK_ADDED = def(9_080_0_00);
public static final TransportVersion ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION = def(9_082_0_00);
public static final TransportVersion ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED = def(9_083_0_00);
public static final TransportVersion INFERENCE_CUSTOM_SERVICE_ADDED = def(9_084_0_00);
public static final TransportVersion ESQL_LIMIT_ROW_SIZE = def(9_085_0_00);
public static final TransportVersion IDP_CUSTOM_SAML_ATTRIBUTES = def(9_087_0_00);
public static final TransportVersion SNAPSHOT_INDEX_SHARD_STATUS_MISSING_STATS = def(9_093_0_00);
public static final TransportVersion SEARCH_LOAD_PER_INDEX_STATS = def(9_095_0_00);
public static final TransportVersion HEAP_USAGE_IN_CLUSTER_INFO = def(9_096_0_00);
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 @@ -306,6 +306,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 @@ -359,8 +363,7 @@ protected Status(StreamInput in) throws IOException {
rowsReceived = 0;
rowsEmitted = 0;
}
if (in.getTransportVersion().onOrAfter(TransportVersions.ESQL_HASH_OPERATOR_STATUS_OUTPUT_TIME)
|| in.getTransportVersion().isPatchFrom(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 @@ -377,8 +380,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)
|| out.getTransportVersion().isPatchFrom(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,8 +33,7 @@ public SamlInitiateSingleSignOnRequest(StreamInput in) throws IOException {
spEntityId = in.readString();
assertionConsumerService = in.readString();
samlAuthenticationState = in.readOptionalWriteable(SamlAuthenticationState::new);
if (in.getTransportVersion().isPatchFrom(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19)
|| in.getTransportVersion().onOrAfter(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES)) {
if (in.getTransportVersion().supports(IDP_CUSTOM_SAML_ATTRIBUTES)) {
attributes = in.readOptionalWriteable(SamlInitiateSingleSignOnAttributes::new);
}
}
Expand Down Expand Up @@ -100,8 +101,7 @@ public void writeTo(StreamOutput out) throws IOException {
out.writeString(spEntityId);
out.writeString(assertionConsumerService);
out.writeOptionalWriteable(samlAuthenticationState);
if (out.getTransportVersion().isPatchFrom(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19)
|| out.getTransportVersion().onOrAfter(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES)) {
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,9 @@

public class SamlInitiateSingleSignOnRequestTests extends ESTestCase {

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

public void testSerializationCurrentVersion() throws Exception {
final SamlInitiateSingleSignOnRequest request = new SamlInitiateSingleSignOnRequest();
request.setSpEntityId("https://kibana_url");
Expand All @@ -43,11 +45,7 @@ public void testSerializationCurrentVersion() throws Exception {
final BytesStreamOutput out = new BytesStreamOutput();
if (randomBoolean()) {
out.setTransportVersion(
TransportVersionUtils.randomVersionBetween(
random(),
TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES,
TransportVersion.current()
)
TransportVersionUtils.randomVersionBetween(random(), IDP_CUSTOM_SAML_ATTRIBUTES, TransportVersion.current())
);
}
request.writeTo(out);
Expand Down Expand Up @@ -79,7 +77,7 @@ public void testSerializationOldButCompatibleTransportVersion() throws Exception
}
assertThat("An invalid request is not guaranteed to serialize correctly", request.validate(), nullValue());
final BytesStreamOutput out = new BytesStreamOutput();
out.setTransportVersion(TransportVersions.IDP_CUSTOM_SAML_ATTRIBUTES_ADDED_8_19);
out.setTransportVersion(IDP_CUSTOM_SAML_ATTRIBUTES_PATCH);
request.writeTo(out);

try (StreamInput in = out.bytes().streamInput()) {
Expand Down Expand Up @@ -113,7 +111,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_PATCH)
)
);
request.writeTo(out);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ public class TextSimilarityRankBuilder extends RankBuilder {
License.OperationMode.ENTERPRISE
);

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

private final String inferenceId;
Expand Down Expand Up @@ -84,8 +85,7 @@ public TextSimilarityRankBuilder(StreamInput in) throws IOException {
this.inferenceText = in.readString();
this.field = in.readString();
this.minScore = in.readOptionalFloat();
if (in.getTransportVersion().isPatchFrom(TransportVersions.RERANKER_FAILURES_ALLOWED_8_19)
|| in.getTransportVersion().onOrAfter(TransportVersions.RERANKER_FAILURES_ALLOWED)) {
if (in.getTransportVersion().supports(RERANKER_FAILURES_ALLOWED)) {
this.failuresAllowed = in.readBoolean();
} else {
this.failuresAllowed = false;
Expand Down Expand Up @@ -114,8 +114,7 @@ public void doWriteTo(StreamOutput out) throws IOException {
out.writeString(inferenceText);
out.writeString(field);
out.writeOptionalFloat(minScore);
if (out.getTransportVersion().isPatchFrom(TransportVersions.RERANKER_FAILURES_ALLOWED_8_19)
|| out.getTransportVersion().onOrAfter(TransportVersions.RERANKER_FAILURES_ALLOWED)) {
if (out.getTransportVersion().supports(RERANKER_FAILURES_ALLOWED)) {
out.writeBoolean(failuresAllowed);
}
if (out.getTransportVersion().supports(RERANK_SNIPPETS)) {
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 @@ -29,6 +28,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 @@ -89,13 +90,12 @@ public String getWriteableName() {
@Override
public TransportVersion getMinimalSupportedVersion() {
assert false : "should never be called when supportsVersion is used";
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
public boolean supportsVersion(TransportVersion version) {
return version.onOrAfter(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED)
|| version.isPatchFrom(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19);
return version.supports(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.cluster.service.ClusterService;
import org.elasticsearch.common.ValidationException;
Expand Down Expand Up @@ -71,6 +70,8 @@ public class CustomService extends SenderService implements RerankingInferenceSe
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 @@ -371,7 +372,7 @@ private static CustomServiceSettings getCustomServiceSettings(CustomModel custom

@Override
public TransportVersion getMinimalSupportedVersion() {
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED;
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 @@ -65,6 +64,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 @@ -427,13 +427,12 @@ public ToXContentObject getFilteredXContentObject() {
@Override
public TransportVersion getMinimalSupportedVersion() {
assert false : "should never be called when supportsVersion is used";
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
public boolean supportsVersion(TransportVersion version) {
return version.onOrAfter(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED)
|| version.isPatchFrom(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19);
return version.supports(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 @@ -31,6 +30,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 @@ -98,13 +99,12 @@ public String getWriteableName() {
@Override
public TransportVersion getMinimalSupportedVersion() {
assert false : "should never be called when supportsVersion is used";
return TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED;
return INFERENCE_CUSTOM_SERVICE_ADDED;
}

@Override
public boolean supportsVersion(TransportVersion version) {
return version.onOrAfter(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED)
|| version.isPatchFrom(TransportVersions.INFERENCE_CUSTOM_SERVICE_ADDED_8_19);
return version.supports(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.Strings;
import org.elasticsearch.common.ValidationException;
import org.elasticsearch.common.io.stream.StreamInput;
Expand Down Expand Up @@ -52,6 +51,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 @@ -266,13 +269,12 @@ public RateLimitSettings rateLimitSettings() {
@Override
public TransportVersion getMinimalSupportedVersion() {
assert false : "should never be called when supportsVersion is used";
return TransportVersions.ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED;
return ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED;
}

@Override
public boolean supportsVersion(TransportVersion version) {
return version.onOrAfter(TransportVersions.ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED)
|| version.isPatchFrom(TransportVersions.ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED_8_19);
return version.supports(ML_INFERENCE_VERTEXAI_CHATCOMPLETION_ADDED);
}

@Override
Expand Down
Loading