Merge pull request #100 from SNQ-2001/feature/update-openai-settings

Allow ChatGPT Model and Reply in Language to be selected in Picker

Author: intitni

Copilot for Xcode/OpenAIView.swift

@@ -30,63 +30,76 @@ struct OpenAIView: View {
 
                 Form {
                     HStack {
-                        Text("OpenAI API Key")
                         TextField(text: $settings.openAIAPIKey, prompt: Text("sk-*")) {
-                            EmptyView()
-                        }.textFieldStyle(.copilot)
+                            Text("OpenAI API Key")
+                        }.textFieldStyle(.roundedBorder)
                         Button(action: {
                             openURL(apiKeyURL)
                         }) {
                             Image(systemName: "questionmark.circle.fill")
-                        }
-                        .buttonStyle(.plain)
+                        }.buttonStyle(.plain)
                     }
 
                     HStack {
-                        Text("ChatGPT Model")
-                        TextField(text: $settings.chatGPTModel, prompt: Text("gpt-3.5-turbo")) {
-                            EmptyView()
-                        }.textFieldStyle(.copilot)
-
+                        Picker(selection: $settings.chatGPTModel) {
+                            ForEach(ChatGPTModel.allCases, id: \.self) { model in
+                                Text(model.rawValue).tag(model.rawValue)
+                            }
+                        } label: {
+                            Text("ChatGPT Model")
+                        }.pickerStyle(.menu)
                         Button(action: {
                             openURL(modelURL)
                         }) {
                             Image(systemName: "questionmark.circle.fill")
+                        }.buttonStyle(.plain)
+                    }
+                    .onChange(of: settings.chatGPTModel) { newValue in
+                        if let model = ChatGPTModel(rawValue: newValue) {
+                            settings.chatGPTEndpoint = model.endpoint
                         }
-                        .buttonStyle(.plain)
                     }
 
-                    HStack {
-                        Text("ChatGPT Endpoint")
-                        TextField(
-                            text: $settings.chatGPTEndpoint,
-                            prompt: Text("https://api.openai.com/v1/chat/completions")
-                        ) {
-                            EmptyView()
-                        }.textFieldStyle(.copilot)
-                    }
-                    
-                    HStack {
+                    TextField(
+                        text: $settings.chatGPTEndpoint,
+                        prompt: Text("https://api.openai.com/v1/chat/completions")
+                    ) {
+                        Text("ChatGPT Server")
+                    }.textFieldStyle(.roundedBorder)
+
+                    Picker(selection: $settings.chatGPTLanguage) {
+                        ForEach(Locale.availableLocalizedLocales, id: \.self) { localizedLocales in
+                            Text(localizedLocales).tag(localizedLocales)
+                        }
+                    } label: {
                         Text("Reply in Language")
-                        TextField(
-                            text: $settings.chatGPTLanguage,
-                            prompt: Text("e.g. English. Leave it blank to let the bot decide.")
-                        ) {
-                            EmptyView()
-                        }.textFieldStyle(.copilot)
-                    }
-                    
-                    HStack {
-                        Text("Max Token")
-                        TextField(
-                            text: .init(get: {
-                                String(settings.chatGPTMaxToken)
-                            }, set: { newValue in
-                                settings.chatGPTMaxToken = Int(newValue) ?? 0
-                            })
-                        ) {
-                            EmptyView()
-                        }.textFieldStyle(.copilot)
+                    }.pickerStyle(.menu)
+
+                    if let model = ChatGPTModel(rawValue: settings.chatGPTModel) {
+                        let binding = Binding(
+                            get: { String(settings.chatGPTMaxToken) },
+                            set: {
+                                if let selectionMaxToken = Int($0) {
+                                    settings.chatGPTMaxToken = model.maxToken < selectionMaxToken ? model.maxToken : selectionMaxToken
+                                } else {
+                                    settings.chatGPTMaxToken = 0
+                                }
+                            }
+                        )
+                        HStack {
+                            Stepper(
+                                value: $settings.chatGPTMaxToken,
+                                in: 0...model.maxToken,
+                                step: 1
+                            ) {
+                                Text("Max Token")
+                            }
+                            TextField(text: binding) {
+                                EmptyView()
+                            }
+                            .labelsHidden()
+                            .textFieldStyle(.roundedBorder)
+                        }
                     }
                 }
             }

Core/Sources/Preferences/ChatGPTModel.swift

@@ -0,0 +1,40 @@
+import Foundation
+
+public enum ChatGPTModel: String {
+    case gpt4 = "gpt-4"
+
+    case gpt40314 = "gpt-4-0314"
+
+    case gpt432k = "gpt-4-32k"
+
+    case gpt432k0314 = "gpt-4-32k-0314"
+
+    case gpt35Turbo = "gpt-3.5-turbo"
+
+    case gpt35Turbo0301 = "gpt-3.5-turbo-0301"
+}
+
+public extension ChatGPTModel {
+    var endpoint: String {
+        "https://api.openai.com/v1/chat/completions"
+    }
+
+    var maxToken: Int {
+        switch self {
+        case .gpt4:
+            return 8192
+        case .gpt40314:
+            return 8192
+        case .gpt432k:
+            return 32768
+        case .gpt432k0314:
+            return 32768
+        case .gpt35Turbo:
+            return 4096
+        case .gpt35Turbo0301:
+            return 4096
+        }
+    }
+}
+
+extension ChatGPTModel: CaseIterable {}

Core/Sources/Preferences/Keys.swift

@@ -80,7 +80,7 @@ public struct UserDefaultPreferenceKeys {
     public var chatGPTEndpoint: ChatGPTEndpoint { .init() }
 
     public struct ChatGPTModel: UserDefaultPreferenceKey {
-        public let defaultValue = ""
+        public let defaultValue = Preferences.ChatGPTModel.gpt35Turbo.rawValue
         public let key = "ChatGPTModel"
     }
 

Core/Sources/Preferences/Locale.swift

@@ -0,0 +1,16 @@
+import Foundation
+
+public extension Locale {
+    static var availableLocalizedLocales: [String] {
+        var localizedLocales = Locale.availableIdentifiers.compactMap {
+            Locale(identifier: $0).localizedString(forIdentifier: $0)
+        }
+        .sorted()
+        localizedLocales.insert("Auto-detected by ChatGPT", at: 0)
+        return localizedLocales
+    }
+
+    var languageName: String {
+        localizedString(forLanguageCode: languageCode ?? "") ?? ""
+    }
+}