diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecAwsVectorizer.java b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecAwsVectorizer.java index f8cf23629..28569b43c 100644 --- a/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecAwsVectorizer.java +++ b/src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecAwsVectorizer.java @@ -18,6 +18,7 @@ public record Text2VecAwsVectorizer( @SerializedName("service") Service service, @SerializedName("targetModel") String targetModel, @SerializedName("targetVariant") String targetVariant, + @SerializedName("dimensions") Integer dimensions, /** * Weaviate defaults to {@code true} if the value is not provided. @@ -83,6 +84,7 @@ public Text2VecAwsVectorizer( Service service, String targetModel, String targetVariant, + Integer dimensions, boolean vectorizeCollectionName, List sourceProperties, @@ -94,6 +96,7 @@ public Text2VecAwsVectorizer( this.service = service; this.targetModel = targetModel; this.targetVariant = targetVariant; + this.dimensions = dimensions; this.vectorizeCollectionName = false; this.sourceProperties = sourceProperties; @@ -109,6 +112,7 @@ public Text2VecAwsVectorizer(Builder builder) { builder.service, builder.targetModel, builder.targetVariant, + builder.dimensions, builder.vectorizeCollectionName, builder.sourceProperties, @@ -128,6 +132,7 @@ public abstract static class Builder implements ObjectBuilder v.dimensions(512)), """ { "vectorIndexType": "hnsw", @@ -180,6 +180,7 @@ public static Object[][] testCases() { "text2vec-aws": { "service": "bedrock", "model": "amazon.titan-embed-text-v2:0", + "dimensions": 512, "vectorizeClassName": false } }