File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -56,7 +56,9 @@ def create_mock_openai_chat_completion(
5656 # or if get_tokens_counts_from_response had different fallback logic.
5757 completion .usage .prompt_tokens = prompt_tokens
5858 completion .usage .completion_tokens = completion_tokens
59- completion .usage .prompt_tokens_details .cached_tokens = 0
59+ prompt_tokens_details_mock = MagicMock ()
60+ prompt_tokens_details_mock .cached_tokens = 0
61+ completion .usage .prompt_tokens_details = prompt_tokens_details_mock
6062
6163 completion .model_dump .return_value = {
6264 "id" : "chatcmpl-xxxx" ,
@@ -211,7 +213,9 @@ def create_mock_openai_responses_api_response(
211213 response_mock .usage .output_tokens = output_tokens
212214 response_mock .usage .prompt_tokens = input_tokens
213215 response_mock .usage .completion_tokens = output_tokens
214- response_mock .usage .input_tokens_details .cached_tokens = 0
216+ input_tokens_details_mock = MagicMock ()
217+ input_tokens_details_mock .cached_tokens = 0
218+ response_mock .usage .input_tokens_details = input_tokens_details_mock
215219
216220 return response_mock
217221
You can’t perform that action at this time.
0 commit comments