Skip to content

Commit c25fd30

Browse files
committed
Update ToolCallback providers to use client.getCurrentInitializationResult()
Signed-off-by: Ilayaperumal Gopinathan <[email protected]>
1 parent 2ae92fd commit c25fd30

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

auto-configurations/mcp/spring-ai-autoconfigure-mcp-client-common/src/test/java/org/springframework/ai/mcp/client/common/autoconfigure/McpToolCallbackAutoConfigurationConditionTests.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,10 @@
2626
import reactor.core.publisher.Mono;
2727

2828
import org.springframework.ai.mcp.AsyncMcpToolCallbackProvider;
29-
import org.springframework.ai.mcp.McpToolFilter;
3029
import org.springframework.ai.mcp.McpClientMetadata;
3130
import org.springframework.ai.mcp.McpMetadata;
3231
import org.springframework.ai.mcp.McpServerMetadata;
32+
import org.springframework.ai.mcp.McpToolFilter;
3333
import org.springframework.ai.mcp.SyncMcpToolCallbackProvider;
3434
import org.springframework.ai.mcp.client.common.autoconfigure.McpToolCallbackAutoConfiguration.McpToolCallbackAutoConfigurationCondition;
3535
import org.springframework.boot.test.context.runner.ApplicationContextRunner;

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ public ToolCallback[] getToolCallbacks() {
147147
.stream()
148148
.filter(tool -> this.toolFilter.test(new McpMetadata(
149149
new McpClientMetadata(mcpClient.getClientCapabilities(), mcpClient.getClientInfo()),
150-
new McpServerMetadata(mcpClient.initialize().block())), tool))
150+
new McpServerMetadata(mcpClient.getCurrentInitializationResult())), tool))
151151
.map(tool -> new AsyncMcpToolCallback(mcpClient, tool))
152152
.toArray(ToolCallback[]::new))
153153
.block();

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ public ToolCallback[] getToolCallbacks() {
133133
.stream()
134134
.filter(tool -> this.toolFilter.test(new McpMetadata(
135135
new McpClientMetadata(mcpClient.getClientCapabilities(), mcpClient.getClientInfo()),
136-
new McpServerMetadata(mcpClient.initialize())), tool))
136+
new McpServerMetadata(mcpClient.getCurrentInitializationResult())), tool))
137137
.map(tool -> new SyncMcpToolCallback(mcpClient, tool)))
138138
.toArray(ToolCallback[]::new);
139139
validateToolCallbacks(array);

0 commit comments

Comments
 (0)