File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -313,7 +313,7 @@ function callPerformance(app) {
313
313
async function callAI ( app ) {
314
314
console . log ( '[AI] start' ) ;
315
315
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' } ) ;
317
317
const result = await model . countTokens ( 'abcdefg' ) ;
318
318
console . log ( `[AI] counted tokens: ${ result . totalTokens } ` ) ;
319
319
}
Original file line number Diff line number Diff line change @@ -314,8 +314,8 @@ describe('MODULAR', () => {
314
314
ai = getAI ( app , { backend : new VertexAIBackend ( ) } ) ;
315
315
} ) ;
316
316
it ( 'getGenerativeModel() and countTokens()' , async ( ) => {
317
- const model = getGenerativeModel ( ai , { model : 'gemini-1 .5-flash' } ) ;
318
- expect ( model . model ) . toMatch ( / g e m i n i - 1 .5 - f l a s h $ / ) ;
317
+ const model = getGenerativeModel ( ai , { model : 'gemini-2 .5-flash' } ) ;
318
+ expect ( model . model ) . toMatch ( / g e m i n i - 2 .5 - f l a s h $ / ) ;
319
319
const result = await model . countTokens ( 'abcdefg' ) ;
320
320
expect ( result . totalTokens ) . toBeTruthy ;
321
321
} ) ;
You can’t perform that action at this time.
0 commit comments