Skip to content

Commit bb02ca7

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

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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)