feat: add rerank argument to other vector-based queries

Author: bevzzz

src/main/java/io/weaviate/client6/v1/api/collections/query/NearDepth.java

@@ -10,7 +10,12 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearDepth(Target searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearDepth(
+    Target searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static NearDepth of(String depth) {
@@ -34,6 +39,7 @@ public NearDepth(Builder builder) {
         builder.media,
         builder.distance,
         builder.certainty,
+        builder.rerank,
         builder.baseOptions());
   }
 

src/main/java/io/weaviate/client6/v1/api/collections/query/NearImage.java

@@ -10,7 +10,12 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearImage(Target searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearImage(
+    Target searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static NearImage of(String image) {
@@ -34,6 +39,7 @@ public NearImage(Builder builder) {
         builder.media,
         builder.distance,
         builder.certainty,
+        builder.rerank,
         builder.baseOptions());
   }
 

src/main/java/io/weaviate/client6/v1/api/collections/query/NearImu.java

@@ -10,7 +10,12 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearImu(Target searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearImu(
+    Target searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static NearImu of(String imu) {
@@ -34,6 +39,7 @@ public NearImu(Builder builder) {
         builder.media,
         builder.distance,
         builder.certainty,
+        builder.rerank,
         builder.baseOptions());
   }
 

src/main/java/io/weaviate/client6/v1/api/collections/query/NearObject.java

@@ -8,7 +8,12 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearObject(String uuid, Float distance, Float certainty, BaseQueryOptions common)
+public record NearObject(
+    String uuid,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static final NearObject of(String uuid) {
@@ -20,7 +25,12 @@ public static final NearObject of(String uuid, Function<Builder, ObjectBuilder<N
   }
 
   public NearObject(Builder builder) {
-    this(builder.uuid, builder.distance, builder.certainty, builder.baseOptions());
+    this(
+        builder.uuid,
+        builder.distance,
+        builder.certainty,
+        builder.rerank,
+        builder.baseOptions());
   }
 
   public static class Builder extends BaseVectorSearchBuilder<Builder, NearObject> {

src/main/java/io/weaviate/client6/v1/api/collections/query/NearThermal.java

@@ -10,7 +10,11 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearThermal(Target searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearThermal(Target searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static NearThermal of(String thermal) {
@@ -34,6 +38,7 @@ public NearThermal(Builder builder) {
         builder.media,
         builder.distance,
         builder.certainty,
+        builder.rerank,
         builder.baseOptions());
   }
 

src/main/java/io/weaviate/client6/v1/api/collections/query/NearVector.java

@@ -8,7 +8,11 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearVector(NearVectorTarget searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearVector(NearVectorTarget searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static final NearVector of(float[] vector) {
@@ -36,7 +40,11 @@ public static final NearVector of(NearVectorTarget searchTarget, Function<Builde
   }
 
   public NearVector(Builder builder) {
-    this(builder.searchTarget, builder.distance, builder.certainty, builder.baseOptions());
+    this(builder.searchTarget,
+        builder.distance,
+        builder.certainty,
+        builder.rerank,
+        builder.baseOptions());
   }
 
   public static class Builder extends BaseVectorSearchBuilder<Builder, NearVector> {

src/main/java/io/weaviate/client6/v1/api/collections/query/NearVideo.java

@@ -10,7 +10,12 @@
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoBaseSearch;
 import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoSearchGet;
 
-public record NearVideo(Target searchTarget, Float distance, Float certainty, BaseQueryOptions common)
+public record NearVideo(
+    Target searchTarget,
+    Float distance,
+    Float certainty,
+    Rerank rerank,
+    BaseQueryOptions common)
     implements QueryOperator, AggregateObjectFilter {
 
   public static NearVideo of(String video) {
@@ -34,6 +39,7 @@ public NearVideo(Builder builder) {
         builder.media,
         builder.distance,
         builder.certainty,
+        builder.rerank,
         builder.baseOptions());
   }