From 6d6bffcabd81ee0e0c13c0568ba38944c10c601d Mon Sep 17 00:00:00 2001 From: Adithya Suresh Date: Fri, 4 Apr 2025 16:58:10 +1100 Subject: [PATCH] Revert "Handle cache token counts being none" This reverts commit 87934be342e438cc270ba04150d0189a870164f1. --- src/pipecat/services/anthropic/llm.py | 24 +++++++----------------- 1 file changed, 7 insertions(+), 17 deletions(-) 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 )