diff --git a/genai/count_tokens/counttoken_with_txt_img_vid.py b/genai/count_tokens/counttoken_with_txt_vid.py similarity index 96% rename from genai/count_tokens/counttoken_with_txt_img_vid.py rename to genai/count_tokens/counttoken_with_txt_vid.py index d5979d7e56b..1fe770cd115 100644 --- a/genai/count_tokens/counttoken_with_txt_img_vid.py +++ b/genai/count_tokens/counttoken_with_txt_vid.py @@ -37,7 +37,7 @@ def count_tokens() -> int: print(response) # Example output: - # total_tokens=10 cached_content_token_count=None + # total_tokens=16252 cached_content_token_count=None # [END googlegenaisdk_counttoken_with_txt_vid] # [END googlegenaisdk_count_tokens_with_txt_img_vid] diff --git a/genai/count_tokens/test_count_tokens_examples.py b/genai/count_tokens/test_count_tokens_examples.py index 06b2d04dd52..014e0418d64 100644 --- a/genai/count_tokens/test_count_tokens_examples.py +++ b/genai/count_tokens/test_count_tokens_examples.py @@ -21,7 +21,7 @@ import counttoken_compute_with_txt import counttoken_resp_with_txt import counttoken_with_txt -import counttoken_with_txt_img_vid +import counttoken_with_txt_vid os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True" os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1" @@ -41,5 +41,5 @@ def test_counttoken_with_txt() -> None: assert counttoken_with_txt.count_tokens() -def test_counttoken_with_txt_img_vid() -> None: - assert counttoken_with_txt_img_vid.count_tokens() +def test_counttoken_with_txt_vid() -> None: + assert counttoken_with_txt_vid.count_tokens()