From 1e998894f5549b8e2d54c77289a0ca60a660ef3f Mon Sep 17 00:00:00 2001
From: Qiao Wang <qiaowang@google.com>
Date: Tue, 17 Jun 2025 11:24:51 -0700
Subject: [PATCH] test: update system test to use 2.0 models

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/googleapis/nodejs-vertexai/pull/490 from googleapis:release-please--branches--main--components--vertexai 85ffaaee8823979122fb229513f1377b7d147116
PiperOrigin-RevId: 772551170
---
 system_test/end_to_end_sample_test.ts | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/system_test/end_to_end_sample_test.ts b/system_test/end_to_end_sample_test.ts
index af9d59c4..b11471c6 100644
--- a/system_test/end_to_end_sample_test.ts
+++ b/system_test/end_to_end_sample_test.ts
@@ -134,7 +134,7 @@ const vertexAI = new VertexAI({
   location: LOCATION,
 });
 
-const TEXT_MODEL_NAME = 'gemini-1.0-pro';
+const TEXT_MODEL_NAME = 'gemini-2.0-flash';
 const generativeTextModel = vertexAI.getGenerativeModel({
   model: TEXT_MODEL_NAME,
   generationConfig: {
@@ -148,30 +148,30 @@ const generativeTextModelPreview = vertexAI.preview.getGenerativeModel({
   },
 });
 const generativeTextModelWithPrefix = vertexAI.getGenerativeModel({
-  model: 'models/gemini-1.0-pro',
+  model: 'models/gemini-2.0-flash',
   generationConfig: {
     maxOutputTokens: 256,
   },
 });
 const generativeTextModelWithPrefixPreview =
   vertexAI.preview.getGenerativeModel({
-    model: 'models/gemini-1.0-pro',
+    model: 'models/gemini-2.0-flash',
     generationConfig: {
       maxOutputTokens: 256,
     },
   });
 const generativeVisionModel = vertexAI.getGenerativeModel({
-  model: 'gemini-1.0-pro-vision',
+  model: 'gemini-2.0-flash',
 });
 const generativeVisionModelPreview = vertexAI.preview.getGenerativeModel({
-  model: 'gemini-1.0-pro-vision',
+  model: 'gemini-2.0-flash',
 });
 const generativeVisionModelWithPrefix = vertexAI.getGenerativeModel({
-  model: 'models/gemini-1.0-pro-vision',
+  model: 'gemini-2.0-flash',
 });
 const generativeVisionModelWithPrefixPreview =
   vertexAI.preview.getGenerativeModel({
-    model: 'models/gemini-1.0-pro-vision',
+    model: 'models/gemini-2.0-flash',
   });
 describe('generateContentStream', () => {
   beforeEach(() => {