feat(a2a): enhance A2A service initialization and error handling #443

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/a2a/A2AClientConsumer.kt

@@ -1,5 +1,6 @@
 package cc.unitmesh.devti.a2a
 
+import com.intellij.openapi.diagnostic.logger
 import io.a2a.A2A
 import io.a2a.client.*
 import io.a2a.client.config.ClientConfig
@@ -63,7 +64,8 @@ class A2AClientConsumer {
     }
 
     fun sendMessage(agentName: String, msgText: String): String {
-        val client = clientMap[agentName] ?: throw IllegalArgumentException("No client found for $agentName")
+        val client = clientMap[agentName]
+            ?: return "Failed to find A2A client for $agentName. Available clients: ${clientMap.keys}"
 
         return try {
             val message = A2A.toUserMessage(msgText)
@@ -76,7 +78,8 @@ class A2AClientConsumer {
             responseFuture.get(30, java.util.concurrent.TimeUnit.SECONDS)
         } catch (e: Exception) {
             responseMap.remove(agentName)
-            throw RuntimeException("Failed to send message to agent $agentName: ${e.message}", e)
+            logger<A2AClientConsumer>().error("Failed to send message to $agentName: ${e.message}", e)
+            return e.message ?: "Unknown error"
         }
     }
 

core/src/main/kotlin/cc/unitmesh/devti/a2a/A2AService.kt

@@ -1,5 +1,7 @@
 package cc.unitmesh.devti.a2a
 
+import cc.unitmesh.devti.mcp.client.McpServer
+import cc.unitmesh.devti.settings.customize.customizeSetting
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
 import io.a2a.spec.AgentCard
@@ -11,17 +13,46 @@ import io.a2a.spec.AgentCard
 class A2AService(private val project: Project) {
     private var a2aClientConsumer: A2AClientConsumer? = null
     private var availableAgents: List<AgentCard> = emptyList()
-    
+    private val cached = mutableMapOf<String, List<AgentCard>>()
+
     companion object {
         fun getInstance(project: Project): A2AService {
             return project.getService(A2AService::class.java)
         }
     }
-    
+
+    /**
+     * Initialize A2A service by reading configuration from project settings
+     */
+    fun initialize() {
+        val mcpServerConfig = project.customizeSetting.mcpServerConfig
+        if (mcpServerConfig.isEmpty()) {
+            a2aClientConsumer = null
+            availableAgents = emptyList()
+            return
+        }
+
+        if (cached.containsKey(mcpServerConfig)) {
+            availableAgents = cached[mcpServerConfig] ?: emptyList()
+            return
+        }
+
+        val mcpConfig = McpServer.load(mcpServerConfig)
+        if (mcpConfig?.a2aServers.isNullOrEmpty()) {
+            a2aClientConsumer = null
+            availableAgents = emptyList()
+            return
+        }
+
+        val servers = mcpConfig.a2aServers.values.toList()
+        initialize(servers)
+        cached[mcpServerConfig] = availableAgents
+    }
+
     /**
      * Initialize A2A service with server configurations
      */
-    fun initialize(servers: List<A2aServer>) {
+    private fun initialize(servers: List<A2aServer>) {
         try {
             a2aClientConsumer = A2AClientConsumer()
             a2aClientConsumer?.init(servers)
@@ -57,7 +88,20 @@ class A2AService(private val project: Project) {
     fun isAvailable(): Boolean {
         return a2aClientConsumer != null && availableAgents.isNotEmpty()
     }
-    
+
+    /**
+     * Get enabled A2A servers from configuration content
+     */
+    fun getEnabledServers(content: String): Map<String, A2aServer>? {
+        val mcpConfig = McpServer.load(content)
+        return mcpConfig?.a2aServers?.filter { entry ->
+            // A2A servers don't have a disabled flag like MCP servers, so all are enabled
+            true
+        }?.mapValues { entry ->
+            entry.value
+        }
+    }
+
     /**
      * Refresh the list of available agents
      */

core/src/main/kotlin/cc/unitmesh/devti/a2a/A2ASketchToolchainProvider.kt

@@ -20,6 +20,10 @@ class A2ASketchToolchainProvider : SketchToolchainProvider {
         fun collectA2ATools(project: Project): List<AgentTool> {
             return try {
                 val a2aService = project.getService(A2AService::class.java)
+
+                // Initialize A2A service from configuration
+                a2aService.initialize()
+
                 val agentCards = a2aService.getAvailableAgents()
 
                 agentCards.map { agentCard ->

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/exec/A2AInsCommand.kt

@@ -39,6 +39,7 @@ class A2AInsCommand(
 
     override suspend fun execute(): String? {
         val a2aService = project.getService(A2AService::class.java)
+        a2aService.initialize()
 
         if (!a2aService.isAvailable()) {
             return "A2A service is not available. Please check your A2A configuration."