File tree Expand file tree Collapse file tree 1 file changed +13
-10
lines changed
Expand file tree Collapse file tree 1 file changed +13
-10
lines changed Original file line number Diff line number Diff line change @@ -121,14 +121,16 @@ def _step(
121121
122122 step_span .set_attributes (
123123 {
124- "cache_creation_input_tokens" : response_message .usage .cache_creation_input_tokens
125- or 0
126- if response_message .usage
127- else 0 ,
128- "cache_read_input_tokens" : response_message .usage .cache_read_input_tokens
129- or 0
130- if response_message .usage
131- else 0 ,
124+ "cache_creation_input_tokens" : (
125+ response_message .usage .cache_creation_input_tokens or 0
126+ if response_message .usage
127+ else 0
128+ ),
129+ "cache_read_input_tokens" : (
130+ response_message .usage .cache_read_input_tokens or 0
131+ if response_message .usage
132+ else 0
133+ ),
132134 "input_tokens" : response_message .usage .input_tokens or 0
133135 if response_message .usage
134136 else 0 ,
@@ -219,8 +221,9 @@ def act(
219221 {
220222 "input_tokens" : accumulated_usage .input_tokens or 0 ,
221223 "output_tokens" : accumulated_usage .output_tokens or 0 ,
222- "cache_creation_input_tokens" : accumulated_usage .cache_creation_input_tokens
223- or 0 ,
224+ "cache_creation_input_tokens" : (
225+ accumulated_usage .cache_creation_input_tokens or 0
226+ ),
224227 "cache_read_input_tokens" : accumulated_usage .cache_read_input_tokens
225228 or 0 ,
226229 }
You can’t perform that action at this time.
0 commit comments