From 5753db604dcc79d8317ddb8d4c08f47ff8c7e0a7 Mon Sep 17 00:00:00 2001
From: Christina Holland <chholland@google.com>
Date: Mon, 29 Sep 2025 11:28:17 -0700
Subject: [PATCH] Update ai model name used for e2e tests

---
 e2e/smoke-tests/sample-apps/modular.js | 2 +-
 e2e/smoke-tests/tests/modular.test.ts  | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/e2e/smoke-tests/sample-apps/modular.js b/e2e/smoke-tests/sample-apps/modular.js
index 2d66b752081..20ff0ce7bd2 100644
--- a/e2e/smoke-tests/sample-apps/modular.js
+++ b/e2e/smoke-tests/sample-apps/modular.js
@@ -313,7 +313,7 @@ function callPerformance(app) {
 async function callAI(app) {
   console.log('[AI] start');
   const ai = getAI(app, { backend: new VertexAIBackend() });
-  const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
+  const model = getGenerativeModel(ai, { model: 'gemini-2.5-flash' });
   const result = await model.countTokens('abcdefg');
   console.log(`[AI] counted tokens: ${result.totalTokens}`);
 }
diff --git a/e2e/smoke-tests/tests/modular.test.ts b/e2e/smoke-tests/tests/modular.test.ts
index c8f6afe779d..536a271ca5e 100644
--- a/e2e/smoke-tests/tests/modular.test.ts
+++ b/e2e/smoke-tests/tests/modular.test.ts
@@ -314,8 +314,8 @@ describe('MODULAR', () => {
       ai = getAI(app, { backend: new VertexAIBackend() });
     });
     it('getGenerativeModel() and countTokens()', async () => {
-      const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
-      expect(model.model).toMatch(/gemini-1.5-flash$/);
+      const model = getGenerativeModel(ai, { model: 'gemini-2.5-flash' });
+      expect(model.model).toMatch(/gemini-2.5-flash$/);
       const result = await model.countTokens('abcdefg');
       expect(result.totalTokens).toBeTruthy;
     });