diff --git a/tests/litellm_utils_tests/test_bedrock_token_counter.py b/tests/litellm_utils_tests/test_bedrock_token_counter.py index 7239ec0bcf..b0e37914af 100644 --- a/tests/litellm_utils_tests/test_bedrock_token_counter.py +++ b/tests/litellm_utils_tests/test_bedrock_token_counter.py @@ -150,7 +150,7 @@ class TestBedrockCountTokensEndpoint: api_base=api_base, aws_bedrock_runtime_endpoint=runtime_endpoint, ) - assert url.startswith(api_base) + assert url == f"{api_base}/model/amazon.nova-lite-v1:0/count-tokens" def test_env_var_overrides_default(self, monkeypatch): monkeypatch.setenv(