feat: Add some async extension methods

Author: amanda-tarafa

apis/Google.Cloud.VertexAI.Extensions/Google.Cloud.VertexAI.Extensions/VertexAIExtensions.cs

@@ -16,6 +16,8 @@
 using Google.Cloud.AIPlatform.V1;
 using Microsoft.Extensions.AI;
 using System;
+using System.Threading;
+using System.Threading.Tasks;
 
 namespace Google.Cloud.VertexAI.Extensions;
 
@@ -52,7 +54,8 @@ public static IChatClient AsIChatClient(
     /// <returns>An <see cref="IChatClient"/> that wraps the built client.</returns>
     /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
     public static IChatClient BuildIChatClient(
-        this PredictionServiceClientBuilder builder, IServiceProvider? provider = null, string? defaultModelId = null)
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null)
     {
         GaxPreconditions.CheckNotNull(builder, nameof(builder));
 
@@ -63,6 +66,32 @@ public static IChatClient BuildIChatClient(
         return client.AsIChatClient(defaultModelId);
     }
 
+    /// <summary>
+    /// Builds a <see cref="PredictionServiceClient"/> and creates an <see cref="IChatClient"/> wrapper around it.
+    /// </summary>
+    /// <param name="builder">The <see cref="PredictionServiceClientBuilder"/> with which to build the <see cref="PredictionServiceClient"/>.</param>
+    /// <param name="provider">An optional <see cref="IServiceProvider"/> from which services are requested when building the client.</param>
+    /// <param name="defaultModelId">
+    /// The default model ID to use for chat requests if not specified in <see cref="ChatOptions.ModelId"/>.
+    /// This must be the full resource name of the model, e.g. "projects/{projectId}/locations/{location}/publishers/{publisher}/models/{model}".
+    /// </param>
+    /// <param name="cancellationToken">A token to cancel the async operation.</param>
+    /// <returns>An <see cref="IChatClient"/> that wraps the built client.</returns>
+    /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
+    public static async Task<IChatClient> BuildIChatClientAsync(
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null,
+        CancellationToken cancellationToken = default)
+    {
+        GaxPreconditions.CheckNotNull(builder, nameof(builder));
+
+        PredictionServiceClient client = await (provider is not null ?
+            builder.BuildAsync(provider, cancellationToken) :
+            builder.BuildAsync(cancellationToken)).ConfigureAwait(false);
+
+        return client.AsIChatClient(defaultModelId);
+    }
+
     /// <summary>
     /// Creates an <see cref="IEmbeddingGenerator{String, Embedding}"/> wrapper around the specified <see cref="PredictionServiceClient"/>.
     /// </summary>
@@ -94,7 +123,8 @@ public static IEmbeddingGenerator<string, Embedding<float>> AsIEmbeddingGenerato
     /// <returns>An <see cref="IEmbeddingGenerator{String, Embedding}"/> that wraps the built client.</returns>
     /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
     public static IEmbeddingGenerator<string, Embedding<float>> BuildIEmbeddingGenerator(
-        this PredictionServiceClientBuilder builder, IServiceProvider? provider = null, string? defaultModelId = null)
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null)
     {
         GaxPreconditions.CheckNotNull(builder, nameof(builder));
 
@@ -105,6 +135,32 @@ public static IEmbeddingGenerator<string, Embedding<float>> BuildIEmbeddingGener
         return client.AsIEmbeddingGenerator(defaultModelId);
     }
 
+    /// <summary>
+    /// Builds a <see cref="PredictionServiceClient"/> and creates an <see cref="IEmbeddingGenerator{String, Embedding}"/> wrapper around it.
+    /// </summary>
+    /// <param name="builder">The <see cref="PredictionServiceClientBuilder"/> with which to build the <see cref="PredictionServiceClient"/>.</param>
+    /// <param name="provider">An optional <see cref="IServiceProvider"/> from which services are requested when building the client.</param>
+    /// <param name="defaultModelId">
+    /// The default model ID to use for chat requests if not specified in <see cref="EmbeddingGenerationOptions.ModelId"/>.
+    /// This must be the full resource name of the model, e.g. "projects/{projectId}/locations/{location}/publishers/{publisher}/models/{model}".
+    /// </param>
+    /// <param name="cancellationToken">A token to cancel the async operation.</param>
+    /// <returns>An <see cref="IEmbeddingGenerator{String, Embedding}"/> that wraps the built client.</returns>
+    /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
+    public static async Task<IEmbeddingGenerator<string, Embedding<float>>> BuildIEmbeddingGeneratorAsync(
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null,
+        CancellationToken cancellationToken = default)
+    {
+        GaxPreconditions.CheckNotNull(builder, nameof(builder));
+
+        PredictionServiceClient client = await (provider is not null ?
+            builder.BuildAsync(provider, cancellationToken) :
+            builder.BuildAsync(cancellationToken)).ConfigureAwait(false);
+
+        return client.AsIEmbeddingGenerator(defaultModelId);
+    }
+
     /// <summary>
     /// Creates an <see cref="IImageGenerator"/> wrapper around the specified <see cref="PredictionServiceClient"/>.
     /// </summary>
@@ -135,7 +191,8 @@ public static IImageGenerator AsIImageGenerator(
     /// <returns>An <see cref="IImageGenerator"/> that wraps the built client.</returns>
     /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
     public static IImageGenerator BuildIImageGenerator(
-        this PredictionServiceClientBuilder builder, IServiceProvider? provider = null, string? defaultModelId = null)
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null)
     {
         GaxPreconditions.CheckNotNull(builder, nameof(builder));
 
@@ -146,6 +203,32 @@ public static IImageGenerator BuildIImageGenerator(
         return client.AsIImageGenerator(defaultModelId);
     }
 
+    /// <summary>
+    /// Builds a <see cref="PredictionServiceClient"/> and creates an <see cref="IImageGenerator"/> wrapper around it.
+    /// </summary>
+    /// <param name="builder">The <see cref="PredictionServiceClientBuilder"/> with which to build the <see cref="PredictionServiceClient"/>.</param>
+    /// <param name="defaultModelId">
+    /// The default model ID to use for chat requests if not specified in <see cref="ImageGenerationOptions.ModelId"/>.
+    /// This must be the full resource name of the model, e.g. "projects/{projectId}/locations/{location}/publishers/{publisher}/models/{model}".
+    /// </param>
+    /// <param name="provider">An optional <see cref="IServiceProvider"/> from which services are requested when building the client.</param>
+    /// <param name="cancellationToken">A token to cancel the async operation.</param>
+    /// <returns>An <see cref="IImageGenerator"/> that wraps the built client.</returns>
+    /// <exception cref="ArgumentNullException"><paramref name="builder"/> is <see langword="null"/>.</exception>
+    public static async Task<IImageGenerator> BuildIImageGeneratorAsync(
+        this PredictionServiceClientBuilder builder,
+        IServiceProvider? provider = null, string? defaultModelId = null,
+        CancellationToken cancellationToken = default)
+    {
+        GaxPreconditions.CheckNotNull(builder, nameof(builder));
+
+        PredictionServiceClient client = await (provider is not null ?
+            builder.BuildAsync(provider, cancellationToken) :
+            builder.BuildAsync(cancellationToken)).ConfigureAwait(false);
+
+        return client.AsIImageGenerator(defaultModelId);
+    }
+
     /// <summary>Creates an <see cref="AITool"/> to represent a raw <see cref="Tool"/>.</summary>
     /// <param name="tool">The tool to wrap as an <see cref="AITool"/>.</param>
     /// <returns>The <paramref name="tool"/> wrapped as an <see cref="AITool"/>.</returns>