Skip to content

Commit fb14ce0

Browse files
committed
adapt naming
1 parent 5a40df2 commit fb14ce0

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

‎sentry_sdk/integrations/anthropic.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,8 +89,8 @@ def _get_token_usage(result: "Messages") -> "tuple[int, int, int, int]":
8989
output_tokens = usage.output_tokens
9090
if hasattr(usage, "cache_read_input_tokens") and isinstance(usage.cache_read_input_tokens, int):
9191
cache_read_input_tokens = usage.cache_read_input_tokens
92-
if hasattr(usage, "cache_creation_input_tokens") and isinstance(usage.cache_creation_input_tokens, int):
93-
cache_write_input_tokens = usage.cache_creation_input_tokens
92+
if hasattr(usage, "cache_write_input_tokens") and isinstance(usage.cache_write_input_tokens, int):
93+
cache_write_input_tokens = usage.cache_write_input_tokens
9494

9595
return input_tokens, output_tokens, cache_read_input_tokens, cache_write_input_tokens
9696

@@ -115,8 +115,8 @@ def _collect_ai_data(
115115
output_tokens += usage.output_tokens
116116
if hasattr(usage, "cache_read_input_tokens") and isinstance(usage.cache_read_input_tokens, int):
117117
cache_read_input_tokens += usage.cache_read_input_tokens
118-
if hasattr(usage, "cache_creation_input_tokens") and isinstance(usage.cache_creation_input_tokens, int):
119-
cache_write_input_tokens += usage.cache_creation_input_tokens
118+
if hasattr(usage, "cache_write_input_tokens") and isinstance(usage.cache_write_input_tokens, int):
119+
cache_write_input_tokens += usage.cache_write_input_tokens
120120
model = event.message.model or model
121121
elif event.type == "content_block_start":
122122
pass

‎sentry_sdk/integrations/langchain.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -634,7 +634,7 @@ def _extract_tokens(
634634

635635
cache_write_tokens = (
636636
_get_value(token_usage, "cache_write_tokens") or # Pydantic-AI
637-
_get_value(token_usage, "cache_creation_input_tokens") # Anthropic
637+
_get_value(token_usage, "cache_write_input_tokens") # Anthropic
638638
)
639639

640640
return input_tokens, output_tokens, total_tokens, cached_tokens, cache_write_tokens

0 commit comments

Comments
 (0)