Skip to content

Commit 2480c40

Browse files
matthew29tangcopybara-github
authored andcommitted
chore: Update Imagen tests to use Imagen 4
PiperOrigin-RevId: 832387149
1 parent 7862ce3 commit 2480c40

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -612,7 +612,7 @@ public void testGenerateImagesAsync(boolean vertexAI) throws Exception {
612612

613613
// Act
614614
CompletableFuture<GenerateImagesResponse> responseFuture =
615-
client.async.models.generateImages("imagen-3.0-generate-002", "Red skateboard", config);
615+
client.async.models.generateImages("imagen-4.0-generate-001", "Red skateboard", config);
616616
GenerateImagesResponse response = responseFuture.join();
617617

618618
// Assert
@@ -648,7 +648,7 @@ public void testUpscaleImageAsync(boolean vertexAI) throws Exception {
648648
// Act
649649
if (vertexAI) {
650650
CompletableFuture<UpscaleImageResponse> responseFuture =
651-
client.async.models.upscaleImage("imagen-3.0-generate-002", image, "x2", config);
651+
client.async.models.upscaleImage("imagen-4.0-upscale-preview", image, "x2", config);
652652
UpscaleImageResponse response = responseFuture.join();
653653

654654
// Assert
@@ -661,7 +661,7 @@ public void testUpscaleImageAsync(boolean vertexAI) throws Exception {
661661
client
662662
.async
663663
.models
664-
.upscaleImage("imagen-3.0-generate-001", image, "x2", config)
664+
.upscaleImage("imagen-4.0-upscale-preview", image, "x2", config)
665665
.join());
666666
// Assert
667667
assertEquals(

0 commit comments

Comments
 (0)