Skip to content

Commit 39a8bc6

Browse files
author
wenhaozhao
committed
feat: pass timeout config while init mcp client
1 parent d204294 commit 39a8bc6

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

core/src/main/java/com/google/adk/tools/mcp/McpSessionManager.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,16 +51,21 @@ public static McpSyncClient initializeSession(Object connectionParams) {
5151

5252
public static McpSyncClient initializeSession(
5353
Object connectionParams, McpTransportBuilder transportBuilder) {
54+
Duration initializationTimeout = null;
55+
Duration requestTimeout = null;
5456
McpClientTransport transport = transportBuilder.build(connectionParams);
55-
57+
if (connectionParams instanceof SseServerParameters sseServerParams) {
58+
initializationTimeout = sseServerParams.timeout();
59+
requestTimeout = sseServerParams.sseReadTimeout();
60+
}
5661
McpSyncClient client =
5762
McpClient.sync(transport)
58-
.requestTimeout(Duration.ofSeconds(10))
63+
.initializationTimeout(initializationTimeout == null ? Duration.ofSeconds(10) : initializationTimeout)
64+
.requestTimeout(requestTimeout == null ? Duration.ofSeconds(10) : requestTimeout)
5965
.capabilities(ClientCapabilities.builder().build())
6066
.build();
6167
InitializeResult initResult = client.initialize();
6268
logger.debug("Initialize Client Result: {}", initResult);
63-
6469
return client;
6570
}
6671
}

0 commit comments

Comments
 (0)