Skip to content

Commit f60b314

Browse files
Fix Reponses API mock object specification in test_responses_api.py
1 parent 16d334c commit f60b314

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/llm/test_response_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ def create_mock_openai_responses_api_response(
184184
Compatible with OpenAIResponseModel and TrackAPIPricingMixin.
185185
"""
186186

187-
response_mock = MagicMock(openai.types.responses.response)
187+
response_mock = MagicMock(spec=openai.types.responses.response.Response)
188188
response_mock.type = "response"
189189
response_mock.output = []
190190

@@ -208,7 +208,7 @@ def create_mock_openai_responses_api_response(
208208
response_mock.output.append(output_item_mock)
209209

210210
# Token usage for pricing tracking
211-
response_mock.usage = MagicMock()
211+
response_mock.usage = MagicMock(spec=openai.types.responses.response.ResponseUsage)
212212
response_mock.usage.input_tokens = input_tokens
213213
response_mock.usage.output_tokens = output_tokens
214214
response_mock.usage.prompt_tokens = input_tokens

0 commit comments

Comments
 (0)