feat: add token length config to #19

Author: phodal

src/main/kotlin/cc/unitmesh/devti/settings/AppSettingsComponent.kt

@@ -26,6 +26,7 @@ class AppSettingsComponent {
     val customEngineServer = JBTextField()
     val customEngineToken = JBTextField()
     val language = ComboBox(HUMAN_LANGUAGES)
+    val maxTokenLengthInput = JBTextField(MAX_TOKEN_LENGTH)
 
     private var myEditor: EditorEx? = null
     private val customEnginePrompt by lazy {
@@ -57,6 +58,7 @@ class AppSettingsComponent {
         panel = FormBuilder.createFormBuilder()
             .addLabeledComponent(JBLabel("Language: "), language, 1, false)
             .addLabeledComponent(JBLabel("AI Engine: "), aiEngine, 1, false)
+            .addLabeledComponent(JBLabel("Max Token Length: "), maxTokenLengthInput, 1, false)
             .addSeparator()
             .addTooltip("GitHub Token is for AutoDev")
             .addLabeledComponent(JBLabel("GitHub Token: "), githubToken, 1, false)
@@ -148,16 +150,25 @@ class AppSettingsComponent {
         language.selectedItem = newText
     }
 
-    fun isModified(origineSettings: AutoDevSettingsState): Boolean {
+    fun getMaxTokenLength(): String {
+        return maxTokenLengthInput.text
+    }
+
+    fun setMaxTokenLength(newText: String) {
+        maxTokenLengthInput.text = newText
+    }
+
+    fun isModified(settings: AutoDevSettingsState): Boolean {
         // TODO use data class to avoid manually write this
-        return origineSettings.openAiKey != getOpenAiKey() ||
-                origineSettings.githubToken != getGithubToken() ||
-                origineSettings.openAiModel != getOpenAiModel() ||
-                origineSettings.customOpenAiHost != getOpenAiHost() ||
-                origineSettings.aiEngine != getAiEngine() ||
-                origineSettings.customEngineServer != getCustomEngineServer() ||
-                origineSettings.customEngineToken != getCustomEngineToken() ||
-                origineSettings.customEnginePrompts != getCustomEnginePrompt() ||
-                origineSettings.language != getLanguage()
+        return settings.openAiKey != getOpenAiKey() ||
+                settings.githubToken != getGithubToken() ||
+                settings.openAiModel != getOpenAiModel() ||
+                settings.customOpenAiHost != getOpenAiHost() ||
+                settings.aiEngine != getAiEngine() ||
+                settings.customEngineServer != getCustomEngineServer() ||
+                settings.customEngineToken != getCustomEngineToken() ||
+                settings.customEnginePrompts != getCustomEnginePrompt() ||
+                settings.language != getLanguage() ||
+                settings.maxTokenLength != getMaxTokenLength()
     }
 }

src/main/kotlin/cc/unitmesh/devti/settings/AutoDevSettingsConfigurable.kt

@@ -39,6 +39,7 @@ class AutoDevSettingsConfigurable : Configurable {
         settings.customEngineToken = component.getCustomEngineToken()
         settings.customEnginePrompts = component.getCustomEnginePrompt()
         settings.language = component.getLanguage()
+        settings.maxTokenLength = component.getMaxTokenLength().ifEmpty(MAX_TOKEN_LENGTH::toString)
     }
 
     override fun reset() {
@@ -52,5 +53,7 @@ class AutoDevSettingsConfigurable : Configurable {
         component.setCustomEngineServer(settings.customEngineServer)
         component.setCustomEngineToken(settings.customEngineToken)
         component.setCustomEnginePrompt(settings.customEnginePrompts)
+        component.setLanguage(settings.language)
+        component.setMaxTokenLength(settings.maxTokenLength)
     }
 }

src/main/kotlin/cc/unitmesh/devti/settings/AutoDevSettingsState.kt

@@ -18,6 +18,7 @@ class AutoDevSettingsState : PersistentStateComponent<AutoDevSettingsState> {
     var customEngineToken = ""
     var customEnginePrompts = ""
     var language = DEFAULT_HUMAN_LANGUAGE
+    var maxTokenLength = MAX_TOKEN_LENGTH.toString()
 
     @Synchronized
     override fun getState(): AutoDevSettingsState {

src/main/kotlin/cc/unitmesh/devti/settings/Constants.kt

@@ -5,4 +5,5 @@ val AI_ENGINES = arrayOf("OpenAI", "Custom", "Azure")
 val DEFAULT_AI_ENGINE = AI_ENGINES[0]
 
 val HUMAN_LANGUAGES = arrayOf("English", "中文")
-val DEFAULT_HUMAN_LANGUAGE = HUMAN_LANGUAGES[0]
+val DEFAULT_HUMAN_LANGUAGE = HUMAN_LANGUAGES[0]
+val MAX_TOKEN_LENGTH = 4000