scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

Merge pull request #27017 from BerriAI/fix/bedrock-count-tokens-url-encoded-test-assertions

Browse source 
[Fix] Tests: Align Bedrock count-tokens endpoint assertions with URL-encoded model id
This commit is contained in:
yuneng-jiang 2026-05-01 15:34:21 -07:00 • committed by GitHub
commit 2573fcdbb2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/litellm_utils_tests/test_bedrock_token_counter.py
View file

@ -130,7 +130,7 @@ class TestBedrockCountTokensEndpoint:
)
assert (
url
== "https://bedrock-runtime.us-east-1.amazonaws.com/model/amazon.nova-lite-v1:0/count-tokens"
== "https://bedrock-runtime.us-east-1.amazonaws.com/model/amazon.nova-lite-v1%3A0/count-tokens"
)
def test_api_base_overrides_default(self):
@ -141,7 +141,7 @@ class TestBedrockCountTokensEndpoint:
aws_region_name="us-east-1",
api_base=custom_base,
)
assert url == f"{custom_base}/model/amazon.nova-lite-v1:0/count-tokens"
assert url == f"{custom_base}/model/amazon.nova-lite-v1%3A0/count-tokens"
def test_aws_bedrock_runtime_endpoint_overrides_default(self):
handler = self._make_handler()
@ -153,7 +153,7 @@ class TestBedrockCountTokensEndpoint:
aws_region_name="eu-west-1",
aws_bedrock_runtime_endpoint=custom_endpoint,
)
assert url == f"{custom_endpoint}/model/amazon.nova-lite-v1:0/count-tokens"
assert url == f"{custom_endpoint}/model/amazon.nova-lite-v1%3A0/count-tokens"
def test_api_base_takes_priority_over_aws_bedrock_runtime_endpoint(self):
handler = self._make_handler()
@ -165,7 +165,7 @@ class TestBedrockCountTokensEndpoint:
api_base=api_base,
aws_bedrock_runtime_endpoint=runtime_endpoint,
)
assert url == f"{api_base}/model/amazon.nova-lite-v1:0/count-tokens"
assert url == f"{api_base}/model/amazon.nova-lite-v1%3A0/count-tokens"
def test_env_var_overrides_default(self, monkeypatch):
monkeypatch.setenv(