File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
tests/test_litellm/responses Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -168,6 +168,7 @@ def test_transform_response_api_usage_to_chat_usage(self):
168
168
"input_tokens" : 10 ,
169
169
"output_tokens" : 20 ,
170
170
"total_tokens" : 30 ,
171
+ "input_tokens_details" : {"cached_tokens" : 2 },
171
172
"output_tokens_details" : {"reasoning_tokens" : 5 },
172
173
}
173
174
@@ -181,6 +182,7 @@ def test_transform_response_api_usage_to_chat_usage(self):
181
182
assert result .prompt_tokens == 10
182
183
assert result .completion_tokens == 20
183
184
assert result .total_tokens == 30
185
+ assert result .prompt_tokens_details and result .prompt_tokens_details .cached_tokens == 2
184
186
185
187
def test_transform_response_api_usage_with_none_values (self ):
186
188
"""Test transformation handles None values properly"""
You can’t perform that action at this time.
0 commit comments