From 25085266af0d2ea63c0c539fd0bd5979fa0082fc Mon Sep 17 00:00:00 2001 From: Otavio Brito Date: Wed, 1 Apr 2026 20:37:01 -0300 Subject: [PATCH] rollback file --- .../vertex_ai_partner_models/count_tokens/handler.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/litellm/llms/vertex_ai/vertex_ai_partner_models/count_tokens/handler.py b/litellm/llms/vertex_ai/vertex_ai_partner_models/count_tokens/handler.py index 9e22f54f112..c6914ac3d6b 100644 --- a/litellm/llms/vertex_ai/vertex_ai_partner_models/count_tokens/handler.py +++ b/litellm/llms/vertex_ai/vertex_ai_partner_models/count_tokens/handler.py @@ -144,6 +144,13 @@ class VertexAIPartnerModelsTokenCounter(VertexBase): timeout=30.0, ) + # Check for errors + if response.status_code != 200: + error_text = response.text + raise ValueError( + f"Token counting request failed with status {response.status_code}: {error_text}" + ) + # Parse response result = response.json()