feat(devti): add MCP group support and refactor toolchain functions

- Add MCP group field to AgentTool
- Refactor toolInfos() method to use project parameter
- Update allToolchains() method to handle null project
- Modify database and MCP function providers to use new method signature
- Update completion provider to use new allToolchains method

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/agent/tool/AgentTool.kt

@@ -9,6 +9,7 @@ data class AgentTool(
     val example: String,
     val isMcp: Boolean = false,
     val completion: String = "",
+    val mcpGroup: String = "",
 ) : Tool {
     override fun toString(): String {
         val string = if (description.isEmpty()) "" else """desc: $description"""

core/src/main/kotlin/cc/unitmesh/devti/bridge/BridgeToolProvider.kt

@@ -19,8 +19,9 @@ object BridgeToolProvider {
         val functions = ToolchainFunctionProvider.all()
 
         commonTools += functions.flatMap {
-            if (it.toolInfos().isNotEmpty()) {
-                return@flatMap it.toolInfos()
+            val toolInfos = it.toolInfos(project)
+            if (toolInfos.isNotEmpty()) {
+                return@flatMap toolInfos
             }
 
             val funcNames = it.funcNames()

core/src/main/kotlin/cc/unitmesh/devti/devin/dataprovider/BuiltinCommand.kt

@@ -6,6 +6,7 @@ import cc.unitmesh.devti.agent.Tool
 import cc.unitmesh.devti.agent.tool.AgentTool
 import cc.unitmesh.devti.provider.toolchain.ToolchainFunctionProvider
 import com.intellij.icons.AllIcons
+import com.intellij.openapi.project.Project
 import com.intellij.openapi.project.ProjectManager
 import java.nio.charset.StandardCharsets
 import javax.swing.Icon
@@ -178,9 +179,11 @@ enum class BuiltinCommand(
             return providerName
         }
 
-        suspend fun allToolchains(): List<AgentTool> {
+        suspend fun allToolchains(project: Project?): List<AgentTool> {
+            if (project == null) return emptyList()
+
             return ToolchainFunctionProvider.all().map {
-                val toolInfo: List<AgentTool> = it.toolInfos()
+                val toolInfo: List<AgentTool> = it.toolInfos(project)
                 if (toolInfo.isNotEmpty()) {
                     return@map toolInfo
                 }

core/src/main/kotlin/cc/unitmesh/devti/mcp/client/McpFunctionProvider.kt

@@ -14,10 +14,8 @@ class McpFunctionProvider : ToolchainFunctionProvider {
         return CustomMcpServerManager.instance(project).collectServerInfos().map { it.name }
     }
 
-    override suspend fun toolInfos(): List<AgentTool> {
-        val manager = CustomMcpServerManager.instance(
-            ProjectManager.getInstance().openProjects.firstOrNull() ?: return emptyList()
-        )
+    override suspend fun toolInfos(project: Project): List<AgentTool> {
+        val manager = CustomMcpServerManager.instance(project)
         return manager.collectServerInfos().map {
             val schemaJson = Json.encodeToString<Input>(it.inputSchema)
             val mockData = Json.encodeToString(MockDataGenerator.generateMockData(it.inputSchema))
@@ -26,6 +24,7 @@ class McpFunctionProvider : ToolchainFunctionProvider {
                 it.description ?: "",
                 "Here is command and JSON schema\n/${it.name}\n```json\n$schemaJson\n```",
                 isMcp = true,
+                mcpGroup = it.name,
                 completion = mockData
             )
         }

core/src/main/kotlin/cc/unitmesh/devti/provider/toolchain/ToolchainFunctionProvider.kt

@@ -5,7 +5,7 @@ import com.intellij.openapi.extensions.ExtensionPointName
 import com.intellij.openapi.project.Project
 
 interface ToolchainFunctionProvider {
-    suspend fun toolInfos(): List<AgentTool> = emptyList()
+    suspend fun toolInfos(project: Project): List<AgentTool> = emptyList()
 
     suspend fun funcNames(): List<String>
 

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/completion/provider/ToolchainCommandCompletion.kt

@@ -16,7 +16,7 @@ class ToolchainCommandCompletion : CompletionProvider<CompletionParameters>() {
         result: CompletionResultSet,
     ) {
         runBlocking {
-            BuiltinCommand.allToolchains().forEach {
+            BuiltinCommand.allToolchains(parameters.originalFile.project).forEach {
                 val lookupElement = createCommandCompletionCandidate(it)
                 result.addElement(lookupElement)
             }
@@ -30,12 +30,20 @@ class ToolchainCommandCompletion : CompletionProvider<CompletionParameters>() {
         }
 
         val element = LookupElementBuilder.create(tool.name).withIcon(AutoDevIcons.MCP)
-            .withTailText(tool.description)
+            .withTailText(getText(tool))
             .withInsertHandler { context, _ ->
                 context.editor.caretModel.moveToOffset(context.tailOffset)
                 context.editor.document.insertString(context.tailOffset, "\n````json\n${tool.completion}\n```")
             }
 
         return PrioritizedLookupElement.withPriority(element, 97.0)
     }
+
+    private fun getText(tool: AgentTool): String {
+        return return if (tool.mcpGroup.isNotEmpty()) {
+            " ${tool.mcpGroup}: ${tool.description}"
+        } else {
+            tool.description
+        }
+    }
 }

exts/ext-database/src/main/kotlin/cc/unitmesh/database/provider/DatabaseFunctionProvider.kt

@@ -13,7 +13,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.openapi.util.NlsSafe
 
 class DatabaseFunctionProvider : ToolchainFunctionProvider {
-    override suspend fun toolInfos(): List<AgentTool> {
+    override suspend fun toolInfos(project: Project): List<AgentTool> {
         val example = BuiltinCommand.example("database")
         return listOf(AgentTool("database", "Database schema and query tool", example))
     }