diff --git a/firebase-vertexai/CHANGELOG.md b/firebase-vertexai/CHANGELOG.md index 56606cf1740..29083bb4e3a 100644 --- a/firebase-vertexai/CHANGELOG.md +++ b/firebase-vertexai/CHANGELOG.md @@ -1,4 +1,5 @@ # Unreleased +* [changed] Added support for Civic Integrity in Harm Category. (#6502) * [fixed] Improved error message when using an invalid location. (#6428) * [fixed] Fixed issue where Firebase App Check error tokens were unintentionally missing from the requests. (#6409) * [fixed] Clarified in the documentation that `Schema.integer` and `Schema.float` only provide hints to the model. (#6420) diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/shared/Types.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/shared/Types.kt index c34af862def..b32772e995c 100644 --- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/shared/Types.kt +++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/common/shared/Types.kt @@ -38,7 +38,8 @@ internal enum class HarmCategory { @SerialName("HARM_CATEGORY_HARASSMENT") HARASSMENT, @SerialName("HARM_CATEGORY_HATE_SPEECH") HATE_SPEECH, @SerialName("HARM_CATEGORY_SEXUALLY_EXPLICIT") SEXUALLY_EXPLICIT, - @SerialName("HARM_CATEGORY_DANGEROUS_CONTENT") DANGEROUS_CONTENT + @SerialName("HARM_CATEGORY_DANGEROUS_CONTENT") DANGEROUS_CONTENT, + @SerialName("HARM_CATEGORY_CIVIC_INTEGRITY") CIVIC_INTEGRITY, } internal typealias Base64 = String diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/conversions.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/conversions.kt index 80542eebe85..f8388054260 100644 --- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/conversions.kt +++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/internal/util/conversions.kt @@ -137,6 +137,8 @@ internal fun HarmCategory.toInternal() = com.google.firebase.vertexai.common.shared.HarmCategory.SEXUALLY_EXPLICIT HarmCategory.DANGEROUS_CONTENT -> com.google.firebase.vertexai.common.shared.HarmCategory.DANGEROUS_CONTENT + HarmCategory.CIVIC_INTEGRITY -> + com.google.firebase.vertexai.common.shared.HarmCategory.CIVIC_INTEGRITY HarmCategory.UNKNOWN -> com.google.firebase.vertexai.common.shared.HarmCategory.UNKNOWN else -> throw makeMissingCaseException("HarmCategory", ordinal) } @@ -315,6 +317,8 @@ internal fun com.google.firebase.vertexai.common.shared.HarmCategory.toPublic() HarmCategory.SEXUALLY_EXPLICIT com.google.firebase.vertexai.common.shared.HarmCategory.DANGEROUS_CONTENT -> HarmCategory.DANGEROUS_CONTENT + com.google.firebase.vertexai.common.shared.HarmCategory.CIVIC_INTEGRITY -> + HarmCategory.CIVIC_INTEGRITY else -> HarmCategory.UNKNOWN } diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/HarmCategory.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/HarmCategory.kt index 9783ca48923..d19de2e1568 100644 --- a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/HarmCategory.kt +++ b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/HarmCategory.kt @@ -33,5 +33,8 @@ public class HarmCategory private constructor(public val ordinal: Int) { /** Dangerous content. */ @JvmField public val DANGEROUS_CONTENT: HarmCategory = HarmCategory(4) + + /** Content that may be used to harm civic integrity. */ + @JvmField public val CIVIC_INTEGRITY: HarmCategory = HarmCategory(5) } }