feat: add generative w/ nearObject query

Author: bevzzz

src/main/java/io/weaviate/client6/v1/api/collections/generate/AbstractGenerateClient.java

@@ -8,6 +8,7 @@
 import io.weaviate.client6.v1.api.collections.query.FetchObjects;
 import io.weaviate.client6.v1.api.collections.query.GroupBy;
 import io.weaviate.client6.v1.api.collections.query.Hybrid;
+import io.weaviate.client6.v1.api.collections.query.NearObject;
 import io.weaviate.client6.v1.api.collections.query.NearVector;
 import io.weaviate.client6.v1.api.collections.query.NearVectorTarget;
 import io.weaviate.client6.v1.api.collections.query.QueryOperator;
@@ -542,4 +543,103 @@ public GroupedResponseT nearVector(NearVectorTarget searchTarget,
   public GroupedResponseT nearVector(NearVector query, GenerativeTask generate, GroupBy groupBy) {
     return performRequest(query, generate, groupBy);
   }
+
+  // NearObject queries -------------------------------------------------------
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param uuid       Query object UUID.
+   * @param generateFn Lambda expression for generative task parameters.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   */
+  public ResponseT nearObject(String uuid,
+      Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn) {
+    return nearObject(NearObject.of(uuid), GenerativeTask.of(generateFn));
+  }
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param uuid       Query object UUID.
+   * @param fn         Lambda expression for optional search parameters.
+   * @param generateFn Lambda expression for generative task parameters.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   */
+  public ResponseT nearObject(String uuid,
+      Function<NearObject.Builder, ObjectBuilder<NearObject>> fn,
+      Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn) {
+    return nearObject(NearObject.of(uuid, fn), GenerativeTask.of(generateFn));
+  }
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param query    Near object query request.
+   * @param generate Generative task.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   */
+  public ResponseT nearObject(NearObject query, GenerativeTask generate) {
+    return performRequest(query, generate);
+  }
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param uuid       Query object UUID.
+   * @param generateFn Lambda expression for generative task parameters.
+   * @param groupBy    Group-by clause.
+   * @return Grouped query result.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   *
+   * @see GroupBy
+   * @see QueryResponseGrouped
+   */
+  public GroupedResponseT nearObject(String uuid,
+      Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn,
+      GroupBy groupBy) {
+    return nearObject(NearObject.of(uuid), GenerativeTask.of(generateFn), groupBy);
+  }
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param uuid       Query object UUID.
+   * @param fn         Lambda expression for optional search parameters.
+   * @param generateFn Lambda expression for generative task parameters.
+   * @param groupBy    Group-by clause.
+   * @return Grouped query result.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   *
+   * @see GroupBy
+   * @see QueryResponseGrouped
+   */
+  public GroupedResponseT nearObject(String uuid,
+      Function<NearObject.Builder, ObjectBuilder<NearObject>> fn,
+      Function<GenerativeTask.Builder, ObjectBuilder<GenerativeTask>> generateFn,
+      GroupBy groupBy) {
+    return nearObject(NearObject.of(uuid, fn), GenerativeTask.of(generateFn), groupBy);
+  }
+
+  /**
+   * Query collection objects using near object search.
+   *
+   * @param query    Near object query request.
+   * @param generate Generative task.
+   * @param groupBy  Group-by clause.
+   * @return Grouped query result.
+   * @throws WeaviateApiException in case the server returned with an
+   *                              error status code.
+   *
+   * @see GroupBy
+   * @see QueryResponseGrouped
+   */
+  public GroupedResponseT nearObject(NearObject query, GenerativeTask generate, GroupBy groupBy) {
+    return performRequest(query, generate, groupBy);
+  }
 }