From de53776cb071b2b1cd65a7335073f21f6acee909 Mon Sep 17 00:00:00 2001
From: Ayush Agrawal <ayushagra@google.com>
Date: Tue, 28 Jul 2026 23:54:44 -0700
Subject: [PATCH] feat: Support per-request retryOptions in the JS GenAI SDK
 API Client

PiperOrigin-RevId: 955694582
---
 src/test/java/com/google/genai/TableTest.java | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/test/java/com/google/genai/TableTest.java b/src/test/java/com/google/genai/TableTest.java
index 84a2a6d7daf..0c1c1bb7fb5 100644
--- a/src/test/java/com/google/genai/TableTest.java
+++ b/src/test/java/com/google/genai/TableTest.java
@@ -360,6 +360,13 @@ private static Map<String, Object> prepareParameters(TestTableItem testTableItem
 
   private static void handleException(
       Throwable cause, TestTableItem testTableItem, Client client, String testName) {
+    if (client.clientMode().equals("api") && cause instanceof com.google.genai.errors.ApiException) {
+      if (((com.google.genai.errors.ApiException) cause).code() == 429) {
+        org.junit.jupiter.api.Assumptions.assumeTrue(
+            false, "Resource Exhausted (429). Skipping test instead of failing: " + cause.getMessage());
+        return;
+      }
+    }
     Optional<String> exceptionIfMldev = testTableItem.exceptionIfMldev();
     Optional<String> exceptionIfVertex = testTableItem.exceptionIfVertex();
     if (exceptionIfMldev.isPresent() && !client.vertexAI()) {