Skip to content

Commit ccbf7ba

Browse files
authored
Update ai model name used for e2e tests (#9282)
1 parent 0bb2fe6 commit ccbf7ba

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

e2e/smoke-tests/sample-apps/modular.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ function callPerformance(app) {
313313
async function callAI(app) {
314314
console.log('[AI] start');
315315
const ai = getAI(app, { backend: new VertexAIBackend() });
316-
const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
316+
const model = getGenerativeModel(ai, { model: 'gemini-2.5-flash' });
317317
const result = await model.countTokens('abcdefg');
318318
console.log(`[AI] counted tokens: ${result.totalTokens}`);
319319
}

e2e/smoke-tests/tests/modular.test.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -314,8 +314,8 @@ describe('MODULAR', () => {
314314
ai = getAI(app, { backend: new VertexAIBackend() });
315315
});
316316
it('getGenerativeModel() and countTokens()', async () => {
317-
const model = getGenerativeModel(ai, { model: 'gemini-1.5-flash' });
318-
expect(model.model).toMatch(/gemini-1.5-flash$/);
317+
const model = getGenerativeModel(ai, { model: 'gemini-2.5-flash' });
318+
expect(model.model).toMatch(/gemini-2.5-flash$/);
319319
const result = await model.countTokens('abcdefg');
320320
expect(result.totalTokens).toBeTruthy;
321321
});

0 commit comments

Comments
 (0)