Add more settings

relentless-dev-purchases · Mar 21, 2023 · 7c051e6 · 7c051e6
1 parent c323ba8
commit 7c051e6
Show file tree

Hide file tree

Showing 7 changed files with 44 additions and 2 deletions.
diff --git a/ChatGPT.CLI/Program.cs b/ChatGPT.CLI/Program.cs
@@ -25,6 +25,7 @@ You are a helpful assistant named Clippy.
     Settings = new ChatSettingsViewModel
     {
         Temperature = 0.7m,
+        TopP = 1m,
         MaxTokens = 2000,
         Model = "gpt-3.5-turbo",
         Directions = directions

diff --git a/ChatGPT.Core/Defaults.cs b/ChatGPT.Core/Defaults.cs
@@ -6,6 +6,12 @@ public static class Defaults
 
     public const decimal DefaultTemperature = 0.7m;
 
+    public const decimal DefaultTopP = 1m;
+
+    public const decimal DefaultPresencePenalty = 0m;
+
+    public const decimal DefaultFrequencyPenalty = 0m;
+
     public const int DefaultMaxTokens = 256;
 
     public const string DefaultModel = "gpt-3.5-turbo";

diff --git a/ChatGPT.Core/ViewModels/Chat/ChatSettingsViewModel.cs b/ChatGPT.Core/ViewModels/Chat/ChatSettingsViewModel.cs
@@ -6,6 +6,9 @@ namespace ChatGPT.ViewModels.Chat;
 public partial class ChatSettingsViewModel : ObservableObject
 {
     private decimal _temperature;
+    private decimal _topP;
+    private decimal _presencePenalty;
+    private decimal _frequencyPenalty;
     private int _maxTokens;
     private string? _apiKey;
     private string? _model;
@@ -19,6 +22,27 @@ public decimal Temperature
         set => SetProperty(ref _temperature, value);
     }
 
+    [JsonPropertyName("top_p")]
+    public decimal TopP
+    {
+        get => _topP;
+        set => SetProperty(ref _topP, value);
+    }
+
+    [JsonPropertyName("presence_penalty")]
+    public decimal PresencePenalty
+    {
+        get => _presencePenalty;
+        set => SetProperty(ref _presencePenalty, value);
+    }
+
+    [JsonPropertyName("frequency_penalty")]
+    public decimal FrequencyPenalty
+    {
+        get => _frequencyPenalty;
+        set => SetProperty(ref _frequencyPenalty, value);
+    }
+
     [JsonPropertyName("maxTokens")]
     public int MaxTokens
     {
@@ -59,6 +83,9 @@ public ChatSettingsViewModel Copy()
         return new ChatSettingsViewModel
         {
             Temperature = _temperature,
+            TopP = _topP,
+            PresencePenalty = _presencePenalty,
+            FrequencyPenalty = _frequencyPenalty,
             MaxTokens = _maxTokens,
             ApiKey = _apiKey,
             Model = _model,

diff --git a/ChatGPT.Core/ViewModels/MainViewModel.Chats.cs b/ChatGPT.Core/ViewModels/MainViewModel.Chats.cs
@@ -58,6 +58,9 @@ private ChatSettingsViewModel CreateDefaultChatSettings()
         return new ChatSettingsViewModel
         {
             Temperature = Defaults.DefaultTemperature,
+            TopP = Defaults.DefaultTopP,
+            PresencePenalty = Defaults.DefaultPresencePenalty,
+            FrequencyPenalty = Defaults.DefaultFrequencyPenalty,
             MaxTokens = Defaults.DefaultMaxTokens,
             Model = Defaults.DefaultModel,
             ApiKey = null,

diff --git a/ChatGPT.UI/Views/Chat/ChatSettingsView.axaml b/ChatGPT.UI/Views/Chat/ChatSettingsView.axaml
@@ -10,6 +10,9 @@
   <Design.DataContext>
     <vm:ChatSettingsViewModel>
       <vm:ChatSettingsViewModel.Temperature>0.6</vm:ChatSettingsViewModel.Temperature>
+      <vm:ChatSettingsViewModel.TopP>1</vm:ChatSettingsViewModel.TopP>
+      <vm:ChatSettingsViewModel.FrequencyPenalty>0</vm:ChatSettingsViewModel.FrequencyPenalty>
+      <vm:ChatSettingsViewModel.PresencePenalty>0</vm:ChatSettingsViewModel.PresencePenalty>
       <vm:ChatSettingsViewModel.MaxTokens>100</vm:ChatSettingsViewModel.MaxTokens>
       <vm:ChatSettingsViewModel.ApiKey><x:Null/></vm:ChatSettingsViewModel.ApiKey>
     </vm:ChatSettingsViewModel>

diff --git a/ChatGPT/Model/Services/ChatServiceSettings.cs b/ChatGPT/Model/Services/ChatServiceSettings.cs
@@ -10,5 +10,7 @@ public class ChatServiceSettings
     public decimal Temperature { get; set; }
     public int MaxTokens { get; set; }
     public decimal TopP { get; set; }
+    public decimal PresencePenalty { get; set; }
+    public decimal FrequencyPenalty { get; set; }
     public string? Stop { get; set; }
 }
diff --git a/ChatGPT/Services/ChatService.cs b/ChatGPT/Services/ChatService.cs
@@ -35,8 +35,8 @@ private static string GetRequestBodyJson(ChatServiceSettings settings)
             N = 1,
             Stream = false,
             Stop = settings.Stop,
-            FrequencyPenalty = 0.0m,
-            PresencePenalty = 0.0m,
+            FrequencyPenalty = settings.FrequencyPenalty,
+            PresencePenalty = settings.PresencePenalty,
             User = null
         };