Skip to content

Commit 9b0eaf0

Browse files
committed
test: add a testcase for cached_tokens
Signed-off-by: Adam Tao <[email protected]>
1 parent f11083e commit 9b0eaf0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_litellm/responses/test_responses_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,7 @@ def test_transform_response_api_usage_to_chat_usage(self):
168168
"input_tokens": 10,
169169
"output_tokens": 20,
170170
"total_tokens": 30,
171+
"input_tokens_details": {"cached_tokens": 2},
171172
"output_tokens_details": {"reasoning_tokens": 5},
172173
}
173174

@@ -181,6 +182,7 @@ def test_transform_response_api_usage_to_chat_usage(self):
181182
assert result.prompt_tokens == 10
182183
assert result.completion_tokens == 20
183184
assert result.total_tokens == 30
185+
assert result.prompt_tokens_details and result.prompt_tokens_details.cached_tokens == 2
184186

185187
def test_transform_response_api_usage_with_none_values(self):
186188
"""Test transformation handles None values properly"""

0 commit comments

Comments
 (0)