Fix testGenerateContentStream_successUnknownSafetyEnum

firebase · Oct 9, 2024 · ae00ec5 · ae00ec5
1 parent 2a51f09
commit ae00ec5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift b/FirebaseVertexAI/Tests/Unit/GenerativeModelTests.swift
@@ -977,7 +977,7 @@ final class GenerativeModelTests: XCTestCase {
       )
     let unknownSafetyRating = SafetyRating(
       category: HarmCategory(rawValue: "HARM_CATEGORY_DANGEROUS_CONTENT_NEW_ENUM"),
-      probability: SafetyRating.HarmProbability(rawValue: "FAKE_NEW_HARM_PROBABILITY")
+      probability: SafetyRating.HarmProbability(rawValue: "NEGLIGIBLE_UNKNOWN_ENUM")
     )
 
     var foundUnknownSafetyRating = false