feat: add tests for translation model

Author: ikesnowy

src/Cnblogs.DashScope.Core/ITextGenerationParameters.cs

@@ -41,6 +41,11 @@ public interface ITextGenerationParameters
     /// </summary>
     public bool? EnableSearch { get; }
 
+    /// <summary>
+    /// Search options. <see cref="EnableSearch"/> should set to true.
+    /// </summary>
+    public TextGenerationSearchOptions? SearchOptions { get; set; }
+
     /// <summary>
     /// Thinking option. Valid for supported models.(e.g. qwen3)
     /// </summary>

src/Cnblogs.DashScope.Core/TextGenerationOutput.cs

@@ -19,4 +19,9 @@ public class TextGenerationOutput
     /// Not null when <see cref="TextGenerationParameters"/>.<see cref="TextGenerationParameters.ResultFormat"/> is "message".
     /// </summary>
     public List<TextGenerationChoice>? Choices { get; set; }
+
+    /// <summary>
+    /// Not null when <see cref="TextGenerationParameters"/>.<see cref="TextGenerationParameters.SearchOptions"/> configured to show source.
+    /// </summary>
+    public TextGenerationWebSearchInfo? SearchInfo { get; set; }
 }

src/Cnblogs.DashScope.Core/TextGenerationParameters.cs

@@ -38,6 +38,9 @@ public class TextGenerationParameters : ITextGenerationParameters
     /// <inheritdoc />
     public bool? EnableSearch { get; set; }
 
+    /// <inheritdoc />
+    public TextGenerationSearchOptions? SearchOptions { get; set; }
+
     /// <inheritdoc />
     public bool? EnableThinking { get; set; }
 

src/Cnblogs.DashScope.Core/TextGenerationSearchOptions.cs

@@ -0,0 +1,32 @@
+namespace Cnblogs.DashScope.Core;
+
+/// <summary>
+/// Web search options
+/// </summary>
+public class TextGenerationSearchOptions
+{
+    /// <summary>
+    /// Show search result in response. Defaults to false.
+    /// </summary>
+    public bool? EnableSource { get; set; }
+
+    /// <summary>
+    /// Include citation in output. Defaults to false.
+    /// </summary>
+    public bool? EnableCitation { get; set; }
+
+    /// <summary>
+    /// Citation format. Defaults to "[&lt;number&gt;]"
+    /// </summary>
+    public string? CitationFormat { get; set; }
+
+    /// <summary>
+    /// Force model to use web search. Defaults to false.
+    /// </summary>
+    public bool? ForcedSearch { get; set; }
+
+    /// <summary>
+    /// How many search records should be provided to model. "standard" - 5 records. "pro" - 10 records.
+    /// </summary>
+    public string? SearchStrategy { get; set; }
+}

src/Cnblogs.DashScope.Core/TextGenerationWebSearchInfo.cs

@@ -0,0 +1,7 @@
+namespace Cnblogs.DashScope.Core;
+
+/// <summary>
+/// Web search information.
+/// </summary>
+/// <param name="SearchResults">Web search results.</param>
+public record TextGenerationWebSearchInfo(List<TextGenerationWebSearchResult> SearchResults);

src/Cnblogs.DashScope.Core/TextGenerationWebSearchResult.cs

@@ -0,0 +1,11 @@
+namespace Cnblogs.DashScope.Core;
+
+/// <summary>
+/// Represents one web search record.
+/// </summary>
+/// <param name="SiteName">Source site name.</param>
+/// <param name="Icon">Source site favicon url.</param>
+/// <param name="Index">Serial number of search records.</param>
+/// <param name="Title">Page title.</param>
+/// <param name="Url">Page url.</param>
+public record TextGenerationWebSearchResult(string SiteName, string Icon, int Index, string Title, string Url);

test/Cnblogs.DashScope.Sdk.UnitTests/TextGenerationSerializationTests.cs

@@ -156,7 +156,8 @@ public async Task ConversationCompletion_MessageFormatSse_SuccessAsync(
         Snapshots.TextGeneration.MessageFormat.SingleMessageReasoning,
         Snapshots.TextGeneration.MessageFormat.SingleMessageWithTools,
         Snapshots.TextGeneration.MessageFormat.SingleMessageJson,
-        Snapshots.TextGeneration.MessageFormat.SingleMessageLogprobs);
+        Snapshots.TextGeneration.MessageFormat.SingleMessageLogprobs,
+        Snapshots.TextGeneration.MessageFormat.SingleMessageTranslation);
 
     public static readonly TheoryData<RequestSnapshot<ModelRequest<TextGenerationInput, ITextGenerationParameters>,
         ModelResponse<TextGenerationOutput, TextGenerationTokenUsage>>> SingleGenerationMessageSseFormatData = new(

test/Cnblogs.DashScope.Tests.Shared/RawHttpData/single-generation-message-translation-nosse.request.body.json

@@ -0,0 +1,32 @@
+{
+  "model": "qwen-mt-plus",
+  "input": {
+    "messages": [
+      {
+        "role": "user",
+        "content": "博客园的理念是代码改变世界"
+      }
+    ]
+  },
+  "parameters": {
+    "result_format": "message",
+    "incremental_output": false,
+    "translation_options": {
+      "source_lang": "Chinese",
+      "target_lang": "English",
+      "terms": [
+        {
+          "source": "博客园",
+          "target": "cnblogs"
+        }
+      ],
+      "tm_list": [
+        {
+          "source": "代码改变世界",
+          "target": "Coding changes world"
+        }
+      ],
+      "domains": "This text is a promotion."
+    }
+  }
+}

test/Cnblogs.DashScope.Tests.Shared/RawHttpData/single-generation-message-translation-nosse.request.header.txt

@@ -0,0 +1,8 @@
+POST /api/v1/services/aigc/text-generation/generation HTTP/1.1
+Content-Type: application/json
+Accept: */*
+Cache-Control: no-cache
+Host: dashscope.aliyuncs.com
+Accept-Encoding: gzip, deflate, br
+Connection: keep-alive
+Content-Length: 85

test/Cnblogs.DashScope.Tests.Shared/RawHttpData/single-generation-message-translation-nosse.response.body.txt

@@ -0,0 +1 @@
+{"output":{"finish_reason":"stop","model_name":"qwen-mt-plus","choices":[{"finish_reason":"stop","message":{"role":"assistant","content":"The concept of cnblogs is that coding changes world "}}]},"usage":{"total_tokens":122,"output_tokens":11,"input_tokens":111},"request_id":"bf86e0f9-a8a2-9b32-be8d-ea3cae47c8ea"}