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) {
313313async 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}
Original file line number Diff line number Diff 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 ( / 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 $ / ) ;
319319 const result = await model . countTokens ( 'abcdefg' ) ;
320320 expect ( result . totalTokens ) . toBeTruthy ;
321321 } ) ;
You can’t perform that action at this time.
0 commit comments