diff --git a/litellm/llms/anthropic/count_tokens/handler.py b/litellm/llms/anthropic/count_tokens/handler.py index 422c6f0e559..5b5354228f9 100644 --- a/litellm/llms/anthropic/count_tokens/handler.py +++ b/litellm/llms/anthropic/count_tokens/handler.py @@ -101,12 +101,8 @@ class AnthropicCountTokensHandler(AnthropicCountTokensConfig): verbose_logger.debug(f"Anthropic response: {anthropic_response}") - # Transform response - final_response = self.transform_response(anthropic_response) - - verbose_logger.debug(f"Final response: {final_response}") - - return final_response + # Return Anthropic response directly - no transformation needed + return anthropic_response except AnthropicError: # Re-raise Anthropic exceptions as-is diff --git a/litellm/llms/anthropic/count_tokens/transformation.py b/litellm/llms/anthropic/count_tokens/transformation.py index beffd3beeec..c3ad72436b4 100644 --- a/litellm/llms/anthropic/count_tokens/transformation.py +++ b/litellm/llms/anthropic/count_tokens/transformation.py @@ -53,24 +53,6 @@ class AnthropicCountTokensConfig: "messages": messages, } - def transform_response(self, response: Dict[str, Any]) -> Dict[str, Any]: - """ - Transform Anthropic CountTokens response. - - Input (Anthropic response): - { - "input_tokens": 123 - } - - Output: - { - "input_tokens": 123 - } - """ - return { - "input_tokens": response.get("input_tokens", 0), - } - def get_required_headers(self, api_key: str) -> Dict[str, str]: """ Get the required headers for the CountTokens API. diff --git a/litellm/llms/azure_ai/anthropic/count_tokens/handler.py b/litellm/llms/azure_ai/anthropic/count_tokens/handler.py index 099f92144ee..52a0bb8bb09 100644 --- a/litellm/llms/azure_ai/anthropic/count_tokens/handler.py +++ b/litellm/llms/azure_ai/anthropic/count_tokens/handler.py @@ -106,12 +106,8 @@ class AzureAIAnthropicCountTokensHandler(AzureAIAnthropicCountTokensConfig): verbose_logger.debug(f"Azure AI Anthropic response: {azure_response}") - # Transform response - final_response = self.transform_response(azure_response) - - verbose_logger.debug(f"Final response: {final_response}") - - return final_response + # Return Anthropic-compatible response directly - no transformation needed + return azure_response except AnthropicError: # Re-raise Anthropic exceptions as-is