From bbd06a049e570a3912325ea86e5faa99d983b3ad Mon Sep 17 00:00:00 2001
From: Demis Bellot <demis.bellot@gmail.com>
Date: Sun, 4 Aug 2024 20:59:50 +0800
Subject: [PATCH] Add deepseek-coder-v2:236b user

---
 MyApp.ServiceInterface/Data/AppConfig.cs |  1 +
 MyApp.Tests/Top1KQuestionTasks.cs        | 22 ++++++++++++++++++++++
 MyApp/Configure.Db.Migrations.cs         | 10 ++++++++++
 3 files changed, 33 insertions(+)

diff --git a/MyApp.ServiceInterface/Data/AppConfig.cs b/MyApp.ServiceInterface/Data/AppConfig.cs
index b8bcaef..a630957 100644
--- 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
index 92a9d34..2a332a6 100644
--- 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 = "servicestack.mail+deepseek-coder2-236b@gmail.com",
+            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
index 7d70a87..1276bd4 100644
--- 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 = "servicestack.mail+deepseek-coder2-236b@gmail.com",
+            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",