feat(settings): add option to select OpenAI model

This commit adds the ability to select an OpenAI model from a dropdown in the settings panel. The selected model will be used for generating commit messages. The available models are `gpt-3.5-turbo` and `gpt-4`. The default model is `gpt-3.5-turbo`.

The changes include adding a new enum class `OpenAIModel`, updating `AppSettings` to include a `openAIModel` property, updating the `AppSettingsConfigurable` panel to include a dropdown to select the model, and updating `OpenAIService` to use the selected model for generating commit messages.

Author: nservais

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

@@ -8,21 +8,22 @@ import com.github.blarc.ai.commits.intellij.plugin.settings.AppSettings
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.components.Service
 
+
 @Service
 class OpenAIService {
 
     companion object {
-        const val model = "gpt-3.5-turbo"
         val instance: OpenAIService
             get() = ApplicationManager.getApplication().getService(OpenAIService::class.java)
     }
 
     @OptIn(BetaOpenAI::class)
     suspend fun generateCommitMessage(prompt: String, completions: Int): String {
+        val openAIModel = AppSettings.instance.openAIModel.id
         val openAI = OpenAI(AppSettings.instance.getOpenAIConfig())
 
         val chatCompletionRequest = ChatCompletionRequest(
-            ModelId(model),
+            ModelId(openAIModel),
             listOf(
                 ChatMessage(
                     role = ChatRole.User,

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

@@ -0,0 +1,6 @@
+package com.github.blarc.ai.commits.intellij.plugin.openaiModel
+
+enum class OpenAIModel(val id: String) {
+    GPT_3_5_TURBO("gpt-3.5-turbo"),
+    GPT_4("gpt-4");
+}

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

@@ -4,6 +4,7 @@ import com.aallam.openai.client.OpenAIConfig
 import com.aallam.openai.client.ProxyConfig
 import com.github.blarc.ai.commits.intellij.plugin.notifications.Notification
 import com.github.blarc.ai.commits.intellij.plugin.notifications.sendNotification
+import com.github.blarc.ai.commits.intellij.plugin.openaiModel.OpenAIModel
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.Prompt
 import com.intellij.credentialStore.CredentialAttributes
 import com.intellij.credentialStore.Credentials
@@ -35,6 +36,8 @@ class AppSettings : PersistentStateComponent<AppSettings> {
     var prompts: MutableMap<String, Prompt> = initPrompts()
     var currentPrompt: Prompt = prompts["basic"]!!
 
+    var openAIModel: OpenAIModel = OpenAIModel.GPT_3_5_TURBO
+
     companion object {
         const val SERVICE_NAME = "com.github.blarc.ai.commits.intellij.plugin.settings.AppSettings"
         val instance: AppSettings

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

@@ -4,11 +4,10 @@ import com.aallam.openai.api.exception.OpenAIAPIException
 import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle
 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.openaiModel.OpenAIModel
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.Prompt
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.PromptTable
 import com.intellij.icons.AllIcons
-import com.intellij.openapi.actionSystem.AnAction
-import com.intellij.openapi.actionSystem.AnActionEvent
 import com.intellij.openapi.options.BoundConfigurable
 import com.intellij.openapi.progress.runBackgroundableTask
 import com.intellij.openapi.ui.ComboBox
@@ -17,15 +16,12 @@ import com.intellij.ui.ToolbarDecorator
 import com.intellij.ui.components.JBLabel
 import com.intellij.ui.dsl.builder.*
 import com.intellij.ui.util.minimumWidth
-import com.intellij.ui.util.preferredWidth
 import kotlinx.coroutines.DelicateCoroutinesApi
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.GlobalScope
 import kotlinx.coroutines.launch
 import java.util.*
-import javax.swing.JComponent
 import javax.swing.JPasswordField
-import javax.swing.JScrollPane
 
 class AppSettingsConfigurable : BoundConfigurable(message("settings.general.group.title")) {
 
@@ -37,6 +33,12 @@ class AppSettingsConfigurable : BoundConfigurable(message("settings.general.grou
     override fun createPanel() = panel {
 
         group(JBLabel("OpenAI")) {
+            row {
+                comboBox(OpenAIModel.values().toList(),
+                        AppSettingsListCellRenderer())
+                        .label(message("settings.openAIModel"))
+                        .bindItem(AppSettings.instance::openAIModel.toNullableProperty())
+            }
             row {
                 cell(tokenPasswordField)
                         .label(message("settings.openAIToken"))

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

@@ -1,6 +1,6 @@
 package com.github.blarc.ai.commits.intellij.plugin.settings
 
-import ai.grazie.utils.capitalize
+import com.github.blarc.ai.commits.intellij.plugin.openaiModel.OpenAIModel
 import com.github.blarc.ai.commits.intellij.plugin.settings.prompt.Prompt
 import java.awt.Component
 import java.util.*
@@ -22,6 +22,9 @@ class AppSettingsListCellRenderer : DefaultListCellRenderer() {
         if (value is Prompt) {
             text = value.name
         }
+        if (value is OpenAIModel) {
+            text = value.id
+        }
         return component
     }
 }

src/main/resources/messages/MyBundle.properties

@@ -5,6 +5,7 @@ settings.general.group.title=AI Commits
 settings.openAIToken=OpenAI token
 settings.locale=Locale
 settings.prompt=Prompt
+settings.openAIModel=OpenAI model
 settings.openAITokenComment=\
   <p>You can get your token <a href="https://platform.openai.com/account/api-keys">here.</a/></p>
 settings.report-bug=Report bug