@@ -49,6 +49,7 @@ struct GenerateContentIntegrationTests {
4949
5050 @Test ( arguments: [
5151 ( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2FlashLite) ,
52+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2FlashLite) ,
5253 ( InstanceConfig . vertexAI_v1beta_staging, ModelNames . gemini2FlashLite) ,
5354 ( InstanceConfig . googleAI_v1beta, ModelNames . gemini2FlashLite) ,
5455 ( InstanceConfig . googleAI_v1beta, ModelNames . gemma3_4B) ,
@@ -136,9 +137,8 @@ struct GenerateContentIntegrationTests {
136137 @Test ( arguments: [
137138 ( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2_5_FlashPreview, 0 ) ,
138139 ( 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 ) ,
142142 ( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_FlashPreview, 0 ) ,
143143 ( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_FlashPreview, 24576 ) ,
144144 ( InstanceConfig . googleAI_v1beta, ModelNames . gemini2_5_ProPreview, 128 ) ,
@@ -250,6 +250,7 @@ struct GenerateContentIntegrationTests {
250250
251251 @Test ( arguments: [
252252 ( InstanceConfig . vertexAI_v1beta, ModelNames . gemini2FlashLite) ,
253+ ( InstanceConfig . vertexAI_v1beta_global, ModelNames . gemini2FlashLite) ,
253254 ( InstanceConfig . vertexAI_v1beta_staging, ModelNames . gemini2FlashLite) ,
254255 ( InstanceConfig . googleAI_v1beta, ModelNames . gemini2FlashLite) ,
255256 ( InstanceConfig . googleAI_v1beta, ModelNames . gemma3_4B) ,
0 commit comments