File tree Expand file tree Collapse file tree 1 file changed +8
-3
lines changed
core/src/main/java/com/google/adk/tools/mcp Expand file tree Collapse file tree 1 file changed +8
-3
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments