feat: extend generative functionality to async client

Author: bevzzz

src/main/java/io/weaviate/client6/v1/api/collections/CollectionHandleAsync.java

@@ -8,6 +8,7 @@
 import io.weaviate.client6.v1.api.collections.aggregate.WeaviateAggregateClientAsync;
 import io.weaviate.client6.v1.api.collections.config.WeaviateConfigClientAsync;
 import io.weaviate.client6.v1.api.collections.data.WeaviateDataClientAsync;
+import io.weaviate.client6.v1.api.collections.generate.WeaviateGenerateClientAsync;
 import io.weaviate.client6.v1.api.collections.pagination.AsyncPaginator;
 import io.weaviate.client6.v1.api.collections.query.ConsistencyLevel;
 import io.weaviate.client6.v1.api.collections.query.WeaviateQueryClientAsync;
@@ -21,6 +22,7 @@ public class CollectionHandleAsync<PropertiesT> {
   public final WeaviateConfigClientAsync config;
   public final WeaviateDataClientAsync<PropertiesT> data;
   public final WeaviateQueryClientAsync<PropertiesT> query;
+  public final WeaviateGenerateClientAsync<PropertiesT> generate;
   public final WeaviateAggregateClientAsync aggregate;
   public final WeaviateTenantsClientAsync tenants;
 
@@ -35,6 +37,7 @@ public CollectionHandleAsync(
     this.config = new WeaviateConfigClientAsync(collection, restTransport, grpcTransport, defaults);
     this.aggregate = new WeaviateAggregateClientAsync(collection, grpcTransport, defaults);
     this.query = new WeaviateQueryClientAsync<>(collection, grpcTransport, defaults);
+    this.generate = new WeaviateGenerateClientAsync<>(collection, grpcTransport, defaults);
     this.data = new WeaviateDataClientAsync<>(collection, restTransport, grpcTransport, defaults);
     this.defaults = defaults;
 
@@ -46,6 +49,7 @@ private CollectionHandleAsync(CollectionHandleAsync<PropertiesT> c, CollectionHa
     this.config = new WeaviateConfigClientAsync(c.config, defaults);
     this.aggregate = new WeaviateAggregateClientAsync(c.aggregate, defaults);
     this.query = new WeaviateQueryClientAsync<>(c.query, defaults);
+    this.generate = new WeaviateGenerateClientAsync<>(c.generate, defaults);
     this.data = new WeaviateDataClientAsync<>(c.data, defaults);
     this.defaults = defaults;
 

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

@@ -0,0 +1,41 @@
+package io.weaviate.client6.v1.api.collections.generate;
+
+import java.util.concurrent.CompletableFuture;
+
+import io.weaviate.client6.v1.api.collections.CollectionHandleDefaults;
+import io.weaviate.client6.v1.api.collections.query.GroupBy;
+import io.weaviate.client6.v1.api.collections.query.QueryOperator;
+import io.weaviate.client6.v1.internal.grpc.GrpcTransport;
+import io.weaviate.client6.v1.internal.orm.CollectionDescriptor;
+
+public class WeaviateGenerateClientAsync<PropertiesT>
+    extends
+    AbstractGenerateClient<PropertiesT, CompletableFuture<GenerativeResponse<PropertiesT>>, CompletableFuture<GenerativeResponseGrouped<PropertiesT>>> {
+
+  public WeaviateGenerateClientAsync(
+      CollectionDescriptor<PropertiesT> collection,
+      GrpcTransport grpcTransport,
+      CollectionHandleDefaults defaults) {
+    super(collection, grpcTransport, defaults);
+  }
+
+  /** Copy constructor that sets new defaults. */
+  public WeaviateGenerateClientAsync(WeaviateGenerateClientAsync<PropertiesT> c, CollectionHandleDefaults defaults) {
+    super(c, defaults);
+  }
+
+  @Override
+  protected final CompletableFuture<GenerativeResponse<PropertiesT>> performRequest(QueryOperator operator,
+      GenerativeTask generate) {
+    var request = new GenerativeRequest(operator, generate, null);
+    return this.grpcTransport.performRequestAsync(request, GenerativeRequest.rpc(collection, defaults));
+  }
+
+  @Override
+  protected final CompletableFuture<GenerativeResponseGrouped<PropertiesT>> performRequest(QueryOperator operator,
+      GenerativeTask generate,
+      GroupBy groupBy) {
+    var request = new GenerativeRequest(operator, generate, groupBy);
+    return this.grpcTransport.performRequestAsync(request, GenerativeRequest.grouped(collection, defaults));
+  }
+}