diff --git a/src/main/java/io/weaviate/client6/v1/api/collections/rerankers/CohereReranker.java b/src/main/java/io/weaviate/client6/v1/api/collections/rerankers/CohereReranker.java index 5cdf60c16..e762b3174 100644 --- a/src/main/java/io/weaviate/client6/v1/api/collections/rerankers/CohereReranker.java +++ b/src/main/java/io/weaviate/client6/v1/api/collections/rerankers/CohereReranker.java @@ -8,6 +8,7 @@ import io.weaviate.client6.v1.internal.ObjectBuilder; public record CohereReranker( + @SerializedName("baseURL") String baseUrl, @SerializedName("model") String model) implements Reranker { public static final String RERANK_ENGLISH_V2 = "rerank-english-v2.0"; @@ -32,12 +33,18 @@ public static CohereReranker of(Function> } public CohereReranker(Builder builder) { - this(builder.model); + this(builder.baseUrl, builder.model); } public static class Builder implements ObjectBuilder { + private String baseUrl; private String model; + public Builder baseUrl(String baseUrl) { + this.baseUrl = baseUrl; + return this; + } + public Builder model(String model) { this.model = model; return this; diff --git a/src/test/java/io/weaviate/client6/v1/internal/json/JSONTest.java b/src/test/java/io/weaviate/client6/v1/internal/json/JSONTest.java index ffb39f6cd..56ceafc12 100644 --- a/src/test/java/io/weaviate/client6/v1/internal/json/JSONTest.java +++ b/src/test/java/io/weaviate/client6/v1/internal/json/JSONTest.java @@ -1362,10 +1362,12 @@ public static Object[][] testCases() { { Reranker.class, Reranker.cohere(rerank -> rerank + .baseUrl("example.com") .model(CohereReranker.RERANK_ENGLISH_V2)), """ { "reranker-cohere": { + "baseURL": "example.com", "model": "rerank-english-v2.0" } }