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 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 exceptionIfMldev = testTableItem.exceptionIfMldev(); Optional exceptionIfVertex = testTableItem.exceptionIfVertex(); if (exceptionIfMldev.isPresent() && !client.vertexAI()) {