From aa6b2af19cacaba21928861b8e32f2e7ae30eec3 Mon Sep 17 00:00:00 2001 From: dyma solovei Date: Wed, 8 Apr 2026 14:17:41 +0200 Subject: [PATCH] feat(gh-522): add baseURL parameter to cohere reranker --- .../v1/api/collections/rerankers/CohereReranker.java | 9 ++++++++- .../io/weaviate/client6/v1/internal/json/JSONTest.java | 2 ++ 2 files changed, 10 insertions(+), 1 deletion(-) 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" } }