Skip to content

Commit 5b5c0dd

Browse files
chore(tracing): pdm format
1 parent 3a134b1 commit 5b5c0dd

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/askui/models/shared/agent.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff 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
}

0 commit comments

Comments
 (0)