scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix: apply repo ruff format (line-length 120) to token_counter.py

Browse source
This commit is contained in:
SIDDARTHA REDDY 2026-09-20 04:27:37 +00:00
parent dd40137463
commit edf783fb6c
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/llms/anthropic/count_tokens/token_counter.py
View file

@ -54,6 +54,16 @@ class AnthropicTokenCounter(BaseTokenCounter):
deployment = deployment or {}
litellm_params: Final = deployment.get("litellm_params", {})
# Skip the provider CountTokens API when the deployment points at a
# third-party Anthropic-compatible endpoint: the handler targets
# https://api.anthropic.com, so with a custom api_base the request
# would go to the wrong destination. Deployments that explicitly
# configure the official endpoint keep the provider API. Returning
# None lets the caller fall back to local token counting.
api_base: Final = litellm_params.get("api_base")
if api_base and api_base.rstrip("/") != "https://api.anthropic.com":
return None
# Get Anthropic API key from deployment config or environment
api_key = litellm_params.get("api_key")
if not api_key:
@ -81,7 +91,11 @@ class AnthropicTokenCounter(BaseTokenCounter):
original_response=result,
)
except AnthropicError as e:
verbose_logger.warning("Anthropic CountTokens API error: status=%s, message=%s", e.status_code, e.message)
verbose_logger.warning(
"Anthropic CountTokens API error: status=%s, message=%s",
e.status_code,
e.message,
)
return TokenCountResponse(
total_tokens=0,
request_model=request_model,