Rename Modality to ContentModality

There are other possible classes that will use the same Modality name

Author: rlazo

firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/Modality.kt renamed to firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/type/ContentModality.kt

@@ -21,7 +21,7 @@ import kotlinx.serialization.KSerializer
 import kotlinx.serialization.SerialName
 import kotlinx.serialization.Serializable
 
-public class Modality private constructor(public val ordinal: Int) {
+public class ContentModality private constructor(public val ordinal: Int) {
 
   @Serializable(Internal.Serializer::class)
   internal enum class Internal {
@@ -36,31 +36,31 @@ public class Modality private constructor(public val ordinal: Int) {
 
     internal fun toPublic() =
       when (this) {
-        TEXT -> Modality.TEXT
-        IMAGE -> Modality.IMAGE
-        VIDEO -> Modality.VIDEO
-        AUDIO -> Modality.AUDIO
-        DOCUMENT -> Modality.DOCUMENT
-        else -> Modality.UNSPECIFIED
+        TEXT -> ContentModality.TEXT
+        IMAGE -> ContentModality.IMAGE
+        VIDEO -> ContentModality.VIDEO
+        AUDIO -> ContentModality.AUDIO
+        DOCUMENT -> ContentModality.DOCUMENT
+        else -> ContentModality.UNSPECIFIED
       }
   }
 
   public companion object {
-    @JvmField public val UNSPECIFIED: Modality = Modality(0)
+    @JvmField public val UNSPECIFIED: ContentModality = ContentModality(0)
 
     /** Plain text. */
-    @JvmField public val TEXT: Modality = Modality(1)
+    @JvmField public val TEXT: ContentModality = ContentModality(1)
 
     /** Image. */
-    @JvmField public val IMAGE: Modality = Modality(2)
+    @JvmField public val IMAGE: ContentModality = ContentModality(2)
 
     /** Video. */
-    @JvmField public val VIDEO: Modality = Modality(3)
+    @JvmField public val VIDEO: ContentModality = ContentModality(3)
 
     /** Audio. */
-    @JvmField public val AUDIO: Modality = Modality(4)
+    @JvmField public val AUDIO: ContentModality = ContentModality(4)
 
     /** Document, e.g. PDF. */
-    @JvmField public val DOCUMENT: Modality = Modality(5)
+    @JvmField public val DOCUMENT: ContentModality = ContentModality(5)
   }
 }

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

@@ -19,10 +19,13 @@ package com.google.firebase.vertexai.type
 import kotlinx.serialization.Serializable
 
 public class ModalityTokenCount
-private constructor(public val modality: Modality, public val tokenCount: Int) {
+private constructor(public val modality: ContentModality, public val tokenCount: Int) {
 
   @Serializable
-  internal data class Internal(val modality: Modality.Internal, val tokenCount: Int? = null) {
+  internal data class Internal(
+    val modality: ContentModality.Internal,
+    val tokenCount: Int? = null
+  ) {
     internal fun toPublic() = ModalityTokenCount(modality.toPublic(), tokenCount ?: 0)
   }
 }

firebase-vertexai/src/test/java/com/google/firebase/vertexai/UnarySnapshotTests.kt

@@ -17,13 +17,13 @@
 package com.google.firebase.vertexai
 
 import com.google.firebase.vertexai.type.BlockReason
+import com.google.firebase.vertexai.type.ContentModality
 import com.google.firebase.vertexai.type.FinishReason
 import com.google.firebase.vertexai.type.FunctionCallPart
 import com.google.firebase.vertexai.type.HarmCategory
 import com.google.firebase.vertexai.type.HarmProbability
 import com.google.firebase.vertexai.type.HarmSeverity
 import com.google.firebase.vertexai.type.InvalidAPIKeyException
-import com.google.firebase.vertexai.type.Modality
 import com.google.firebase.vertexai.type.PromptBlockedException
 import com.google.firebase.vertexai.type.ResponseStoppedException
 import com.google.firebase.vertexai.type.SerializationException
@@ -83,11 +83,11 @@ internal class UnarySnapshotTests {
           totalTokenCount shouldBe 1913
           candidatesTokenCount shouldBe 76
           promptTokensDetails?.forAtLeastOne {
-            it.modality shouldBe Modality.IMAGE
+            it.modality shouldBe ContentModality.IMAGE
             it.tokenCount shouldBe 1806
           }
           candidatesTokensDetails?.forAtLeastOne {
-            it.modality shouldBe Modality.TEXT
+            it.modality shouldBe ContentModality.TEXT
             it.tokenCount shouldBe 76
           }
         }
@@ -491,7 +491,7 @@ internal class UnarySnapshotTests {
         response.totalBillableCharacters shouldBe 117
         response.promptTokensDetails shouldNotBe null
         response.promptTokensDetails?.forAtLeastOne {
-          it.modality shouldBe Modality.IMAGE
+          it.modality shouldBe ContentModality.IMAGE
           it.tokenCount shouldBe 1806
         }
       }