Merge pull request #558 from weaviate/feat/cohere-reranker

feat(config): add baseURL parameter to Cohere reranker

Author: bevzzz

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<Builder, ObjectBuilder<CohereReranker>>
   }
 
   public CohereReranker(Builder builder) {
-    this(builder.model);
+    this(builder.baseUrl, builder.model);
   }
 
   public static class Builder implements ObjectBuilder<CohereReranker> {
+    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;

src/test/java/io/weaviate/client6/v1/internal/json/JSONTest.java

@@ -1363,10 +1363,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"
                   }
                 }