Skip to content

Commit 16d334c

Browse files
fix nested mock object in test.
1 parent 8f3105f commit 16d334c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/llm/test_response_api.py

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

0 commit comments

Comments
 (0)