feat(settings): add OpenAI temperature setting

Add new setting for OpenAI temperature in AppSettings with input validation and context help describing the functionality in the settings GUI. Use the temperature value from the settings in the OpenAIService for chat completion requests.

Author: aronbraun

CHANGELOG.md

@@ -2,6 +2,9 @@
 
 ## [Unreleased]
 
+### Added
+- Add OpenAI temperature setting.
+
 ## [1.1.0] - 2023-04-28
 
 ### Added

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/AICommitsExtensions.kt

@@ -16,6 +16,18 @@ fun <T> createColumn(name: String, formatter: (T) -> String) : ColumnInfo<T, Str
 fun ValidationInfoBuilder.notBlank(value: String): ValidationInfo? =
         if (value.isBlank()) error(message("validation.required")) else null
 
+
+fun ValidationInfoBuilder.temperatureValid(value: String): ValidationInfo?  {
+    if (value.isNotBlank()) {
+        value.toFloatOrNull().let {
+            if (it != null && it in 0.0..2.0) {
+                return null
+            }
+        }
+    }
+    return error(message("validation.temperature"))
+}
+
 fun ValidationInfoBuilder.unique(value: String, existingValues: Set<String>): ValidationInfo? =
         if (existingValues.contains(value)) error(message("validation.unique")) else null
 

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/OpenAIService.kt

@@ -32,7 +32,7 @@ class OpenAIService {
                     content = prompt
                 )
             ),
-            temperature = 0.7,
+            temperature = AppSettings.instance.openAITemperature.toDouble(),
             topP = 1.0,
             frequencyPenalty = 0.0,
             presencePenalty = 0.0,
@@ -42,7 +42,6 @@ class OpenAIService {
 
         val completion: ChatCompletion = openAI.chatCompletion(chatCompletionRequest)
         return completion.choices[0].message!!.content
-
     }
 
     suspend fun refreshOpenAIModelIds() {

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/AppSettings.kt

@@ -41,6 +41,7 @@ class AppSettings : PersistentStateComponent<AppSettings> {
 
     var openAIModelId = "gpt-3.5-turbo"
     var openAIModelIds = listOf("gpt-3.5-turbo", "gpt-4")
+    var openAITemperature = "0.7"
 
     var appExclusions: Set<String> = setOf()
 

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/AppSettingsConfigurable.kt

@@ -6,6 +6,7 @@ import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle.message
 import com.github.blarc.ai.commits.intellij.plugin.OpenAIService
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.Prompt
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.PromptTable
+import com.github.blarc.ai.commits.intellij.plugin.temperatureValid
 import com.intellij.icons.AllIcons
 import com.intellij.openapi.options.BoundConfigurable
 import com.intellij.openapi.progress.runBackgroundableTask
@@ -106,6 +107,25 @@ class AppSettingsConfigurable : BoundConfigurable(message("settings.general.grou
                         .align(AlignX.RIGHT)
                         .widthGroup("button")
             }
+
+            row {
+                label(message("settings.openAITemperature"))
+                        .widthGroup("label")
+
+                textField()
+                        .bindText(AppSettings.instance::openAITemperature)
+                        .applyToComponent { minimumWidth = 400 }
+                        .resizableColumn()
+                        .widthGroup("input")
+                        .validationOnInput { temperatureValid(it.text) }
+
+                contextHelp(message("settings.openAITemperatureComment"))
+            }
+
+            row {
+                cell(verifyLabel)
+                        .align(AlignX.RIGHT)
+            }
         }
 
         group(JBLabel("Prompt")) {

src/main/resources/messages/MyBundle.properties

@@ -6,8 +6,11 @@ settings.openAIToken=OpenAI token
 settings.locale=Locale
 settings.prompt=Prompt
 settings.openAIModel=OpenAI model
+settings.openAITemperature=OpenAI temperature
 settings.openAITokenComment=\
   <p>You can get your token <a href="https://platform.openai.com/account/api-keys">here.</a/></p>
+settings.openAITemperatureComment=What sampling temperature to use, between 0 and 2. Higher values like 0.8 will make \
+  the output more random,while lower values like 0.2 will make it more focused and deterministic.
 settings.report-bug=Report bug
 settings.verifyToken=Verify
 settings.verify.valid=Open AI configuration is valid.
@@ -41,6 +44,7 @@ settings.prompt.name=Name
 settings.prompt.content=Content
 validation.required=This value is required.
 validation.number=Value is not a number.
+validation.temperature=Temperature should be between 0 and 2.
 settings.prompt.comment=You can use {locale}, {diff}, {branch} variables to customise your prompt.
 actions.update=Update
 actions.add=Add