diff --git a/src/main/java/com/google/genai/ApiClient.java b/src/main/java/com/google/genai/ApiClient.java index 90ca5bd56c2..5b1e457053a 100644 --- a/src/main/java/com/google/genai/ApiClient.java +++ b/src/main/java/com/google/genai/ApiClient.java @@ -231,7 +231,7 @@ private OkHttpClient createHttpClient( httpOptions .timeout() - .ifPresent(connectTimeout -> builder.connectTimeout(Duration.ofMillis(connectTimeout))); + .ifPresent(timeout -> builder.callTimeout(Duration.ofMillis(timeout))); HttpRetryOptions retryOptions = httpOptions.retryOptions().orElse(HttpRetryOptions.builder().build()); diff --git a/src/test/java/com/google/genai/HttpApiClientTest.java b/src/test/java/com/google/genai/HttpApiClientTest.java index d76a3d7958e..451d4b5b571 100644 --- a/src/test/java/com/google/genai/HttpApiClientTest.java +++ b/src/test/java/com/google/genai/HttpApiClientTest.java @@ -872,7 +872,7 @@ public void testHttpClientMLDevTimeout() throws Exception { OkHttpClient httpClient = client.httpClient(); assertNotNull(httpClient); - assertEquals(5000, httpClient.connectTimeoutMillis()); + assertEquals(5000, httpClient.callTimeoutMillis()); assertEquals(API_KEY, client.apiKey()); assertFalse(client.vertexAI()); } @@ -892,7 +892,7 @@ public void testHttpClientVertexTimeout() throws Exception { OkHttpClient httpClient = client.httpClient(); assertNotNull(httpClient); - assertEquals(4999, httpClient.connectTimeoutMillis()); + assertEquals(4999, httpClient.callTimeoutMillis()); assertEquals(PROJECT, client.project()); assertEquals(LOCATION, client.location()); assertTrue(client.vertexAI());