@@ -159,7 +159,7 @@ describe('VertexAI', () => {
159
159
project : PROJECT ,
160
160
location : LOCATION ,
161
161
} ) ;
162
- vertexai . preview [ 'tokenInternalPromise' ] = Promise . resolve ( TEST_TOKEN ) ;
162
+ spyOnProperty ( vertexai . preview , 'token' , 'get' ) . and . resolveTo ( TEST_TOKEN ) ;
163
163
model = vertexai . preview . getGenerativeModel ( { model : 'gemini-pro' } ) ;
164
164
} ) ;
165
165
@@ -247,8 +247,7 @@ describe('VertexAI', () => {
247
247
location : LOCATION ,
248
248
apiEndpoint : TEST_ENDPOINT_BASE_PATH ,
249
249
} ) ;
250
- vertexaiWithBasePath . preview [ 'tokenInternalPromise' ] =
251
- Promise . resolve ( TEST_TOKEN ) ;
250
+ spyOnProperty ( vertexaiWithBasePath . preview , 'token' , 'get' ) . and . resolveTo ( TEST_TOKEN ) ;
252
251
model = vertexaiWithBasePath . preview . getGenerativeModel ( {
253
252
model : 'gemini-pro' ,
254
253
} ) ;
@@ -280,8 +279,8 @@ describe('VertexAI', () => {
280
279
project : PROJECT ,
281
280
location : LOCATION ,
282
281
} ) ;
283
- vertexaiWithoutBasePath . preview [ 'tokenInternalPromise' ] =
284
- Promise . resolve ( TEST_TOKEN ) ;
282
+ spyOnProperty ( vertexaiWithoutBasePath . preview , 'token' , 'get' )
283
+ . and . resolveTo ( TEST_TOKEN ) ;
285
284
model = vertexaiWithoutBasePath . preview . getGenerativeModel ( {
286
285
model : 'gemini-pro' ,
287
286
} ) ;
@@ -461,7 +460,7 @@ describe('ChatSession', () => {
461
460
462
461
beforeEach ( ( ) => {
463
462
vertexai = new VertexAI ( { project : PROJECT , location : LOCATION } ) ;
464
- vertexai . preview [ 'tokenInternalPromise' ] = Promise . resolve ( TEST_TOKEN ) ;
463
+ spyOnProperty ( vertexai . preview , 'token' , 'get' ) . and . resolveTo ( TEST_TOKEN ) ;
465
464
model = vertexai . preview . getGenerativeModel ( { model : 'gemini-pro' } ) ;
466
465
chatSession = model . startChat ( {
467
466
history : TEST_USER_CHAT_MESSAGE ,
0 commit comments