Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

chore: Use stream flatmap when build toolcallbacks. (#2802)

Browse Source 
Signed-off-by: 虎鸣 <hepin.p@alibaba-inc.com>
This commit is contained in:
He-Pin(kerr)
2025-04-30 01:25:26 +08:00
committed by GitHub
parent c6ccc1a4a3
commit 5aa22c4187
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
mcp/common/src/main/java/org/springframework/ai/mcp/SyncMcpToolCallbackProvider.java
View File

@@ -51,7 +51,7 @@ import org.springframework.util.CollectionUtils;
* // Get all available tools
* ToolCallback[] tools = provider.getToolCallbacks();
* }</pre>
*
* <p>
* Example usage with multiple clients:
*
* <pre>{@code
@@ -63,10 +63,10 @@ import org.springframework.util.CollectionUtils;
* }</pre>
*
* @author Christian Tzolov
* @since 1.0.0
* @see ToolCallbackProvider
* @see SyncMcpToolCallback
* @see McpSyncClient
* @since 1.0.0
*/
public class SyncMcpToolCallbackProvider implements ToolCallbackProvider {
@@ -130,17 +130,13 @@ public class SyncMcpToolCallbackProvider implements ToolCallbackProvider {
*/
@Override
public ToolCallback[] getToolCallbacks() {
var toolCallbacks = new ArrayList<>();
this.mcpClients.stream()
.forEach(mcpClient -> toolCallbacks.addAll(mcpClient.listTools()
var array = this.mcpClients.stream()
.flatMap(mcpClient -> mcpClient.listTools()
.tools()
.stream()
.filter(tool -> this.toolFilter.test(mcpClient, tool))
.map(tool -> new SyncMcpToolCallback(mcpClient, tool))
.toList()));
var array = toolCallbacks.toArray(new ToolCallback[0]);
.map(tool -> new SyncMcpToolCallback(mcpClient, tool)))
.toArray(ToolCallback[]::new);
validateToolCallbacks(array);
return array;
}