Skip to content

Commit 8601aa9

Browse files
chore(genai): replace deprecated genai image generation model (#10254)
* chore(genai): replace deprecated genai image generation model * Fix lint issue * Update veo generate model to fix Kokoro CI builds * remove unused model
1 parent dc24cee commit 8601aa9

6 files changed

Lines changed: 7 additions & 9 deletions

File tree

genai/snippets/pom.xml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,6 @@
5656
<dependency>
5757
<groupId>com.google.cloud</groupId>
5858
<artifactId>google-cloud-storage</artifactId>
59-
<scope>test</scope>
6059
</dependency>
6160
<dependency>
6261
<groupId>com.openai</groupId>

genai/snippets/src/main/java/genai/imagegeneration/ImageGenVirtualTryOnWithTextAndImage.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ public class ImageGenVirtualTryOnWithTextAndImage {
3636

3737
public static void main(String[] args) throws IOException {
3838
// TODO(developer): Replace these variables before running the sample.
39-
String modelId = "virtual-try-on-preview-08-04";
39+
String modelId = "virtual-try-on-001";
4040
String outputFile = "resources/output/man_in_sweater.png";
4141
generateContent(modelId, outputFile);
4242
}
@@ -84,4 +84,4 @@ public static Image generateContent(String modelId, String outputFile) throws IO
8484
}
8585
}
8686
}
87-
// [END googlegenaisdk_imggen_virtual_try_on_with_txt_img]
87+
// [END googlegenaisdk_imggen_virtual_try_on_with_txt_img]

genai/snippets/src/main/java/genai/videogeneration/VideoGenWithImg.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ public class VideoGenWithImg {
3232

3333
public static void main(String[] args) throws InterruptedException {
3434
// TODO(developer): Replace these variables before running the sample.
35-
String modelId = "veo-3.0-generate-preview";
35+
String modelId = "veo-3.1-generate-001";
3636
String outputGcsUri = "gs://your-bucket/your-prefix";
3737
generateContent(modelId, outputGcsUri);
3838
}

genai/snippets/src/main/java/genai/videogeneration/VideoGenWithTxt.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ public class VideoGenWithTxt {
3232

3333
public static void main(String[] args) throws InterruptedException {
3434
// TODO(developer): Replace these variables before running the sample.
35-
String modelId = "veo-3.0-generate-001";
35+
String modelId = "veo-3.1-generate-001";
3636
String outputGcsUri = "gs://your-bucket/your-prefix";
3737
generateContent(modelId, outputGcsUri);
3838
}

genai/snippets/src/test/java/genai/imagegeneration/ImageGenerationIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ public class ImageGenerationIT {
4545
private static final String PREFIX = "genai-img-generation-" + UUID.randomUUID();
4646
private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX);
4747
private static final String IMAGEN_4_MODEL = "imagen-4.0-generate-001";
48-
private static final String VIRTUAL_TRY_ON_MODEL = "virtual-try-on-preview-08-04";
48+
private static final String VIRTUAL_TRY_ON_MODEL = "virtual-try-on-001";
4949

5050
private ByteArrayOutputStream bout;
5151
private PrintStream out;

genai/snippets/src/test/java/genai/videogeneration/VideoGenerationIT.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,7 @@
3737
@RunWith(JUnit4.class)
3838
public class VideoGenerationIT {
3939

40-
private static final String VIDEO_GEN_MODEL = "veo-3.0-generate-001";
41-
private static final String VIDEO_GEN_PREVIEW_MODEL = "veo-3.0-generate-preview";
40+
private static final String VIDEO_GEN_MODEL = "veo-3.1-generate-001";
4241
private static final String BUCKET_NAME = "java-docs-samples-testing";
4342
private static final String PREFIX = "genai-video-generation-" + UUID.randomUUID();
4443
private static final String OUTPUT_GCS_URI = String.format("gs://%s/%s", BUCKET_NAME, PREFIX);
@@ -82,7 +81,7 @@ public void tearDown() {
8281

8382
@Test
8483
public void testVideoGenWithImg() throws InterruptedException {
85-
String response = VideoGenWithImg.generateContent(VIDEO_GEN_PREVIEW_MODEL, OUTPUT_GCS_URI);
84+
String response = VideoGenWithImg.generateContent(VIDEO_GEN_MODEL, OUTPUT_GCS_URI);
8685
assertThat(response).isNotEmpty();
8786
assertThat(bout.toString()).contains(OUTPUT_GCS_URI);
8887
}

0 commit comments

Comments
 (0)