tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-01 12:22:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Fix] Tests: Align Bedrock count-tokens endpoint assertions with URL-encoded model id

Browse Source 
The endpoint builder in BedrockCountTokensConfig.get_bedrock_count_tokens_endpoint
percent-encodes the model id as a single path segment (d4dd865b1a, path-traversal
hardening). Update the four endpoint-URL assertions in TestBedrockCountTokensEndpoint
to expect `amazon.nova-lite-v1%3A0` instead of the literal `:0`, matching production
behavior already covered by test_count_tokens_endpoint_encodes_model_id.
This commit is contained in:
Yuneng Jiang
2026-05-01 15:22:40 -07:00
parent d07cdd4481
commit 8eca93470c
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/litellm_utils_tests/test_bedrock_token_counter.py
+4 -4
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(