@@ -49,6 +49,7 @@ struct GenerateContentIntegrationTests {
49
49
50
50
@Test ( arguments: [
51
51
( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2FlashLite) ,
52
+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2FlashLite) ,
52
53
( InstanceConfig . vertexAI_v1beta_staging, ModelNames . gemini2FlashLite) ,
53
54
( InstanceConfig . googleAI_v1beta, ModelNames . gemini2FlashLite) ,
54
55
( InstanceConfig . googleAI_v1beta, ModelNames . gemma3_4B) ,
@@ -136,9 +137,8 @@ struct GenerateContentIntegrationTests {
136
137
@Test ( arguments: [
137
138
( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2_5_FlashPreview, 0 ) ,
138
139
( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2_5_FlashPreview, 24576 ) ,
139
- // TODO: Add Vertex AI Gemini 2.5 Pro tests when available.
140
- // (InstanceConfig.vertexAI_v1beta, ModelNames.gemini2_5_ProPreview, 128),
141
- // (InstanceConfig.vertexAI_v1beta, ModelNames.gemini2_5_ProPreview, 32768),
140
+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2_5_ProPreview, 128 ) ,
141
+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2_5_ProPreview, 32768 ) ,
142
142
( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_FlashPreview, 0 ) ,
143
143
( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_FlashPreview, 24576 ) ,
144
144
( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_ProPreview, 128 ) ,
@@ -250,6 +250,7 @@ struct GenerateContentIntegrationTests {
250
250
251
251
@Test ( arguments: [
252
252
( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2FlashLite) ,
253
+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2FlashLite) ,
253
254
( InstanceConfig . vertexAI_v1beta_staging, ModelNames . gemini2FlashLite) ,
254
255
( InstanceConfig . googleAI_v1beta, ModelNames . gemini2FlashLite) ,
255
256
( InstanceConfig . googleAI_v1beta, ModelNames . gemma3_4B) ,
0 commit comments