# common_utils.py:1064
vertex_location = (
    count_tokens_params_request.get("vertex_count_tokens_location")
    or vertex_location
)
# → correctly resolves to "europe-west1" ✅

This value is passed to VertexAIPartnerModels.count_tokens() as vertex_location="europe-west1", which copies it into _litellm_params["vertex_location"] and calls
handle_count_tokens_request().

However, inside handler.py, the location is re-read from litellm_params and then unconditionally overridden because "claude" is always present in the model name:

# handler.py:108-113 — BROKEN
vertex_location = self.get_vertex_ai_location(litellm_params)
# → reads "europe-west1"

if not vertex_location or "claude" in model.lower():
    vertex_location = "us-central1"
# → "claude" is ALWAYS in Claude model names → always overrides to "us-central1" ❌
# → vertex_count_tokens_location is never consulted here

---
Expected Behavior

vertex_count_tokens_location should take precedence over the us-central1 default. The default override should only apply when no explicit count-tokens location is configured.

---
Proposed Fix

In litellm/llms/vertex_ai/vertex_ai_partner_models/count_tokens/handler.py, change lines 108–113:

# Current (broken):
vertex_location = self.get_vertex_ai_location(litellm_params)
if not vertex_location or "claude" in model.lower():
    vertex_location = "us-central1"

# Fixed:
vertex_location = (
    litellm_params.get("vertex_count_tokens_location")
    or self.get_vertex_ai_location(litellm_params)
)
if not vertex_location or "claude" in model.lower():
    vertex_location = "us-central1"

This ensures vertex_count_tokens_location is checked first. If set, the us-central1 override is bypassed. If not set and the model is Claude, it falls back to us-central1 as before.

---
Steps to Reproduce

1. Configure a Claude model on Vertex AI with vertex_location: europe-west1 and vertex_count_tokens_location: europe-west1
2. Make a request that triggers token counting
3. Observe: request routes to us-central1 instead of europe-west1 → 500 error

LiteLLM config: