File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 11"""Event loop-related type definitions for the SDK."""
22
3- from typing import Literal , Optional
3+ from typing import Literal
44
55from 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
2626class Metrics (TypedDict ):
You can’t perform that action at this time.
0 commit comments