Update SyncMcpToolCallbackProvider.java

Signed-off-by: Sizhe Fan <[email protected]>

Author: paoxiaomooo

mcp/common/src/main/java/org/springframework/ai/mcp/SyncMcpToolCallbackProvider.java

@@ -17,6 +17,7 @@
 package org.springframework.ai.mcp;
 
 import java.util.List;
+import java.util.Set;
 import java.util.function.BiPredicate;
 
 import io.modelcontextprotocol.client.McpSyncClient;
@@ -115,6 +116,33 @@ public SyncMcpToolCallbackProvider(McpSyncClient... mcpClients) {
 		this(List.of(mcpClients));
 	}
 
+	/**
+	 * Creates a new {@code SyncMcpToolCallbackProvider} instance that includes only
+	 * clients from the specified allowed servers.
+	 * <p>
+	 * This constructor:
+	 * <ol>
+	 * <li>Filters the provided MCP clients to only those matching allowed server
+	 * names</li>
+	 * <li>Retains all tools from the selected clients (no additional tool filtering)</li>
+	 * <li>Ensures no null parameters are passed</li>
+	 * </ol>
+	 * @param mcpClients complete list of available MCP clients
+	 * @param allowedServerNames set of server names to include (case-sensitive)
+	 * @throws IllegalArgumentException if parameters are null or empty
+	 * @since 1.1.0
+	 */
+	public SyncMcpToolCallbackProvider(List<McpSyncClient> mcpClients, Set<String> allowedServerNames) {
+		Assert.notNull(mcpClients, "MCP clients list must not be null");
+		Assert.notNull(allowedServerNames, "Allowed server names set must not be null");
+		Assert.notEmpty(allowedServerNames, "Allowed server names set must not be empty");
+
+		this.mcpClients = mcpClients.stream()
+			.filter(client -> allowedServerNames.contains(client.getServerInfo().name()))
+			.toList();
+		this.toolFilter = (client, tool) -> true; // No additional filtering
+	}
+
 	/**
 	 * Discovers and returns all available tools from all connected MCP servers.
 	 * <p>