Skip to content

Commit 7f3013c

Browse files
chore(tracing): pdm lint
1 parent 5b5c0dd commit 7f3013c

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

src/askui/models/shared/agent.py

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

0 commit comments

Comments
 (0)