Skip to content

Commit 0ae0ebd

Browse files
committed
Remove unnecessary field and add componentN for ContentModality
1 parent e5a993b commit 0ae0ebd

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,10 @@ import kotlinx.serialization.Serializable
2121
public class ModalityTokenCount
2222
private constructor(public val modality: ContentModality, public val tokenCount: Int) {
2323

24+
public operator fun component1(): ContentModality = modality
25+
26+
public operator fun component2(): Int = tokenCount
27+
2428
@Serializable
2529
internal data class Internal(
2630
val modality: ContentModality.Internal,

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

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@ public class UsageMetadata(
3636
public val candidatesTokenCount: Int?,
3737
public val totalTokenCount: Int,
3838
public val promptTokensDetails: List<ModalityTokenCount>?,
39-
public val cacheTokensDetails: List<ModalityTokenCount>?,
4039
public val candidatesTokensDetails: List<ModalityTokenCount>?,
4140
) {
4241

@@ -46,7 +45,6 @@ public class UsageMetadata(
4645
val candidatesTokenCount: Int? = null,
4746
val totalTokenCount: Int? = null,
4847
val promptTokensDetails: List<ModalityTokenCount.Internal>? = null,
49-
val cacheTokensDetails: List<ModalityTokenCount.Internal>? = null,
5048
val candidatesTokensDetails: List<ModalityTokenCount.Internal>? = null,
5149
) {
5250

@@ -56,7 +54,6 @@ public class UsageMetadata(
5654
candidatesTokenCount ?: 0,
5755
totalTokenCount ?: 0,
5856
promptTokensDetails = promptTokensDetails?.map { it.toPublic() },
59-
cacheTokensDetails = cacheTokensDetails?.map { it.toPublic() },
6057
candidatesTokensDetails = candidatesTokensDetails?.map { it.toPublic() }
6158
)
6259
}

0 commit comments

Comments
 (0)