Skip to content

Commit 219b227

Browse files
author
Workshop Participant
committed
Removes optional cache metric
1 parent 5a961e4 commit 219b227

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/strands/telemetry/tracer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -492,8 +492,8 @@ def end_agent_span(
492492
"gen_ai.usage.input_tokens": accumulated_usage["inputTokens"],
493493
"gen_ai.usage.output_tokens": accumulated_usage["outputTokens"],
494494
"gen_ai.usage.total_tokens": accumulated_usage["totalTokens"],
495-
"gen_ai.usage.cache_read_input_tokens": accumulated_usage["cacheReadInputTokens"],
496-
"gen_ai.usage.cache_write_input_tokens": accumulated_usage["cacheWriteInputTokens"],
495+
"gen_ai.usage.cache_read_input_tokens": accumulated_usage.get("cacheReadInputTokens", 0),
496+
"gen_ai.usage.cache_write_input_tokens": accumulated_usage.get("cacheWriteInputTokens", 0),
497497
}
498498
)
499499

src/strands/types/event_loop.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""Event loop-related type definitions for the SDK."""
22

3-
from typing import Literal, Optional
3+
from typing import Literal
44

55
from typing_extensions import TypedDict
66

@@ -19,8 +19,8 @@ class Usage(TypedDict, total=False):
1919
inputTokens: int
2020
outputTokens: int
2121
totalTokens: int
22-
cacheReadInputTokens: Optional[int]
23-
cacheWriteInputTokens: Optional[int]
22+
cacheReadInputTokens: int
23+
cacheWriteInputTokens: int
2424

2525

2626
class Metrics(TypedDict):

0 commit comments

Comments
 (0)