diff --git a/build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/transport/CollectTransportVersionReferencesTask.java b/build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/transport/CollectTransportVersionReferencesTask.java index d572abb848d1c..5b2cb960f7d47 100644 --- a/build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/transport/CollectTransportVersionReferencesTask.java +++ b/build-tools-internal/src/main/java/org/elasticsearch/gradle/internal/transport/CollectTransportVersionReferencesTask.java @@ -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)) { diff --git a/server/src/main/java/org/elasticsearch/TransportVersions.java b/server/src/main/java/org/elasticsearch/TransportVersions.java index 9c24eaf3b416b..d0c1491090f7f 100644 --- a/server/src/main/java/org/elasticsearch/TransportVersions.java +++ b/server/src/main/java/org/elasticsearch/TransportVersions.java @@ -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, diff --git a/server/src/main/resources/transport/definitions/referable/esql_hash_operator_status_output_time.csv b/server/src/main/resources/transport/definitions/referable/esql_hash_operator_status_output_time.csv new file mode 100644 index 0000000000000..3e0813793dbb1 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/esql_hash_operator_status_output_time.csv @@ -0,0 +1 @@ +9077000,8841034 diff --git a/server/src/main/resources/transport/definitions/referable/idp_custom_saml_attributes.csv b/server/src/main/resources/transport/definitions/referable/idp_custom_saml_attributes.csv new file mode 100644 index 0000000000000..9cf553bbaf0b8 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/idp_custom_saml_attributes.csv @@ -0,0 +1 @@ +9087000,8841040 diff --git a/server/src/main/resources/transport/definitions/referable/inference_custom_service_added.csv b/server/src/main/resources/transport/definitions/referable/inference_custom_service_added.csv new file mode 100644 index 0000000000000..f3df322636d71 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/inference_custom_service_added.csv @@ -0,0 +1 @@ +9084000,8841039 diff --git a/server/src/main/resources/transport/definitions/referable/ml_inference_hugging_face_rerank_added.csv b/server/src/main/resources/transport/definitions/referable/ml_inference_hugging_face_rerank_added.csv new file mode 100644 index 0000000000000..d902e29af78c2 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/ml_inference_hugging_face_rerank_added.csv @@ -0,0 +1 @@ +9080000,8841036 diff --git a/server/src/main/resources/transport/definitions/referable/ml_inference_sagemaker_chat_completion.csv b/server/src/main/resources/transport/definitions/referable/ml_inference_sagemaker_chat_completion.csv new file mode 100644 index 0000000000000..c81bb0fc03562 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/ml_inference_sagemaker_chat_completion.csv @@ -0,0 +1 @@ +9082000,8841037 diff --git a/server/src/main/resources/transport/definitions/referable/ml_inference_vertexai_chatcompletion_added.csv b/server/src/main/resources/transport/definitions/referable/ml_inference_vertexai_chatcompletion_added.csv new file mode 100644 index 0000000000000..eef6a4174a9dc --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/ml_inference_vertexai_chatcompletion_added.csv @@ -0,0 +1 @@ +9083000,8841038 diff --git a/server/src/main/resources/transport/definitions/referable/reranker_failures_allowed.csv b/server/src/main/resources/transport/definitions/referable/reranker_failures_allowed.csv new file mode 100644 index 0000000000000..c0efde0a3bca5 --- /dev/null +++ b/server/src/main/resources/transport/definitions/referable/reranker_failures_allowed.csv @@ -0,0 +1 @@ +9013000,8841035 diff --git a/server/src/test/java/org/elasticsearch/TransportVersionTests.java b/server/src/test/java/org/elasticsearch/TransportVersionTests.java index f73f622fb8083..3b2a474e0d7f2 100644 --- a/server/src/test/java/org/elasticsearch/TransportVersionTests.java +++ b/server/src/test/java/org/elasticsearch/TransportVersionTests.java @@ -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) ); } } diff --git a/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/HashAggregationOperator.java b/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/HashAggregationOperator.java index 113891151b0ed..df31e331a892c 100644 --- a/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/HashAggregationOperator.java +++ b/x-pack/plugin/esql/compute/src/main/java/org/elasticsearch/compute/operator/HashAggregationOperator.java @@ -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. */ @@ -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; @@ -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); } } diff --git a/x-pack/plugin/identity-provider/src/main/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequest.java b/x-pack/plugin/identity-provider/src/main/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequest.java index 1834d46319096..6610606a37264 100644 --- a/x-pack/plugin/identity-provider/src/main/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequest.java +++ b/x-pack/plugin/identity-provider/src/main/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequest.java @@ -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; @@ -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; @@ -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); } } @@ -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); } } diff --git a/x-pack/plugin/identity-provider/src/test/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequestTests.java b/x-pack/plugin/identity-provider/src/test/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequestTests.java index 4aa609f595578..17ddf4cd9c8f4 100644 --- a/x-pack/plugin/identity-provider/src/test/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequestTests.java +++ b/x-pack/plugin/identity-provider/src/test/java/org/elasticsearch/xpack/idp/action/SamlInitiateSingleSignOnRequestTests.java @@ -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; @@ -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"); @@ -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); @@ -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); diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankBuilder.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankBuilder.java index 62dab66c4cce0..4f9763b563599 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankBuilder.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/rank/textsimilarity/TextSimilarityRankBuilder.java @@ -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; @@ -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; @@ -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); } } diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomSecretSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomSecretSettings.java index 37fc9f946d4ff..42c718a16ff5b 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomSecretSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomSecretSettings.java @@ -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; @@ -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 map) { if (map == null) { return null; @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomService.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomService.java index 8a9f9b377989b..4242e89ec3f4f 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomService.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomService.java @@ -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; @@ -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 supportedTaskTypes = EnumSet.of( TaskType.TEXT_EMBEDDING, TaskType.SPARSE_EMBEDDING, @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceSettings.java index 191475b0cc0db..731328da2cd90 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomServiceSettings.java @@ -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; @@ -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" ); @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomTaskSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomTaskSettings.java index b353450191e07..ca93322a1ba61 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomTaskSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/custom/CustomTaskSettings.java @@ -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; @@ -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 map) { ValidationException validationException = new ValidationException(); if (map == null || map.isEmpty()) { @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/googlevertexai/completion/GoogleVertexAiChatCompletionServiceSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/googlevertexai/completion/GoogleVertexAiChatCompletionServiceSettings.java index 3e6f2c5d45238..f3227ab72605c 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/googlevertexai/completion/GoogleVertexAiChatCompletionServiceSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/googlevertexai/completion/GoogleVertexAiChatCompletionServiceSettings.java @@ -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; @@ -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; @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankServiceSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankServiceSettings.java index bba7afdabda75..0275c22fb6ed2 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankServiceSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankServiceSettings.java @@ -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; @@ -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 map, ConfigurationParseContext context) { ValidationException validationException = new ValidationException(); var uri = extractUri(map, URL, validationException); @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankTaskSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankTaskSettings.java index 87640d52fb189..fd61ed8dd20f1 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankTaskSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/huggingface/rerank/HuggingFaceRerankTaskSettings.java @@ -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; @@ -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 map) { ValidationException validationException = new ValidationException(); @@ -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 diff --git a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/SageMakerOpenAiTaskSettings.java b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/SageMakerOpenAiTaskSettings.java index 0200783e8f1ef..43d06ef8e9b5b 100644 --- a/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/SageMakerOpenAiTaskSettings.java +++ b/x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/SageMakerOpenAiTaskSettings.java @@ -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; @@ -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()); } @@ -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