Add AnthropicModel count_tokens
#3458
Merged
+357
−20
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
nathan-gage
force-pushed
the
anthropic-count-tokens
branch
from
338b3d1 to
b9f3094
Compare
Add comprehensive test coverage for the Anthropic `count_tokens` implementation: - test_anthropic_model_usage_limit_exceeded: Verifies UsageLimitExceeded is raised when input_tokens_limit would be exceeded (18 < 19 tokens) - test_anthropic_model_usage_limit_not_exceeded: Tests successful execution when within token limits (25 > 19 tokens) - test_anthropic_count_tokens_error: Tests ModelHTTPError handling for invalid model names (404 response) Tests follow the same pattern as Bedrock's count_tokens tests, using the standard "quick brown fox" prompt which tokenizes to 19 input tokens with Anthropic's tokenizer. All tests include VCR cassettes for the `/v1/messages/count_tokens?beta=true` API endpoint. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
nathan-gage
force-pushed
the
anthropic-count-tokens
branch
from
b9f3094 to
8f47956
Compare
Contributor
Author
|
It's my first time using the cassette/vcr for tests, so would love some feedback on if I got that part right. |
nathan-gage
changed the title
count_tokens
nathan-gage
changed the title
count_tokens count_tokens
Collaborator
|
@nathan-gage Looks great, thanks Nathan! |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Closes #3429
AFAIK, bedrock client does not support count_tokens, so I have to raise a UserError in that case.