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(
492
492
"gen_ai.usage.input_tokens" : accumulated_usage ["inputTokens" ],
493
493
"gen_ai.usage.output_tokens" : accumulated_usage ["outputTokens" ],
494
494
"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 ) ,
497
497
}
498
498
)
499
499
Original file line number Diff line number Diff line change 1
1
"""Event loop-related type definitions for the SDK."""
2
2
3
- from typing import Literal , Optional
3
+ from typing import Literal
4
4
5
5
from typing_extensions import TypedDict
6
6
@@ -19,8 +19,8 @@ class Usage(TypedDict, total=False):
19
19
inputTokens : int
20
20
outputTokens : int
21
21
totalTokens : int
22
- cacheReadInputTokens : Optional [ int ]
23
- cacheWriteInputTokens : Optional [ int ]
22
+ cacheReadInputTokens : int
23
+ cacheWriteInputTokens : int
24
24
25
25
26
26
class Metrics (TypedDict ):
You can’t perform that action at this time.
0 commit comments