Skip to content

Commit 5f3e83d

Browse files
committed
Update openai mocks to support the model_dump
1 parent 2de1593 commit 5f3e83d

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

tests/_utils/llm.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -89,13 +89,11 @@ def mk_mock_openai_response(messages: list[str]) -> mock.MagicMock:
8989
response = mock.MagicMock()
9090

9191
response.choices = [_mk_mock_choice(message) for message in messages]
92-
response.dict.return_value = {
93-
"choices": [{
94-
'message': {
95-
'role': 'assistant', 'content': message
96-
}
97-
} for message in messages]
98-
}
92+
93+
response_dict = {"choices": [{'message': {'role': 'assistant', 'content': message}} for message in messages]}
94+
95+
response.dict.return_value = response_dict
96+
response.model_dump.return_value = response_dict
9997

10098
return response
10199

0 commit comments

Comments
 (0)