feat: Add button for verifying Open AI token in settings.

Author: Blarc

CHANGELOG.md

@@ -1,6 +1,9 @@
 # Changelog
 
 ## [Unreleased]
+### Added
+- Add button for verifying Open AI token in settings.
+
 
 ## [0.4.0] - 2023-03-29
 

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

@@ -2,6 +2,7 @@ package com.github.blarc.ai.commits.intellij.plugin
 
 import com.aallam.openai.api.BetaOpenAI
 import com.aallam.openai.api.chat.*
+import com.aallam.openai.api.completion.CompletionRequest
 import com.aallam.openai.api.model.ModelId
 import com.aallam.openai.client.OpenAI
 import com.github.blarc.ai.commits.intellij.plugin.settings.AppSettings
@@ -46,4 +47,8 @@ class OpenAIService {
         return completion.choices[0].message!!.content
 
     }
+    @Throws(Exception::class)
+    suspend fun verifyToken(token: String){
+        OpenAI(token).models()
+    }
 }

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

@@ -1,32 +1,81 @@
 package com.github.blarc.ai.commits.intellij.plugin.settings
 
+import com.aallam.openai.api.exception.OpenAIAPIException
 import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle.message
+import com.github.blarc.ai.commits.intellij.plugin.OpenAIService
+import com.intellij.icons.AllIcons
 import com.intellij.openapi.options.BoundConfigurable
+import com.intellij.openapi.progress.runBackgroundableTask
+import com.intellij.ui.components.JBLabel
 import com.intellij.ui.dsl.builder.*
-import java.awt.Component
+import kotlinx.coroutines.DelicateCoroutinesApi
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.GlobalScope
+import kotlinx.coroutines.launch
 import java.util.*
-import javax.swing.DefaultListCellRenderer
-import javax.swing.JList
+import javax.swing.JPasswordField
 
 class AppSettingsConfigurable : BoundConfigurable(message("settings.general.group.title")) {
+
+    private val tokenPasswordField = JPasswordField()
+    private val verifyLabel = JBLabel()
     override fun createPanel() = panel {
 
         row {
-            passwordField()
+            cell(tokenPasswordField)
                 .label(message("settings.openAIToken"))
                 .bindText(
                     { AppSettings.instance.getOpenAIToken().orEmpty() },
                     { AppSettings.instance.saveOpenAIToken(it)}
                 )
                 .align(Align.FILL)
+                .resizableColumn()
+                .focused()
+            button(message("settings.verifyToken")) {
+                verifyToken()
+            }.align(AlignX.RIGHT)
         }
         row {
             comment(message("settings.openAITokenComment"))
+                .align(AlignX.LEFT)
+            cell(verifyLabel)
+                .align(AlignX.RIGHT)
         }
         row {
             comboBox(Locale.getAvailableLocales().toList(), AppSettingsListCellRenderer())
                 .label(message("settings.locale"))
                 .bindItem(AppSettings.instance::locale.toNullableProperty())
         }
     }
+
+    @OptIn(DelicateCoroutinesApi::class)
+    private fun verifyToken() {
+        runBackgroundableTask(message("settings.verify.running")) {
+            if (tokenPasswordField.password.isEmpty()) {
+                verifyLabel.icon = AllIcons.General.InspectionsError
+                verifyLabel.text = message("settings.verify.token-is-empty")
+            } else {
+                verifyLabel.icon = AllIcons.General.InlineRefreshHover
+                verifyLabel.text = message("settings.verify.running")
+
+                GlobalScope.launch(Dispatchers.IO) {
+                    try {
+                        println(String(tokenPasswordField.password))
+                        OpenAIService.instance.verifyToken(String(tokenPasswordField.password))
+                        verifyLabel.text = message("settings.verify.valid")
+                        verifyLabel.icon = AllIcons.General.InspectionsOK
+                    }
+                    catch (e: OpenAIAPIException) {
+                        verifyLabel.text = message("settings.verify.invalid", e.statusCode)
+                        verifyLabel.icon = AllIcons.General.InspectionsError
+                    }
+                    catch (e: Exception) {
+                        verifyLabel.text = message("settings.verify.invalid", "Unknown")
+                        verifyLabel.icon = AllIcons.General.InspectionsError
+                    }
+                }
+            }
+        }
+
+    }
 }

src/main/resources/messages/MyBundle.properties

@@ -6,10 +6,14 @@ settings.openAIToken=OpenAI token
 settings.locale=Locale
 settings.openAITokenComment=\
   <p>You can get your token <a href="https://platform.openai.com/account/api-keys">here.</a/></p>
+settings.verifyToken=Verify
+settings.verify.valid=Open AI token is valid.
+settings.verify.invalid=Open AI token is not valid ({0}).
+settings.verify.running=Verifying Open AI token...
+settings.verify.token-is-empty=Open AI token is empty.
 
 action.background=Generating commit message
 
-
 notifications.title=AI Commits
 notifications.welcome=Thanks for installing AI Commits <strong>{0}</strong>
 notifications.unsuccessful-request=Error occurred: {0}