No public description

PiperOrigin-RevId: 743353678

Author: google-genai-bot

src/main/java/com/google/genai/ApiClient.java

@@ -20,6 +20,7 @@
 
 import com.google.auth.oauth2.GoogleCredentials;
 import com.google.common.collect.ImmutableMap;
+import com.google.genai.errors.GenAiIOException;
 import com.google.genai.types.HttpOptions;
 import java.io.IOException;
 import java.util.Map;
@@ -103,13 +104,7 @@ abstract class ApiClient {
       throw new IllegalArgumentException("Location must not be empty.");
     }
 
-    try {
-      this.credentials = Optional.of(credentials.orElse(defaultCredentials()));
-    } catch (IOException e) {
-      throw new IllegalArgumentException(
-          "Failed to get application default credentials, please explicitly provide credentials.",
-          e);
-    }
+    this.credentials = Optional.of(credentials.orElse(defaultCredentials()));
 
     this.httpOptions = defaultHttpOptions(/* vertexAI= */ true, this.location);
 
@@ -133,7 +128,7 @@ private CloseableHttpClient createHttpClient(Optional<Integer> timeout) {
   }
 
   /** Sends a Http Post request given the path and request json string. */
-  public abstract ApiResponse post(String path, String requestJson) throws IOException;
+  public abstract ApiResponse post(String path, String requestJson);
 
   /** Returns the library version. */
   static String libraryVersion() {
@@ -216,8 +211,14 @@ static HttpOptions defaultHttpOptions(boolean vertexAI, Optional<String> locatio
     return defaultHttpOptionsBuilder.build();
   }
 
-  GoogleCredentials defaultCredentials() throws IOException {
-    return GoogleCredentials.getApplicationDefault()
-        .createScoped("https://www.googleapis.com/auth/cloud-platform");
+  GoogleCredentials defaultCredentials() {
+    try {
+      return GoogleCredentials.getApplicationDefault()
+          .createScoped("https://www.googleapis.com/auth/cloud-platform");
+    } catch (IOException e) {
+      throw new GenAiIOException(
+          "Failed to get application default credentials, please explicitly provide credentials.",
+          e);
+    }
   }
 }

src/main/java/com/google/genai/ApiResponse.java

@@ -16,14 +16,13 @@
 
 package com.google.genai;
 
-import java.io.IOException;
 import org.apache.http.HttpEntity;
-import org.apache.http.HttpException;
 
 /** The API response contains a response to a call to the GenAI APIs. */
 public abstract class ApiResponse implements AutoCloseable {
   /** Gets the HttpEntity. */
-  public abstract HttpEntity getEntity() throws HttpException;
+  public abstract HttpEntity getEntity();
 
-  public abstract void close() throws IOException;
+  @Override
+  public abstract void close();
 }

src/main/java/com/google/genai/AsyncLive.java

@@ -20,6 +20,7 @@
 
 import com.fasterxml.jackson.databind.node.ObjectNode;
 import com.google.auth.oauth2.GoogleCredentials;
+import com.google.genai.errors.GenAiIOException;
 import com.google.genai.types.LiveClientSetup;
 import com.google.genai.types.LiveConnectConfig;
 import com.google.genai.types.LiveServerMessage;
@@ -120,7 +121,7 @@ private Map<String, String> getWebSocketHeaders() {
         credentials.refreshIfExpired();
         headers.put("Authorization", "Bearer " + credentials.getAccessToken().getTokenValue());
       } catch (IOException e) {
-        throw new IllegalStateException("Failed to get credentials for Vertex AI.", e);
+        throw new GenAiIOException("Failed to refresh credentials for Vertex AI.", e);
       }
     }
     return headers;
@@ -230,7 +231,7 @@ public void onClose(int code, String reason, boolean remote) {
       System.out.println("Live session closed with code: " + code + " and reason: " + reason);
       if (!sessionFuture.isDone()) {
         sessionFuture.completeExceptionally(
-            new IOException("WebSocket closed unexpectedly: " + reason));
+            new GenAiIOException("WebSocket closed unexpectedly: " + reason));
       }
     }
 

src/main/java/com/google/genai/AsyncModels.java

@@ -31,10 +31,8 @@
 import com.google.genai.types.Image;
 import com.google.genai.types.UpscaleImageConfig;
 import com.google.genai.types.UpscaleImageResponse;
-import java.io.IOException;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
-import org.apache.http.HttpException;
 
 public final class AsyncModels {
   Models models;
@@ -46,14 +44,7 @@ public AsyncModels(ApiClient apiClient) {
   @BetaApi
   public CompletableFuture<GenerateVideosOperation> generateVideos(
       String model, String prompt, Image image, GenerateVideosConfig config) {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.generateVideos(model, prompt, image, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+    return CompletableFuture.supplyAsync(() -> models.generateVideos(model, prompt, image, config));
   }
 
   /**
@@ -65,19 +56,10 @@ public CompletableFuture<GenerateVideosOperation> generateVideos(
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<GenerateContentResponse> generateContent(
       String model, List<Content> contents, GenerateContentConfig config) {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.generateContent(model, contents, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+    return CompletableFuture.supplyAsync(() -> models.generateContent(model, contents, config));
   }
 
   /**
@@ -89,12 +71,9 @@ public CompletableFuture<GenerateContentResponse> generateContent(
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<GenerateContentResponse> generateContent(
-      String model, Content content, GenerateContentConfig config)
-      throws IOException, HttpException {
+      String model, Content content, GenerateContentConfig config) {
     return generateContent(model, Transformers.tContents(null, (Object) content), config);
   }
 
@@ -107,11 +86,9 @@ public CompletableFuture<GenerateContentResponse> generateContent(
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<GenerateContentResponse> generateContent(
-      String model, String text, GenerateContentConfig config) throws IOException, HttpException {
+      String model, String text, GenerateContentConfig config) {
     return generateContent(model, Transformers.tContents(null, (Object) text), config);
   }
 
@@ -125,19 +102,11 @@ public CompletableFuture<GenerateContentResponse> generateContent(
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<ResponseStream<GenerateContentResponse>> generateContentStream(
       String model, List<Content> contents, GenerateContentConfig config) {
     return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.generateContentStream(model, contents, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+        () -> models.generateContentStream(model, contents, config));
   }
 
   /**
@@ -150,12 +119,9 @@ public CompletableFuture<ResponseStream<GenerateContentResponse>> generateConten
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<ResponseStream<GenerateContentResponse>> generateContentStream(
-      String model, Content content, GenerateContentConfig config)
-      throws IOException, HttpException {
+      String model, Content content, GenerateContentConfig config) {
     return generateContentStream(model, Transformers.tContents(null, (Object) content), config);
   }
 
@@ -168,11 +134,9 @@ public CompletableFuture<ResponseStream<GenerateContentResponse>> generateConten
    *     the optional configurations
    * @return a {@link com.google.genai.types.GenerateContentResponse} instance that contains
    *     response contents and other metadata
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<ResponseStream<GenerateContentResponse>> generateContentStream(
-      String model, String text, GenerateContentConfig config) throws IOException, HttpException {
+      String model, String text, GenerateContentConfig config) {
     return generateContentStream(model, Transformers.tContents(null, (Object) text), config);
   }
 
@@ -185,19 +149,10 @@ public CompletableFuture<ResponseStream<GenerateContentResponse>> generateConten
    *     optional configurations
    * @return a {@link com.google.genai.types.GenerateImagesResponse} instance that contains the
    *     generated images.
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<GenerateImagesResponse> generateImages(
-      String model, String prompt, GenerateImagesConfig config) throws IOException, HttpException {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.generateImages(model, prompt, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+      String model, String prompt, GenerateImagesConfig config) {
+    return CompletableFuture.supplyAsync(() -> models.generateImages(model, prompt, config));
   }
 
   /**
@@ -210,20 +165,11 @@ public CompletableFuture<GenerateImagesResponse> generateImages(
    *     optional configurations
    * @return a {@link com.google.genai.types.UpscaleImageResponse} instance that contains the
    *     upscaled image.
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<UpscaleImageResponse> upscaleImage(
-      String model, Image image, String upscaleFactor, UpscaleImageConfig config)
-      throws IOException, HttpException {
+      String model, Image image, String upscaleFactor, UpscaleImageConfig config) {
     return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.upscaleImage(model, image, upscaleFactor, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+        () -> models.upscaleImage(model, image, upscaleFactor, config));
   }
 
   /**
@@ -233,19 +179,10 @@ public CompletableFuture<UpscaleImageResponse> upscaleImage(
    * @param text the text string to send to the embedding model
    * @return a {@link com.google.genai.types.EmbedContentResponse} instance that contains the
    *     embedding.
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<EmbedContentResponse> embedContent(
-      String model, String text, EmbedContentConfig config) throws IOException, HttpException {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.embedContent(model, text, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+      String model, String text, EmbedContentConfig config) {
+    return CompletableFuture.supplyAsync(() -> models.embedContent(model, text, config));
   }
 
   /**
@@ -255,19 +192,9 @@ public CompletableFuture<EmbedContentResponse> embedContent(
    * @param texts the list of text strings to send to the embedding model
    * @return a {@link com.google.genai.types.EmbedContentResponse} instance that contains the
    *     embedding.
-   * @throws IOException if an I/O error occurs while making the API call
-   * @throws HttpException if an HTTP error occurs while making the API call
    */
   public CompletableFuture<EmbedContentResponse> embedContent(
-      String model, List<String> texts, EmbedContentConfig config)
-      throws IOException, HttpException {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return models.embedContent(model, texts, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+      String model, List<String> texts, EmbedContentConfig config) {
+    return CompletableFuture.supplyAsync(() -> models.embedContent(model, texts, config));
   }
 }

src/main/java/com/google/genai/AsyncOperations.java

@@ -21,9 +21,7 @@
 import com.google.api.core.BetaApi;
 import com.google.genai.types.GenerateVideosOperation;
 import com.google.genai.types.GetOperationConfig;
-import java.io.IOException;
 import java.util.concurrent.CompletableFuture;
-import org.apache.http.HttpException;
 
 public final class AsyncOperations {
   Operations operations;
@@ -38,20 +36,10 @@ public AsyncOperations(ApiClient apiClient) {
    * @param operation A GenerateVideosOperation.
    * @param config The configuration for getting the operation.
    * @return A GenerateVideosOperation with the updated status of the operation.
-   * @throws IOException If an I/O error occurs while reading the operation.
-   * @throws HttpException If an HTTP error occurs while reading the operation.
    */
   @BetaApi
   public CompletableFuture<GenerateVideosOperation> getVideoOperation(
-      GenerateVideosOperation operation, GetOperationConfig config)
-      throws IOException, HttpException {
-    return CompletableFuture.supplyAsync(
-        () -> {
-          try {
-            return operations.getVideoOperation(operation, config);
-          } catch (IOException | HttpException e) {
-            throw new RuntimeException(e);
-          }
-        });
+      GenerateVideosOperation operation, GetOperationConfig config) {
+    return CompletableFuture.supplyAsync(() -> operations.getVideoOperation(operation, config));
   }
 }

src/main/java/com/google/genai/AsyncSession.java

@@ -27,7 +27,6 @@
 import com.google.genai.types.LiveSendToolResponseParameters;
 import com.google.genai.types.LiveServerMessage;
 import java.util.concurrent.CompletableFuture;
-import java.util.concurrent.CompletionException;
 import java.util.function.Consumer;
 
 /**
@@ -101,18 +100,7 @@ public CompletableFuture<Void> sendToolResponse(LiveSendToolResponseParameters t
    *     will fail if the message cannot be sent.
    */
   private CompletableFuture<Void> send(LiveClientMessage input) {
-    return CompletableFuture.runAsync(
-        () -> {
-          try {
-            websocket.send(input.toJson());
-          } catch (RuntimeException e) {
-            throw new CompletionException("Failed to send message to live session.", e);
-          }
-        });
-  }
-
-  public CompletableFuture<Void> sendContent(LiveClientMessage input) {
-    return send(input);
+    return CompletableFuture.runAsync(() -> websocket.send(input.toJson()));
   }
 
   /**
@@ -133,17 +121,12 @@ public CompletableFuture<Void> receive(Consumer<LiveServerMessage> onMessage) {
   /**
    * Closes the WebSocket connection.
    *
-   * @return A {@link CompletableFuture} that completes when the connection has been closed. Fails
-   *     with a {@link CompletionException} wrapping the underlying cause if closing fails.
+   * @return A {@link CompletableFuture} that completes when the connection has been closed.
    */
   public CompletableFuture<Void> close() {
     return CompletableFuture.runAsync(
         () -> {
-          try {
-            websocket.close();
-          } catch (RuntimeException e) {
-            throw new CompletionException("Failed to close websocket connection.", e);
-          }
+          websocket.close();
         });
   }
 }