File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments