Skip to content

Commit b9bf3ad

Browse files
authored
[Firebase AI] Remove .civicIntegrity from generateImage test (#15169)
1 parent b030a13 commit b9bf3ad

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

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

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,11 @@ struct GenerateContentIntegrationTests {
209209
topK: 1,
210210
responseModalities: [.text, .image]
211211
)
212+
let safetySettings = safetySettings.filter {
213+
// HARM_CATEGORY_CIVIC_INTEGRITY is deprecated in Vertex AI but only rejected when using the
214+
// 'gemini-2.0-flash-preview-image-generation' model.
215+
$0.harmCategory != .civicIntegrity
216+
}
212217
let model = FirebaseAI.componentInstance(config).generativeModel(
213218
modelName: ModelNames.gemini2FlashPreviewImageGeneration,
214219
generationConfig: generationConfig,

0 commit comments

Comments
 (0)