diff --git a/src/pipecat/services/anthropic/llm.py b/src/pipecat/services/anthropic/llm.py index eba6a5041..277e29f83 100644 --- a/src/pipecat/services/anthropic/llm.py +++ b/src/pipecat/services/anthropic/llm.py @@ -250,24 +250,14 @@ class AnthropicLLMService(LLMService): if hasattr(event.message.usage, "output_tokens") else 0 ) - cache_creation_input_tokens += ( - event.message.usage.cache_creation_input_tokens - if ( - hasattr(event.message.usage, "cache_creation_input_tokens") - and event.message.usage.cache_creation_input_tokens is not None + if hasattr(event.message.usage, "cache_creation_input_tokens"): + cache_creation_input_tokens += ( + event.message.usage.cache_creation_input_tokens ) - else 0 - ) - logger.debug(f"Cache creation input tokens: {cache_creation_input_tokens}") - cache_read_input_tokens += ( - event.message.usage.cache_read_input_tokens - if ( - hasattr(event.message.usage, "cache_read_input_tokens") - and event.message.usage.cache_read_input_tokens is not None - ) - else 0 - ) - logger.debug(f"Cache read input tokens: {cache_read_input_tokens}") + logger.debug(f"Cache creation input tokens: {cache_creation_input_tokens}") + if hasattr(event.message.usage, "cache_read_input_tokens"): + cache_read_input_tokens += event.message.usage.cache_read_input_tokens + logger.debug(f"Cache read input tokens: {cache_read_input_tokens}") total_input_tokens = ( prompt_tokens + cache_creation_input_tokens + cache_read_input_tokens )