chore: chore

PiperOrigin-RevId: 897263266

Author: speedstorm1

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

@@ -94,6 +94,40 @@ public java.time.Duration deserialize(JsonParser p, DeserializationContext ctxt)
     }
   }
 
+  /** Custom Jackson serializer for {@code byte[]} to output standard base64 without linefeeds. */
+  static class CustomByteArraySerializer extends JsonSerializer<byte[]> {
+    @Override
+    public void serialize(byte[] value, JsonGenerator gen, SerializerProvider serializers)
+        throws java.io.IOException {
+      if (value == null) {
+        gen.writeNull();
+      } else {
+        gen.writeString(java.util.Base64.getEncoder().encodeToString(value));
+      }
+    }
+  }
+
+  /** Custom Jackson deserializer for {@code byte[]} to support URL-safe base64 strings. */
+  static class CustomByteArrayDeserializer extends JsonDeserializer<byte[]> {
+    @Override
+    public byte[] deserialize(JsonParser p, DeserializationContext ctxt)
+        throws java.io.IOException, JsonProcessingException {
+      String value = p.getValueAsString();
+      if (value == null) {
+        return null;
+      }
+      try {
+        if (value.contains("-") || value.contains("_")) {
+          return java.util.Base64.getUrlDecoder().decode(value);
+        } else {
+          return java.util.Base64.getDecoder().decode(value);
+        }
+      } catch (IllegalArgumentException e) {
+        throw ctxt.weirdStringException(value, byte[].class, "Failed to decode base64 string");
+      }
+    }
+  }
+
   /** Configures the stream read constraints for the JSON parser. */
   private static void configureStreamReadConstraints(int maxReadLength) {
     if (maxReadLength <= 0) {
@@ -116,6 +150,8 @@ private static void configureStreamReadConstraints(int maxReadLength) {
     SimpleModule customModule = new SimpleModule();
     customModule.addSerializer(java.time.Duration.class, new CustomDurationSerializer());
     customModule.addDeserializer(java.time.Duration.class, new CustomDurationDeserializer());
+    customModule.addSerializer(byte[].class, new CustomByteArraySerializer());
+    customModule.addDeserializer(byte[].class, new CustomByteArrayDeserializer());
 
     // Register JavaTimeModule for other java.time types *before* the custom module
     // This ensures our custom Duration handling takes precedence over the default one

src/test/java/com/google/genai/TableTest.java

@@ -200,10 +200,7 @@ private static Collection<DynamicTest> createTestCases(
       String msg = " => Test skipped: replay tests are not supported for edit_image";
       return Collections.singletonList(DynamicTest.dynamicTest(testName + msg, () -> {}));
     }
-    if (testName.contains("models.embed_content.test_new_api_inline_pdf")) {
-      String msg = " => Test skipped: inline byte deserialization fails";
-      return Collections.singletonList(DynamicTest.dynamicTest(testName + msg, () -> {}));
-    }
+
     // TODO(b/457846189): Support models.list filter parameter
     if (testName.contains("models.list.test_tuned_models_with_filter")
         || testName.contains("models.list.test_tuned_models.vertex")) {