Skip to content

Commit 5e42130

Browse files
committed
review
1 parent 33c057f commit 5e42130

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

FirebaseAI/Tests/TestApp/Tests/Integration/GenerativeModelSessionTests.swift

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,15 +150,15 @@
150150
modelName: ModelNames.gemini2_5_FlashLite,
151151
)
152152
let session = GenerativeModelSession(model: model)
153-
let prompt = "Generate a recipe for a vegetarian pasta dish."
153+
let prompt = "Generate a recipe for a pasta dish with meat."
154154

155155
let response = try await session.respond(to: prompt, generating: Recipe.self)
156156

157157
let recipe = response.content
158158
#expect(!recipe.name.isEmpty)
159159
#expect(recipe.preparationTime >= 5)
160160
#expect(recipe.preparationTime <= 120)
161-
#expect(recipe.isVegetarian == true)
161+
#expect(!recipe.isVegetarian)
162162
#expect(recipe.rating >= 1.0)
163163
#expect(recipe.rating <= 5.0)
164164
#expect(!recipe.ingredients.isEmpty)
@@ -185,6 +185,7 @@
185185
#expect(recipe.preparationTime >= 5)
186186
#expect(recipe.preparationTime <= 120)
187187
#expect(!recipe.ingredients.isEmpty)
188+
#expect(recipe.isVegetarian)
188189
}
189190
}
190191
}

0 commit comments

Comments
 (0)