Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion genai/count_tokens/counttoken_with_txt.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def count_tokens() -> int:
)
print(response)
# Example output:
# total_tokens=10
# total_tokens=9
# cached_content_token_count=None
# [END googlegenaisdk_counttoken_with_txt]
return response.total_tokens
Expand Down
1 change: 1 addition & 0 deletions genai/count_tokens/requirements.txt
Original file line number Diff line number Diff line change
@@ -1 +1,2 @@
google-genai==1.42.0
sentencepiece==0.2.1
10 changes: 10 additions & 0 deletions genai/count_tokens/test_count_tokens_examples.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,8 @@
import os

import counttoken_compute_with_txt
import counttoken_localtokenizer_compute_with_txt
import counttoken_localtokenizer_with_txt
import counttoken_resp_with_txt
import counttoken_with_txt
import counttoken_with_txt_vid
Expand All @@ -43,3 +45,11 @@ def test_counttoken_with_txt() -> None:

def test_counttoken_with_txt_vid() -> None:
assert counttoken_with_txt_vid.count_tokens()


def test_counttoken_localtokenizer_with_txt() -> None:
assert counttoken_localtokenizer_with_txt.counttoken_localtokenizer_with_txt()


def test_counttoken_localtokenizer_compute_with_txt() -> None:
assert counttoken_localtokenizer_compute_with_txt.counttoken_localtokenizer_compute_with_txt()