Add text embedding processor

java-client/src/main/java/org/opensearch/client/opensearch/ingest/Processor.java

@@ -127,6 +127,8 @@ public enum Kind implements JsonEnum {
 
         Inference("inference"),
 
+        TextEmbedding("text_embedding"),
+
         ;
 
         private final String jsonValue;
@@ -735,6 +737,23 @@ public InferenceProcessor inference() {
         return TaggedUnionUtils.get(this, Kind.Inference);
     }
 
+    /**
+     * Is this variant instance of kind {@code text_embedding}?
+     */
+    public boolean isTextEmbedding() {
+        return _kind == Kind.TextEmbedding;
+    }
+
+    /**
+     * Get the {@code text_embedding} variant value.
+     *
+     * @throws IllegalStateException
+     *             if the current variant is not of the {@code text_embedding} kind.
+     */
+    public TextEmbeddingProcessor textEmbedding() {
+        return TaggedUnionUtils.get(this, Kind.TextEmbedding);
+    }
+
     @Override
     @SuppressWarnings("unchecked")
     public void serialize(JsonGenerator generator, JsonpMapper mapper) {
@@ -1086,6 +1105,16 @@ public ObjectBuilder<Processor> inference(Function<InferenceProcessor.Builder, O
             return this.inference(fn.apply(new InferenceProcessor.Builder()).build());
         }
 
+        public ObjectBuilder<Processor> textEmbedding(TextEmbeddingProcessor v) {
+            this._kind = Kind.TextEmbedding;
+            this._value = v;
+            return this;
+        }
+
+        public ObjectBuilder<Processor> textEmbedding(Function<TextEmbeddingProcessor.Builder, ObjectBuilder<TextEmbeddingProcessor>> fn) {
+            return this.textEmbedding(fn.apply(new TextEmbeddingProcessor.Builder()).build());
+        }
+
         public Processor build() {
             _checkSingleUse();
             return new Processor(this);
@@ -1128,6 +1157,7 @@ protected static void setupProcessorDeserializer(ObjectDeserializer<Builder> op)
         op.add(Builder::drop, DropProcessor._DESERIALIZER, "drop");
         op.add(Builder::circle, CircleProcessor._DESERIALIZER, "circle");
         op.add(Builder::inference, InferenceProcessor._DESERIALIZER, "inference");
+        op.add(Builder::textEmbedding, TextEmbeddingProcessor._DESERIALIZER, "text_embedding");
 
     }
 

java-client/src/main/java/org/opensearch/client/opensearch/ingest/ProcessorBuilders.java

@@ -301,4 +301,12 @@ public static InferenceProcessor.Builder inference() {
         return new InferenceProcessor.Builder();
     }
 
+    /**
+     * Creates a builder for the {@link TextEmbeddingProcessor text_embedding}
+     * {@code Processor} variant.
+     */
+    public static TextEmbeddingProcessor.Builder textEmbedding() {
+        return new TextEmbeddingProcessor.Builder();
+    }
+
 }

java-client/src/main/java/org/opensearch/client/opensearch/ingest/TextEmbeddingProcessor.java

@@ -0,0 +1,214 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ *
+ * The OpenSearch Contributors require contributions made to
+ * this file be licensed under the Apache-2.0 license or a
+ * compatible open source license.
+ */
+
+/*
+ * Licensed to Elasticsearch B.V. under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch B.V. licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+/*
+ * Modifications Copyright OpenSearch Contributors. See
+ * GitHub history for details.
+ */
+
+package org.opensearch.client.opensearch.ingest;
+
+import jakarta.json.stream.JsonGenerator;
+import java.util.Map;
+import java.util.function.Function;
+import javax.annotation.Nullable;
+import org.opensearch.client.json.JsonData;
+import org.opensearch.client.json.JsonpDeserializable;
+import org.opensearch.client.json.JsonpDeserializer;
+import org.opensearch.client.json.JsonpMapper;
+import org.opensearch.client.json.ObjectBuilderDeserializer;
+import org.opensearch.client.json.ObjectDeserializer;
+import org.opensearch.client.util.ApiTypeHelper;
+import org.opensearch.client.util.ObjectBuilder;
+
+// typedef: ingest._types.TextEmbeddingProcessor
+
+@JsonpDeserializable
+public class TextEmbeddingProcessor extends ProcessorBase implements ProcessorVariant {
+    private final String modelId;
+
+    private final Map<String, JsonData> fieldMap;
+
+    @Nullable
+    private final String description;
+
+    // ---------------------------------------------------------------------------------------------
+
+    private TextEmbeddingProcessor(Builder builder) {
+        super(builder);
+
+        this.modelId = ApiTypeHelper.requireNonNull(builder.modelId, this, "modelId");
+        this.fieldMap = ApiTypeHelper.unmodifiableRequired(builder.fieldMap, this, "fieldMap");
+        this.description = builder.description;
+
+    }
+
+    public static TextEmbeddingProcessor of(Function<Builder, ObjectBuilder<TextEmbeddingProcessor>> fn) {
+        return fn.apply(new Builder()).build();
+    }
+
+    /**
+     * Processor variant kind.
+     */
+    @Override
+    public Processor.Kind _processorKind() {
+        return Processor.Kind.Inference;
+    }
+
+    /**
+     * Required - API name: {@code model_id}
+     */
+    public final String modelId() {
+        return this.modelId;
+    }
+
+    /**
+     * API name: {@code field_map}
+     */
+    public final Map<String, JsonData> fieldMap() {
+        return this.fieldMap;
+    }
+
+    /**
+    * API name: {@code description}
+    */
+    @Nullable
+    public final String description() {
+        return this.description;
+    }
+
+    protected void serializeInternal(JsonGenerator generator, JsonpMapper mapper) {
+
+        super.serializeInternal(generator, mapper);
+        generator.writeKey("model_id");
+        generator.write(this.modelId);
+
+        if (ApiTypeHelper.isDefined(this.fieldMap)) {
+            generator.writeKey("field_map");
+            generator.writeStartObject();
+            for (Map.Entry<String, JsonData> item0 : this.fieldMap.entrySet()) {
+                generator.writeKey(item0.getKey());
+                item0.getValue().serialize(generator, mapper);
+
+            }
+            generator.writeEnd();
+
+        }
+        if (this.description != null) {
+            generator.writeKey("description");
+            generator.write(this.description);
+        }
+
+    }
+
+    // ---------------------------------------------------------------------------------------------
+
+    /**
+     * Builder for {@link TextEmbeddingProcessor}.
+     */
+
+    public static class Builder extends ProcessorBase.AbstractBuilder<Builder> implements ObjectBuilder<TextEmbeddingProcessor> {
+        private String modelId;
+
+        @Nullable
+        private Map<String, JsonData> fieldMap;
+
+        @Nullable
+        private String description;
+
+        /**
+         * Required - API name: {@code model_id}
+         */
+        public final Builder modelId(String value) {
+            this.modelId = value;
+            return this;
+        }
+
+        /**
+         * API name: {@code field_map}
+         * <p>
+         * Adds all entries of <code>map</code> to <code>fieldMap</code>.
+         */
+        public final Builder fieldMap(Map<String, JsonData> map) {
+            this.fieldMap = _mapPutAll(this.fieldMap, map);
+            return this;
+        }
+
+        /**
+         * API name: {@code field_map}
+         * <p>
+         * Adds an entry to <code>fieldMap</code>.
+         */
+        public final Builder fieldMap(String key, JsonData value) {
+            this.fieldMap = _mapPut(this.fieldMap, key, value);
+            return this;
+        }
+
+        /**
+         * API name: {@code description}
+         */
+        public final Builder description(@Nullable String value) {
+            this.description = value;
+            return this;
+        }
+
+        @Override
+        protected Builder self() {
+            return this;
+        }
+
+        /**
+         * Builds a {@link TextEmbeddingProcessor}.
+         *
+         * @throws NullPointerException
+         *             if some of the required fields are null.
+         */
+        public TextEmbeddingProcessor build() {
+            _checkSingleUse();
+
+            return new TextEmbeddingProcessor(this);
+        }
+    }
+
+    // ---------------------------------------------------------------------------------------------
+
+    /**
+     * Json deserializer for {@link TextEmbeddingProcessor}
+     */
+    public static final JsonpDeserializer<TextEmbeddingProcessor> _DESERIALIZER = ObjectBuilderDeserializer.lazy(
+        Builder::new,
+        TextEmbeddingProcessor::setupTextEmbeddingProcessorDeserializer
+    );
+
+    protected static void setupTextEmbeddingProcessorDeserializer(ObjectDeserializer<TextEmbeddingProcessor.Builder> op) {
+        ProcessorBase.setupProcessorBaseDeserializer(op);
+        op.add(Builder::modelId, JsonpDeserializer.stringDeserializer(), "model_id");
+        op.add(Builder::fieldMap, JsonpDeserializer.stringMapDeserializer(JsonData._DESERIALIZER), "field_map");
+        op.add(Builder::description, JsonpDeserializer.stringDeserializer(), "description");
+    }
+
+}