@@ -20,20 +20,18 @@ import Testing
2020
2121struct ServerPromptTemplateIntegrationTests {
2222 private static let testConfigs : [ InstanceConfig ] = [
23- . vertexAI_v1beta,
23+ . googleAI_v1beta,
24+ // .vertexAI_v1beta, // Waiting for backend location propagation.
2425 . vertexAI_v1beta_global,
2526 ]
2627 private static let imageGenerationTestConfigs : [ InstanceConfig ] = [ . vertexAI_v1beta]
2728
28- @Test ( arguments: [
29- // The "greeting2" template is only available in the `global` location.
30- InstanceConfig . vertexAI_v1beta_global,
31- ] )
29+ @Test ( arguments: testConfigs)
3230 func generateContentWithText( _ config: InstanceConfig ) async throws {
3331 let model = FirebaseAI . componentInstance ( config) . templateGenerativeModel ( )
3432 let userName = " paul "
3533 let response = try await model. generateContent (
36- templateID: " greeting4 " ,
34+ templateID: " greeting-5 " ,
3735 inputs: [
3836 " name " : userName,
3937 " language " : " Spanish " ,
@@ -43,10 +41,7 @@ struct ServerPromptTemplateIntegrationTests {
4341 #expect( text. contains ( " Paul " ) )
4442 }
4543
46- @Test ( arguments: [
47- // The "greeting2" template is only available in the `global` location.
48- InstanceConfig . vertexAI_v1beta_global,
49- ] )
44+ @Test ( arguments: testConfigs)
5045 func generateContentStream( _ config: InstanceConfig ) async throws {
5146 let model = FirebaseAI . componentInstance ( config) . templateGenerativeModel ( )
5247 let userName = " paul "
@@ -141,10 +136,7 @@ struct ServerPromptTemplateIntegrationTests {
141136 #expect( !resultText. isEmpty)
142137 }
143138
144- @Test ( arguments: [
145- // The "greeting2" template is only available in the `global` location.
146- InstanceConfig . vertexAI_v1beta_global,
147- ] )
139+ @Test ( arguments: testConfigs)
148140 func chat( _ config: InstanceConfig ) async throws {
149141 let model = FirebaseAI . componentInstance ( config) . templateGenerativeModel ( )
150142 let initialHistory = [
0 commit comments