add changes from #6011

David Motsonashvili · David Motsonashvili · commit 3edae05a4b5f · 2024-06-12T13:55:25.000-07:00
diff --git a/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/GenerativeModel.kt b/firebase-vertexai/src/main/kotlin/com/google/firebase/vertexai/GenerativeModel.kt
@@ -230,7 +230,7 @@ internal constructor(
     )
 
   private fun constructCountTokensRequest(vararg prompt: Content) =
-    CountTokensRequest(modelName, prompt.map { it.toInternal() })
+    CountTokensRequest.forVertexAI(constructRequest(*prompt))
 
   private fun GenerateContentResponse.validate() = apply {
     if (candidates.isEmpty() && promptFeedback == null) {

Original file line number	Diff line number	Diff line change
`@@ -230,7 +230,7 @@ internal constructor(`
`230`	`230`	`)`
`231`	`231`
`232`	`232`	`private fun constructCountTokensRequest(vararg prompt: Content) =`
`233`		`- CountTokensRequest(modelName, prompt.map { it.toInternal() })`
	`233`	`+ CountTokensRequest.forVertexAI(constructRequest(*prompt))`
`234`	`234`
`235`	`235`	`private fun GenerateContentResponse.validate() = apply {`
`236`	`236`	`if (candidates.isEmpty() && promptFeedback == null) {`