feat: pass timeout config while init mcp client

Author: wenhaozhao

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

@@ -46,29 +46,26 @@ public McpSyncClient createSession() {
   }
 
   public static McpSyncClient initializeSession(Object connectionParams) {
-    McpClientTransport transport = new DefaultMcpTransportBuilder().build(connectionParams);
-    McpSyncClient client =
-        McpClient.sync(transport)
-            .requestTimeout(Duration.ofSeconds(10))
-            .capabilities(ClientCapabilities.builder().build())
-            .build();
-    InitializeResult initResult = client.initialize();
-    logger.debug("Initialize Client Result: {}", initResult);
-    return client;
+    return initializeSession(connectionParams, new DefaultMcpTransportBuilder());
   }
 
   public static McpSyncClient initializeSession(
       Object connectionParams, McpTransportBuilder transportBuilder) {
+    Duration initializationTimeout = null;
+    Duration requestTimeout = null;
     McpClientTransport transport = transportBuilder.build(connectionParams);
-
+    if (connectionParams instanceof SseServerParameters sseServerParams) {
+      initializationTimeout = sseServerParams.timeout();
+      requestTimeout = sseServerParams.sseReadTimeout();
+    }
     McpSyncClient client =
         McpClient.sync(transport)
-            .requestTimeout(Duration.ofSeconds(10))
+            .initializationTimeout(initializationTimeout == null ? Duration.ofSeconds(10) : initializationTimeout)
+            .requestTimeout(requestTimeout == null ? Duration.ofSeconds(10) : requestTimeout)
             .capabilities(ClientCapabilities.builder().build())
             .build();
     InitializeResult initResult = client.initialize();
     logger.debug("Initialize Client Result: {}", initResult);
-
     return client;
   }
 }