feat(custom-agent): add support for authentication types in custom agent execution #51

The commit adds support for different authentication types when executing custom agents, including Bearer tokens.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentExecutor.kt

@@ -1,6 +1,7 @@
 package cc.unitmesh.devti.counit
 
 import cc.unitmesh.devti.counit.configurable.customAgentSetting
+import cc.unitmesh.devti.counit.model.AuthType
 import cc.unitmesh.devti.counit.model.CustomAgentConfig
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
@@ -13,11 +14,21 @@ import okhttp3.RequestBody
 class CustomAgentExecutor(val project: Project) {
     private var client = OkHttpClient()
 
-    fun execute(input: String, selectedAgent: CustomAgentConfig): String? {
+    fun execute(input: String, agent: CustomAgentConfig): String? {
         val serverAddress = project.customAgentSetting.serverAddress ?: return null
         val body = RequestBody.create("application/json; charset=utf-8".toMediaTypeOrNull(), input)
         val builder = Request.Builder()
 
+        val auth = agent.auth
+        when (auth?.type) {
+            AuthType.Bearer -> {
+                builder.addHeader("Authorization", "Bearer ${auth.token}")
+                builder.addHeader("Content-Type", "application/json")
+            }
+
+            null -> TODO()
+        }
+
         client = client.newBuilder().build()
         val call = client.newCall(builder.url(serverAddress).post(body).build())
 

src/main/kotlin/cc/unitmesh/devti/counit/model/CustomAgentConfig.kt

@@ -53,11 +53,16 @@ data class CustomAgentConfig(
     val responseAction: ResponseAction = ResponseAction.Direct,
     val customFlowTransition: List<CustomFlowTransition> = emptyList(),
     val interactive: InteractionType = InteractionType.ChatPanel,
-    val auth: CustomAgentAuth = CustomAgentAuth()
+    val auth: CustomAgentAuth? = null
 )
 
 @Serializable
 data class CustomAgentAuth(
-    val type: String = "",
+    val type: AuthType = AuthType.Bearer,
     val token: String = "",
 )
+
+@Serializable
+enum class AuthType {
+    Bearer,
+}