Skip to content

Commit a62384b

Browse files
committed
feat(mcp): filter enabled servers only #371
Change `getServerConfigs` to `getEnabledServers` and modify logic to return only non-disabled servers. The method now filters out servers marked as disabled before returning the map.
1 parent 731966a commit a62384b

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,13 @@ class CustomMcpServerManager(val project: Project) {
4848
return toolsMap
4949
}
5050

51-
fun getServerConfigs(content: String): Map<String, McpServer>? {
51+
fun getEnabledServers(content: String): Map<String, McpServer>? {
5252
val mcpConfig = McpServer.load(content)
53-
return mcpConfig?.mcpServers
53+
return mcpConfig?.mcpServers?.filter { entry ->
54+
entry.value.disabled != true
55+
}?.mapValues { entry ->
56+
entry.value
57+
}
5458
}
5559

5660
suspend fun collectServerInfo(serverKey: String, serverConfig: McpServer): List<Tool> {

core/src/main/kotlin/cc/unitmesh/devti/mcp/ui/McpToolListPanel.kt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ class McpToolListPanel(private val project: Project) : JPanel() {
5151
}
5252

5353
loadingJob = CoroutineScope(Dispatchers.IO).launch {
54-
val serverConfigs = mcpServerManager.getServerConfigs(content)
54+
val serverConfigs = mcpServerManager.getEnabledServers(content)
5555

5656
if (serverConfigs.isNullOrEmpty()) {
5757
SwingUtilities.invokeLater {

0 commit comments

Comments
 (0)