feat: support wanx image generation

Author: ikesnowy

Cnblogs.DashScope.Sdk.sln.DotSettings

@@ -1,2 +1,3 @@
 <wpf:ResourceDictionary xml:space="preserve" xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" xmlns:s="clr-namespace:System;assembly=mscorlib" xmlns:ss="urn:shemas-jetbrains-com:settings-storage-xaml" xmlns:wpf="http://schemas.microsoft.com/winfx/2006/xaml/presentation">
-	<s:Boolean x:Key="/Default/Environment/Filtering/ExcludeCoverageFilters/=Cnblogs_002EDashScope_002ESample_003B_002A_003B_002A_003B_002A/@EntryIndexedValue">True</s:Boolean></wpf:ResourceDictionary>
+	<s:Boolean x:Key="/Default/Environment/Filtering/ExcludeCoverageFilters/=Cnblogs_002EDashScope_002ESample_003B_002A_003B_002A_003B_002A/@EntryIndexedValue">True</s:Boolean>
+	<s:Boolean x:Key="/Default/Environment/Filtering/ExcludeCoverageFilters/=Cnblogs_002EDashScope_002ESdk_002EUnitTests_003B_002A_003B_002A_003B_002A/@EntryIndexedValue">True</s:Boolean></wpf:ResourceDictionary>

src/Cnblogs.DashScope.Sdk/DashScopeClientCore.cs

@@ -180,6 +180,16 @@ public async Task<ModelResponse<TokenizationOutput, TokenizationUsage>> Tokenize
         return (await SendAsync<ModelResponse<TokenizationOutput, TokenizationUsage>>(request, cancellationToken))!;
     }
 
+    /// <inheritdoc />
+    public async Task<ModelResponse<ImageGenerationOutput, ImageGenerationUsage>> CreateImageGenerationTaskAsync(
+        ModelRequest<ImageGenerationInput> input,
+        CancellationToken cancellationToken = default)
+    {
+        var request = BuildRequest(HttpMethod.Post, ApiLinks.ImageGeneration, input, isTask: true);
+        return (await SendAsync<ModelResponse<ImageGenerationOutput, ImageGenerationUsage>>(request, cancellationToken))
+            !;
+    }
+
     private static HttpRequestMessage BuildSseRequest<TPayload>(HttpMethod method, string url, TPayload payload)
         where TPayload : class
     {

src/Cnblogs.DashScope.Sdk/IDashScopeClient.cs

@@ -128,4 +128,14 @@ Task<DashScopeTaskList> ListTasksAsync(
     Task<ModelResponse<TokenizationOutput, TokenizationUsage>> TokenizeAsync(
         ModelRequest<TextGenerationInput, TextGenerationParameters> input,
         CancellationToken cancellationToken = default);
+
+    /// <summary>
+    /// Create a image generation task.
+    /// </summary>
+    /// <param name="input">The input of task.</param>
+    /// <param name="cancellationToken">The cancellation token to use.</param>
+    /// <returns></returns>
+    Task<ModelResponse<ImageGenerationOutput, ImageGenerationUsage>> CreateImageGenerationTaskAsync(
+        ModelRequest<ImageGenerationInput> input,
+        CancellationToken cancellationToken = default);
 }

src/Cnblogs.DashScope.Sdk/ImageGenerationInput.cs

@@ -0,0 +1,17 @@
+namespace Cnblogs.DashScope.Sdk;
+
+/// <summary>
+/// The input of image generation.
+/// </summary>
+public class ImageGenerationInput
+{
+    /// <summary>
+    /// The image url to generation new image from.
+    /// </summary>
+    public required string ImageUrl { get; set; }
+
+    /// <summary>
+    /// The style the new image should use, checkout docs for sample image of different indexes: https://help.aliyun.com/zh/dashscope/developer-reference/tongyi-wanxiang-style-repaint
+    /// </summary>
+    public required int StyleIndex { get; set; }
+}

src/Cnblogs.DashScope.Sdk/ImageGenerationOutput.cs

@@ -0,0 +1,36 @@
+using System.Text.Json.Serialization;
+using Cnblogs.DashScope.Sdk.Internals;
+
+namespace Cnblogs.DashScope.Sdk;
+
+/// <summary>
+/// The output of image generation task.
+/// </summary>
+public record ImageGenerationOutput : DashScopeTaskOutput
+{
+    /// <summary>
+    /// The generated image url.
+    /// </summary>
+    public List<ImageGenerationResult>? Results { get; set; }
+
+    /// <summary>
+    /// The style index of generated image.
+    /// </summary>
+    public int StyleIndex { get; set; }
+
+    /// <summary>
+    /// The generation start time.
+    /// </summary>
+    [JsonConverter(typeof(DashScopeDateTimeConvertor))]
+    public DateTime? StartTime { get; set; }
+
+    /// <summary>
+    /// The error code, will be 0 when succeeded.
+    /// </summary>
+    public int? ErrorCode { get; set; }
+
+    /// <summary>
+    /// The error message, will be 'Success' when succeeded.
+    /// </summary>
+    public string? ErrorMessage { get; set; }
+}

src/Cnblogs.DashScope.Sdk/ImageGenerationResult.cs

@@ -0,0 +1,7 @@
+namespace Cnblogs.DashScope.Sdk;
+
+/// <summary>
+/// The result of image generation task.
+/// </summary>
+/// <param name="Url">The url of generated image.</param>
+public record ImageGenerationResult(string Url);

src/Cnblogs.DashScope.Sdk/ImageGenerationUsage.cs

@@ -0,0 +1,7 @@
+namespace Cnblogs.DashScope.Sdk;
+
+/// <summary>
+/// The usage of one image generation request.
+/// </summary>
+/// <param name="ImageCount">The generated image count.</param>
+public record ImageGenerationUsage(int ImageCount);

src/Cnblogs.DashScope.Sdk/ModelRequest.cs

@@ -4,10 +4,8 @@
 /// Represents a request for model generation.
 /// </summary>
 /// <typeparam name="TInput">The input type for this request.</typeparam>
-/// <typeparam name="TParameter">The option type for this request.</typeparam>
-public class ModelRequest<TInput, TParameter>
+public class ModelRequest<TInput>
     where TInput : class
-    where TParameter : class
 {
     /// <summary>
     /// The model to use.
@@ -18,7 +16,17 @@ public class ModelRequest<TInput, TParameter>
     /// Input of this request.
     /// </summary>
     public required TInput Input { get; init; }
+}
 
+/// <summary>
+/// Represents a request for model generation.
+/// </summary>
+/// <typeparam name="TInput">The input type for this request.</typeparam>
+/// <typeparam name="TParameter">The option type for this request.</typeparam>
+public class ModelRequest<TInput, TParameter> : ModelRequest<TInput>
+    where TInput : class
+    where TParameter : class
+{
     /// <summary>
     /// Optional configuration of this request.
     /// </summary>

src/Cnblogs.DashScope.Sdk/Wanx/WanxGenerationApi.cs

@@ -75,4 +75,57 @@ public static Task<DashScopeTask<ImageSynthesisOutput, ImageSynthesisUsage>> Get
     {
         return dashScopeClient.GetTaskAsync<ImageSynthesisOutput, ImageSynthesisUsage>(taskId, cancellationToken);
     }
+
+    /// <summary>
+    /// Creates a wanx image generation task with given model and input.
+    /// </summary>
+    /// <param name="dashScopeClient">The <see cref="IDashScopeClient"/>.</param>
+    /// <param name="model">The model to use.</param>
+    /// <param name="input">The input for image to be generate from.</param>
+    /// <param name="cancellationToken">The cancellation token to use.</param>
+    /// <returns>The generated task.</returns>
+    public static Task<DashScopeTaskOutput> CreateWanxImageGenerationTaskAsync(
+        this IDashScopeClient dashScopeClient,
+        WanxStyleRepaintModel model,
+        ImageGenerationInput input,
+        CancellationToken cancellationToken = default)
+    {
+        return dashScopeClient.CreateWanxImageGenerationTaskAsync(model.GetModelName(), input, cancellationToken);
+    }
+
+    /// <summary>
+    /// Creates a wanx image generation task with given model and input.
+    /// </summary>
+    /// <param name="dashScopeClient">The <see cref="IDashScopeClient"/>.</param>
+    /// <param name="model">The model to use.</param>
+    /// <param name="input">The input for image to be generate from.</param>
+    /// <param name="cancellationToken">The cancellation token to use.</param>
+    /// <returns>The generated task.</returns>
+    public static async Task<DashScopeTaskOutput> CreateWanxImageGenerationTaskAsync(
+        this IDashScopeClient dashScopeClient,
+        string model,
+        ImageGenerationInput input,
+        CancellationToken cancellationToken = default)
+    {
+        var response = await dashScopeClient.CreateImageGenerationTaskAsync(
+            new ModelRequest<ImageGenerationInput> { Model = model, Input = input },
+            cancellationToken);
+        return response.Output;
+    }
+
+    /// <summary>
+    /// Get wanx image generation task detail.
+    /// </summary>
+    /// <param name="dashScopeClient">The <see cref="IDashScopeClient"/>.</param>
+    /// <param name="taskId">The task id to query.</param>
+    /// <param name="cancellationToken">The cancellation token to use.</param>
+    /// <returns></returns>
+    public static Task<DashScopeTask<ImageGenerationOutput, ImageGenerationUsage>>
+        GetWanxImageGenerationTaskAsync(
+            this IDashScopeClient dashScopeClient,
+            string taskId,
+            CancellationToken cancellationToken = default)
+    {
+        return dashScopeClient.GetTaskAsync<ImageGenerationOutput, ImageGenerationUsage>(taskId, cancellationToken);
+    }
 }

src/Cnblogs.DashScope.Sdk/Wanx/WanxModelNames.cs

@@ -12,4 +12,13 @@ public static string GetModelName(this WanxModel model)
             _ => ThrowHelper.UnknownModelName(nameof(model), model)
         };
     }
+
+    public static string GetModelName(this WanxStyleRepaintModel model)
+    {
+        return model switch
+        {
+            WanxStyleRepaintModel.WanxStyleRepaintingV1 => "wanx-style-repaint-v1",
+            _ => ThrowHelper.UnknownModelName(nameof(model), model)
+        };
+    }
 }