From 7e03ab062d09e35e220924846a4dd1792a3e55ff Mon Sep 17 00:00:00 2001 From: Donghee Na Date: Mon, 27 Oct 2025 15:31:21 +0900 Subject: [PATCH] fix: Fix invalid cache_creation_tokens metric key (#195) --- src/api/models/bedrock.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/api/models/bedrock.py b/src/api/models/bedrock.py index 66b7424..4253e1f 100644 --- a/src/api/models/bedrock.py +++ b/src/api/models/bedrock.py @@ -375,7 +375,7 @@ class BedrockModel(BaseChatModel): # Extract prompt caching metrics if available cache_read_tokens = usage.get("cacheReadInputTokens", 0) - cache_creation_tokens = usage.get("cacheCreationInputTokens", 0) + cache_creation_tokens = usage.get("cacheWriteInputTokens", 0) # Calculate actual prompt tokens # Bedrock's totalTokens includes all: inputTokens + cacheRead + cacheWrite + outputTokens @@ -1035,7 +1035,7 @@ class BedrockModel(BaseChatModel): # Extract prompt caching metrics if available cache_read_tokens = usage_data.get("cacheReadInputTokens", 0) - cache_creation_tokens = usage_data.get("cacheCreationInputTokens", 0) + cache_creation_tokens = usage_data.get("cacheWriteInputTokens", 0) # Create prompt_tokens_details if cache metrics are available prompt_tokens_details = None