Skip to content

Commit 60f32ef

Browse files
committed
Convert tests to use test project prompt templates
1 parent 5c784c7 commit 60f32ef

File tree

1 file changed

+6
-14
lines changed

1 file changed

+6
-14
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/ServerPromptTemplateIntegrationTests.swift

Lines changed: 6 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -20,20 +20,18 @@ import Testing
2020

2121
struct 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

Comments
 (0)