Add deepseek-coder-v2:236b user

ServiceStack · Aug 4, 2024 · bbd06a0 · bbd06a0
1 parent 25e0eb8
commit bbd06a0
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/MyApp.ServiceInterface/Data/AppConfig.cs b/MyApp.ServiceInterface/Data/AppConfig.cs
@@ -79,6 +79,7 @@ public static (string Model, int Questions)[] ModelsForQuestions =
         ("gemma-2b", -1),
         ("qwen-4b", -1),
         ("deepseek-coder-33b", -1),
+        ("deepseek-coder2-236b", -1),
         ("mistral", -1),
 
         ("phi", 0),

diff --git a/MyApp.Tests/Top1KQuestionTasks.cs b/MyApp.Tests/Top1KQuestionTasks.cs
@@ -245,6 +245,28 @@ public async Task Create_Mistral_Nemo_User()
         api.ThrowIfError();
     }
 
+    [Test]
+    public async Task Create_DeepSeek236B_User()
+    {
+        // var client = await TestUtils.CreateAdminDevClientAsync();
+        var client = await TestUtils.CreateAdminProdClientAsync();
+        var api = await client.ApiAsync(new AdminCreateUser
+        {
+            UserName = "deepseek-coder2-236b",
+            DisplayName = "DeepSeek Coder2 236B",
+            Email = "[email protected]",
+            UserAuthProperties = new()
+            {
+                [nameof(ApplicationUser.Model)] = "deepseek-coder-v2:236b",
+                [nameof(ApplicationUser.DisplayName)] = "DeepSeek Coder2 236B",
+                [nameof(ApplicationUser.ProfilePath)] = "/profiles/de/deepseek-coder2-236b/deepseek-coder2-236b.jpg",
+            },
+            Password = Environment.GetEnvironmentVariable("AUTH_SECRET"),
+        });
+        api.Response.PrintDump();
+        api.ThrowIfError();
+    }
+
     [Test]
     public async Task Generate_top_10k_answers_for_gemini_flash()
     {

diff --git a/MyApp/Configure.Db.Migrations.cs b/MyApp/Configure.Db.Migrations.cs
@@ -263,6 +263,16 @@ await EnsureUserAsync(new ApplicationUser
             Model = "gemini-pro",
         }, "p@55wOrd");
 
+        await EnsureUserAsync(new ApplicationUser
+        {
+            UserName = "deepseek-coder2-236b",
+            Email = "[email protected]",
+            DisplayName = "DeepSeek Coder2 236B",
+            EmailConfirmed = true,
+            ProfilePath = "/profiles/de/deepseek-coder2-236b/deepseek-coder2-236b.jpg",
+            Model = "deepseek-coder-v2:236b",
+        }, "p@55wOrd");
+
         await EnsureUserAsync(new ApplicationUser
         {
             UserName = "deepseek-coder-33b",