feat: support llama2 text generation

Author: ikesnowy

src/Cnblogs.DashScope.Sdk/Llama2/Llama2Model.cs

@@ -0,0 +1,17 @@
+namespace Cnblogs.DashScope.Sdk.Llama2;
+
+/// <summary>
+/// Supported models for LLaMa2.
+/// </summary>
+public enum Llama2Model
+{
+    /// <summary>
+    /// llama2-7b-chat-v2
+    /// </summary>
+    Chat7Bv2 = 1,
+
+    /// <summary>
+    /// llama2-13b-chat-v2
+    /// </summary>
+    Chat13Bv2 = 2
+}

src/Cnblogs.DashScope.Sdk/Llama2/Llama2ModelNames.cs

@@ -0,0 +1,17 @@
+namespace Cnblogs.DashScope.Sdk.Llama2;
+
+internal static class Llama2ModelNames
+{
+    public static string GetModelName(this Llama2Model model)
+    {
+        return model switch
+        {
+            Llama2Model.Chat7Bv2 => "llama2-7b-chat-v2",
+            Llama2Model.Chat13Bv2 => "llama2-13b-chat-v2",
+            _ => throw new ArgumentOutOfRangeException(
+                nameof(model),
+                model,
+                "Unknown model type, please use the overload that accepts a string ‘model’ parameter.")
+        };
+    }
+}

src/Cnblogs.DashScope.Sdk/Llama2/Llama2TextGenerationApi.cs

@@ -0,0 +1,51 @@
+namespace Cnblogs.DashScope.Sdk.Llama2;
+
+/// <summary>
+/// Extensions for llama2 text generation, docs: https://help.aliyun.com/zh/dashscope/developer-reference/api-details-11
+/// </summary>
+public static class Llama2TextGenerationApi
+{
+    /// <summary>
+    /// Get text completion from llama2 model.
+    /// </summary>
+    /// <param name="client">The <see cref="IDashScopeClient"/>.</param>
+    /// <param name="model">The model name.</param>
+    /// <param name="messages">The context messages.</param>
+    /// <param name="resultFormat">Can be 'text' or 'message'. Call <see cref="ResultFormats"/> to get available options.</param>
+    /// <returns></returns>
+    public static async Task<ModelResponse<TextGenerationOutput, TextGenerationTokenUsage>>
+        GetLlama2TextCompletionAsync(
+            this IDashScopeClient client,
+            Llama2Model model,
+            IEnumerable<ChatMessage> messages,
+            string? resultFormat = null)
+    {
+        return await client.GetLlama2TextCompletionAsync(model.GetModelName(), messages, resultFormat);
+    }
+
+    /// <summary>
+    /// Get text completion from llama2 model.
+    /// </summary>
+    /// <param name="client">The <see cref="IDashScopeClient"/>.</param>
+    /// <param name="model">The model name.</param>
+    /// <param name="messages">The context messages.</param>
+    /// <param name="resultFormat">Can be 'text' or 'message'. Call <see cref="ResultFormats"/> to get available options.</param>
+    /// <returns></returns>
+    public static async Task<ModelResponse<TextGenerationOutput, TextGenerationTokenUsage>>
+        GetLlama2TextCompletionAsync(
+            this IDashScopeClient client,
+            string model,
+            IEnumerable<ChatMessage> messages,
+            string? resultFormat = null)
+    {
+        return await client.GetTextCompletionAsync(
+            new ModelRequest<TextGenerationInput, TextGenerationParameters>
+            {
+                Model = model,
+                Input = new TextGenerationInput { Messages = messages },
+                Parameters = resultFormat != null
+                    ? new TextGenerationParameters { ResultFormat = resultFormat }
+                    : null
+            });
+    }
+}

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

@@ -0,0 +1,46 @@
+using Cnblogs.DashScope.Sdk.Llama2;
+using NSubstitute;
+
+namespace Cnblogs.DashScope.Sdk.UnitTests;
+
+public class Llama2TextGenerationTests
+{
+    private static readonly List<ChatMessage> Messages =
+        [new("system", "you are a helpful assistant"), new("user", "hello")];
+
+    [Fact]
+    public async Task Llama2_UseEnum_SuccessAsync()
+    {
+        // Arrange
+        var client = Substitute.For<IDashScopeClient>();
+
+        // Act
+        _ = await client.GetLlama2TextCompletionAsync(Llama2Model.Chat13Bv2, Messages, ResultFormats.Message);
+
+        // Assert
+        _ = await client.Received().GetTextCompletionAsync(
+            Arg.Is<ModelRequest<TextGenerationInput, TextGenerationParameters>>(
+                s => s.Input.Messages == Messages
+                     && s.Model == "llama2-13b-chat-v2"
+                     && s.Parameters != null
+                     && s.Parameters.ResultFormat == ResultFormats.Message));
+    }
+
+    [Fact]
+    public async Task Llama2_CustomModel_SuccessAsync()
+    {
+        // Arrange
+        var client = Substitute.For<IDashScopeClient>();
+
+        // Act
+        _ = await client.GetLlama2TextCompletionAsync("custom-model", Messages, ResultFormats.Message);
+
+        // Assert
+        _ = await client.Received().GetTextCompletionAsync(
+            Arg.Is<ModelRequest<TextGenerationInput, TextGenerationParameters>>(
+                s => s.Input.Messages == Messages
+                     && s.Model == "custom-model"
+                     && s.Parameters != null
+                     && s.Parameters.ResultFormat == ResultFormats.Message));
+    }
+}

test/Cnblogs.DashScope.Sdk.UnitTests/QWenMultimodalCompletionTests.cs renamed to test/Cnblogs.DashScope.Sdk.UnitTests/QWenMultimodalGenerationTests.cs

@@ -3,7 +3,7 @@
 
 namespace Cnblogs.DashScope.Sdk.UnitTests;
 
-public class QWenMultimodalCompletionTests
+public class QWenMultimodalGenerationTests
 {
     private const string CustomModel = "custom-model";
 

test/Cnblogs.DashScope.Sdk.UnitTests/QWenTextCompletionTests.cs renamed to test/Cnblogs.DashScope.Sdk.UnitTests/QWenTextGenerationTests.cs

@@ -3,7 +3,7 @@
 
 namespace Cnblogs.DashScope.Sdk.UnitTests;
 
-public class QWenTextCompletionTests
+public class QWenTextGenerationTests
 {
     private const string CustomModel = "custom-model";