diff --git a/generative_ai/token_count/multimodal_example.py b/generative_ai/token_count/multimodal_token_count_example.py similarity index 100% rename from generative_ai/token_count/multimodal_example.py rename to generative_ai/token_count/multimodal_token_count_example.py diff --git a/generative_ai/token_count/token_count_test.py b/generative_ai/token_count/token_count_test.py index c3d975c912ed..32e2b273c213 100644 --- a/generative_ai/token_count/token_count_test.py +++ b/generative_ai/token_count/token_count_test.py @@ -14,7 +14,7 @@ import api_example import local_sdk_example -import multimodal_example +import multimodal_token_count_example import simple_example @@ -28,6 +28,6 @@ def test_gemini_count_token_example() -> None: assert response assert response.usage_metadata - response = multimodal_example.count_tokens_multimodal() + response = multimodal_token_count_example.count_tokens_multimodal() assert response assert response.usage_metadata