-
Notifications
You must be signed in to change notification settings - Fork 250
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
8 changed files
with
524 additions
and
26 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
183 changes: 183 additions & 0 deletions
183
java-client/src/main/java/co/elastic/clients/elasticsearch/_types/QueryVectorBuilder.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,183 @@ | ||
| /* | ||
| * 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. | ||
| */ | ||
|
|
||
| //---------------------------------------------------- | ||
| // THIS CODE IS GENERATED. MANUAL EDITS WILL BE LOST. | ||
| //---------------------------------------------------- | ||
|
|
||
| package co.elastic.clients.elasticsearch._types; | ||
|
|
||
| import co.elastic.clients.json.JsonEnum; | ||
| import co.elastic.clients.json.JsonpDeserializable; | ||
| import co.elastic.clients.json.JsonpDeserializer; | ||
| import co.elastic.clients.json.JsonpMapper; | ||
| import co.elastic.clients.json.JsonpSerializable; | ||
| import co.elastic.clients.json.JsonpUtils; | ||
| import co.elastic.clients.json.ObjectBuilderDeserializer; | ||
| import co.elastic.clients.json.ObjectDeserializer; | ||
| import co.elastic.clients.util.ApiTypeHelper; | ||
| import co.elastic.clients.util.ObjectBuilder; | ||
| import co.elastic.clients.util.TaggedUnion; | ||
| import co.elastic.clients.util.TaggedUnionUtils; | ||
| import co.elastic.clients.util.WithJsonObjectBuilderBase; | ||
| import jakarta.json.stream.JsonGenerator; | ||
| import java.lang.Object; | ||
| import java.util.Objects; | ||
| import java.util.function.Function; | ||
| import javax.annotation.Nullable; | ||
|
|
||
| // typedef: _types.QueryVectorBuilder | ||
|
|
||
| /** | ||
| * | ||
| * @see <a href="../doc-files/api-spec.html#_types.QueryVectorBuilder">API | ||
| * specification</a> | ||
| */ | ||
| @JsonpDeserializable | ||
| public class QueryVectorBuilder implements TaggedUnion<QueryVectorBuilder.Kind, Object>, JsonpSerializable { | ||
|
|
||
| /** | ||
| * {@link QueryVectorBuilder} variant kinds. | ||
| * | ||
| * @see <a href="../doc-files/api-spec.html#_types.QueryVectorBuilder">API | ||
| * specification</a> | ||
| */ | ||
|
|
||
| public enum Kind implements JsonEnum { | ||
| TextEmbedding("text_embedding"), | ||
|
|
||
| ; | ||
|
|
||
| private final String jsonValue; | ||
|
|
||
| Kind(String jsonValue) { | ||
| this.jsonValue = jsonValue; | ||
| } | ||
|
|
||
| public String jsonValue() { | ||
| return this.jsonValue; | ||
| } | ||
|
|
||
| } | ||
|
|
||
| private final Kind _kind; | ||
| private final Object _value; | ||
|
|
||
| @Override | ||
| public final Kind _kind() { | ||
| return _kind; | ||
| } | ||
|
|
||
| @Override | ||
| public final Object _get() { | ||
| return _value; | ||
| } | ||
|
|
||
| public QueryVectorBuilder(QueryVectorBuilderVariant value) { | ||
|
|
||
| this._kind = ApiTypeHelper.requireNonNull(value._queryVectorBuilderKind(), this, "<variant kind>"); | ||
| this._value = ApiTypeHelper.requireNonNull(value, this, "<variant value>"); | ||
|
|
||
| } | ||
|
|
||
| private QueryVectorBuilder(Builder builder) { | ||
|
|
||
| this._kind = ApiTypeHelper.requireNonNull(builder._kind, builder, "<variant kind>"); | ||
| this._value = ApiTypeHelper.requireNonNull(builder._value, builder, "<variant value>"); | ||
|
|
||
| } | ||
|
|
||
| public static QueryVectorBuilder of(Function<Builder, ObjectBuilder<QueryVectorBuilder>> fn) { | ||
| return fn.apply(new Builder()).build(); | ||
| } | ||
|
|
||
| /** | ||
| * 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 TextEmbedding textEmbedding() { | ||
| return TaggedUnionUtils.get(this, Kind.TextEmbedding); | ||
| } | ||
|
|
||
| @Override | ||
| @SuppressWarnings("unchecked") | ||
| public void serialize(JsonGenerator generator, JsonpMapper mapper) { | ||
|
|
||
| generator.writeStartObject(); | ||
|
|
||
| generator.writeKey(_kind.jsonValue()); | ||
| if (_value instanceof JsonpSerializable) { | ||
| ((JsonpSerializable) _value).serialize(generator, mapper); | ||
| } | ||
|
|
||
| generator.writeEnd(); | ||
|
|
||
| } | ||
|
|
||
| @Override | ||
| public String toString() { | ||
| return JsonpUtils.toString(this); | ||
| } | ||
|
|
||
| public static class Builder extends WithJsonObjectBuilderBase<Builder> | ||
| implements | ||
| ObjectBuilder<QueryVectorBuilder> { | ||
| private Kind _kind; | ||
| private Object _value; | ||
|
|
||
| @Override | ||
| protected Builder self() { | ||
| return this; | ||
| } | ||
| public ObjectBuilder<QueryVectorBuilder> textEmbedding(TextEmbedding v) { | ||
| this._kind = Kind.TextEmbedding; | ||
| this._value = v; | ||
| return this; | ||
| } | ||
|
|
||
| public ObjectBuilder<QueryVectorBuilder> textEmbedding( | ||
| Function<TextEmbedding.Builder, ObjectBuilder<TextEmbedding>> fn) { | ||
| return this.textEmbedding(fn.apply(new TextEmbedding.Builder()).build()); | ||
| } | ||
|
|
||
| public QueryVectorBuilder build() { | ||
| _checkSingleUse(); | ||
| return new QueryVectorBuilder(this); | ||
| } | ||
|
|
||
| } | ||
|
|
||
| protected static void setupQueryVectorBuilderDeserializer(ObjectDeserializer<Builder> op) { | ||
|
|
||
| op.add(Builder::textEmbedding, TextEmbedding._DESERIALIZER, "text_embedding"); | ||
|
|
||
| } | ||
|
|
||
| public static final JsonpDeserializer<QueryVectorBuilder> _DESERIALIZER = ObjectBuilderDeserializer | ||
| .lazy(Builder::new, QueryVectorBuilder::setupQueryVectorBuilderDeserializer, Builder::build); | ||
| } |
54 changes: 54 additions & 0 deletions
54
...ent/src/main/java/co/elastic/clients/elasticsearch/_types/QueryVectorBuilderBuilders.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,54 @@ | ||
| /* | ||
| * 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. | ||
| */ | ||
|
|
||
| //---------------------------------------------------- | ||
| // THIS CODE IS GENERATED. MANUAL EDITS WILL BE LOST. | ||
| //---------------------------------------------------- | ||
|
|
||
| package co.elastic.clients.elasticsearch._types; | ||
|
|
||
| import co.elastic.clients.util.ObjectBuilder; | ||
| import java.util.function.Function; | ||
|
|
||
| /** | ||
| * Builders for {@link QueryVectorBuilder} variants. | ||
| */ | ||
| public class QueryVectorBuilderBuilders { | ||
| private QueryVectorBuilderBuilders() { | ||
| } | ||
|
|
||
| /** | ||
| * Creates a builder for the {@link TextEmbedding text_embedding} | ||
| * {@code QueryVectorBuilder} variant. | ||
| */ | ||
| public static TextEmbedding.Builder textEmbedding() { | ||
| return new TextEmbedding.Builder(); | ||
| } | ||
|
|
||
| /** | ||
| * Creates a QueryVectorBuilder of the {@link TextEmbedding text_embedding} | ||
| * {@code QueryVectorBuilder} variant. | ||
| */ | ||
| public static QueryVectorBuilder textEmbedding(Function<TextEmbedding.Builder, ObjectBuilder<TextEmbedding>> fn) { | ||
| QueryVectorBuilder.Builder builder = new QueryVectorBuilder.Builder(); | ||
| builder.textEmbedding(fn.apply(new TextEmbedding.Builder()).build()); | ||
| return builder.build(); | ||
| } | ||
|
|
||
| } |
Oops, something went wrong.