feat(settings): add new LLMParam components

This commit adds new LLMParam components to the LLMSettingsComponent. The new components include a text field, a password field, and a combo box. These components allow users to input and select different settings for the LLM feature. The components are created using the Reactive delegate, which allows for easy updating of the component values.

Author: iptton

src/main/kotlin/cc/unitmesh/devti/llms/LLMProviderFactory.kt

@@ -4,21 +4,21 @@ import cc.unitmesh.devti.llms.azure.AzureOpenAIProvider
 import cc.unitmesh.devti.llms.custom.CustomLLMProvider
 import cc.unitmesh.devti.llms.openai.OpenAIProvider
 import cc.unitmesh.devti.llms.xianghuo.XingHuoProvider
+import cc.unitmesh.devti.settings.AIEngines
 import cc.unitmesh.devti.settings.AutoDevSettingsState
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
 
 @Service
 class LLMProviderFactory {
-    private val aiEngine: String
-        get() = AutoDevSettingsState.getInstance().aiEngine
+    private val aiEngine: AIEngines
+        get() = AIEngines.values().find { it.name.lowercase() == AutoDevSettingsState.getInstance().aiEngine.lowercase() } ?: AIEngines.OpenAI
     fun connector(project: Project): LLMProvider {
         return when (aiEngine) {
-            // TODO use mapping and avoid hard code engine name
-            "OpenAI" -> project.getService(OpenAIProvider::class.java)
-            "Custom" -> project.getService(CustomLLMProvider::class.java)
-            "Azure" -> project.getService(AzureOpenAIProvider::class.java)
-            "XingHuo" -> project.getService(XingHuoProvider::class.java)
+            AIEngines.OpenAI -> project.getService(OpenAIProvider::class.java)
+            AIEngines.Custom -> project.getService(CustomLLMProvider::class.java)
+            AIEngines.Azure -> project.getService(AzureOpenAIProvider::class.java)
+            AIEngines.XingHuo -> project.getService(XingHuoProvider::class.java)
             else -> project.getService(OpenAIProvider::class.java)
         }
     }

src/main/kotlin/cc/unitmesh/devti/llms/xianghuo/XingHuoProvider.kt

@@ -37,12 +37,13 @@ class XingHuoProvider(val project: Project) : LLMProvider {
     private val hmacsha256Algorithms = "hmacsha256"
     private val uid = UUID.randomUUID().toString().substring(0, 32)
 
-    private val hmacsha256 by lazy {
-        val hmac = Mac.getInstance(hmacsha256Algorithms)
-        val keySpec = SecretKeySpec(secrectKey.toByteArray(), hmacsha256Algorithms)
-        hmac.init(keySpec)
-        hmac
-    }
+    private val hmacsha256: Mac
+        get() {
+            val hmac = Mac.getInstance(hmacsha256Algorithms)
+            val keySpec = SecretKeySpec(secrectKey.toByteArray(), hmacsha256Algorithms)
+            hmac.init(keySpec)
+            return hmac
+        }
 
     override fun prompt(promptText: String): String {
         // prompt 接口看似是无用的废弃接口，因为所有 LLM 请求都只能异步返回，不可能直接返回同步结果
@@ -54,12 +55,9 @@ class XingHuoProvider(val project: Project) : LLMProvider {
     override fun stream(promptText: String, systemPrompt: String): Flow<String> {
         return callbackFlow {
             val client = OkHttpClient()
-            client.newWebSocket(request, MyListener(this, onSocketOpend = {
+            client.newWebSocket(request, MyListener(this, onSocketOpen = {
                 val msg = getSendBody(promptText)
-                println("sending $msg")
                 send(msg)
-            }, onSocketClosed = {
-                close()
             }))
             awaitClose()
         }
@@ -68,19 +66,18 @@ class XingHuoProvider(val project: Project) : LLMProvider {
 
     class MyListener(
         private val producerScope: ProducerScope<String>,
-        private val onSocketOpend: WebSocket.() -> Unit,
-        private val onSocketClosed: WebSocket.() -> Unit
+        private val onSocketOpen: WebSocket.() -> Unit,
     ) : WebSocketListener() {
 
         private var sockedOpen = false
         override fun onOpen(webSocket: WebSocket, response: Response) {
-            webSocket.onSocketOpend()
+            webSocket.onSocketOpen()
             sockedOpen = true
         }
 
-        override fun onMessage(webSocket: WebSocket, body: String) {
-            return runCatching {
-                val element = Json.parseToJsonElement(body)
+        override fun onMessage(webSocket: WebSocket, text: String) {
+            runCatching {
+                val element = Json.parseToJsonElement(text)
                 val choices = element.jsonObject["payload"]!!.jsonObject["choices"]!!
                 val statusCode: Int = choices.jsonObject["status"]?.jsonPrimitive?.int!!
                 val message = choices.jsonObject["text"]!!.jsonArray[0]
@@ -96,11 +93,12 @@ class XingHuoProvider(val project: Project) : LLMProvider {
         }
 
         override fun onClosed(webSocket: WebSocket, code: Int, reason: String) {
-            webSocket.onSocketClosed()
+            producerScope.close()
         }
 
         override fun onFailure(webSocket: WebSocket, t: Throwable, response: Response?) {
             // WebSocket connection failed
+            println("failure ${t.message} ${response?.body} ${response?.message} ${response?.code}")
             producerScope.trySend("onFailure ${response?.body} ${response?.message} ${response?.code}")
             producerScope.close()
         }
@@ -131,6 +129,7 @@ class XingHuoProvider(val project: Project) : LLMProvider {
                 urlBuilder.addQueryParameter(it.key, it.value)
             }
             val url = urlBuilder.build().toString().replace("https://", "wss://")
+            println(url)
             return Request.Builder().url(url).build()
         }
 

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

@@ -56,8 +56,8 @@ fun ReactivePasswordField(param: LLMParam, initBlock: JBPasswordField.(LLMParam)
 
     component.initBlock(reactive.value)
     component.document.addUndoableEditListener {
-        if (component.text == param.value) return@addUndoableEditListener
-        reactive.value.value = component.text
+        if (component.password.joinToString("") == param.value) return@addUndoableEditListener
+        reactive.value.value = component.password.joinToString("")
     }
 
     return component
@@ -74,7 +74,6 @@ fun ReactiveComboBox(param: LLMParam, initBlock: ComboBox<String>.(LLMParam) ->
     component.initBlock(reactive)
     component.addItemListener {
         if (it.stateChange == ItemEvent.SELECTED) {
-            println("item changed to ${component.selectedItem}")
             reactive.value = component.selectedItem as String
         }
     }
@@ -111,7 +110,6 @@ class LLMParam(
     val isEditable: Boolean = true,
     val type: ParamType = ParamType.Text,
     var items: List<String> = emptyList(),
-    var visible: Boolean = true,
 ) {
     enum class ParamType {
         Text, Password, ComboBox, Separator
@@ -126,7 +124,6 @@ class LLMParam(
             val changed = field != newValue
             field = newValue
             if (changed) {
-                println("value changed $newValue $value")
                 onChange?.invoke(this, newValue)
             }
         }
@@ -165,8 +162,6 @@ class LLMParam(
 
         fun ComboBox(value: String, items: List<String>) =
             LLMParam(value = value, type = ParamType.ComboBox, items = items.toList())
-
-        fun Separator() = LLMParam(type = ParamType.Separator)
     }
 }
 

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

@@ -88,7 +88,6 @@ class LLMSettingComponent(private val settings: AutoDevSettingsState) {
 
     private val currentLLMParams: List<LLMParam>
         get() {
-            println("current engine: ${_currentSelectedEngine}")
             return llmGroups[_currentSelectedEngine] ?: throw IllegalStateException("Unknown engine: ${settings.aiEngine}")
         }
 
@@ -131,7 +130,6 @@ class LLMSettingComponent(private val settings: AutoDevSettingsState) {
 
 
     fun applySettings(settings: AutoDevSettingsState, updateParams: Boolean = false) {
-        println("applySettings ${panel.components.size}")
         panel.removeAll()
         if (updateParams) {
             updateParams(settings)