Skip to content

Commit a220fb1

Browse files
authored
[Firebase AI] Add integration tests for global endpoint (#14991)
1 parent a14936c commit a220fb1

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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),

FirebaseAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,10 @@ struct InstanceConfig: Equatable, Encodable {
2323
static let vertexAI_v1beta = InstanceConfig(
2424
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyProd), version: .v1beta)
2525
)
26+
static let vertexAI_v1beta_global = InstanceConfig(
27+
location: "global",
28+
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyProd), version: .v1beta)
29+
)
2630
static let vertexAI_v1beta_staging = InstanceConfig(
2731
apiConfig: APIConfig(service: .vertexAI(endpoint: .firebaseProxyStaging), version: .v1beta)
2832
)
@@ -43,6 +47,7 @@ struct InstanceConfig: Equatable, Encodable {
4347

4448
static let allConfigs = [
4549
vertexAI_v1beta,
50+
vertexAI_v1beta_global,
4651
vertexAI_v1beta_staging,
4752
googleAI_v1beta,
4853
googleAI_v1beta_staging,

0 commit comments

Comments
 (0)