Add support for new FinishReason and BlockReason values

Author: David Motsonashvili

firebase-vertexai/CHANGELOG.md

@@ -1,4 +1,5 @@
 # Unreleased
+* [fixed] Added support for new values sent by the server for `FinishReason` and `BlockReason`
 
 
 # 16.1.0

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/Candidate.kt

@@ -247,7 +247,13 @@ public class FinishReason private constructor(public val name: String, public va
     MAX_TOKENS,
     SAFETY,
     RECITATION,
-    OTHER;
+    OTHER,
+    BLOCKLIST,
+    PROHIBITED_CONTENT,
+    SPII,
+    MALFORMED_FUNCTION_CALL,
+    MODEL_ARMOR,
+    IMAGE_SAFETY;
 
     internal object Serializer : KSerializer<Internal> by FirstOrdinalSerializer(Internal::class)
 
@@ -258,6 +264,12 @@ public class FinishReason private constructor(public val name: String, public va
         SAFETY -> FinishReason.SAFETY
         STOP -> FinishReason.STOP
         OTHER -> FinishReason.OTHER
+        BLOCKLIST -> FinishReason.BLOCKLIST
+        PROHIBITED_CONTENT -> FinishReason.PROHIBITED_CONTENT
+        SPII -> FinishReason.SPII
+        MALFORMED_FUNCTION_CALL -> FinishReason.MALFORMED_FUNCTION_CALL
+        MODEL_ARMOR -> FinishReason.MODEL_ARMOR
+        IMAGE_SAFETY -> FinishReason.IMAGE_SAFETY
         else -> FinishReason.UNKNOWN
       }
   }
@@ -281,5 +293,27 @@ public class FinishReason private constructor(public val name: String, public va
 
     /** Model stopped for another reason. */
     @JvmField public val OTHER: FinishReason = FinishReason("OTHER", 5)
+
+    /** Token generation stopped because the content contains forbidden terms. */
+    @JvmField public val BLOCKLIST: FinishReason = FinishReason("BLOCKLIST", 6)
+
+    /** Token generation stopped for potentially containing prohibited content. */
+    @JvmField public val PROHIBITED_CONTENT: FinishReason = FinishReason("PROHIBITED_CONTENT", 7)
+
+    /**
+     * Token generation stopped because the content potentially contains Sensitive Personally
+     * Identifiable Information (SPII).
+     */
+    @JvmField public val SPII: FinishReason = FinishReason("SPII", 8)
+
+    /** The function call generated by the model is invalid. */
+    @JvmField
+    public val MALFORMED_FUNCTION_CALL: FinishReason = FinishReason("MALFORMED_FUNCTION_CALL", 9)
+
+    /** The model response was blocked by Model Armor. */
+    @JvmField public val MODEL_ARMOR: FinishReason = FinishReason("MODEL_ARMOR", 10)
+
+    /** Token generation stopped because generated images has safety violations. */
+    @JvmField public val IMAGE_SAFETY: FinishReason = FinishReason("IMAGE_SAFETY", 11)
   }
 }

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/PromptFeedback.kt

@@ -56,14 +56,22 @@ public class BlockReason private constructor(public val name: String, public val
     UNKNOWN,
     @SerialName("BLOCKED_REASON_UNSPECIFIED") UNSPECIFIED,
     SAFETY,
-    OTHER;
+    OTHER,
+    BLOCKLIST,
+    PROHIBITED_CONTENT,
+    MODEL_ARMOR,
+    IMAGE_SAFETY;
 
     internal object Serializer : KSerializer<Internal> by FirstOrdinalSerializer(Internal::class)
 
     internal fun toPublic() =
       when (this) {
         SAFETY -> BlockReason.SAFETY
         OTHER -> BlockReason.OTHER
+        BLOCKLIST -> BlockReason.BLOCKLIST
+        PROHIBITED_CONTENT -> BlockReason.PROHIBITED_CONTENT
+        MODEL_ARMOR -> BlockReason.MODEL_ARMOR
+        IMAGE_SAFETY -> BlockReason.IMAGE_SAFETY
         else -> BlockReason.UNKNOWN
       }
   }
@@ -76,5 +84,17 @@ public class BlockReason private constructor(public val name: String, public val
 
     /** Content was blocked for another reason. */
     @JvmField public val OTHER: BlockReason = BlockReason("OTHER", 2)
+
+    /** Content was blocked for another reason. */
+    @JvmField public val BLOCKLIST: BlockReason = BlockReason("BLOCKLIST", 3)
+
+    /** Candidates blocked due to the terms which are included from the terminology blocklist. */
+    @JvmField public val PROHIBITED_CONTENT: BlockReason = BlockReason("PROHIBITED_CONTENT", 4)
+
+    /** The user prompt was blocked by Model Armor. */
+    @JvmField public val MODEL_ARMOR: BlockReason = BlockReason("MODEL_ARMOR", 5)
+
+    /** Candidates blocked due to unsafe image generation content. */
+    @JvmField public val IMAGE_SAFETY: BlockReason = BlockReason("IMAGE_SAFETY", 6)
   }
 }