feat(clients): show client exceptions

Author: Blarc

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

@@ -53,3 +53,21 @@ fun <T>Cell<T>.emptyText(emptyText: String) : Cell<T> where T : JComponent, T :
     this.component.emptyText.text = emptyText
     return this
 }
+
+fun String.wrap(length: Int): String {
+    var input = this
+    val wrapped = StringBuilder()
+
+    while (input.length > length) {
+        var index = input.lastIndexOf(' ', length)
+
+        if (index == -1) index = length
+        wrapped.append(input.substring(0, index)).append("<br>")
+
+        input = input.substring(index).trimStart()
+    }
+
+    wrapped.append(input)
+
+    return wrapped.toString()
+}

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

@@ -119,7 +119,10 @@ abstract class LLMClientPanel(
     open fun Panel.verifyRow() {
         row {
             cell(verifyLabel)
-                .applyToComponent { setAllowAutoWrapping(true) }
+                .applyToComponent {
+                    setAllowAutoWrapping(true)
+                    setCopyable(true)
+                }
                 .align(AlignX.LEFT)
 
             button(message("settings.verifyToken")) { verifyConfiguration() }

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

@@ -2,81 +2,59 @@ package com.github.blarc.ai.commits.intellij.plugin.settings.clients
 
 import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle.message
 import com.github.blarc.ai.commits.intellij.plugin.settings.AppSettings2
+import com.github.blarc.ai.commits.intellij.plugin.wrap
 import com.intellij.icons.AllIcons
-import com.intellij.openapi.application.EDT
 import com.intellij.openapi.vcs.ui.CommitMessage
 import com.intellij.ui.components.JBLabel
-import dev.langchain4j.data.message.AiMessage
 import dev.langchain4j.data.message.UserMessage
 import dev.langchain4j.model.chat.ChatLanguageModel
-import dev.langchain4j.model.output.Response
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.withContext
 
-abstract class LLMClientService <T: LLMClientConfiguration>(private val cs: CoroutineScope) {
+abstract class LLMClientService<T : LLMClientConfiguration>(private val cs: CoroutineScope) {
 
     abstract fun buildChatModel(client: T): ChatLanguageModel
 
     fun generateCommitMessage(client: T, prompt: String, commitMessage: CommitMessage) {
-        // TODO @Blarc: catch exceptions
         val model = buildChatModel(client)
-        sendRequest(model, prompt) {
-            withContext(Dispatchers.EDT) {
-                commitMessage.setCommitMessage(it.content().text())
-            }
+        sendRequest(model, prompt, onSuccess = {
+            commitMessage.setCommitMessage(it)
             AppSettings2.instance.recordHit()
-        }
+        }, onError = {
+            commitMessage.setCommitMessage(it)
+        })
     }
 
     fun verifyConfiguration(client: T, label: JBLabel) {
-        // TODO @Blarc: catch exceptions
         val model = buildChatModel(client)
-        sendRequest(model, "test") {
-            withContext(Dispatchers.EDT) {
-                label.text = message("settings.verify.valid")
-                label.icon = AllIcons.General.InspectionsOK
-            }
-        }
-
-//        GlobalScope.launch(Dispatchers.IO) {
-//            try {
-//                client.verifyConfiguration(hostComboBox.item, proxyTextField.text, socketTimeoutTextField.text, modelComboBox.item, String(tokenPasswordField.password))
-//                verifyLabel.text = message("settings.verify.valid")
-//                verifyLabel.icon = AllIcons.General.InspectionsOK
-//            } catch (e: Exception) {
-//                var errorMessage = e.localizedMessage
-//                if (e.cause is OpenAiHttpException) {
-//                    val openAiError = Json.decodeFromString<OpenAiErrorWrapper>((e.cause as OpenAiHttpException).message!!).error
-//                    errorMessage = openAiError.code
-//                }
-//                verifyLabel.text = message("settings.verify.invalid", errorMessage)
-//                verifyLabel.icon = AllIcons.General.InspectionsError
-//            }
-//        }
+        sendRequest(model, "test", onSuccess = {
+            label.text = message("settings.verify.valid")
+            label.icon = AllIcons.General.InspectionsOK
+        }, onError = {
+            label.text = it.wrap(80)
+            label.icon = AllIcons.General.InspectionsError
+        })
     }
 
-//    @Serializable
-//    data class OpenAiError(val message: String?, val type: String?, val param: String?, val code: String?)
-//
-//    @Serializable
-//    data class OpenAiErrorWrapper(val error: OpenAiError)
-
-    private fun sendRequest(model: ChatLanguageModel, text: String, onResponse: suspend (r: Response<AiMessage>) -> Unit ) {
+    private fun sendRequest(model: ChatLanguageModel, text: String, onSuccess: suspend (r: String) -> Unit, onError: suspend (r: String) -> Unit) {
         cs.launch(Dispatchers.Default) {
-            val response = withContext(Dispatchers.IO) {
-                model.generate(
-                    listOf(
-                        UserMessage.from(
-                            "user",
-                            text
+            try {
+                val response = withContext(Dispatchers.IO) {
+                    model.generate(
+                        listOf(
+                            UserMessage.from(
+                                "user",
+                                text
+                            )
                         )
-                    )
-                )
+                    ).content().text()
+                }
+                onSuccess(response)
+            } catch (e: Exception) {
+                onError(e.message ?: "Unknown error.")
             }
-            onResponse(response)
         }
     }
-
 }

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/ollama/OllamaClientService.kt

@@ -1,7 +1,6 @@
 package com.github.blarc.ai.commits.intellij.plugin.settings.clients.ollama
 
 import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientService
-import com.intellij.openapi.application.EDT
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
 import com.intellij.openapi.ui.ComboBox
@@ -38,10 +37,11 @@ class OllamaClientService(private val cs: CoroutineScope) : LLMClientService<Oll
             OllamaClientSharedState.getInstance().modelIds.addAll(availableModels.content()
                 .map { it.name }
             )
-            withContext(Dispatchers.EDT) {
-                comboBox.model = DefaultComboBoxModel(client.getModelIds().naturalSorted().toTypedArray())
-                comboBox.item = client.modelId
-            }
+
+            // This can't be called from EDT thread, because dialog blocks the EDT thread
+            val modelItems = client.getModelIds().naturalSorted().toTypedArray()
+            comboBox.model = DefaultComboBoxModel(modelItems)
+            comboBox.item = client.modelId
         }
     }