remove use of VertexCountTokensRequest

google-gemini · Jun 4, 2024 · ef8cda5 · ef8cda5
1 parent 0e15bb7
commit ef8cda5
Showing 1 changed file with 0 additions and 14 deletions.
diff --git a/common/src/main/kotlin/com/google/ai/client/generativeai/common/APIController.kt b/common/src/main/kotlin/com/google/ai/client/generativeai/common/APIController.kt
@@ -131,24 +131,10 @@ internal constructor(
       throw GoogleGenerativeAIException.from(e)
     }
 
-  suspend fun vertexCountTokens(request: VertexCountTokensRequest): CountTokensResponse =
-    try {
-      client
-        .post("${requestOptions.endpoint}/${requestOptions.apiVersion}/$model:countTokens") {
-          applyCommonConfiguration(request)
-          applyHeaderProvider()
-        }
-        .also { validateResponse(it) }
-        .body()
-    } catch (e: Throwable) {
-      throw GoogleGenerativeAIException.from(e)
-    }
-
   private fun HttpRequestBuilder.applyCommonConfiguration(request: Request) {
     when (request) {
       is GenerateContentRequest -> setBody<GenerateContentRequest>(request)
       is CountTokensRequest -> setBody<CountTokensRequest>(request)
-      is VertexCountTokensRequest -> setBody<VertexCountTokensRequest>(request)
     }
     contentType(ContentType.Application.Json)
     header("x-goog-api-key", key)