From baff2a5bce1830e8c76cee9dfe465775b9ea8847 Mon Sep 17 00:00:00 2001
From: Anayeli Malvaez Colin <anayelim@google.com>
Date: Wed, 11 Feb 2026 19:40:27 +0000
Subject: [PATCH] chore(aiplatform): Change
 CreateBatchPredictionGeminiJobSample to use gemini 2.5

---
 .../batchpredict/CreateBatchPredictionGeminiJobSample.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java b/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
index 2d081403698..1ad1f09f003 100644
--- a/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
+++ b/aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java
@@ -72,7 +72,7 @@ public static BatchPredictionJob createBatchPredictionGeminiJobSample(
       String modelName =
           String.format(
               "projects/%s/locations/%s/publishers/google/models/%s",
-              project, location, "gemini-2.0-flash-001");
+              project, location, "gemini-2.5-flash");
 
       BatchPredictionJob batchPredictionJob =
           BatchPredictionJob.newBuilder()