File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed
Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -121,13 +121,11 @@ def _step(
121121
122122 step_span .set_attributes (
123123 {
124- "cache_creation_input_tokens" :
125- response_message .usage .cache_creation_input_tokens
124+ "cache_creation_input_tokens" : response_message .usage .cache_creation_input_tokens
126125 or 0
127126 if response_message .usage
128127 else 0 ,
129- "cache_read_input_tokens" :
130- response_message .usage .cache_read_input_tokens
128+ "cache_read_input_tokens" : response_message .usage .cache_read_input_tokens
131129 or 0
132130 if response_message .usage
133131 else 0 ,
@@ -221,8 +219,8 @@ def act(
221219 {
222220 "input_tokens" : accumulated_usage .input_tokens or 0 ,
223221 "output_tokens" : accumulated_usage .output_tokens or 0 ,
224- "cache_creation_input_tokens" :
225- accumulated_usage . cache_creation_input_tokens or 0 ,
222+ "cache_creation_input_tokens" : accumulated_usage . cache_creation_input_tokens
223+ or 0 ,
226224 "cache_read_input_tokens" : accumulated_usage .cache_read_input_tokens
227225 or 0 ,
228226 }
You can’t perform that action at this time.
0 commit comments