Skip to content

Commit c08a3ef

Browse files
authored
Merge pull request #504 from weaviate/v6-voyage-dimensions
Add support voyageAI dimensions
2 parents 3aa5bee + 69ec80f commit c08a3ef

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

src/main/java/io/weaviate/client6/v1/api/collections/vectorizers/Text2VecVoyageAiVectorizer.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ public record Text2VecVoyageAiVectorizer(
1717
@SerializedName("baseUrl") String baseUrl,
1818
@SerializedName("model") String model,
1919
@SerializedName("truncate") Boolean truncate,
20+
@SerializedName("dimensions") Integer dimensions,
2021

2122
/**
2223
* Weaviate defaults to {@code true} if the value is not provided.
@@ -57,6 +58,7 @@ public Text2VecVoyageAiVectorizer(
5758
String baseUrl,
5859
String model,
5960
Boolean truncate,
61+
Integer dimensions,
6062

6163
boolean vectorizeCollectionName,
6264
List<String> sourceProperties,
@@ -65,6 +67,7 @@ public Text2VecVoyageAiVectorizer(
6567
this.model = model;
6668
this.truncate = truncate;
6769
this.baseUrl = baseUrl;
70+
this.dimensions = dimensions;
6871

6972
this.vectorizeCollectionName = false;
7073
this.sourceProperties = Collections.emptyList();
@@ -77,6 +80,7 @@ public Text2VecVoyageAiVectorizer(Builder builder) {
7780
builder.baseUrl,
7881
builder.model,
7982
builder.truncate,
83+
builder.dimensions,
8084
builder.vectorizeCollectionName,
8185
builder.sourceProperties,
8286
builder.vectorIndex,
@@ -92,6 +96,7 @@ public static class Builder implements ObjectBuilder<Text2VecVoyageAiVectorizer>
9296
private String model;
9397
private Boolean truncate;
9498
private String baseUrl;
99+
private Integer dimensions;
95100

96101
public Builder baseUrl(String baseUrl) {
97102
this.baseUrl = baseUrl;
@@ -108,6 +113,11 @@ public Builder truncate(boolean truncate) {
108113
return this;
109114
}
110115

116+
public Builder dimensions(Integer dimensions) {
117+
this.dimensions = dimensions;
118+
return this;
119+
}
120+
111121
/** Add properties to include in the embedding. */
112122
public Builder sourceProperties(String... properties) {
113123
return sourceProperties(Arrays.asList(properties));

0 commit comments

Comments
 (0)