From e41323960300b73532bc09e7b233d3a7c0629cc5 Mon Sep 17 00:00:00 2001 From: Andrew Heard Date: Wed, 16 Apr 2025 11:24:26 -0400 Subject: [PATCH] [Vertex AI] Re-enable Developer API staging integration tests --- .../Integration/GenerateContentIntegrationTests.swift | 3 ++- .../Tests/TestApp/Tests/Utilities/InstanceConfig.swift | 8 ++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift index 0f06131ce8a..5c55685a085 100644 --- a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift +++ b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/GenerateContentIntegrationTests.swift @@ -116,8 +116,9 @@ struct GenerateContentIntegrationTests { @Test(arguments: [ InstanceConfig.vertexV1Beta, - // Temporarily disabled due to Firebase Proxy issues. + // TODO(andrewheard): Prod config temporarily disabled due to backend issue. // InstanceConfig.developerV1Beta, + InstanceConfig.developerV1BetaStaging, // Remove after re-enabling `developerV1Beta` config. ]) func generateImage(_ config: InstanceConfig) async throws { let generationConfig = GenerationConfig( diff --git a/FirebaseVertexAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift b/FirebaseVertexAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift index 2bfd135d562..1130597fead 100644 --- a/FirebaseVertexAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift +++ b/FirebaseVertexAI/Tests/TestApp/Tests/Utilities/InstanceConfig.swift @@ -51,9 +51,9 @@ struct InstanceConfig { vertexV1Staging, vertexV1Beta, vertexV1BetaStaging, - // Temporarily disabled due to Firebase Proxy issues: + // TODO(andrewheard): Prod config temporarily disabled due to backend issue: // developerV1Beta, - // developerV1BetaStaging, + developerV1BetaStaging, developerV1Spark, developerV1BetaSpark, ] @@ -63,9 +63,9 @@ struct InstanceConfig { vertexV1Staging, vertexV1Beta, vertexV1BetaStaging, - // Temporarily disabled due to Firebase Proxy issues: + // TODO(andrewheard): Prod config temporarily disabled due to backend issue: // developerV1Beta, - // developerV1BetaStaging, + developerV1BetaStaging, developerV1BetaSpark, ]