From ce8f0e38e7b6da61732daef4819193f99c34c161 Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Fri, 3 Jul 2026 18:00:53 +0000 Subject: [PATCH 1/3] refactor: remove redundant tag from GetMediaTokenCount sample --- .../src/main/java/vertexai/gemini/GetMediaTokenCount.java | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java b/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java index c6f5015d373..56d36d23a43 100644 --- a/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java +++ b/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java @@ -16,7 +16,6 @@ package vertexai.gemini; -// [START generativeaionvertexai_gemini_token_count_advanced] import com.google.cloud.vertexai.VertexAI; import com.google.cloud.vertexai.api.Content; import com.google.cloud.vertexai.api.CountTokensResponse; @@ -57,5 +56,4 @@ public static int getMediaTokenCount(String projectId, String location, String m return tokenCount; } } -} -// [END generativeaionvertexai_gemini_token_count_advanced] +} \ No newline at end of file From be322dc7166d4a07e50138774fd9f1da2c5d345a Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Fri, 3 Jul 2026 18:06:46 +0000 Subject: [PATCH 2/3] refactor: remove GetMediaTokenCount sample from vertexai snippets --- .../vertexai/gemini/GetMediaTokenCount.java | 59 ------------------- 1 file changed, 59 deletions(-) delete mode 100644 vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java diff --git a/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java b/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java deleted file mode 100644 index 56d36d23a43..00000000000 --- a/vertexai/snippets/src/main/java/vertexai/gemini/GetMediaTokenCount.java +++ /dev/null @@ -1,59 +0,0 @@ -/* - * Copyright 2024 Google LLC - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -package vertexai.gemini; - -import com.google.cloud.vertexai.VertexAI; -import com.google.cloud.vertexai.api.Content; -import com.google.cloud.vertexai.api.CountTokensResponse; -import com.google.cloud.vertexai.generativeai.ContentMaker; -import com.google.cloud.vertexai.generativeai.GenerativeModel; -import com.google.cloud.vertexai.generativeai.PartMaker; -import java.io.IOException; - -public class GetMediaTokenCount { - public static void main(String[] args) throws IOException { - // TODO(developer): Replace these variables before running the sample. - String projectId = "your-google-cloud-project-id"; - String location = "us-central1"; - String modelName = "gemini-2.5-flash"; - - getMediaTokenCount(projectId, location, modelName); - } - - // Gets the number of tokens for the prompt with text and video and the model's response. - public static int getMediaTokenCount(String projectId, String location, String modelName) - throws IOException { - // Initialize client that will be used to send requests. - // This client only needs to be created once, and can be reused for multiple requests. - try (VertexAI vertexAI = new VertexAI(projectId, location)) { - GenerativeModel model = new GenerativeModel(modelName, vertexAI); - - Content content = ContentMaker.fromMultiModalData( - "Provide a description of the video.", - PartMaker.fromMimeTypeAndData( - "video/mp4", "gs://cloud-samples-data/generative-ai/video/pixel8.mp4") - ); - - CountTokensResponse response = model.countTokens(content); - - int tokenCount = response.getTotalTokens(); - System.out.println("Token count: " + tokenCount); - - return tokenCount; - } - } -} \ No newline at end of file From 47b90a477f3a9a05c3a610fcc81cb23c53df7a06 Mon Sep 17 00:00:00 2001 From: Kevin Gordillo Date: Fri, 3 Jul 2026 18:53:17 +0000 Subject: [PATCH 3/3] test: remove testMediaTokenCount from SnippetsIT.java --- .../snippets/src/test/java/vertexai/gemini/SnippetsIT.java | 6 ------ 1 file changed, 6 deletions(-) diff --git a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java index 1b496c48c9e..0abb3077894 100644 --- a/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java +++ b/vertexai/snippets/src/test/java/vertexai/gemini/SnippetsIT.java @@ -166,12 +166,6 @@ public void testTokenCount() throws Exception { assertThat(tokenCount).isEqualTo(6); } - @Test - public void testMediaTokenCount() throws Exception { - int tokenCount = GetMediaTokenCount.getMediaTokenCount(PROJECT_ID, LOCATION, GEMINI_FLASH); - assertThat(tokenCount).isNotNull(); - } - @Test public void testFunctionCalling() throws Exception { String textPrompt = "What's the weather in Paris?";