Skip to content

Commit b228a2a

Browse files
authored
test(ai): update vertex ai live model name (#9376)
1 parent 180b1ad commit b228a2a

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

packages/ai/integration/constants.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ const modelNames: readonly string[] = ['gemini-2.0-flash', 'gemini-2.5-flash'];
5757
// The Live API requires a different set of models, and they're different for each backend.
5858
const liveModelNames: Map<BackendType, string[]> = new Map([
5959
[BackendType.GOOGLE_AI, ['gemini-live-2.5-flash-preview']],
60-
[BackendType.VERTEX_AI, ['gemini-2.0-flash-exp']]
60+
[BackendType.VERTEX_AI, ['gemini-2.0-flash-live-preview-04-09']]
6161
]);
6262

6363
/**

packages/ai/integration/live.test.ts

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717

1818
import { expect } from 'chai';
1919
import {
20-
BackendType,
2120
getLiveGenerativeModel,
2221
LiveGenerationConfig,
2322
LiveServerContent,
@@ -81,9 +80,6 @@ describe('Live', function () {
8180
};
8281

8382
liveTestConfigs.forEach(testConfig => {
84-
if (testConfig.ai.backend.backendType === BackendType.VERTEX_AI) {
85-
return;
86-
}
8783
describe(`${testConfig.toString()}`, () => {
8884
describe('Live', () => {
8985
it('should connect, send a message, receive a response, and close', async () => {

0 commit comments

Comments
 (0)