Skip to content

Commit 7b9ca48

Browse files
fix(genai sdk): fix example output and file name (GoogleCloudPlatform#13185)
* fix output * fix filename * Update genai/count_tokens/test_count_tokens_examples.py --------- Co-authored-by: Holt Skinner <13262395+holtskinner@users.noreply.github.com>
1 parent 46663e8 commit 7b9ca48

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

genai/count_tokens/counttoken_with_txt_img_vid.py renamed to genai/count_tokens/counttoken_with_txt_vid.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def count_tokens() -> int:
3737
print(response)
3838

3939
# Example output:
40-
# total_tokens=10 cached_content_token_count=None
40+
# total_tokens=16252 cached_content_token_count=None
4141

4242
# [END googlegenaisdk_counttoken_with_txt_vid]
4343
# [END googlegenaisdk_count_tokens_with_txt_img_vid]

genai/count_tokens/test_count_tokens_examples.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
import counttoken_compute_with_txt
2222
import counttoken_resp_with_txt
2323
import counttoken_with_txt
24-
import counttoken_with_txt_img_vid
24+
import counttoken_with_txt_vid
2525

2626
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
2727
os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
@@ -41,5 +41,5 @@ def test_counttoken_with_txt() -> None:
4141
assert counttoken_with_txt.count_tokens()
4242

4343

44-
def test_counttoken_with_txt_img_vid() -> None:
45-
assert counttoken_with_txt_img_vid.count_tokens()
44+
def test_counttoken_with_txt_vid() -> None:
45+
assert counttoken_with_txt_vid.count_tokens()

0 commit comments

Comments
 (0)