diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Multi2VecCohereVectorizer.java b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Multi2VecCohereVectorizer.java index c6e0baee7..a70e7fa86 100644 --- a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Multi2VecCohereVectorizer.java +++ b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Multi2VecCohereVectorizer.java @@ -18,6 +18,8 @@ public record Multi2VecCohereVectorizer( @SerializedName("baseURL") String baseUrl, /** Inference model to use. */ @SerializedName("model") String model, + /** The number of dimensions for the generated embeddings. */ + @SerializedName("dimensions") Integer dimensions, /** The truncate strategy to use. */ @SerializedName("truncate") String truncate, /** BLOB properties included in the embedding. */ @@ -71,6 +73,7 @@ public static Multi2VecCohereVectorizer of(Function imageFields, List textFields, @@ -81,6 +84,7 @@ public Multi2VecCohereVectorizer( this.vectorizeCollectionName = false; this.baseUrl = baseUrl; this.model = model; + this.dimensions = dimensions; this.truncate = truncate; this.imageFields = imageFields; this.textFields = textFields; @@ -93,6 +97,7 @@ public Multi2VecCohereVectorizer(Builder builder) { this( builder.baseUrl, builder.model, + builder.dimensions, builder.truncate, builder.imageFields.keySet().stream().toList(), builder.textFields.keySet().stream().toList(), @@ -114,6 +119,7 @@ public static class Builder implements ObjectBuilder private String baseUrl; private String model; + private Integer dimensions; private String truncate; /** Set base URL of the embedding service. */ @@ -127,6 +133,11 @@ public Builder model(String model) { return this; } + public Builder dimensions(Integer dimensions) { + this.dimensions = dimensions; + return this; + } + public Builder truncate(String truncate) { this.truncate = truncate; return this; diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecCohereVectorizer.java b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecCohereVectorizer.java index 1cfc239aa..ea0bed5e6 100644 --- a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecCohereVectorizer.java +++ b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecCohereVectorizer.java @@ -15,6 +15,7 @@ public record Text2VecCohereVectorizer( @SerializedName("baseUrl") String baseUrl, @SerializedName("model") String model, + @SerializedName("dimensions") Integer dimensions, @SerializedName("truncate") Truncate truncate, /** @@ -68,6 +69,7 @@ public static Text2VecCohereVectorizer of( public Text2VecCohereVectorizer( String baseUrl, String model, + Integer dimensions, Truncate truncate, boolean vectorizeCollectionName, @@ -75,6 +77,7 @@ public Text2VecCohereVectorizer( VectorIndex vectorIndex, Quantization quantization) { this.model = model; + this.dimensions = dimensions; this.truncate = truncate; this.baseUrl = baseUrl; @@ -88,6 +91,7 @@ public Text2VecCohereVectorizer(Builder builder) { this( builder.baseUrl, builder.model, + builder.dimensions, builder.truncate, builder.vectorizeCollectionName, builder.sourceProperties, @@ -102,6 +106,7 @@ public static class Builder implements ObjectBuilder { private VectorIndex vectorIndex = VectorIndex.DEFAULT_VECTOR_INDEX; private String model; + private Integer dimensions; private Truncate truncate; private String baseUrl; @@ -115,6 +120,11 @@ public Builder model(String model) { return this; } + public Builder dimensions(Integer dimensions) { + this.dimensions = dimensions; + return this; + } + public Builder truncate(Truncate truncate) { this.truncate = truncate; return this;