From e76098043355da613fcbb5db3b15e8185b052f97 Mon Sep 17 00:00:00 2001 From: Ayush Agrawal Date: Thu, 6 Nov 2025 09:56:53 -0800 Subject: [PATCH] chore: Use call timeout for HttpOptions.timeout in Java GenAI SDK. Co-authored-by: Edmund Kohlwey PiperOrigin-RevId: 828998640 --- src/main/java/com/google/genai/ApiClient.java | 2 +- src/test/java/com/google/genai/HttpApiClientTest.java | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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());