[ML] SageMaker supports 8.19 + 9.1 clusters

Overriding version support to include the 8.19 patch version.

Author: prwhelan

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/model/SageMakerServiceSettings.java

@@ -114,6 +114,11 @@ public TransportVersion getMinimalSupportedVersion() {
         return TransportVersions.ML_INFERENCE_SAGEMAKER;
     }
 
+    @Override
+    public boolean supportsVersion(TransportVersion version) {
+        return ServiceSettings.super.supportsVersion(version) || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_8_19);
+    }
+
     @Override
     public void writeTo(StreamOutput out) throws IOException {
         out.writeString(endpointName());

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/model/SageMakerTaskSettings.java

@@ -104,6 +104,11 @@ public TransportVersion getMinimalSupportedVersion() {
         return TransportVersions.ML_INFERENCE_SAGEMAKER;
     }
 
+    @Override
+    public boolean supportsVersion(TransportVersion version) {
+        return TaskSettings.super.supportsVersion(version) || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_8_19);
+    }
+
     @Override
     public void writeTo(StreamOutput out) throws IOException {
         out.writeOptionalString(customAttributes);

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/SageMakerStoredServiceSchema.java

@@ -32,6 +32,12 @@ public TransportVersion getMinimalSupportedVersion() {
             return TransportVersions.ML_INFERENCE_SAGEMAKER;
         }
 
+        @Override
+        public boolean supportsVersion(TransportVersion version) {
+            return SageMakerStoredServiceSchema.super.supportsVersion(version)
+                || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_8_19);
+        }
+
         @Override
         public void writeTo(StreamOutput out) {
 

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/SageMakerStoredTaskSchema.java

@@ -42,6 +42,12 @@ public TransportVersion getMinimalSupportedVersion() {
             return TransportVersions.ML_INFERENCE_SAGEMAKER;
         }
 
+        @Override
+        public boolean supportsVersion(TransportVersion version) {
+            return SageMakerStoredTaskSchema.super.supportsVersion(version)
+                || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_8_19);
+        }
+
         @Override
         public void writeTo(StreamOutput out) {}
 

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/elastic/ElasticTextEmbeddingPayload.java

@@ -253,6 +253,12 @@ public TransportVersion getMinimalSupportedVersion() {
             return TransportVersions.ML_INFERENCE_SAGEMAKER_ELASTIC;
         }
 
+        @Override
+        public boolean supportsVersion(TransportVersion version) {
+            return SageMakerStoredServiceSchema.super.supportsVersion(version)
+                || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_ELASTIC_8_19);
+        }
+
         @Override
         public void writeTo(StreamOutput out) throws IOException {
             out.writeOptionalVInt(dimensions);

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/elastic/SageMakerElasticTaskSettings.java

@@ -53,6 +53,12 @@ public TransportVersion getMinimalSupportedVersion() {
         return TransportVersions.ML_INFERENCE_SAGEMAKER_ELASTIC;
     }
 
+    @Override
+    public boolean supportsVersion(TransportVersion version) {
+        return SageMakerStoredTaskSchema.super.supportsVersion(version)
+            || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_ELASTIC_8_19);
+    }
+
     @Override
     public void writeTo(StreamOutput out) throws IOException {
         out.writeGenericMap(passthroughSettings);

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/OpenAiTextEmbeddingPayload.java

@@ -141,6 +141,12 @@ public TransportVersion getMinimalSupportedVersion() {
             return TransportVersions.ML_INFERENCE_SAGEMAKER;
         }
 
+        @Override
+        public boolean supportsVersion(TransportVersion version) {
+            return SageMakerStoredServiceSchema.super.supportsVersion(version)
+                || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_8_19);
+        }
+
         @Override
         public void writeTo(StreamOutput out) throws IOException {
             out.writeOptionalInt(dimensions);

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/services/sagemaker/schema/openai/SageMakerOpenAiTaskSettings.java

@@ -40,6 +40,12 @@ public TransportVersion getMinimalSupportedVersion() {
         return TransportVersions.ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION;
     }
 
+    @Override
+    public boolean supportsVersion(TransportVersion version) {
+        return SageMakerStoredTaskSchema.super.supportsVersion(version)
+            || version.isPatchFrom(TransportVersions.ML_INFERENCE_SAGEMAKER_CHAT_COMPLETION_8_19);
+    }
+
     @Override
     public void writeTo(StreamOutput out) throws IOException {
         out.writeOptionalString(user);