Add settings for language and max token

Author: intitni

Copilot for Xcode/OpenAIView.swift

@@ -8,6 +8,8 @@ final class OpenAIViewSettings: ObservableObject {
     @AppStorage(\.openAIAPIKey) var openAIAPIKey: String
     @AppStorage(\.chatGPTModel) var chatGPTModel: String
     @AppStorage(\.chatGPTEndpoint) var chatGPTEndpoint: String
+    @AppStorage(\.chatGPTLanguage) var chatGPTLanguage: String
+    @AppStorage(\.chatGPTMaxToken) var chatGPTMaxToken: Int
     init() {}
 }
 
@@ -63,6 +65,29 @@ struct OpenAIView: View {
                             EmptyView()
                         }.textFieldStyle(.copilot)
                     }
+                    
+                    HStack {
+                        Text("Reply in Language")
+                        TextField(
+                            text: $settings.chatGPTLanguage,
+                            prompt: Text("English")
+                        ) {
+                            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)
+                    }
                 }
             }
         }

Core/Sources/Preferences/Keys.swift

@@ -85,4 +85,18 @@ public struct UserDefaultPreferenceKeys {
     }
 
     public var chatGPTModel: ChatGPTModel { .init() }
+    
+    public struct ChatGPTMaxToken: UserDefaultPreferenceKey {
+        public let defaultValue = 2048
+        public let key = "ChatGPTMaxToken"
+    }
+
+    public var chatGPTMaxToken: ChatGPTMaxToken { .init() }
+    
+    public struct ChatGPTLanguage: UserDefaultPreferenceKey {
+        public let defaultValue = ""
+        public let key = "ChatGPTLanguage"
+    }
+
+    public var chatGPTLanguage: ChatGPTLanguage { .init() }
 }

Core/Sources/Service/SuggestionCommandHandler/WindowBaseCommandHandler.swift

@@ -194,23 +194,32 @@ struct WindowBaseCommandHandler: SuggestionCommandHandler {
         let fileURL = try await Environment.fetchCurrentFileURL()
         let endpoint = UserDefaults.shared.value(for: \.chatGPTEndpoint)
         let model = UserDefaults.shared.value(for: \.chatGPTModel)
+        let language = UserDefaults.shared.value(for: \.chatGPTLanguage)
         guard let selection = editor.selections.last else { return }
+
         let service = ChatGPTService(
             systemPrompt: """
-            You are a code explanation engine, you can only explain the code, do not interpret or translate it. Reply in Chinese
+            You are a code explanation engine, you can only explain the code, do not interpret or translate it. Reply in \(language.isEmpty ? language : "English")
             """,
             apiKey: UserDefaults.shared.value(for: \.openAIAPIKey),
             endpoint: endpoint.isEmpty ? nil : endpoint,
             model: model.isEmpty ? nil : model,
-            temperature: 1,
-            maxToken: 2048
+            temperature: 0.6,
+            maxToken: UserDefaults.shared.value(for: \.chatGPTMaxToken)
         )
-        
+
         let code = editor.selectedCode(in: selection)
         Task {
-            try? await service.send(content: code, summary: "Explain selected code.")
+            try? await service.send(
+                content: removeContinousSpaces(from: code),
+                summary: "Explain selected code from \(selection.start.line):\(selection.start.character) to \(selection.end.line):\(selection.end.character)."
+            )
         }
-        
+
         presenter.presentChatGPTConversation(service, fileURL: fileURL)
     }
 }
+
+func removeContinousSpaces(from string: String) -> String {
+    return string.replacingOccurrences(of: " +", with: " ", options: .regularExpression)
+}