Skip to content

Commit 26ca4ad

Browse files
committed
feat: updated cached read/write input token metrics
1 parent a513b75 commit 26ca4ad

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/strands/telemetry/metrics.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -350,9 +350,9 @@ def _metrics_summary_to_lines(event_loop_metrics: EventLoopMetrics, allowed_name
350350

351351
# Add cached token info if present
352352
if summary["accumulated_usage"].get("cacheReadInputTokens"):
353-
token_parts.append(f"cache_read={summary['accumulated_usage']['cacheReadInputTokens']}")
353+
token_parts.append(f"cache_read_input_tokens={summary['accumulated_usage']['cacheReadInputTokens']}")
354354
if summary["accumulated_usage"].get("cacheWriteInputTokens"):
355-
token_parts.append(f"cache_write={summary['accumulated_usage']['cacheWriteInputTokens']}")
355+
token_parts.append(f"cache_write_input_tokens={summary['accumulated_usage']['cacheWriteInputTokens']}")
356356

357357
yield f"├─ Tokens: {', '.join(token_parts)}"
358358
yield f"├─ Bedrock Latency: {summary['accumulated_metrics']['latencyMs']}ms"

tests/strands/telemetry/test_metrics.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def usage(request):
9090
"inputTokens": 1,
9191
"outputTokens": 2,
9292
"totalTokens": 3,
93-
"cacheWriteInputTokens": 10,
93+
"cacheWriteInputTokens": 2,
9494
}
9595
if hasattr(request, "param"):
9696
params.update(request.param)
@@ -316,7 +316,7 @@ def test_event_loop_metrics_update_usage(usage, event_loop_metrics, mock_get_met
316316
event_loop_metrics.update_usage(usage)
317317

318318
tru_usage = event_loop_metrics.accumulated_usage
319-
exp_usage = Usage(inputTokens=3, outputTokens=6, totalTokens=9, cacheWriteInputTokens=30)
319+
exp_usage = Usage(inputTokens=3, outputTokens=6, totalTokens=9, cacheWriteInputTokens=6)
320320

321321
assert tru_usage == exp_usage
322322
mock_get_meter_provider.return_value.get_meter.assert_called()

0 commit comments

Comments
 (0)