add model option to prompt

fmbot-discord · Mar 13, 2024 · 0b7756d · 0b7756d
1 parent ba98839
commit 0b7756d
Show file tree

Hide file tree

Showing 6 changed files with 2,468 additions and 2 deletions.
diff --git a/src/FMBot.Bot/Services/OpenAiService.cs b/src/FMBot.Bot/Services/OpenAiService.cs
@@ -75,7 +75,7 @@ public async Task<OpenAiResponse> GetJudgeResponse(List<string> artists, PromptT
              artistList.Add(artist[..Math.Min(artist.Length, 36)]);
         }
 
-        var model = supporter ? "gpt-4" : "gpt-3.5-turbo";
+        var model = supporter ? prompt.PremiumModel : prompt.FreeModel;
 
         return await SendRequest($"{prompt.Prompt} {string.Join(", ", artistList)}", model);
     }

diff --git a/src/FMBot.Bot/Services/WhoKnows/WhoKnowsService.cs b/src/FMBot.Bot/Services/WhoKnows/WhoKnowsService.cs
@@ -251,7 +251,7 @@ public static StringBuilder GetGlobalWhoKnowsFooter(StringBuilder footer, WhoKno
         }
         if (context.ContextUser.PrivacyLevel != PrivacyLevel.Global)
         {
-            footer.AppendLine($"You are currently not globally visible - use '{context.Prefix}privacy' to enable.");
+            footer.AppendLine($"You're currently not globally visible - use '{context.Prefix}privacy' to enable.");
         }
         if (settings.HidePrivateUsers)
         {

diff --git a/src/FMBot.Persistence.Domain/Models/AiPrompt.cs b/src/FMBot.Persistence.Domain/Models/AiPrompt.cs
@@ -9,4 +9,6 @@ public class AiPrompt
     public int Version { get; set; }
     public string Language { get; set; }
     public string Prompt { get; set; }
+    public string FreeModel { get; set; }
+    public string PremiumModel { get; set; }
 }